CustomNPC-Plus/src/main/java/kamkeel/npcs/command/CommandKamkeelBase.java

package kamkeel.npcs.command;

import net.minecraft.command.CommandBase;
import net.minecraft.command.CommandException;
import net.minecraft.command.ICommandSender;
import net.minecraft.entity.player.EntityPlayer;
import noppes.npcs.CustomNpcsPermissions;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.lang.reflect.Method;
import java.util.*;

public abstract class CommandKamkeelBase extends CommandBase {
    public Map<String, Method> subcommands = new HashMap<String, Method>();
    public Map<String, CommandKamkeelBase> nestedCommands = new HashMap<String, CommandKamkeelBase>();

    public CommandKamkeelBase() {
        for (Method m : this.getClass().getDeclaredMethods()) {
            SubCommand sc = m.getAnnotation(SubCommand.class);
            if (sc != null) {
                String name = sc.name();
                if (name.equals(""))
                    name = m.getName();
                subcommands.put(name.toLowerCase(), m);
            }
        }
    }

    /**
     * Register a nested command class as a subcommand.
     * The nested command's getCommandName() becomes the subcommand name.
     */
    public void registerNestedCommand(CommandKamkeelBase command) {
        nestedCommands.put(command.getCommandName().toLowerCase(), command);
    }

    @Override
    public void processCommand(ICommandSender sender, String[] args) throws CommandException {

    }

    @Override
    public String getCommandUsage(ICommandSender sender) {
        return getDescription();
    }

    public abstract String getDescription();

    /**
     * @return Should return a string in the format of <arg> <arg2> <arg3> [arg4] where <> is a required parameter and [] optional
     */
    public String getUsage() {
        return "";
    }

    public boolean runSubCommands() {
        return !subcommands.isEmpty() || !nestedCommands.isEmpty();
    }

    @Retention(value = RetentionPolicy.RUNTIME)
    @Target(ElementType.METHOD)
    public @interface SubCommand {

        String name() default "";

        /**
         * @return Should return a string in the format of <arg> <arg2> <arg3> [arg4] where <> is a required parameter and [] optional
         */
        String usage() default "";

        String desc();

        int permission() default 2;
    }

    public void processSubCommand(ICommandSender sender, String command, String[] args) throws CommandException {
        // Check nested commands first
        CommandKamkeelBase nested = nestedCommands.get(command.toLowerCase());
        if (nested != null) {
            if (!canSendNestedCommand(sender, nested)) {
                throw new CommandException("You are not allowed to use this command: " + command);
            }

            if (args.length == 0 || !nested.runSubCommands()) {
                nested.processCommand(sender, args);
            } else {
                nested.processSubCommand(sender, args[0], Arrays.copyOfRange(args, 1, args.length));
            }
            return;
        }

        // Fall back to method subcommands
        Method m = subcommands.get(command.toLowerCase());
        if (m == null)
            throw new CommandException("Unknown subcommand " + command);

        SubCommand sc = m.getAnnotation(SubCommand.class);
        if (!canSendCommand(sender, sc, command))
            throw new CommandException("You are not allowed to use this command: " + command);

        canRun(sender, sc.usage(), args);
        try {
            m.invoke(this, sender, args);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void canRun(ICommandSender sender, String usage, String[] args) throws CommandException {
        String[] np = usage.split(" ");
        List<String> required = new ArrayList<String>();
        for (int i = 0; i < np.length; i++) {
            String command = np[i];
            if (command.startsWith("<")) {
                required.add(command);
            }
            if (command.equals("<player>") && args.length > i) {
                CommandBase.getPlayer(sender, args[i]); //throws PlayerNotFoundException if no player is found
            }

        }
        if (args.length < required.size()) {
            throw new CommandException("Missing parameter: " + required.get(args.length));
        }
    }

    public int getRequiredPermissionLevel() {
        return 2;
    }

    public String getSubCommandPermission(String subCommand) {
        return "cnpc.kamkeel." + getCommandName().toLowerCase() + "." + subCommand.toLowerCase();
    }

    public String getSubUniversalPermission() {
        return "cnpc.kamkeel." + getCommandName().toLowerCase() + "*";
    }

    public boolean canSendCommand(ICommandSender sender, SubCommand command, String subCommand) {
        if (sender.canCommandSenderUseCommand(command.permission(), getSubUniversalPermission())) {
            return true;
        }

        if (sender.canCommandSenderUseCommand(command.permission(), getSubCommandPermission(subCommand))) {
            return true;
        }

        if (sender instanceof EntityPlayer) {
            return CustomNpcsPermissions.hasCustomPermission((EntityPlayer) sender, getSubCommandPermission(subCommand));
        }

        return false;
    }

    public boolean canSendNestedCommand(ICommandSender sender, CommandKamkeelBase nested) {
        int permission = nested.getRequiredPermissionLevel();
        String subCommand = nested.getCommandName();

        if (sender.canCommandSenderUseCommand(permission, getSubUniversalPermission())) {
            return true;
        }

        if (sender.canCommandSenderUseCommand(permission, getSubCommandPermission(subCommand))) {
            return true;
        }

        if (sender instanceof EntityPlayer) {
            return CustomNpcsPermissions.hasCustomPermission((EntityPlayer) sender, getSubCommandPermission(subCommand));
        }

        return false;
    }

    // =========================================
    // Tab Completion Helpers
    // =========================================

    /**
     * Get all subcommand names (both method-based and nested)
     */
    public String[] getAllSubCommandNames() {
        Set<String> names = new HashSet<>();
        names.addAll(subcommands.keySet());
        names.addAll(nestedCommands.keySet());
        return names.toArray(new String[0]);
    }

    /**
     * Get tab completions for nested command at given depth
     */
    public List getNestedTabCompletions(ICommandSender sender, String[] args) {
        if (args.length == 0) return null;

        String subCmd = args[0].toLowerCase();
        CommandKamkeelBase nested = nestedCommands.get(subCmd);

        if (nested != null && args.length > 1) {
            // Delegate to nested command
            String[] nestedArgs = Arrays.copyOfRange(args, 1, args.length);
            return nested.addTabCompletionOptions(sender, nestedArgs);
        }

        return null;
    }
}