✨ Command unregister feature.

MarcusSlover · Aug 1, 2024 · 8ffe17f · 8ffe17f
1 parent bc35f66
commit 8ffe17f
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/src/main/java/com/marcusslover/plus/lib/command/CommandManager.java b/src/main/java/com/marcusslover/plus/lib/command/CommandManager.java
@@ -77,6 +77,28 @@ public boolean execute(@NotNull CommandSender sender, @NotNull String commandLab
         return this;
     }
 
+    /**
+     * Unregister a command.
+     * @param command the command
+     */
+    public void unregister(@NotNull ICommand command) {
+        Command commandAnnotation = this.getCommandAnnotation(command);
+        if (commandAnnotation == null) {
+            return;
+        }
+        String name = commandAnnotation.name();
+        CommandMap commandMap = Bukkit.getCommandMap();
+        List<String> keys = new ArrayList<>();
+        commandMap.getKnownCommands().forEach((key, value) -> {
+            if (value.getName().equals(name)) {
+                keys.add(key);
+            }
+        });
+        for (String key : keys) {
+            commandMap.getKnownCommands().remove(key);
+        }
+    }
+
     @NotNull
     public Set<org.bukkit.command.Command> getCommands() {
         return this.commandSet;