From 66ae2a837c9d606388a1b2ba871899c8e590c6f1 Mon Sep 17 00:00:00 2001
From: yoyosource <yoyosource@nidido.de>
Date: Thu, 8 Jul 2021 22:46:32 +0200
Subject: [PATCH] CMDoS because Brigadier

---
 .../steamwar/command/SWCommandBrigadier.java  | 50 +++++++++----------
 1 file changed, 24 insertions(+), 26 deletions(-)

diff --git a/SpigotCore_Main/src/de/steamwar/command/SWCommandBrigadier.java b/SpigotCore_Main/src/de/steamwar/command/SWCommandBrigadier.java
index 9e31c8b..e928f15 100644
--- a/SpigotCore_Main/src/de/steamwar/command/SWCommandBrigadier.java
+++ b/SpigotCore_Main/src/de/steamwar/command/SWCommandBrigadier.java
@@ -47,6 +47,29 @@ class SWCommandBrigadier implements SWCommand.SWCommandInterface {
 
     protected SWCommandBrigadier(SWCommand swCommand, String command, String... aliases) {
         this.swCommand = swCommand;
+        this.command = new Command(command, "", "/" + command, Arrays.asList(aliases)) {
+            @Override
+            public boolean execute(CommandSender sender, String alias, String[] args) {
+                if (commandList.stream().anyMatch(s -> s.invoke(sender, args))) return false;
+                commandHelpList.stream().anyMatch(s -> s.invoke(sender, args));
+                return false;
+            }
+
+            @Override
+            public List<String> tabComplete(CommandSender sender, String alias, String[] args) throws IllegalArgumentException {
+                String string = args[args.length - 1].toLowerCase();
+                return commandList.stream()
+                        .map(s -> s.tabComplete(sender, args))
+                        .filter(Objects::nonNull)
+                        .flatMap(Collection::stream)
+                        .filter(s -> !s.isEmpty())
+                        .filter(s -> s.toLowerCase().startsWith(string))
+                        .collect(Collectors.toList());
+            }
+        };
+        unregister();
+        register();
+
         SWCommandUtils.createList(swCommand, commandList, commandHelpList, localTypeMapper, this::createSubCommand);
         Stream.of(commandList, commandHelpList).flatMap(List::stream).forEach(subCommand -> {
             register(command, subCommand, command, aliases);
@@ -82,10 +105,7 @@ class SWCommandBrigadier implements SWCommand.SWCommandInterface {
                 }
                 argumentBuilder = requiredArgumentBuilder;
                 if (i == current.arguments.length - 1) {
-                    argumentBuilder.executes(commandContext -> {
-                        current.invoke((CommandSender) commandContext.getCommand(), commandContext.getInput().split(" "));
-                        return 0;
-                    });
+                    argumentBuilder.executes(commandContext -> 1);
                 }
             }
             if (current.varArgType != null) {
@@ -117,28 +137,6 @@ class SWCommandBrigadier implements SWCommand.SWCommandInterface {
         if (subCommand.argumentNode == null) {
             return;
         }
-        if (subCommand.normalTabCompleteNeeded && command == null) {
-            this.command = new Command(command, "", "/" + command, Arrays.asList(aliases)) {
-                @Override
-                public boolean execute(CommandSender sender, String alias, String[] args) {
-                    if (commandList.stream().anyMatch(s -> s.invoke(sender, args))) return false;
-                    commandHelpList.stream().anyMatch(s -> s.invoke(sender, args));
-                    return false;
-                }
-
-                @Override
-                public List<String> tabComplete(CommandSender sender, String alias, String[] args) throws IllegalArgumentException {
-                    String string = args[args.length - 1].toLowerCase();
-                    return commandList.stream()
-                            .map(s -> s.tabComplete(sender, args))
-                            .filter(Objects::nonNull)
-                            .flatMap(Collection::stream)
-                            .filter(s -> !s.isEmpty())
-                            .filter(s -> s.toLowerCase().startsWith(string))
-                            .collect(Collectors.toList());
-                }
-            };
-        }
         literalArgumentBuilder.then(subCommand.argumentNode);
         SWCommand.dispatcher.register(literalArgumentBuilder);
     }