ensure Switch is only used on booleans

Revxrsal · Sep 17, 2024 · 67a34de · 67a34de
1 parent 1820e00
commit 67a34de
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/common/src/main/java/revxrsal/commands/node/parser/ParameterNodeImpl.java b/common/src/main/java/revxrsal/commands/node/parser/ParameterNodeImpl.java
@@ -42,6 +42,7 @@
 import revxrsal.commands.stream.MutableStringStream;
 import revxrsal.commands.stream.MutableStringStreamImpl;
 import revxrsal.commands.stream.StringStream;
+import revxrsal.commands.util.Classes;
 
 import java.util.Collection;
 
@@ -75,6 +76,9 @@ public ParameterNodeImpl(
         this.isOptional = isOptional;
         this.switchAnn = parameter.getAnnotation(Switch.class);
         this.flagAnn = parameter.getAnnotation(Flag.class);
+        if (isSwitch() && Classes.wrap(type()) != Boolean.class) {
+            throw new IllegalArgumentException("@Switch can only be used on boolean types!");
+        }
         if (isSwitch() && isFlag()) {
             throw new IllegalArgumentException("A parameter cannot have @Switch and @Flag at the same time!");
         }