优化tab补全

Lings-MC · Sep 10, 2022 · c0695b2 · c0695b2
1 parent 32c85dc
commit c0695b2
Showing 1 changed file with 19 additions and 5 deletions.
diff --git a/src/main/java/cn/lingsmc/lingshttputils/commands/TabComplete.java b/src/main/java/cn/lingsmc/lingshttputils/commands/TabComplete.java
@@ -4,7 +4,7 @@
 import org.bukkit.command.CommandSender;
 import org.jetbrains.annotations.NotNull;
 
-import java.util.Arrays;
+import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
 
@@ -20,12 +20,26 @@ public List<String> onTabComplete(@NotNull CommandSender sender, @NotNull Comman
         final String[] root = new String[]{"workers", "help", "reload"};
         final String[] worker = new String[]{"start", "stop"};
 
-        if (args.length <= 1) {
-            return Arrays.asList(root);
+        if (args.length == 1) {
+            List<String> res = new ArrayList<>();
+            for (String val : root) {
+                if (val.contains(args[0])) {
+                    res.add(val);
+                }
+            }
+            return res;
         }
-        if (args.length <= 2 && root[0].equalsIgnoreCase(args[0])) {
-            return Arrays.asList(worker);
+
+        if (args.length == 2 && root[0].equalsIgnoreCase(args[0])) {
+            List<String> res = new ArrayList<>();
+            for (String val : worker) {
+                if (val.contains(args[1])) {
+                    res.add(val);
+                }
+            }
+            return res;
         }
+
         return Collections.emptyList();
     }
 }