Separate Minecraft dependant and non-dependant code

1 files changed, 11 insertions, 317 deletions

diff --git a/src/main/java/cc/polyfrost/oneconfig/utils/commands/CommandManager.java b/src/main/java/cc/polyfrost/oneconfig/utils/commands/CommandManager.java
index 004f427..16a2f8d 100644
--- a/src/main/java/cc/polyfrost/oneconfig/utils/commands/CommandManager.java
+++ b/src/main/java/cc/polyfrost/oneconfig/utils/commands/CommandManager.java
@@ -1,19 +1,10 @@
 package cc.polyfrost.oneconfig.utils.commands;
 
-import cc.polyfrost.oneconfig.utils.commands.annotations.*;
+import cc.polyfrost.oneconfig.utils.commands.annotations.Command;
+import cc.polyfrost.oneconfig.utils.commands.annotations.Main;
+import cc.polyfrost.oneconfig.utils.commands.annotations.SubCommand;
 import cc.polyfrost.oneconfig.utils.commands.arguments.*;
-import cc.polyfrost.oneconfig.libs.universal.ChatColor;
-import cc.polyfrost.oneconfig.libs.universal.UChat;
-import net.minecraft.command.CommandBase;
-import net.minecraft.command.ICommandSender;
-import net.minecraft.util.BlockPos;
-import net.minecraftforge.client.ClientCommandHandler;
-import org.apache.commons.lang3.tuple.ImmutablePair;
-import org.apache.commons.lang3.tuple.ImmutableTriple;
-import org.apache.commons.lang3.tuple.Pair;
-import org.apache.commons.lang3.tuple.Triple;
 
-import java.lang.reflect.InvocationTargetException;
 import java.lang.reflect.Method;
 import java.lang.reflect.Modifier;
 import java.lang.reflect.Parameter;
