383 lines
19 KiB
Java
383 lines
19 KiB
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) {
|
|
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 {
|
|
String permissionNode = "libsdisguises." + getClass().getSimpleName().replace("Command", "").toLowerCase() + ".";
|
|
ArrayList<String> allowedDisguises = getAllowedDisguises(sender, permissionNode);
|
|
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 (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.iterator().next());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
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);
|
|
}
|