package me.libraryaddict.disguise.utilities; import java.lang.reflect.Method; import java.util.ArrayList; import java.util.Collections; import java.util.HashSet; import java.util.Iterator; 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.Ageable; import org.bukkit.entity.Animals; import org.bukkit.entity.Monster; import org.bukkit.inventory.ItemStack; import org.bukkit.permissions.PermissionAttachmentInfo; import org.bukkit.potion.PotionEffectType; public abstract class BaseDisguiseCommand implements CommandExecutor { protected ArrayList getAllowedDisguises(CommandSender sender) { String permissionNode = "libsdisguises." + getClass().getSimpleName().replace("Command", "").toLowerCase() + "."; return getAllowedDisguises(sender, permissionNode); } protected ArrayList getAllowedDisguises(CommandSender sender, String permissionNode) { HashSet singleAllowed = new HashSet(); HashSet singleForbidden = new HashSet(); HashSet globalForbidden = new HashSet(); HashSet globalAllowed = new HashSet(); for (PermissionAttachmentInfo permission : sender.getEffectivePermissions()) { String perm = permission.getPermission().toLowerCase(); if (perm.startsWith(permissionNode)) { perm = perm.substring(permissionNode.length()); String disguiseType = perm.split("\\.")[0]; HashSet perms; for (DisguiseType type : DisguiseType.values()) { if (type.getEntityType() == null) { continue; } Class entityClass = type.getEntityType().getEntityClass(); String name = type.name().toLowerCase(); if (!perm.equalsIgnoreCase(name)) { perms = (permission.getValue() ? globalAllowed : globalForbidden); } else { perms = (permission.getValue() ? singleAllowed : singleForbidden); } if (perm.equals("mob")) { if (type.isMob()) { perms.add(name); } } else if (perm.equals("animal") || perm.equals("animals")) { if (Animals.class.isAssignableFrom(entityClass)) { perms.add(name); } } else if (perm.equals("monster") || perm.equals("monsters")) { if (Monster.class.isAssignableFrom(entityClass)) { perms.add(name); } } else if (perm.equals("misc")) { if (type.isMisc()) { perms.add(name); } } else if (perm.equals("ageable")) { if (Ageable.class.isAssignableFrom(entityClass)) { perms.add(name); } } else if (disguiseType.equals("*")) { perms.add(name); } else if (disguiseType.equals(name)) { perms.add(name); } } } } // This does the disguises for OP's. for (DisguiseType type : DisguiseType.values()) { if (type.getEntityType() == null) { continue; } if (!globalAllowed.contains(type.name().toLowerCase())) { if (sender.hasPermission(permissionNode + "*") || sender.hasPermission(permissionNode + type.name().toLowerCase())) { globalAllowed.add(type.name().toLowerCase()); } } } globalAllowed.removeAll(globalForbidden); singleAllowed.addAll(globalAllowed); singleAllowed.removeAll(singleForbidden); ArrayList disguiseTypes = new ArrayList(singleAllowed); Collections.sort(disguiseTypes, String.CASE_INSENSITIVE_ORDER); return disguiseTypes; } 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> getPermissions(CommandSender sender, String disguiseType) { HashSet> perms = new HashSet>(); 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 p = new HashSet(); 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 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; if (args[0].equalsIgnoreCase("p")) { disguiseType = DisguiseType.PLAYER; } else { for (DisguiseType type : DisguiseType.values()) { if (type.getEntityType() == null) { continue; } if (args[0].equalsIgnoreCase(type.name()) || args[0].equalsIgnoreCase(type.name().replace("_", ""))) { 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!"); } ArrayList usedOptions = new ArrayList(); 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> 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("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++; } } } } if (miscId != -1) { if (disguiseType == DisguiseType.FALLING_BLOCK) { usedOptions.add("setblock"); doCheck(optionPermissions, usedOptions); } else if (disguiseType == DisguiseType.PAINTING) { usedOptions.add("setpainting"); doCheck(optionPermissions, usedOptions); } else if (disguiseType == DisguiseType.SPLASH_POTION) { usedOptions.add("setpotionid"); doCheck(optionPermissions, usedOptions); } } // 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) && method.getAnnotation(Deprecated.class) == null && method.getParameterTypes().length == 1) { 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.getSimpleName().equals("Color")) { try { value = param.getMethod("valueOf", String.class).invoke(null, valueString.toUpperCase()); } catch (Exception ex) { throw parseToException("a horse color", valueString, methodName); } // Parse to horse style } else if (param.getSimpleName().equals("Style")) { try { value = param.getMethod("valueOf", String.class).invoke(null, valueString.toUpperCase()); } catch (Exception ex) { throw parseToException("a horse style", valueString, methodName); } // Parse to villager profession } else if (param.getSimpleName().equals("Profession")) { try { value = param.getMethod("valueOf", String.class).invoke(null, valueString.toUpperCase()); } catch (Exception ex) { throw parseToException("a villager profession", valueString, methodName); } // Parse to ocelot type } else if (param.getSimpleName().equals("Art")) { try { value = param.getMethod("valueOf", String.class).invoke(null, valueString.toUpperCase()); } catch (Exception ex) { ex.printStackTrace(); throw parseToException("a painting art", valueString, methodName); } // Parse to ocelot type } else if (param.getSimpleName().equals("Type")) { try { value = param.getMethod("valueOf", String.class).invoke(null, valueString.toUpperCase()); } catch (Exception ex) { throw parseToException("a 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("a potioneffect type", valueString, methodName); } } } break; } } if (methodToUse == null) { throw new Exception(ChatColor.RED + "Cannot find the option " + methodName); } if (!usedOptions.contains(methodName.toLowerCase())) { usedOptions.add(methodName.toLowerCase()); } doCheck(optionPermissions, usedOptions); methodToUse.invoke(disguise.getWatcher(), value); } // Alright. We've constructed our disguise. return disguise; } private void doCheck(HashSet> optionPermissions, ArrayList usedOptions) throws Exception { if (!optionPermissions.isEmpty()) { for (HashSet perms : optionPermissions) { HashSet cloned = (HashSet) perms.clone(); Iterator itel = cloned.iterator(); while (itel.hasNext()) { String perm = itel.next(); if (perm.startsWith("-")) { if (usedOptions.contains(perm.substring(1))) { throw new Exception(ChatColor.RED + "You do not have the permission to use the option " + perm.substring(1)); } itel.remove(); } } if (cloned.size() == perms.size() && !cloned.containsAll(usedOptions)) { throw new Exception(ChatColor.RED + "You do not have the permission to use the option " + usedOptions.get(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); }