LibsDisguises/src/me/libraryaddict/disguise/BaseDisguiseCommand.java

package me.libraryaddict.disguise;

import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;

import me.libraryaddict.disguise.disguisetypes.AnimalColor;
import me.libraryaddict.disguise.disguisetypes.Disguise;
import me.libraryaddict.disguise.disguisetypes.DisguiseType;
import me.libraryaddict.disguise.disguisetypes.MiscDisguise;
import me.libraryaddict.disguise.disguisetypes.MobDisguise;
import me.libraryaddict.disguise.disguisetypes.PlayerDisguise;
import org.bukkit.ChatColor;
import org.bukkit.command.CommandExecutor;
import org.bukkit.command.CommandSender;
import org.bukkit.entity.Horse.Color;
import org.bukkit.entity.Horse.Style;
import org.bukkit.entity.Ocelot.Type;
import org.bukkit.entity.Villager.Profession;
import org.bukkit.inventory.ItemStack;
import org.bukkit.permissions.PermissionAttachmentInfo;
import org.bukkit.potion.PotionEffectType;

public abstract class BaseDisguiseCommand implements CommandExecutor {
    protected ArrayList<String> getAllowedDisguises(CommandSender sender) {
        String permissionNode = "libsdisguises." + getClass().getSimpleName().replace("Command", "").toLowerCase() + ".";
        return getAllowedDisguises(sender, permissionNode);
    }

    protected ArrayList<String> getAllowedDisguises(CommandSender sender, String permissionNode) {
        ArrayList<String> names = new ArrayList<String>();
        ArrayList<String> forbiddenDisguises = new ArrayList<String>();
        for (PermissionAttachmentInfo permission : sender.getEffectivePermissions()) {
            String perm = permission.getPermission().toLowerCase();
            if (perm.startsWith(permissionNode)) {
                perm = perm.substring(permissionNode.length());
                for (DisguiseType type : DisguiseType.values()) {
                    String name = type.name().toLowerCase();
                    if (perm.split("\\.")[0].equals("*") && permission.getValue()) {
                        if (!names.contains(name))
                            names.add(name);
                    } else if (perm.split("\\.")[0].equals(name)) {
                        if (permission.getValue()) {
                            if (!names.contains(name))
                                names.add(name);
                        } else {
                            forbiddenDisguises.add(name);
                        }
                    }
                }
            }
        }
        // This does the disguises for ops.
        for (DisguiseType type : DisguiseType.values()) {
            if (!names.contains(type.name().toLowerCase())) {
                if (sender.hasPermission(permissionNode + "*")
                        || sender.hasPermission(permissionNode + type.name().toLowerCase())) {
                    names.add(type.name().toLowerCase());
                }
            }
        }
        names.removeAll(forbiddenDisguises);
        Collections.sort(names, String.CASE_INSENSITIVE_ORDER);
        return names;
    }

    protected boolean isDouble(String string) {
        try {
            Float.parseFloat(string);
            return true;
        } catch (Exception ex) {
            return false;
        }
    }

    protected boolean isNumeric(String string) {
        try {
            Integer.parseInt(string);
            return true;
        } catch (Exception ex) {
            return false;
        }
    }

    /**
     * Returns null if they have all perms. Else they can only use the options in the returned
     */
    protected HashSet<HashSet<String>> getPermissions(CommandSender sender, String disguiseType) {
        HashSet<HashSet<String>> perms = new HashSet<HashSet<String>>();
        String permNode = "libsdisguises." + getClass().getSimpleName().replace("Command", "").toLowerCase() + ".";
        for (PermissionAttachmentInfo permission : sender.getEffectivePermissions()) {
            if (permission.getValue()) {
                String s = permission.getPermission().toLowerCase();
                if (s.startsWith(permNode + disguiseType) || s.startsWith(permNode + "*")) {
                    if (s.startsWith(permNode + disguiseType))
                        s = s.substring((permNode + disguiseType).length());
                    else if (s.startsWith(permNode + "*")) {
                        s = s.substring((permNode + "*").length());
                        if (s.length() == 0)
                            continue;
                    }
                    if (s.length() == 0)
                        return new HashSet<HashSet<String>>();
                    HashSet<String> p = new HashSet<String>();
                    for (String str : s.split("\\.")) {
                        if (str.equals("baby"))
                            str = "setbaby";
                        p.add(str);
                    }
                    perms.add(p);
                }
            }
        }
        return perms;
    }

