Added "set" command to - group

DenizenScript · Jul 15, 2015 · 3171b2b · 3171b2b
1 parent 89845ff
commit 3171b2b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/src/main/java/net/aufdemrand/denizen/scripts/commands/BukkitCommandRegistry.java b/src/main/java/net/aufdemrand/denizen/scripts/commands/BukkitCommandRegistry.java
@@ -1230,10 +1230,10 @@ public void registerCoreMembers() {
 
         // <--[command]
         // @Name Group
-        // @Syntax group [add/remove] [<group>] (<world>)
+        // @Syntax group [add/remove/set] [<group>] (<world>)
         // @Required 2
         // @Stable TODO: Document Command Details
-        // @Short Adds a player to or removes a player from a permissions group.
+        // @Short Adds a player to or removes a player from or sets a players permissions group.
         // @Author GnomeffinWay
         // @Group player
         // @Description
@@ -1246,7 +1246,7 @@ public void registerCoreMembers() {
         // TODO: Document Command Details
         // -->
         registerCoreMember(GroupCommand.class,
-                "GROUP", "group [add/remove] [<group>] (<world>)", 2);
+                "GROUP", "group [add/remove/set] [<group>] (<world>)", 2);
 
 
         // <--[command]

diff --git a/src/main/java/net/aufdemrand/denizen/scripts/commands/player/GroupCommand.java b/src/main/java/net/aufdemrand/denizen/scripts/commands/player/GroupCommand.java
@@ -15,7 +15,7 @@
 
 public class GroupCommand extends AbstractCommand {
 
-    private enum Action {ADD, REMOVE}
+    private enum Action {ADD, REMOVE, SET}
 
     @Override
     public void parseArgs(ScriptEntry scriptEntry) throws InvalidArgumentsException {
@@ -79,7 +79,11 @@ public void execute(ScriptEntry scriptEntry) throws CommandExecutionException {
                 else
                     Depends.permissions.playerRemoveGroup((bukkitWorld == null ? null : bukkitWorld.getName()), player, group.asString());
                 return;
+            case SET:
+                for (String grp : Depends.permissions.getPlayerGroups((bukkitWorld == null ? null : bukkitWorld.getName()), player)) {
+                    Depends.permissions.playerRemoveGroup((bukkitWorld == null ? null : bukkitWorld.getName()), player, grp);
+                }
+                Depends.permissions.playerAddGroup((bukkitWorld == null ? null : bukkitWorld.getName()), player, group.asString());
         }
-
     }
 }