SubCommandHandler: Fix tab complete handling

xDec0de · Nov 2, 2024 · 3cd85bb · 3cd85bb
1 parent b7de8e3
commit 3cd85bb
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/shared/src/main/java/net/codersky/mcutils/cmd/SubCommandHandler.java b/shared/src/main/java/net/codersky/mcutils/cmd/SubCommandHandler.java
@@ -2,6 +2,7 @@
 
 import org.jetbrains.annotations.NotNull;
 
+import java.util.ArrayList;
 import java.util.Collections;
 import java.util.HashSet;
 import java.util.List;
@@ -38,7 +39,18 @@ public boolean onCommand(@NotNull MCCommand<P, S> mainCmd, @NotNull S sender, @N
 	}
 
 	public List<String> onTab(@NotNull MCCommand<P, S> mainCommand, @NotNull S sender, @NotNull String[] args) {
-		return onUsedCommand(mainCommand, sender, args, (cmd, newArgs) -> cmd.onTab(sender, newArgs), List.of(), false);
+		return onUsedCommand(mainCommand, sender, args, (cmd, newArgs) -> {
+			if (subCommands.isEmpty() || args.length > 1)
+				return cmd.onTab(sender, newArgs);
+			final List<String> cmdTabs = cmd.onTab(sender, newArgs);
+			final List<String> tabs = new ArrayList<>(subCommands.size() + cmdTabs.size());
+			tabs.addAll(cmdTabs);
+			for (MCCommand<P, S> subCmd : subCommands) {
+				tabs.add(subCmd.getName());
+				tabs.addAll(subCmd.getAliases());
+			}
+			return tabs;
+		}, List.of(), false);
 	}
 
 	@SafeVarargs