Arg length fix

espidev · Apr 21, 2019 · 8526a69 · 8526a69
1 parent d3dce0e
commit 8526a69
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/main/java/me/vik1395/ProtectionStones/commands/ArgGet.java b/src/main/java/me/vik1395/ProtectionStones/commands/ArgGet.java
@@ -14,6 +14,11 @@ public static boolean argumentGet(Player p, String[] args) {
             return true;
         }
 
+        if (args.length != 2) {
+            p.sendMessage(PSL.GET_HELP.msg());
+            return true;
+        }
+
         // check if argument is valid block
         ConfigProtectBlock cp = ProtectionStones.getProtectBlockFromName(args[1]);
         if (cp == null) {

diff --git a/src/main/java/me/vik1395/ProtectionStones/commands/ArgGive.java b/src/main/java/me/vik1395/ProtectionStones/commands/ArgGive.java
@@ -13,6 +13,11 @@ public static boolean argumentGive(Player p, String[] args) {
             return true;
         }
 
+        if (args.length != 3) {
+            p.sendMessage(PSL.GIVE_HELP.msg());
+            return true;
+        }
+
         // check if argument is valid block
         ConfigProtectBlock cp = ProtectionStones.getProtectBlockFromName(args[1]);
         if (cp == null) {