refactor(events): Validate listeners

discord-jar · Mar 17, 2024 · 2dff812 · 2dff812
1 parent d6d0923
commit 2dff812
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/main/java/com/seailz/discordjar/events/EventDispatcher.java b/src/main/java/com/seailz/discordjar/events/EventDispatcher.java
@@ -55,7 +55,15 @@ public void addListener(DiscordListener... listeners) {
         for (DiscordListener listener : listeners) {
             for (Method method : listener.getClass().getMethods()) {
                 if (method.isAnnotationPresent(EventMethod.class)) {
-                    Class<? extends Event> eventType = (Class<? extends Event>) method.getParameterTypes()[0];
+                    Class<?> maybeEventType = method.getParameterTypes()[0];
+
+                    if (!Event.class.isAssignableFrom(maybeEventType))
+                        throw new IllegalArgumentException(String.format("%s first arg is not of Event", method));
+                    else if (method.getParameterTypes().length > 1)
+                        throw new IllegalArgumentException(String.format("%s#%s is an invalid listener", method.getDeclaringClass(), method.getName()));
+
+                    @SuppressWarnings("unchecked")
+                    Class<? extends Event> eventType = (Class<? extends Event>) maybeEventType;
                     EventMethod eventMethod = method.getAnnotation(EventMethod.class);
 
                     String customId = null;