    /**
     * Returns the disguise if it all parsed correctly. Returns a exception with a complete message if it didn't. The
     * commandsender is purely used for checking permissions. Would defeat the purpose otherwise. To reach this point, the
     * disguise has been feed a proper disguisetype.
     */
    protected Disguise parseDisguise(CommandSender sender, String[] args) throws Exception {
        ArrayList<String> allowedDisguises = getAllowedDisguises(sender);
        if (allowedDisguises.isEmpty()) {
            throw new Exception(ChatColor.RED + "You are forbidden to use this command.");
        }
        if (args.length == 0) {
            sendCommandUsage(sender);
            throw new Exception();
        }
        DisguiseType disguiseType = null;
        for (DisguiseType type : DisguiseType.values()) {
            if (type.getEntityType() == null) {
                continue;
            }
            if (args[0].equalsIgnoreCase(type.name()) || type.name().replace("_", "").equalsIgnoreCase(args[0])) {
                disguiseType = type;
                break;
            }
        }
        if (disguiseType == null) {
            throw new Exception(ChatColor.RED + "Error! The disguise " + ChatColor.GREEN + args[0] + ChatColor.RED
                    + " doesn't exist!");
        }
        if (!allowedDisguises.contains(disguiseType.name().toLowerCase())) {
            throw new Exception(ChatColor.RED + "You are forbidden to use this disguise!");
        }
        HashSet<String> usedOptions = new HashSet<String>();
        Disguise disguise = null;
        // How many args to skip due to the disugise being constructed
        int toSkip = 1;
        // Time to start constructing the disguise.
        // We will need to check between all 3 kinds of disguises

        HashSet<HashSet<String>> optionPermissions = this.getPermissions(sender, disguiseType.name().toLowerCase());
        if (disguiseType.isPlayer()) {// If he is doing a player disguise
            if (args.length == 1) {
                // He needs to give the player name
                throw new Exception(ChatColor.RED + "Error! You need to give a player name!");
            } else {
                // Construct the player disguise
                disguise = new PlayerDisguise(ChatColor.translateAlternateColorCodes('&', args[1]));
                toSkip++;
            }
        } else {
            if (disguiseType.isMob()) { // Its a mob, use the mob constructor
                boolean adult = true;
                if (args.length > 1) {
                    if (args[1].equalsIgnoreCase("true") || args[1].equalsIgnoreCase("false")) {
                        usedOptions.add("setbaby");
                        doCheck(optionPermissions, usedOptions);
                        adult = "false".equalsIgnoreCase(args[1]);
                        sender.sendMessage(ChatColor.RED
                                + "I notice you are using true/false for constructing a mob disguise! This will soon be removed in favor of the simple 'baby'");
                        toSkip++;
                    } else if (args[1].equalsIgnoreCase("baby") || args[1].equalsIgnoreCase("adult")) {
                        usedOptions.add("setbaby");
                        doCheck(optionPermissions, usedOptions);
                        adult = args[1].equalsIgnoreCase("adult");
                        toSkip++;
                    }
                }
                disguise = new MobDisguise(disguiseType, adult);
            } else if (disguiseType.isMisc()) {
                // Its a misc, we are going to use the MiscDisguise constructor.
                int miscId = -1;
                int miscData = -1;
                if (args.length > 1) {
                    // They have defined more arguements!
                    // If the first arg is a number
                    if (isNumeric(args[1])) {
                        miscId = Integer.parseInt(args[1]);
                        toSkip++;
                        // If they also defined a data value
                        if (args.length > 2) {
                            if (isNumeric(args[2])) {
                                miscData = Integer.parseInt(args[2]);
                                toSkip++;
                            }
                        }
                    }
                }
                // Construct the disguise
                disguise = new MiscDisguise(disguiseType, miscId, miscData);
            }
        }
        // Copy strings to their new range
        String[] newArgs = new String[args.length - toSkip];
        for (int i = toSkip; i < args.length; i++) {
            newArgs[i - toSkip] = args[i];
        }
        args = newArgs;
        for (int i = 0; i < args.length; i += 2) {
            String methodName = args[i];
            if (i + 1 >= args.length) {
                throw new Exception(ChatColor.RED + "No value was given for the option " + methodName);
            }
            String valueString = args[i + 1];
            Method methodToUse = null;
            Object value = null;
            for (Method method : disguise.getWatcher().getClass().getMethods()) {
                if (!method.getName().startsWith("get") && method.getName().equalsIgnoreCase(methodName)) {
                    methodToUse = method;
                    methodName = method.getName();
                    Class<?>[] types = method.getParameterTypes();
                    if (types.length == 1) {
                        Class param = types[0];
                        // Parse to number
                        if (int.class == param) {
                            if (isNumeric(valueString)) {
                                value = (int) Integer.parseInt(valueString);
                            } else {
                                throw parseToException("number", valueString, methodName);
                            }
                            // Parse to boolean
                        } else if (float.class == param || double.class == param) {
                            if (isDouble(valueString)) {
                                float obj = Float.parseFloat(valueString);
                                if (param == float.class) {
                                    value = (float) obj;
                                } else if (param == int.class) {
                                    value = (int) obj;
                                } else if (param == double.class) {
                                    value = (double) obj;
                                }
                            } else {
                                throw parseToException("number.0", valueString, methodName);
                            }
                            // Parse to boolean
                        } else if (boolean.class == param) {
                            if (!("true".equalsIgnoreCase(valueString) || "false".equalsIgnoreCase(valueString)))
                                throw parseToException("true/false", valueString, methodName);
                            value = (boolean) "true".equalsIgnoreCase(valueString);
                            // Parse to string
                        } else if (param == String.class) {
                            value = ChatColor.translateAlternateColorCodes('&', valueString);
                            // Parse to animal color
                        } else if (param == AnimalColor.class) {
                            try {
                                value = AnimalColor.valueOf(valueString.toUpperCase());
                            } catch (Exception ex) {
                                throw parseToException("animal color", valueString, methodName);
                            }
                            // Parse to itemstack
                        } else if (param == ItemStack.class) {
                            try {
                                value = parseToItemstack(valueString);
                            } catch (Exception ex) {
                                throw new Exception(String.format(ex.getMessage(), methodName));
                            }
                            // Parse to itemstack array
                        } else if (param == ItemStack[].class) {
                            ItemStack[] items = new ItemStack[4];
                            String[] split = valueString.split(",");
                            if (split.length == 4) {
                                for (int a = 0; a < 4; a++) {
                                    try {
                                        ItemStack item = parseToItemstack(split[a]);
                                        items[a] = item;
                                    } catch (Exception ex) {
                                        throw parseToException("item ID,ID,ID,ID" + ChatColor.RED + " or " + ChatColor.GREEN
                                                + "ID:Data,ID:Data,ID:Data,ID:Data combo", valueString, methodName);
                                    }
                                }
                            } else {
                                throw parseToException("item ID,ID,ID,ID" + ChatColor.RED + " or " + ChatColor.GREEN
                                        + "ID:Data,ID:Data,ID:Data,ID:Data combo", valueString, methodName);
                            }
                            value = items;
                            // Parse to horse color
                        } else if (param == Color.class) {
                            try {
                                value = Color.valueOf(valueString.toUpperCase());
                            } catch (Exception ex) {
                                throw parseToException("horse color", valueString, methodName);
                            }
                            // Parse to horse style
                        } else if (param == Style.class) {
                            try {
                                value = Style.valueOf(valueString.toUpperCase());
                            } catch (Exception ex) {
                                throw parseToException("horse style", valueString, methodName);
                            }
                            // Parse to villager profession
                        } else if (param == Profession.class) {
                            try {
                                value = Profession.valueOf(valueString.toUpperCase());
                            } catch (Exception ex) {
                                throw parseToException("villager profession", valueString, methodName);
                            }
                            // Parse to ocelot type
                        } else if (param == Type.class) {
                            try {
                                value = Type.valueOf(valueString.toUpperCase());
                            } catch (Exception ex) {
                                throw parseToException("ocelot type", valueString, methodName);
                            }

                            // Parse to potion effect
                        } else if (param == PotionEffectType.class) {
                            try {
                                PotionEffectType potionType = PotionEffectType.getByName(valueString.toUpperCase());
                                if (potionType == null && isNumeric(valueString)) {
                                    potionType = PotionEffectType.getById(Integer.parseInt(valueString));
                                }
                                if (potionType == null)
                                    throw new Exception();
                                value = potionType;
                            } catch (Exception ex) {
                                throw parseToException("potioneffect type", valueString, methodName);
                            }
                        }
                    }
                    break;
                }
            }
            if (methodToUse == null) {
                throw new Exception(ChatColor.RED + "Cannot find the option " + methodName);
            }
            usedOptions.add(methodName.toLowerCase());
            doCheck(optionPermissions, usedOptions);
            methodToUse.invoke(disguise.getWatcher(), value);
        }
        // Alright. We've constructed our disguise.
        return disguise;
    }

