Clean up annotation parsing.
This commit is contained in:
parent
e72a876037
commit
56623160e1
10 changed files with 480 additions and 184 deletions
|
|
@ -37,19 +37,14 @@ import com.intellectualsites.commands.meta.CommandMeta;
|
||||||
import javax.annotation.Nonnull;
|
import javax.annotation.Nonnull;
|
||||||
import javax.annotation.Nullable;
|
import javax.annotation.Nullable;
|
||||||
import java.lang.annotation.Annotation;
|
import java.lang.annotation.Annotation;
|
||||||
import java.lang.reflect.InvocationTargetException;
|
|
||||||
import java.lang.reflect.Method;
|
import java.lang.reflect.Method;
|
||||||
import java.lang.reflect.Parameter;
|
import java.lang.reflect.Parameter;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
import java.util.LinkedHashMap;
|
import java.util.LinkedHashMap;
|
||||||
import java.util.List;
|
|
||||||
import java.util.Map;
|
import java.util.Map;
|
||||||
import java.util.StringTokenizer;
|
|
||||||
import java.util.function.Function;
|
import java.util.function.Function;
|
||||||
import java.util.function.Predicate;
|
|
||||||
import java.util.regex.Pattern;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Parser that parses class instances {@link com.intellectualsites.commands.Command commands}
|
* Parser that parses class instances {@link com.intellectualsites.commands.Command commands}
|
||||||
|
|
@ -58,18 +53,13 @@ import java.util.regex.Pattern;
|
||||||
*/
|
*/
|
||||||
public final class AnnotationParser<C> {
|
public final class AnnotationParser<C> {
|
||||||
|
|
||||||
private static final Predicate<String> PATTERN_ARGUMENT_LITERAL = Pattern.compile("([A-Za-z0-9]+)(|([A-Za-z0-9]+))*")
|
private final SyntaxParser syntaxParser = new SyntaxParser();
|
||||||
.asPredicate();
|
private final ArgumentExtractor argumentExtractor = new ArgumentExtractor();
|
||||||
private static final Predicate<String> PATTERN_ARGUMENT_REQUIRED = Pattern.compile("<([A-Za-z0-9]+)>")
|
|
||||||
.asPredicate();
|
|
||||||
private static final Predicate<String> PATTERN_ARGUMENT_OPTIONAL = Pattern.compile("\\[([A-Za-z0-9]+)]")
|
|
||||||
.asPredicate();
|
|
||||||
|
|
||||||
|
|
||||||
private final Function<ParserParameters, CommandMeta> metaMapper;
|
|
||||||
private final CommandManager<C> manager;
|
private final CommandManager<C> manager;
|
||||||
private final Map<Class<? extends Annotation>, Function<? extends Annotation, ParserParameters>> annotationMappers;
|
private final Map<Class<? extends Annotation>, Function<? extends Annotation, ParserParameters>> annotationMappers;
|
||||||
private final Class<C> commandSenderClass;
|
private final Class<C> commandSenderClass;
|
||||||
|
private final MetaFactory metaFactory;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Construct a new annotation parser
|
* Construct a new annotation parser
|
||||||
|
|
@ -86,7 +76,7 @@ public final class AnnotationParser<C> {
|
||||||
@Nonnull final Function<ParserParameters, CommandMeta> metaMapper) {
|
@Nonnull final Function<ParserParameters, CommandMeta> metaMapper) {
|
||||||
this.commandSenderClass = commandSenderClass;
|
this.commandSenderClass = commandSenderClass;
|
||||||
this.manager = manager;
|
this.manager = manager;
|
||||||
this.metaMapper = metaMapper;
|
this.metaFactory = new MetaFactory(this, metaMapper);
|
||||||
this.annotationMappers = Maps.newHashMap();
|
this.annotationMappers = Maps.newHashMap();
|
||||||
this.registerAnnotationMapper(Description.class, d -> ParserParameters.single(StandardParameters.DESCRIPTION, d.value()));
|
this.registerAnnotationMapper(Description.class, d -> ParserParameters.single(StandardParameters.DESCRIPTION, d.value()));
|
||||||
}
|
}
|
||||||
|
|
@ -103,20 +93,6 @@ public final class AnnotationParser<C> {
|
||||||
this.annotationMappers.put(annotation, mapper);
|
this.annotationMappers.put(annotation, mapper);
|
||||||
}
|
}
|
||||||
|
|
||||||
@Nonnull
|
|
||||||
private CommandMeta createMeta(@Nonnull final Annotation[] annotations) {
|
|
||||||
final ParserParameters parameters = ParserParameters.empty();
|
|
||||||
for (final Annotation annotation : annotations) {
|
|
||||||
@SuppressWarnings("ALL") final Function function = this.annotationMappers.get(annotation.annotationType());
|
|
||||||
if (function == null) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
//noinspection unchecked
|
|
||||||
parameters.merge((ParserParameters) function.apply(annotation));
|
|
||||||
}
|
|
||||||
return this.metaMapper.apply(parameters);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Scan a class instance of {@link CommandMethod} annotations and attempt to
|
* Scan a class instance of {@link CommandMethod} annotations and attempt to
|
||||||
* compile them into {@link Command} instances
|
* compile them into {@link Command} instances
|
||||||
|
|
@ -160,15 +136,15 @@ public final class AnnotationParser<C> {
|
||||||
for (final CommandMethodPair commandMethodPair : methodPairs) {
|
for (final CommandMethodPair commandMethodPair : methodPairs) {
|
||||||
final CommandMethod commandMethod = commandMethodPair.getCommandMethod();
|
final CommandMethod commandMethod = commandMethodPair.getCommandMethod();
|
||||||
final Method method = commandMethodPair.getMethod();
|
final Method method = commandMethodPair.getMethod();
|
||||||
final LinkedHashMap<String, SyntaxFragment> tokens = this.parseSyntax(commandMethod.value());
|
final LinkedHashMap<String, SyntaxFragment> tokens = this.syntaxParser.apply(commandMethod.value());
|
||||||
/* Determine command name */
|
/* Determine command name */
|
||||||
final String commandToken = commandMethod.value().split(" ")[0].split("\\|")[0];
|
final String commandToken = commandMethod.value().split(" ")[0].split("\\|")[0];
|
||||||
@SuppressWarnings("ALL") final CommandManager manager = this.manager;
|
@SuppressWarnings("ALL") final CommandManager manager = this.manager;
|
||||||
@SuppressWarnings("ALL")
|
@SuppressWarnings("ALL")
|
||||||
Command.Builder builder = manager.commandBuilder(commandToken,
|
Command.Builder builder = manager.commandBuilder(commandToken,
|
||||||
tokens.get(commandToken).getMinor(),
|
tokens.get(commandToken).getMinor(),
|
||||||
this.createMeta(method.getAnnotations()));
|
this.metaFactory.apply(method.getAnnotations()));
|
||||||
final Collection<ArgumentParameterPair> arguments = this.getArguments(method);
|
final Collection<ArgumentParameterPair> arguments = this.argumentExtractor.apply(method);
|
||||||
final Map<String, CommandArgument<C, ?>> commandArguments = Maps.newHashMap();
|
final Map<String, CommandArgument<C, ?>> commandArguments = Maps.newHashMap();
|
||||||
final Map<CommandArgument<C, ?>, String> argumentDescriptions = Maps.newHashMap();
|
final Map<CommandArgument<C, ?>, String> argumentDescriptions = Maps.newHashMap();
|
||||||
/* Go through all annotated parameters and build up the argument tree */
|
/* Go through all annotated parameters and build up the argument tree */
|
||||||
|
|
@ -198,7 +174,6 @@ public final class AnnotationParser<C> {
|
||||||
}
|
}
|
||||||
|
|
||||||
final String description = argumentDescriptions.getOrDefault(argument, "");
|
final String description = argumentDescriptions.getOrDefault(argument, "");
|
||||||
|
|
||||||
builder = builder.argument(argument, description);
|
builder = builder.argument(argument, description);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -220,37 +195,14 @@ public final class AnnotationParser<C> {
|
||||||
} else if (senderType != null) {
|
} else if (senderType != null) {
|
||||||
builder = builder.withSenderType(senderType);
|
builder = builder.withSenderType(senderType);
|
||||||
}
|
}
|
||||||
/* Construct the handler */
|
try {
|
||||||
final CommandExecutionHandler<C> commandExecutionHandler = commandContext -> {
|
/* Construct the handler */
|
||||||
final List<Object> parameters = new ArrayList<>(method.getParameterCount());
|
final CommandExecutionHandler<C> commandExecutionHandler
|
||||||
for (final Parameter parameter : method.getParameters()) {
|
= new MethodCommandExecutionHandler<>(instance, commandArguments, method);
|
||||||
if (parameter.isAnnotationPresent(Argument.class)) {
|
builder = builder.handler(commandExecutionHandler);
|
||||||
final Argument argument = parameter.getAnnotation(Argument.class);
|
} catch (final Exception e) {
|
||||||
final CommandArgument<C, ?> commandArgument = commandArguments.get(argument.value());
|
throw new RuntimeException("Failed to construct command execution handler", e);
|
||||||
if (commandArgument.isRequired()) {
|
}
|
||||||
parameters.add(commandContext.getRequired(argument.value()));
|
|
||||||
} else {
|
|
||||||
final Object optional = commandContext.get(argument.value()).orElse(null);
|
|
||||||
parameters.add(optional);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if (parameter.getType().isAssignableFrom(commandContext.getSender().getClass())) {
|
|
||||||
parameters.add(commandContext.getSender());
|
|
||||||
} else {
|
|
||||||
throw new IllegalArgumentException(String.format(
|
|
||||||
"Unknown command parameter '%s' in method '%s'",
|
|
||||||
parameter.getName(), method.getName()
|
|
||||||
));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
try {
|
|
||||||
method.invoke(instance, parameters.toArray());
|
|
||||||
} catch (final IllegalAccessException | InvocationTargetException e) {
|
|
||||||
e.printStackTrace();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
builder = builder.handler(commandExecutionHandler);
|
|
||||||
commands.add(builder.build());
|
commands.add(builder.build());
|
||||||
}
|
}
|
||||||
return commands;
|
return commands;
|
||||||
|
|
@ -309,126 +261,8 @@ public final class AnnotationParser<C> {
|
||||||
}
|
}
|
||||||
|
|
||||||
@Nonnull
|
@Nonnull
|
||||||
LinkedHashMap<String, SyntaxFragment> parseSyntax(@Nonnull final String syntax) {
|
Map<Class<? extends Annotation>, Function<? extends Annotation, ParserParameters>> getAnnotationMappers() {
|
||||||
final StringTokenizer stringTokenizer = new StringTokenizer(syntax, " ");
|
return this.annotationMappers;
|
||||||
final LinkedHashMap<String, SyntaxFragment> map = new LinkedHashMap<>();
|
|
||||||
while (stringTokenizer.hasMoreTokens()) {
|
|
||||||
final String token = stringTokenizer.nextToken();
|
|
||||||
String major;
|
|
||||||
List<String> minor = new ArrayList<>();
|
|
||||||
ArgumentMode mode;
|
|
||||||
if (PATTERN_ARGUMENT_REQUIRED.test(token)) {
|
|
||||||
major = token.substring(1, token.length() - 1);
|
|
||||||
mode = ArgumentMode.REQUIRED;
|
|
||||||
} else if (PATTERN_ARGUMENT_OPTIONAL.test(token)) {
|
|
||||||
major = token.substring(1, token.length() - 1);
|
|
||||||
mode = ArgumentMode.OPTIONAL;
|
|
||||||
} else if (PATTERN_ARGUMENT_LITERAL.test(token)) {
|
|
||||||
final String[] literals = token.split("\\|");
|
|
||||||
/* Actually use the other literals as well */
|
|
||||||
major = literals[0];
|
|
||||||
minor.addAll(Arrays.asList(literals).subList(1, literals.length));
|
|
||||||
mode = ArgumentMode.LITERAL;
|
|
||||||
} else {
|
|
||||||
throw new IllegalArgumentException(String.format("Unrecognizable syntax token '%s'", syntax));
|
|
||||||
}
|
|
||||||
map.put(major, new SyntaxFragment(major, minor, mode));
|
|
||||||
}
|
|
||||||
return map;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Nonnull
|
|
||||||
private Collection<ArgumentParameterPair> getArguments(@Nonnull final Method method) {
|
|
||||||
final Collection<ArgumentParameterPair> arguments = new ArrayList<>();
|
|
||||||
for (final Parameter parameter : method.getParameters()) {
|
|
||||||
if (!parameter.isAnnotationPresent(Argument.class)) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
arguments.add(new ArgumentParameterPair(parameter, parameter.getAnnotation(Argument.class)));
|
|
||||||
}
|
|
||||||
return arguments;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
enum ArgumentMode {
|
|
||||||
LITERAL,
|
|
||||||
OPTIONAL,
|
|
||||||
REQUIRED
|
|
||||||
}
|
|
||||||
|
|
||||||
private static final class CommandMethodPair {
|
|
||||||
|
|
||||||
private final Method method;
|
|
||||||
private final CommandMethod commandMethod;
|
|
||||||
|
|
||||||
private CommandMethodPair(@Nonnull final Method method, @Nonnull final CommandMethod commandMethod) {
|
|
||||||
this.method = method;
|
|
||||||
this.commandMethod = commandMethod;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Nonnull
|
|
||||||
private Method getMethod() {
|
|
||||||
return this.method;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Nonnull
|
|
||||||
private CommandMethod getCommandMethod() {
|
|
||||||
return this.commandMethod;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
private static final class ArgumentParameterPair {
|
|
||||||
|
|
||||||
private final Parameter parameter;
|
|
||||||
private final Argument argument;
|
|
||||||
|
|
||||||
private ArgumentParameterPair(@Nonnull final Parameter parameter, @Nonnull final Argument argument) {
|
|
||||||
this.parameter = parameter;
|
|
||||||
this.argument = argument;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Nonnull
|
|
||||||
private Parameter getParameter() {
|
|
||||||
return this.parameter;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Nonnull
|
|
||||||
private Argument getArgument() {
|
|
||||||
return this.argument;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
private static final class SyntaxFragment {
|
|
||||||
|
|
||||||
private final String major;
|
|
||||||
private final List<String> minor;
|
|
||||||
private final ArgumentMode argumentMode;
|
|
||||||
|
|
||||||
private SyntaxFragment(@Nonnull final String major,
|
|
||||||
@Nonnull final List<String> minor,
|
|
||||||
@Nonnull final ArgumentMode argumentMode) {
|
|
||||||
this.major = major;
|
|
||||||
this.minor = minor;
|
|
||||||
this.argumentMode = argumentMode;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Nonnull
|
|
||||||
private String getMajor() {
|
|
||||||
return this.major;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Nonnull
|
|
||||||
private List<String> getMinor() {
|
|
||||||
return this.minor;
|
|
||||||
}
|
|
||||||
|
|
||||||
@Nonnull
|
|
||||||
private ArgumentMode getArgumentMode() {
|
|
||||||
return this.argumentMode;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,51 @@
|
||||||
|
//
|
||||||
|
// MIT License
|
||||||
|
//
|
||||||
|
// Copyright (c) 2020 Alexander Söderberg
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in all
|
||||||
|
// copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
// SOFTWARE.
|
||||||
|
//
|
||||||
|
package com.intellectualsites.commands.annotations;
|
||||||
|
|
||||||
|
import javax.annotation.Nonnull;
|
||||||
|
import java.lang.reflect.Method;
|
||||||
|
import java.lang.reflect.Parameter;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Collection;
|
||||||
|
import java.util.function.Function;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Utility that extract {@link Argument arguments} from
|
||||||
|
* {@link java.lang.reflect.Method method} {@link java.lang.reflect.Parameter parameters}
|
||||||
|
*/
|
||||||
|
class ArgumentExtractor implements Function<Method, Collection<ArgumentParameterPair>> {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Collection<ArgumentParameterPair> apply(@Nonnull final Method method) {
|
||||||
|
final Collection<ArgumentParameterPair> arguments = new ArrayList<>();
|
||||||
|
for (final Parameter parameter : method.getParameters()) {
|
||||||
|
if (!parameter.isAnnotationPresent(Argument.class)) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
arguments.add(new ArgumentParameterPair(parameter, parameter.getAnnotation(Argument.class)));
|
||||||
|
}
|
||||||
|
return arguments;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,30 @@
|
||||||
|
//
|
||||||
|
// MIT License
|
||||||
|
//
|
||||||
|
// Copyright (c) 2020 Alexander Söderberg
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in all
|
||||||
|
// copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
// SOFTWARE.
|
||||||
|
//
|
||||||
|
package com.intellectualsites.commands.annotations;
|
||||||
|
|
||||||
|
enum ArgumentMode {
|
||||||
|
LITERAL,
|
||||||
|
OPTIONAL,
|
||||||
|
REQUIRED
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,49 @@
|
||||||
|
//
|
||||||
|
// MIT License
|
||||||
|
//
|
||||||
|
// Copyright (c) 2020 Alexander Söderberg
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in all
|
||||||
|
// copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
// SOFTWARE.
|
||||||
|
//
|
||||||
|
package com.intellectualsites.commands.annotations;
|
||||||
|
|
||||||
|
import javax.annotation.Nonnull;
|
||||||
|
import java.lang.reflect.Parameter;
|
||||||
|
|
||||||
|
final class ArgumentParameterPair {
|
||||||
|
|
||||||
|
private final Parameter parameter;
|
||||||
|
private final Argument argument;
|
||||||
|
|
||||||
|
ArgumentParameterPair(@Nonnull final Parameter parameter, @Nonnull final Argument argument) {
|
||||||
|
this.parameter = parameter;
|
||||||
|
this.argument = argument;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Nonnull
|
||||||
|
Parameter getParameter() {
|
||||||
|
return this.parameter;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Nonnull
|
||||||
|
Argument getArgument() {
|
||||||
|
return this.argument;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,49 @@
|
||||||
|
//
|
||||||
|
// MIT License
|
||||||
|
//
|
||||||
|
// Copyright (c) 2020 Alexander Söderberg
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in all
|
||||||
|
// copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
// SOFTWARE.
|
||||||
|
//
|
||||||
|
package com.intellectualsites.commands.annotations;
|
||||||
|
|
||||||
|
import javax.annotation.Nonnull;
|
||||||
|
import java.lang.reflect.Method;
|
||||||
|
|
||||||
|
final class CommandMethodPair {
|
||||||
|
|
||||||
|
private final Method method;
|
||||||
|
private final CommandMethod commandMethod;
|
||||||
|
|
||||||
|
CommandMethodPair(@Nonnull final Method method, @Nonnull final CommandMethod commandMethod) {
|
||||||
|
this.method = method;
|
||||||
|
this.commandMethod = commandMethod;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Nonnull
|
||||||
|
Method getMethod() {
|
||||||
|
return this.method;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Nonnull
|
||||||
|
CommandMethod getCommandMethod() {
|
||||||
|
return this.commandMethod;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,59 @@
|
||||||
|
//
|
||||||
|
// MIT License
|
||||||
|
//
|
||||||
|
// Copyright (c) 2020 Alexander Söderberg
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in all
|
||||||
|
// copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
// SOFTWARE.
|
||||||
|
//
|
||||||
|
package com.intellectualsites.commands.annotations;
|
||||||
|
|
||||||
|
import com.intellectualsites.commands.arguments.parser.ParserParameters;
|
||||||
|
import com.intellectualsites.commands.meta.CommandMeta;
|
||||||
|
|
||||||
|
import javax.annotation.Nonnull;
|
||||||
|
import java.lang.annotation.Annotation;
|
||||||
|
import java.util.function.Function;
|
||||||
|
|
||||||
|
class MetaFactory implements Function<Annotation[], CommandMeta> {
|
||||||
|
|
||||||
|
private final AnnotationParser<?> annotationParser;
|
||||||
|
private final Function<ParserParameters, CommandMeta> metaMapper;
|
||||||
|
|
||||||
|
MetaFactory(@Nonnull final AnnotationParser<?> annotationParser,
|
||||||
|
@Nonnull final Function<ParserParameters, CommandMeta> metaMapper) {
|
||||||
|
this.annotationParser = annotationParser;
|
||||||
|
this.metaMapper = metaMapper;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public CommandMeta apply(@Nonnull final Annotation[] annotations) {
|
||||||
|
final ParserParameters parameters = ParserParameters.empty();
|
||||||
|
for (final Annotation annotation : annotations) {
|
||||||
|
@SuppressWarnings("ALL") final Function function = this.annotationParser.getAnnotationMappers()
|
||||||
|
.get(annotation.annotationType());
|
||||||
|
if (function == null) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
//noinspection unchecked
|
||||||
|
parameters.merge((ParserParameters) function.apply(annotation));
|
||||||
|
}
|
||||||
|
return this.metaMapper.apply(parameters);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,89 @@
|
||||||
|
//
|
||||||
|
// MIT License
|
||||||
|
//
|
||||||
|
// Copyright (c) 2020 Alexander Söderberg
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in all
|
||||||
|
// copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
// SOFTWARE.
|
||||||
|
//
|
||||||
|
package com.intellectualsites.commands.annotations;
|
||||||
|
|
||||||
|
import com.intellectualsites.commands.arguments.CommandArgument;
|
||||||
|
import com.intellectualsites.commands.context.CommandContext;
|
||||||
|
import com.intellectualsites.commands.execution.CommandExecutionHandler;
|
||||||
|
|
||||||
|
import javax.annotation.Nonnull;
|
||||||
|
import java.lang.invoke.MethodHandle;
|
||||||
|
import java.lang.invoke.MethodHandles;
|
||||||
|
import java.lang.reflect.Method;
|
||||||
|
import java.lang.reflect.Parameter;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
class MethodCommandExecutionHandler<C> implements CommandExecutionHandler<C> {
|
||||||
|
|
||||||
|
private final Parameter[] parameters;
|
||||||
|
private final MethodHandle methodHandle;
|
||||||
|
private final Map<String, CommandArgument<C, ?>> commandArguments;
|
||||||
|
|
||||||
|
MethodCommandExecutionHandler(@Nonnull final Object instance,
|
||||||
|
@Nonnull final Map<String, CommandArgument<C, ?>> commandArguments,
|
||||||
|
@Nonnull final Method method) throws Exception {
|
||||||
|
this.commandArguments = commandArguments;
|
||||||
|
method.setAccessible(true);
|
||||||
|
this.methodHandle = MethodHandles.lookup().unreflect(method).bindTo(instance);
|
||||||
|
this.parameters = method.getParameters();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void execute(@Nonnull final CommandContext<C> commandContext) {
|
||||||
|
final List<Object> arguments = new ArrayList<>(this.parameters.length);
|
||||||
|
|
||||||
|
/* Bind parameters to context */
|
||||||
|
for (final Parameter parameter : this.parameters) {
|
||||||
|
if (parameter.isAnnotationPresent(Argument.class)) {
|
||||||
|
final Argument argument = parameter.getAnnotation(Argument.class);
|
||||||
|
final CommandArgument<C, ?> commandArgument = this.commandArguments.get(argument.value());
|
||||||
|
if (commandArgument.isRequired()) {
|
||||||
|
arguments.add(commandContext.getRequired(argument.value()));
|
||||||
|
} else {
|
||||||
|
final Object optional = commandContext.get(argument.value()).orElse(null);
|
||||||
|
arguments.add(optional);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if (parameter.getType().isAssignableFrom(commandContext.getSender().getClass())) {
|
||||||
|
arguments.add(commandContext.getSender());
|
||||||
|
} else {
|
||||||
|
throw new IllegalArgumentException(String.format(
|
||||||
|
"Unknown command parameter '%s' in method '%s'",
|
||||||
|
parameter.getName(), this.methodHandle.toString()
|
||||||
|
));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Invoke the command method */
|
||||||
|
try {
|
||||||
|
this.methodHandle.invokeWithArguments(arguments);
|
||||||
|
} catch (final Throwable e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,58 @@
|
||||||
|
//
|
||||||
|
// MIT License
|
||||||
|
//
|
||||||
|
// Copyright (c) 2020 Alexander Söderberg
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in all
|
||||||
|
// copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
// SOFTWARE.
|
||||||
|
//
|
||||||
|
package com.intellectualsites.commands.annotations;
|
||||||
|
|
||||||
|
import javax.annotation.Nonnull;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
final class SyntaxFragment {
|
||||||
|
|
||||||
|
private final String major;
|
||||||
|
private final List<String> minor;
|
||||||
|
private final ArgumentMode argumentMode;
|
||||||
|
|
||||||
|
SyntaxFragment(@Nonnull final String major,
|
||||||
|
@Nonnull final List<String> minor,
|
||||||
|
@Nonnull final ArgumentMode argumentMode) {
|
||||||
|
this.major = major;
|
||||||
|
this.minor = minor;
|
||||||
|
this.argumentMode = argumentMode;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Nonnull
|
||||||
|
String getMajor() {
|
||||||
|
return this.major;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Nonnull
|
||||||
|
List<String> getMinor() {
|
||||||
|
return this.minor;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Nonnull
|
||||||
|
ArgumentMode getArgumentMode() {
|
||||||
|
return this.argumentMode;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,77 @@
|
||||||
|
//
|
||||||
|
// MIT License
|
||||||
|
//
|
||||||
|
// Copyright (c) 2020 Alexander Söderberg
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in all
|
||||||
|
// copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||||
|
// SOFTWARE.
|
||||||
|
//
|
||||||
|
package com.intellectualsites.commands.annotations;
|
||||||
|
|
||||||
|
import javax.annotation.Nonnull;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.Arrays;
|
||||||
|
import java.util.LinkedHashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.StringTokenizer;
|
||||||
|
import java.util.function.Function;
|
||||||
|
import java.util.function.Predicate;
|
||||||
|
import java.util.regex.Pattern;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Parses command syntax into syntax fragments
|
||||||
|
*/
|
||||||
|
final class SyntaxParser implements Function<String, LinkedHashMap<String, SyntaxFragment>> {
|
||||||
|
|
||||||
|
private static final Predicate<String> PATTERN_ARGUMENT_LITERAL = Pattern.compile("([A-Za-z0-9]+)(|([A-Za-z0-9]+))*")
|
||||||
|
.asPredicate();
|
||||||
|
private static final Predicate<String> PATTERN_ARGUMENT_REQUIRED = Pattern.compile("<([A-Za-z0-9]+)>")
|
||||||
|
.asPredicate();
|
||||||
|
private static final Predicate<String> PATTERN_ARGUMENT_OPTIONAL = Pattern.compile("\\[([A-Za-z0-9]+)]")
|
||||||
|
.asPredicate();
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public LinkedHashMap<String, SyntaxFragment> apply(@Nonnull final String syntax) {
|
||||||
|
final StringTokenizer stringTokenizer = new StringTokenizer(syntax, " ");
|
||||||
|
final LinkedHashMap<String, SyntaxFragment> map = new LinkedHashMap<>();
|
||||||
|
while (stringTokenizer.hasMoreTokens()) {
|
||||||
|
final String token = stringTokenizer.nextToken();
|
||||||
|
String major;
|
||||||
|
List<String> minor = new ArrayList<>();
|
||||||
|
ArgumentMode mode;
|
||||||
|
if (PATTERN_ARGUMENT_REQUIRED.test(token)) {
|
||||||
|
major = token.substring(1, token.length() - 1);
|
||||||
|
mode = ArgumentMode.REQUIRED;
|
||||||
|
} else if (PATTERN_ARGUMENT_OPTIONAL.test(token)) {
|
||||||
|
major = token.substring(1, token.length() - 1);
|
||||||
|
mode = ArgumentMode.OPTIONAL;
|
||||||
|
} else if (PATTERN_ARGUMENT_LITERAL.test(token)) {
|
||||||
|
final String[] literals = token.split("\\|");
|
||||||
|
/* Actually use the other literals as well */
|
||||||
|
major = literals[0];
|
||||||
|
minor.addAll(Arrays.asList(literals).subList(1, literals.length));
|
||||||
|
mode = ArgumentMode.LITERAL;
|
||||||
|
} else {
|
||||||
|
throw new IllegalArgumentException(String.format("Unrecognizable syntax token '%s'", syntax));
|
||||||
|
}
|
||||||
|
map.put(major, new SyntaxFragment(major, minor, mode));
|
||||||
|
}
|
||||||
|
return map;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
@ -55,7 +55,7 @@ class AnnotationParserTest {
|
||||||
final Collection<Command<TestCommandSender>> commands = annotationParser.parse(this);
|
final Collection<Command<TestCommandSender>> commands = annotationParser.parse(this);
|
||||||
Assertions.assertFalse(commands.isEmpty());
|
Assertions.assertFalse(commands.isEmpty());
|
||||||
manager.executeCommand(new TestCommandSender(), "test 10").join();
|
manager.executeCommand(new TestCommandSender(), "test 10").join();
|
||||||
manager.executeCommand(new TestCommandSender(), "t 10").join();
|
manager.executeCommand(new TestCommandSender(), "t 10 o").join();
|
||||||
Assertions.assertThrows(CompletionException.class, () ->
|
Assertions.assertThrows(CompletionException.class, () ->
|
||||||
manager.executeCommand(new TestCommandSender(), "test 101").join());
|
manager.executeCommand(new TestCommandSender(), "test 101").join());
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue