io.github.llewvallis.commandbuilder.arguments.IntegerArgument Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of command-builder Show documentation

A framework for constructing Bukkit commands

The newest version!

package io.github.llewvallis.commandbuilder.arguments;

import io.github.llewvallis.commandbuilder.*;
import lombok.AccessLevel;
import lombok.AllArgsConstructor;
import lombok.NoArgsConstructor;
import lombok.With;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.util.List;
import java.util.Set;

/**
 * An argument that matches optionally bounded integers.
 */
@With
@NoArgsConstructor
@AllArgsConstructor(access = AccessLevel.PRIVATE)
public class IntegerArgument implements ArgumentParser {

    private int min = Integer.MIN_VALUE;
    private int max = Integer.MAX_VALUE;

    @ParserAnnotation(IntegerArgument.class)
    @Retention(RetentionPolicy.RUNTIME)
    @Target(ElementType.PARAMETER)
    public @interface Arg {

        int min() default Integer.MIN_VALUE;
        int max() default Integer.MAX_VALUE;
    }

    @ArgumentInferenceFactory
    private static IntegerArgument createParserFromAnnotation(ArgumentInferenceContext ctx) {
        return new IntegerArgument(ctx.getAnnotation().min(), ctx.getAnnotation().max());
    }

    @Override
    public Integer parse(String argument, int position, CommandContext context) throws ArgumentParseException {
        int value;
        try {
            value = Integer.parseInt(argument);
        } catch (NumberFormatException e) {
            throw new ArgumentParseException("not a valid integer");
        }

        if (value < min) {
            throw new ArgumentParseException("should be greater or equal to " + min);
        }

        if (value > max) {
            throw new ArgumentParseException("should be lesser or equal to " + max);
        }

        return value;
    }

    @Override
    public Set complete(List