LibsDisguises/src/me/libraryaddict/disguise/commands/DisguiseEntityCommand.java

package me.libraryaddict.disguise.commands;

import java.util.ArrayList;
import java.util.Collections;


import me.libraryaddict.disguise.DisguiseListener;
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.apache.commons.lang.StringUtils;
import org.bukkit.ChatColor;
import org.bukkit.command.Command;
import org.bukkit.command.CommandExecutor;
import org.bukkit.command.CommandSender;


public class DisguiseEntityCommand implements CommandExecutor {

    private DisguiseListener listener;

    public DisguiseEntityCommand(DisguiseListener listener) {
        this.listener = listener;
    }

    private ArrayList<String> allowedDisguises(CommandSender sender) {
        ArrayList<String> names = new ArrayList<String>();
        for (DisguiseType type : DisguiseType.values()) {
            String name = type.name().toLowerCase();
            if (sender.hasPermission("libsdisguises.disguiseentity.*")
                    || sender.hasPermission("libsdisguises.disguiseentity." + name))
                names.add(name);
        }
        Collections.sort(names, String.CASE_INSENSITIVE_ORDER);
        return names;
    }

    private ArrayList<String> forbiddenDisguises(CommandSender sender) {
        ArrayList<String> names = new ArrayList<String>();
        if (sender.hasPermission("libsdisguises.disguiseentity.*"))
            return names;
        for (DisguiseType type : DisguiseType.values()) {
            String name = type.name().toLowerCase();
            if (!sender.hasPermission("libsdisguises.disguiseentity." + name))
                names.add(name);
        }
        Collections.sort(names, String.CASE_INSENSITIVE_ORDER);
        return names;
    }

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

    @Override
    public boolean onCommand(CommandSender sender, Command cmd, String label, String[] args) {
        if (sender.getName().equals("CONSOLE")) {
            sender.sendMessage(ChatColor.RED + "You may not use this command from the console!");
            return true;
        }
        // What disguises can he use
        ArrayList<String> allowedDisguises = allowedDisguises(sender);
        // If he owns at least one disguise
        if (allowedDisguises.size() > 0) {
            // Get his forbidden disguises (Disguises he can't use) for later use
            ArrayList<String> forbiddenDisguises = forbiddenDisguises(sender);
            // If he is attempting to do something
            if (args.length > 0) {
                // If he owns the disguise
                if (allowedDisguises.contains(args[0].toLowerCase())) {
                    Disguise disguise = null;
                    // Time to start constructing the disguise.
                    // We will need to check between all 3 kinds of disguises
                    if (args[0].equalsIgnoreCase("player")) { // If he is doing a player disguise
                        if (args.length == 1) {
                            // He needs to give the player name
                            sender.sendMessage(ChatColor.RED + "Error! You need to give a player name!");
                            return true;
                        } else {
                            // Construct the player disguise
                            disguise = new PlayerDisguise(ChatColor.translateAlternateColorCodes('&', args[1]));
                        }
                    } else {
                        // Grab the disguise type so we know what constructor to use
                        DisguiseType disguiseType = DisguiseType.valueOf(args[0].toUpperCase());
                        if (disguiseType.isMob()) { // Its a mob, use the mob constructor
                            boolean adult = true;
                            if (args.length > 1) {
                                // Seems they want to make this a baby disguise!
                                if (!args[1].equalsIgnoreCase("false") && !args[1].equalsIgnoreCase("true")) {
                                    sender.sendMessage(ChatColor.RED + "Error! " + ChatColor.GREEN + args[1] + ChatColor.RED
                                            + " isn't true or false!");
                                    return true;
                                }
                                adult = args[1].equalsIgnoreCase("false"); // Adult = !arg
                            }
                            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]);
                                } else {
                                    // Send them a error
                                    sender.sendMessage(ChatColor.RED + "Error! " + ChatColor.GREEN + args[1] + ChatColor.RED
                                            + " is not a number!");
                                    return true;
                                }
                                // If they also defined a data value
                                if (args.length > 2) {
                                    if (isNumeric(args[1])) {
                                        miscData = Integer.parseInt(args[2]);
                                    } else {
                                        // Send them a error
                                        sender.sendMessage(ChatColor.RED + "Error! " + ChatColor.GREEN + args[2] + ChatColor.RED
                                                + " is not a number!");
                                        return true;
                                    }
                                }
                            }
                            // Construct the disguise
                            disguise = new MiscDisguise(disguiseType, true, miscId, miscData);
                        }
                    }
                    // Alright. We've constructed our disguise.
                    // Time to use it!
                    listener.setSlap(sender.getName(), disguise);
                    sender.sendMessage(ChatColor.RED + "Right click a entity in the next 10 seconds to disguise it as a "
                            + toReadable(disguise.getType().name()) + "!");
                } else {
                    // He doesn't. Either tell him its incorrect or he isn't allowed to use it
                    if (forbiddenDisguises.contains(args[0].toLowerCase())) {
                        // He isn't allowed to use it..
                        sender.sendMessage(ChatColor.RED + "You are forbidden to use this disguise!");
                    } else {
                        sender.sendMessage(ChatColor.RED + "Error! The disguise " + ChatColor.GREEN + args[0] + ChatColor.RED
                                + " doesn't exist!");
                    }
                }
            } else {
                // Just send the disguises information.
                sendDisguises(sender, allowedDisguises, forbiddenDisguises);
            }
        } else
            sender.sendMessage(ChatColor.RED + "You are forbidden to use this command!");
        return true;
    }

    /**
     * Send the player the information
     */
    private void sendDisguises(CommandSender sender, ArrayList<String> allowedDisguises, ArrayList<String> forbiddenDisguises) {
        sender.sendMessage(ChatColor.DARK_GREEN + "Choose a disguise then slap a entity to disguise it!");
        sender.sendMessage(ChatColor.DARK_GREEN + "You can use the disguises: " + ChatColor.GREEN
                + StringUtils.join(allowedDisguises, ChatColor.RED + ", " + ChatColor.GREEN));
        if (allowedDisguises.contains("player"))
            sender.sendMessage(ChatColor.DARK_GREEN + "/disguiseentity player <Name>");
        sender.sendMessage(ChatColor.DARK_GREEN + "/disguise <DisguiseType> <Baby>");
        if (allowedDisguises.contains("dropped_item") || allowedDisguises.contains("falling_block"))
            sender.sendMessage(ChatColor.DARK_GREEN + "/disguiseentity <Dropped_Item/Falling_Block> <Id> <Durability>");
    }

    private String toReadable(String name) {
        String[] split = name.split("_");
        for (int i = 0; i < split.length; i++)
            split[i] = split[i].substring(0, 1) + split[i].substring(1).toLowerCase();
        return StringUtils.join(split, " ");
    }
}