FINALLY revamped the commandhandler (but it still isn't fully complete, eventually I want there to be a hierarchy system where you can specify "parents" to comamnds for a sub-command system)

This commit is contained in:
batchprogrammer314
2021-09-02 15:46:00 -05:00
parent e7d8bf1b55
commit fadcc88946
14 changed files with 269 additions and 187 deletions

View File

@@ -1,13 +1,23 @@
package net.nuggetmc.tplus.command;
import net.md_5.bungee.api.ChatColor;
import net.nuggetmc.tplus.command.annotation.Arg;
import net.nuggetmc.tplus.command.annotation.OptArg;
import net.nuggetmc.tplus.command.annotation.TextArg;
import net.nuggetmc.tplus.command.exception.ArgCountException;
import net.nuggetmc.tplus.command.exception.ArgParseException;
import net.nuggetmc.tplus.command.exception.NonPlayerException;
import net.nuggetmc.tplus.utils.ChatUtils;
import org.apache.commons.lang.StringUtils;
import org.bukkit.command.CommandSender;
import org.bukkit.command.defaults.BukkitCommand;
import org.bukkit.entity.Player;
import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Parameter;
import java.util.*;
import java.util.stream.Collectors;
@@ -61,8 +71,136 @@ public abstract class CommandInstance extends BukkitCommand {
arguments.remove(0);
}
List<Object> parsedArguments = new ArrayList<>();
int index = 0;
try {
method.getMethod().invoke(method.getHandler(), sender, arguments);
for (Parameter parameter : method.getMethod().getParameters()) {
Class<?> type = parameter.getType();
boolean required = !parameter.isAnnotationPresent(OptArg.class);
if (type == CommandSender.class) {
parsedArguments.add(sender);
} else if (type == Player.class) {
if (!(sender instanceof Player)) {
throw new NonPlayerException();
}
parsedArguments.add(sender);
} else if (type == List.class) {
parsedArguments.add(arguments);
}
else {
if (parameter.isAnnotationPresent(TextArg.class)) {
if (index >= arguments.size()) {
parsedArguments.add("");
} else {
parsedArguments.add(StringUtils.join(arguments.subList(index, arguments.size()), " "));
}
continue;
}
if (index >= arguments.size() && required) {
throw new ArgCountException();
}
String arg;
if (index >= arguments.size()) {
arg = null;
} else {
arg = arguments.get(index);
}
index++;
if (type == String.class) {
parsedArguments.add(arg);
} else if (type == int.class) {
if (arg == null) {
parsedArguments.add(0);
continue;
}
try {
parsedArguments.add(Integer.parseInt(arg));
} catch (NumberFormatException e) {
throw new ArgParseException(parameter);
}
} else if (type == double.class) {
if (arg == null) {
parsedArguments.add(0);
continue;
}
try {
parsedArguments.add(Double.parseDouble(arg));
} catch (NumberFormatException e) {
throw new ArgParseException(parameter);
}
} else if (type == float.class) {
if (arg == null) {
parsedArguments.add(0);
continue;
}
try {
parsedArguments.add(Float.parseFloat(arg));
} catch (NumberFormatException e) {
throw new ArgParseException(parameter);
}
} else if (type == boolean.class) {
if (arg == null) {
parsedArguments.add(false);
continue;
}
if (arg.equalsIgnoreCase("true") || arg.equalsIgnoreCase("false")) {
parsedArguments.add(Boolean.parseBoolean(arg));
} else {
throw new ArgParseException(parameter);
}
} else {
parsedArguments.add(arg);
}
}
}
}
catch (NonPlayerException e) {
sender.sendMessage("This is a player-only command.");
return true;
}
catch (ArgParseException e) {
Parameter parameter = e.getParameter();
String name = getArgumentName(parameter);
sender.sendMessage("The parameter " + ChatColor.YELLOW + name + ChatColor.RESET + " must be of type " + ChatColor.YELLOW + parameter.getType().toString() + ChatColor.RESET + ".");
return true;
}
catch (ArgCountException e) {
List<String> usageArgs = new ArrayList<>();
Arrays.stream(method.getMethod().getParameters()).forEach(parameter -> {
Class<?> type = parameter.getType();
if (type != CommandSender.class && type != Player.class){
usageArgs.add(getArgumentName(parameter));
}
});
sender.sendMessage("Command Usage: " + org.bukkit.ChatColor.YELLOW + "/" + getName() + (method.getName().isEmpty() ? "" : " " + method.getName())
+ " " + StringUtils.join(usageArgs, " "));
return true;
}
try {
method.getMethod().invoke(method.getHandler(), parsedArguments.toArray());
} catch (InvocationTargetException | IllegalAccessException e) {
sender.sendMessage(ChatColor.RED + "Failed to perform command.");
e.printStackTrace();
@@ -71,6 +209,24 @@ public abstract class CommandInstance extends BukkitCommand {
return true;
}
public static String getArgumentName(Parameter parameter) {
if (parameter.isAnnotationPresent(OptArg.class)) {
OptArg arg = parameter.getAnnotation(OptArg.class);
if (!arg.value().isEmpty()) {
return "[" + ChatUtils.camelToDashed(arg.value()) + "]";
}
} else if (parameter.isAnnotationPresent(Arg.class)) {
Arg arg = parameter.getAnnotation(Arg.class);
if (!arg.value().isEmpty()) {
return "<" + ChatUtils.camelToDashed(arg.value()) + ">";
}
}
return "<" + ChatUtils.camelToDashed(parameter.getName()) + ">";
}
@Override
@Nonnull
@SuppressWarnings("unchecked")