SteamWar/SpigotCore
SteamWar/SpigotCore
Archiviert
13
0
Fork 0
Code Issues 4 Pull-Requests 1 Releases Wiki Aktivität

Add SWCommand.Mapper

Quellcode durchsuchen 
Add mapper support to internal reflections
Add better tab complete support for to InternalCommand
Dieser Commit ist enthalten in:
yoyosource 2021-03-13 19:42:28 +01:00
Ursprung add406a8aa
Commit f0bf20e960
5 geänderte Dateien mit 74 neuen und 31 gelöschten Zeilen
Statistiken anzeigen Patch-Datei herunterladen Vergleichs-Datei herunterladen Alle Dateien ausklappen Alle Dateien einklappen

20
SpigotCore_Main/src/de/steamwar/command/InternalCommand.java
Datei anzeigen

@ -43,7 +43,9 @@ class InternalCommand {
} }
boolean invoke(CommandSender commandSender, String[] args) { boolean invoke(CommandSender commandSender, String[] args) {
if (args.length < parameters.length - 1) return false; if (args.length < parameters.length - 1) {
return false;
}
try { try {
Object[] objects = SWCommandUtils.generateArgumentArray(commandSender, method, parameters, args); Object[] objects = SWCommandUtils.generateArgumentArray(commandSender, method, parameters, args);
method.setAccessible(true); method.setAccessible(true);
@ -59,18 +61,26 @@ class InternalCommand {
List<String> tabComplete(String[] args) { List<String> tabComplete(String[] args) {
if (args.length > parameters.length - increment) { if (args.length > parameters.length - increment) {
if (parameters[parameters.length - 1].isVarArgs()) { if (parameters[parameters.length - 1].isVarArgs()) {
Class<?> clazz = parameters[parameters.length - 1].getType().getComponentType(); String name = parameters[parameters.length - 1].getType().getComponentType().getTypeName();
SWCommand.Mapper mapper = parameters[parameters.length - 1].getAnnotation(SWCommand.Mapper.class);
if (mapper != null) {
name = mapper.mapper();
}
try { try {
return SWCommandUtils.MAPPER_FUNCTIONS.getOrDefault(clazz, SWCommandUtils.ERROR_FUNCTION).tabCompletes(args[args.length - 1]); return SWCommandUtils.MAPPER_FUNCTIONS.getOrDefault(name, SWCommandUtils.ERROR_FUNCTION).tabCompletes(args[args.length - 1]);
} catch (Exception e) { } catch (Exception e) {
// Ignored // Ignored
} }
} }
return Collections.emptyList(); return Collections.emptyList();
} }
Class<?> clazz = parameters[args.length - increment + 1].getType(); String name = parameters[args.length - increment + 1].getType().getTypeName();
SWCommand.Mapper mapper = parameters[args.length - increment + 1].getAnnotation(SWCommand.Mapper.class);
if (mapper != null) {
name = mapper.mapper();
}
try { try {
return SWCommandUtils.MAPPER_FUNCTIONS.getOrDefault(clazz, SWCommandUtils.ERROR_FUNCTION).tabCompletes(args[args.length - 1]); return SWCommandUtils.MAPPER_FUNCTIONS.getOrDefault(name, SWCommandUtils.ERROR_FUNCTION).tabCompletes(args[args.length - 1]);
} catch (Exception e) { } catch (Exception e) {
return Collections.emptyList(); return Collections.emptyList();
} }

4
SpigotCore_Main/src/de/steamwar/command/InternalTabComplete.java
Datei anzeigen

@ -38,7 +38,9 @@ class InternalTabComplete {
} }
SWCommandUtils.TabComplete invoke(CommandSender commandSender, String[] args) { SWCommandUtils.TabComplete invoke(CommandSender commandSender, String[] args) {
if (args.length < parameters.length - 1) return null; if (args.length < parameters.length - 1) {
return null;
}
try { try {
Object[] objects = SWCommandUtils.generateArgumentArray(commandSender, method, parameters, args); Object[] objects = SWCommandUtils.generateArgumentArray(commandSender, method, parameters, args);
method.setAccessible(true); method.setAccessible(true);

36
SpigotCore_Main/src/de/steamwar/command/SWCommand.java
Datei anzeigen

@ -36,7 +36,8 @@ public abstract class SWCommand {
private final Set<InternalCommand> commandSet = new HashSet<>(); private final Set<InternalCommand> commandSet = new HashSet<>();
private final Set<InternalTabComplete> tabCompleteSet = new HashSet<>(); private final Set<InternalTabComplete> tabCompleteSet = new HashSet<>();
private final Map<Integer, Set<String>> subCommandTabCompletes = new HashMap<>(); private final Map<Integer, Set<String>> subCommandTabCompletes = new HashMap<>();
private Consumer<CommandSender> helpMessage = sender -> {}; private Consumer<CommandSender> helpMessage = sender -> {
};
protected SWCommand(String command) { protected SWCommand(String command) {
this(command, new String[0]); this(command, new String[0]);
@ -82,12 +83,16 @@ public abstract class SWCommand {
for (Method method : getClass().getDeclaredMethods()) { for (Method method : getClass().getDeclaredMethods()) {
Register register = method.getDeclaredAnnotation(Register.class); Register register = method.getDeclaredAnnotation(Register.class);
if (register == null) continue; if (register == null) {
continue;
}
for (int i = 0; i < register.subCommand().length; i++) { for (int i = 0; i < register.subCommand().length; i++) {
subCommandTabCompletes.computeIfAbsent(i, integer -> new HashSet<>()).add(register.subCommand()[i]); subCommandTabCompletes.computeIfAbsent(i, integer -> new HashSet<>()).add(register.subCommand()[i]);
} }
if (!validMethod(method)) continue; if (!validMethod(method)) {
continue;
}
if (method.getReturnType() == Void.TYPE) { if (method.getReturnType() == Void.TYPE) {
commandSet.add(new InternalCommand(this, method)); commandSet.add(new InternalCommand(this, method));
} }
@ -99,16 +104,27 @@ public abstract class SWCommand {
private boolean validMethod(Method method) { private boolean validMethod(Method method) {
Parameter[] parameters = method.getParameters(); Parameter[] parameters = method.getParameters();
if (parameters.length == 0) return false; if (parameters.length == 0) {
if (!CommandSender.class.isAssignableFrom(parameters[0].getType())) return false; return false;
}
if (!CommandSender.class.isAssignableFrom(parameters[0].getType())) {
return false;
}
for (int i = 1; i < parameters.length; i++) { for (int i = 1; i < parameters.length; i++) {
Parameter parameter = parameters[i]; Parameter parameter = parameters[i];
Class<?> clazz = parameter.getType(); Class<?> clazz = parameter.getType();
if (parameter.isVarArgs() && i == parameters.length - 1) { if (parameter.isVarArgs() && i == parameters.length - 1) {
clazz = parameter.getType().getComponentType(); clazz = parameter.getType().getComponentType();
} }
if (clazz.isEnum()) continue; if (clazz.isEnum()) {
if (!SWCommandUtils.MAPPER_FUNCTIONS.containsKey(clazz)) return false; continue;
}
String name = clazz.getTypeName();
Mapper mapper = parameter.getAnnotation(Mapper.class);
if (mapper != null) {
name = mapper.mapper();
}
if (!SWCommandUtils.MAPPER_FUNCTIONS.containsKey(name)) return false;
} }
return true; return true;
} }
@ -128,4 +144,10 @@ public abstract class SWCommand {
String[] subCommand() default {}; String[] subCommand() default {};
} }
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.PARAMETER})
protected @interface Mapper {
String mapper();
}
} }

42
SpigotCore_Main/src/de/steamwar/command/SWCommandUtils.java
Datei anzeigen

@ -38,7 +38,7 @@ class SWCommandUtils {
throw new IllegalStateException("Utility Class"); throw new IllegalStateException("Utility Class");
} }
static final Map<Class<?>, TypeMapper<?>> MAPPER_FUNCTIONS = new HashMap<>(); static final Map<String, TypeMapper<?>> MAPPER_FUNCTIONS = new HashMap<>();
static final TypeMapper<?> ERROR_FUNCTION = new TypeMapper<>() { static final TypeMapper<?> ERROR_FUNCTION = new TypeMapper<>() {
@Override @Override
@ -120,7 +120,7 @@ class SWCommandUtils {
} }
} }
}); });
MAPPER_FUNCTIONS.put(String.class, new TypeMapper<String>() { MAPPER_FUNCTIONS.put(String.class.getTypeName(), new TypeMapper<String>() {
@Override @Override
public String map(String s) { public String map(String s) {
return s; return s;
@ -131,7 +131,7 @@ class SWCommandUtils {
return Collections.singletonList(s); return Collections.singletonList(s);
} }
}); });
MAPPER_FUNCTIONS.put(StringBuilder.class, new TypeMapper<StringBuilder>() { MAPPER_FUNCTIONS.put(StringBuilder.class.getTypeName(), new TypeMapper<StringBuilder>() {
@Override @Override
public StringBuilder map(String s) { public StringBuilder map(String s) {
return new StringBuilder(s); return new StringBuilder(s);
@ -142,7 +142,7 @@ class SWCommandUtils {
return Collections.singletonList(s); return Collections.singletonList(s);
} }
}); });
MAPPER_FUNCTIONS.put(Player.class, new TypeMapper<Player>() { MAPPER_FUNCTIONS.put(Player.class.getTypeName(), new TypeMapper<Player>() {
@Override @Override
public Player map(String s) { public Player map(String s) {
return Bukkit.getPlayer(s); return Bukkit.getPlayer(s);
@ -156,8 +156,8 @@ class SWCommandUtils {
} }
private static void addMapper(Class<?> clazz, Class<?> alternativeClazz, TypeMapper<?> mapper) { private static void addMapper(Class<?> clazz, Class<?> alternativeClazz, TypeMapper<?> mapper) {
MAPPER_FUNCTIONS.put(clazz, mapper); MAPPER_FUNCTIONS.put(clazz.getTypeName(), mapper);
MAPPER_FUNCTIONS.put(alternativeClazz, mapper); MAPPER_FUNCTIONS.put(alternativeClazz.getTypeName(), mapper);
} }
static final CommandMap commandMap; static final CommandMap commandMap;
@ -189,14 +189,13 @@ class SWCommandUtils {
} }
for (int i = subCommandIndex; i < parameters.length - (varArgs ? 1 : 0); i++) { for (int i = subCommandIndex; i < parameters.length - (varArgs ? 1 : 0); i++) {
Class<?> clazz = parameters[i].getType(); arguments[i] = mapper(parameters[i]).apply(args[i - 1]);
arguments[i] = mapper(clazz).apply(args[i - 1]);
} }
if (varArgs) { if (varArgs) {
Object[] varArgument = new Object[args.length - parameters.length + 2]; Object[] varArgument = new Object[args.length - parameters.length + 2];
arguments[arguments.length - 1] = varArgument; arguments[arguments.length - 1] = varArgument;
Function<String, Object> mapper = mapper(parameters[parameters.length - 1].getType().getComponentType()); Function<String, Object> mapper = mapper(parameters[parameters.length - 1]);
int index = 0; int index = 0;
for (int i = parameters.length - 2; i < args.length; i++) { for (int i = parameters.length - 2; i < args.length; i++) {
@ -208,13 +207,25 @@ class SWCommandUtils {
} }
@SuppressWarnings("unchecked") @SuppressWarnings("unchecked")
private static Function<String, Object> mapper(Class<?> clazz) { private static Function<String, Object> mapper(Parameter parameter) {
Class<?> clazz = parameter.getType();
if (parameter.isVarArgs()) {
clazz = clazz.getComponentType();
}
if (clazz.isEnum()) { if (clazz.isEnum()) {
Class<Enum<?>> enumClass = (Class<Enum<?>>) clazz; Class<Enum<?>> enumClass = (Class<Enum<?>>) clazz;
return s -> ENUM_MAPPER.apply(enumClass, s); return s -> ENUM_MAPPER.apply(enumClass, s);
} else {
return s -> MAPPER_FUNCTIONS.getOrDefault(clazz, ERROR_FUNCTION).map(s);
} }
String name = clazz.getTypeName();
SWCommand.Mapper mapper = parameter.getAnnotation(SWCommand.Mapper.class);
if (mapper != null) {
name = mapper.mapper();
}
TypeMapper<?> typeMapper = MAPPER_FUNCTIONS.getOrDefault(name, ERROR_FUNCTION);
return typeMapper::map;
} }
public static class TabComplete { public static class TabComplete {
@ -225,10 +236,9 @@ class SWCommandUtils {
} }
} }
public static void addMapper(Class<?> clazz, TypeMapper<?> mapper) { public static void addMapper(String name, TypeMapper<?> mapper) {
if (clazz.isEnum()) return; if (MAPPER_FUNCTIONS.containsKey(name)) return;
if (MAPPER_FUNCTIONS.containsKey(clazz)) return; MAPPER_FUNCTIONS.put(name, mapper);
MAPPER_FUNCTIONS.put(clazz, mapper);
} }
} }

3
SpigotCore_Main/src/de/steamwar/command/TypeMapper.java
Datei anzeigen

@ -22,8 +22,7 @@ package de.steamwar.command;
import java.util.List; import java.util.List;
public interface TypeMapper<T> { public interface TypeMapper<T> {
T map(String s); T map(String s);
List<String> tabCompletes(String s);
List<String> tabCompletes(String s);
} }