@@ -25,11 +16,12 @@ import java.util.*;
  * @see Command
  */
 public class CommandManager {
+    private static final PlatformCommandManager platform = ServiceLoader.load(PlatformCommandManager.class, PlatformCommandManager.class.getClassLoader()).iterator().next();
     public static final CommandManager INSTANCE = new CommandManager();
-    private static final String NOT_FOUND_TEXT = "Command not found! Type /@ROOT_COMMAND@ help for help.";
-    private static final String TOO_MANY_PARAMETERS = "There were too many / little parameters for this command! Type /@ROOT_COMMAND@ help for help.";
-    private static final String METHOD_RUN_ERROR = "Error while running @ROOT_COMMAND@ method! Please report this to the developer.";
-    private final HashMap<Class<?>, ArgumentParser<?>> parsers = new HashMap<>();
+    static final String NOT_FOUND_TEXT = "Command not found! Type /@ROOT_COMMAND@ help for help.";
+    static final String TOO_MANY_PARAMETERS = "There were too many / little parameters for this command! Type /@ROOT_COMMAND@ help for help.";
+    static final String METHOD_RUN_ERROR = "Error while running @ROOT_COMMAND@ method! Please report this to the developer.";
+    final HashMap<Class<?>, ArgumentParser<?>> parsers = new HashMap<>();
 
     private CommandManager() {
         addParser(new StringParser());
@@ -79,305 +71,7 @@ public class CommandManager {
                 }
             }
             addToInvokers(clazz.getDeclaredClasses(), root);
-            ClientCommandHandler.instance.registerCommand(new CommandBase() {
-                @Override
-                public String getCommandName() {
-                    return annotation.value();
-                }
-
-                @Override
-                public String getCommandUsage(ICommandSender sender) {
-                    return "/" + annotation.value();
-                }
-
-                @Override
-                public void processCommand(ICommandSender sender, String[] args) {
-                    handleCommand(root, annotation, args);
-                }
-
-                @Override
-                public int getRequiredPermissionLevel() {
-                    return -1;
-                }
-
-                @Override
-                public List<String> addTabCompletionOptions(ICommandSender sender, String[] args, BlockPos pos) {
-                    return handleTabCompletion(root, annotation, args);
-                }
-            });
-        }
-    }
-
-    private void handleCommand(InternalCommand root, Command annotation, String[] args) {
-        if (args.length == 0) {
-            if (!root.invokers.isEmpty()) {
-                try {
-                    root.invokers.get(0).method.invoke(null);
-                } catch (IllegalAccessException | InvocationTargetException | IllegalArgumentException |
-                        ExceptionInInitializerError e) {
-                    e.printStackTrace();
-                    UChat.chat(ChatColor.RED.toString() + ChatColor.BOLD + METHOD_RUN_ERROR);
-                }
-            }
-        } else {
-            if (annotation.helpCommand() && args[0].equalsIgnoreCase("help")) {
-                UChat.chat(sendHelpCommand(root));
-            } else {
-                List<InternalCommand.InternalCommandInvoker> commands = new ArrayList<>();
-                int depth = 0;
-                for (InternalCommand command : root.children) {
-                    int newDepth = loopThroughCommands(commands, 0, command, args);
-                    if (newDepth != -1) {
-                        depth = newDepth;
-                        break;
-                    }
-                }
-                if (commands.isEmpty()) {
-                    if (depth == -2) {
-                        UChat.chat(ChatColor.RED.toString() + ChatColor.BOLD + TOO_MANY_PARAMETERS.replace("@ROOT_COMMAND@", annotation.value()));
-                    } else {
-                        UChat.chat(ChatColor.RED.toString() + ChatColor.BOLD + NOT_FOUND_TEXT.replace("@ROOT_COMMAND@", annotation.value()));
-                    }
-                } else {
-                    List<CustomError> errors = new ArrayList<>();
-                    for (InternalCommand.InternalCommandInvoker invoker : commands) {
-                        try {
-                            List<Object> params = getParametersForInvoker(invoker, depth, args);
-                            if (params.size() == 1) {
-                                Object first = params.get(0);
-                                if (first instanceof CustomError) {
-                                    errors.add((CustomError) first);
-                                    continue;
-                                }
-                            }
-                            invoker.method.invoke(null, params.toArray());
-                            return;
-                        } catch (Exception e) {
-                            e.printStackTrace();
-                            UChat.chat(ChatColor.RED.toString() + ChatColor.BOLD + METHOD_RUN_ERROR);
-                            return;
-                        }
-                    }
-                    //noinspection ConstantConditions
-                    if (!errors.isEmpty()) {
-                        UChat.chat(ChatColor.RED.toString() + ChatColor.BOLD + "Multiple errors occurred:");
-                        for (CustomError error : errors) {
-                            UChat.chat("    " + ChatColor.RED + ChatColor.BOLD + error.message);
-                        }
-                    }
-                }
-            }
-        }
-    }
-
-    private List<String> handleTabCompletion(InternalCommand root, Command annotation, String[] args) {
-        try {
-            Set<Pair<InternalCommand.InternalCommandInvoker, Integer>> commands = new HashSet<>();
-            for (InternalCommand command : root.children) {
-                loopThroughCommandsTab(commands, 0, command, args);
-            }
-            if (!commands.isEmpty() || annotation.helpCommand()) {
-                List<Triple<InternalCommand.InternalCommandInvoker, Integer, Integer>> validCommands = new ArrayList<>(); // command, depth, and all processed params
-                for (Pair<InternalCommand.InternalCommandInvoker, Integer> pair : commands) {
-                    InternalCommand.InternalCommandInvoker invoker = pair.getLeft();
-                    int depth = pair.getRight();
-                    int currentParam = 0;
-                    boolean failed = false;
-                    while (args.length - depth > 1) {
-                        Parameter param = invoker.method.getParameters()[currentParam];
-                        if (param.isAnnotationPresent(Greedy.class) && currentParam + 1 != invoker.parameterTypes.length) {
-                            failed = true;
-                            break;
-                        }
-                        ArgumentParser<?> parser = parsers.get(param.getType());
-                        if (parser == null) {
-                            failed = true;
-                            break;
-                        }
-                        try {
-                            Arguments arguments = new Arguments(Arrays.copyOfRange(args, depth, args.length), param.isAnnotationPresent(Greedy.class));
-                            if (parser.parse(arguments) != null) {
-                                depth += arguments.getPosition();
-                                currentParam++;
-                            } else {
-                                failed = true;
-                                break;
-                            }
-                        } catch (Exception e) {
-                            failed = true;
-                            break;
-                        }
-                    }
-                    if (!failed) {
-                        validCommands.add(new ImmutableTriple<>(pair.getLeft(), depth, currentParam));
-                    }
-                }
-                if (!validCommands.isEmpty() || annotation.helpCommand()) {
-                    Set<String> completions = new HashSet<>();
-                    for (Triple<InternalCommand.InternalCommandInvoker, Integer, Integer> valid : validCommands) {
-                        if (valid.getMiddle() == args.length) {
-                            completions.add(valid.getLeft().name);
-                            completions.addAll(Arrays.asList(valid.getLeft().aliases));
-                            continue;
-                        }
-                        if (valid.getRight() + 1 > valid.getLeft().parameterTypes.length) continue;
-                        Parameter param = valid.getLeft().method.getParameters()[valid.getRight()];
-                        if (param.isAnnotationPresent(Greedy.class) && valid.getRight() + 1 != valid.getLeft().parameterTypes.length) {
-                            continue;
-                        }
-                        ArgumentParser<?> parser = parsers.get(param.getType());
-                        if (parser == null) {
-                            continue;
-                        }
-                        try {
-                            Arguments arguments = new Arguments(Arrays.copyOfRange(args, valid.getMiddle(), args.length), param.isAnnotationPresent(Greedy.class));
-                            List<String> possibleCompletions = parser.complete(arguments, param);
-                            if (possibleCompletions != null) {
-                                completions.addAll(possibleCompletions);
-                            }
-                        } catch (Exception ignored) {
-
-                        }
-                    }
-                    if (args.length == 1 && annotation.helpCommand()) {
-                        if ("help".startsWith(args[0].toLowerCase(Locale.ENGLISH))) {
-                            completions.add("help");
-                        }
-                    }
-                    return new ArrayList<>(completions);
-                }
-            }
-        } catch (Exception ignored) {
-
-        }
-        return null;
-    }
-
-    private List<Object> getParametersForInvoker(InternalCommand.InternalCommandInvoker invoker, int depth, String[] args) {
-        List<Object> parameters = new ArrayList<>();
-        int processed = depth;
-        int currentParam = 0;
-        while (processed < args.length) {
-            Parameter param = invoker.method.getParameters()[currentParam];
-            if (param.isAnnotationPresent(Greedy.class) && currentParam + 1 != invoker.parameterTypes.length) {
-                return Collections.singletonList(new CustomError("Parsing failed: Greedy parameter must be the last one."));
-            }
-            ArgumentParser<?> parser = parsers.get(param.getType());
-            if (parser == null) {
-                return Collections.singletonList(new CustomError("No parser for " + invoker.method.getParameterTypes()[currentParam].getSimpleName() + "! Please report this to the mod author."));
-            }
-            try {
-                Arguments arguments = new Arguments(Arrays.copyOfRange(args, processed, args.length), param.isAnnotationPresent(Greedy.class));
-                try {
-                    Object a = parser.parse(arguments);
-                    if (a != null) {
-                        parameters.add(a);
-                        processed += arguments.getPosition();
-                        currentParam++;
-                    } else {
-                        return Collections.singletonList(new CustomError("Failed to parse " + param.getType().getSimpleName() + "! Please report this to the mod author."));
-                    }
-                } catch (Exception e) {
-                    return Collections.singletonList(new CustomError("A " + e.getClass().getSimpleName() + " has occured while try to parse " + param.getType().getSimpleName() + "! Please report this to the mod author."));
-                }
-            } catch (Exception e) {
-                return Collections.singletonList(new CustomError("A " + e.getClass().getSimpleName() + " has occured while try to parse " + param.getType().getSimpleName() + "! Please report this to the mod author."));
-            }
-        }
-        return parameters;
-    }
-
-    private int loopThroughCommands(List<InternalCommand.InternalCommandInvoker> commands, int depth, InternalCommand command, String[] args) {
-        int nextDepth = depth + 1;
-        boolean thatOneSpecialError = false;
-        if (command.isValid(args[depth], false)) {
-            for (InternalCommand child : command.children) {
-                if (args.length > nextDepth && child.isValid(args[nextDepth], false)) {
-                    int result = loopThroughCommands(commands, nextDepth, child, args);
-                    if (result > -1) {
-                        return result;
-                    } else if (result == -2) {
-                        thatOneSpecialError = true;
-                    }
-                }
-            }
-            boolean added = false;
-            for (InternalCommand.InternalCommandInvoker invoker : command.invokers) {
-                if (args.length - nextDepth == invoker.parameterTypes.length) {
-                    commands.add(invoker);
-                    added = true;
-                } else {
-                    thatOneSpecialError = true;
-                }
-            }
-            if (added) {
-                return nextDepth;
-            }
-        }
-        return thatOneSpecialError ? -2 : -1;
-    }
-
-    private void loopThroughCommandsTab(Set<Pair<InternalCommand.InternalCommandInvoker, Integer>> commands, int depth, InternalCommand command, String[] args) {
-        int nextDepth = depth + 1;
-        if (command.isValid(args[depth], args.length == nextDepth)) {
-            if (args.length != nextDepth) {
-                for (InternalCommand child : command.children) {
-                    if (child.isValid(args[nextDepth], args.length == nextDepth + 1)) {
-                        loopThroughCommandsTab(commands, nextDepth, child, args);
-                    }
-                }
-            }
-            for (InternalCommand.InternalCommandInvoker invoker : command.invokers) {
-                commands.add(new ImmutablePair<>(invoker, nextDepth));
-            }
-        }
-    }
-
-    //TODO: someone make the help command actually look nice lmao
-    private String sendHelpCommand(InternalCommand root) {
-        StringBuilder builder = new StringBuilder();
-        builder.append(ChatColor.GOLD).append("Help for ").append(ChatColor.BOLD).append(root.name).append(ChatColor.RESET).append(ChatColor.GOLD).append(":\n");
-        if (!root.description.isEmpty()) {
-            builder.append("\n").append(ChatColor.GOLD).append("Description: ").append(ChatColor.BOLD).append(root.description);
-        }
-        for (InternalCommand command : root.children) {
-            runThroughCommandsHelp(root.name, command, builder);
-        }
-        builder.append("\n").append(ChatColor.GOLD).append("Aliases: ").append(ChatColor.BOLD);
-        int index = 0;
-        for (String alias : root.aliases) {
-            ++index;
-            builder.append(alias).append(index < root.aliases.length ? ", " : "");
-        }
-        builder.append("\n");
-        return builder.toString();
-    }
-
-    private void runThroughCommandsHelp(String append, InternalCommand command, StringBuilder builder) {
-        if (!command.invokers.isEmpty()) {
-            Class<?> declaringClass = command.invokers.get(0).method.getDeclaringClass();
-            if (declaringClass.isAnnotationPresent(SubCommand.class)) {
-                String description = declaringClass.getAnnotation(SubCommand.class).description();
-                if (!description.isEmpty()) {
-                    builder.append("\n").append(ChatColor.GOLD).append("Description: ").append(ChatColor.BOLD).append(description);
-                }
-            }
-        }
-        for (InternalCommand.InternalCommandInvoker invoker : command.invokers) {
-            builder.append("\n").append(ChatColor.GOLD).append("/").append(append).append(" ").append(command.name);
-            for (Parameter parameter : invoker.method.getParameters()) {
-                String name = parameter.getName();
-                if (parameter.isAnnotationPresent(Name.class)) {
-                    name = parameter.getAnnotation(Name.class).value();
-                }
-                builder.append(" <").append(name).append(">");
-            }
-            if (!command.description.trim().isEmpty()) {
-                builder.append(": ").append(ChatColor.BOLD).append(command.description);
-            }
-        }
-        for (InternalCommand subCommand : command.children) {
-            runThroughCommandsHelp(append + " " + command.name, subCommand, builder);
+            platform.createCommand(root, annotation);
         }
     }
 
@@ -397,7 +91,7 @@ public class CommandManager {
         }
     }
 
-    private static class CustomError {
+    static class CustomError {
         public String message;
 
         public CustomError(String message) {
@@ -405,7 +99,7 @@ public class CommandManager {
         }
     }
 
-    private static class InternalCommand {
+    static class InternalCommand {
         public final String name;
         public final String[] aliases;
         public final String description;