zhdev/oblak
1
0
Fork 0
Code Issues Pull requests Activity

✨ Allow interception of command builders based on annotations in AnnotationParser

Browse source 
Fixes #179
This commit is contained in:
Alexander Söderberg 2020-12-18 18:59:59 +01:00 • committed by Alexander Söderberg
parent c684c6607f
commit 9ed40a698a
3 changed files with 66 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

36
cloud-annotations/src/main/java/cloud/commandframework/annotations/AnnotationParser.java
View file

@ -83,6 +83,8 @@ public final class AnnotationParser<C> {
private final Map<Class<? extends Annotation>, Function<? extends Annotation, ParserParameters>> annotationMappers;
private final Map<Class<? extends Annotation>, Function<? extends Annotation, BiFunction<@NonNull CommandContext<C>,
@NonNull Queue<@NonNull String>, @NonNull ArgumentParseResult<Boolean>>>> preprocessorMappers;
private final Map<Class<? extends Annotation>, BiFunction<? extends Annotation, Command.Builder<C>, Command.Builder<C>>>
builderModifiers;
private final Class<C> commandSenderClass;
private final MetaFactory metaFactory;
private final FlagExtractor flagExtractor;
@ -107,6 +109,7 @@ public final class AnnotationParser<C> {
this.metaFactory = new MetaFactory(this, metaMapper);
this.annotationMappers = new HashMap<>();
this.preprocessorMappers = new HashMap<>();
this.builderModifiers = new HashMap<>();
this.flagExtractor = new FlagExtractor(manager);
this.registerAnnotationMapper(CommandDescription.class, d ->
ParserParameters.single(StandardParameters.DESCRIPTION, d.value()));
@ -177,6 +180,25 @@ public final class AnnotationParser<C> {
return getMethodOrClassAnnotation(method, clazz) != null;
}
/**
* Register a builder modifier for a specific annotation. The builder modifiers are
* allowed to act on a {@link Command.Builder} after all arguments have been added
* to the builder. This allows for modifications of the builder instance before
* the command is registered to the command manager.
*
* @param annotation Annotation (class) that the builder modifier reacts to
* @param builderModifier Modifier that acts on the given annotation and the incoming builder. Command builders
* are immutable, so the modifier should return the instance of the command builder that is
* returned as a result of any operation on the builder
* @param <A> Annotation type
*/
public <A extends Annotation> void registerBuilderModifier(
final @NonNull Class<A> annotation,
final @NonNull BiFunction<A, Command.Builder<C>, Command.Builder<C>> builderModifier
) {
this.builderModifiers.put(annotation, builderModifier);
}
/**
* Register an annotation mapper
*
@ -275,9 +297,9 @@ public final class AnnotationParser<C> {
method.setAccessible(true);
}
if (method.getParameterCount() != 2
|| !method.getReturnType().equals(List.class)
|| !method.getParameters()[0].getType().equals(CommandContext.class)
|| !method.getParameters()[1].getType().equals(String.class)
|| !method.getReturnType().equals(List.class)
|| !method.getParameters()[0].getType().equals(CommandContext.class)
|| !method.getParameters()[1].getType().equals(String.class)
) {
throw new IllegalArgumentException(String.format(
"@Suggestions annotated method '%s' in class '%s' does not have the correct signature",
@ -457,6 +479,14 @@ public final class AnnotationParser<C> {
for (final CommandFlag<?> flag : flags) {
builder = builder.flag(flag);
}
for (final Annotation annotation : method.getDeclaredAnnotations()) {
@SuppressWarnings("ALL")
final BiFunction builderModifier = this.builderModifiers.get(annotation.annotationType());
if (builderModifier == null) {
continue;
}
builder = (Command.Builder<C>) builderModifier.apply(annotation, builder);
}
/* Construct and register the command */
final Command<C> builtCommand = builder.build();
commands.add(builtCommand);