    private void doCheck(HashSet<HashSet<String>> optionPermissions, HashSet<String> usedOptions) throws Exception {
        if (!optionPermissions.isEmpty()) {
            for (HashSet<String> perms : optionPermissions) {
                if (!perms.containsAll(usedOptions)) {
                    throw new Exception(ChatColor.RED + "You do not have the permission to use the option "
                            + usedOptions.toArray(new String[usedOptions.size()])[usedOptions.size() - 1]);
                }
            }
        }
    }

    private Exception parseToException(String expectedValue, String receivedInstead, String methodName) {
        return new Exception(ChatColor.RED + "Expected " + ChatColor.GREEN + expectedValue + ChatColor.RED + ", received "
                + ChatColor.GREEN + receivedInstead + ChatColor.RED + " instead for " + ChatColor.GREEN + methodName);
    }

    private ItemStack parseToItemstack(String string) throws Exception {
        String[] split = string.split(":", -1);
        if (isNumeric(split[0])) {
            int itemId = Integer.parseInt(split[0]);
            short itemDura = 0;
            if (split.length > 1) {
                if (isNumeric(split[1])) {
                    itemDura = Short.parseShort(split[1]);
                } else {
                    throw parseToException("item ID:Durability combo", string, "%s");
                }
            }
            return new ItemStack(itemId, 1, itemDura);
        } else {
            if (split.length == 1) {
                throw parseToException("item ID", string, "%s");
            } else {
                throw parseToException("item ID:Durability combo", string, "%s");
            }
        }
    }

    protected abstract void sendCommandUsage(CommandSender sender);
}