• Home
• io.github.jbock-java
• jbock-compiler
• 5.17
• source code
• OptionStatesMethod.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

net.jbock.writing.OptionStatesMethod Maven / Gradle / Ivy

package net.jbock.writing;

import io.jbock.javapoet.ClassName;
import io.jbock.javapoet.CodeBlock;
import io.jbock.javapoet.MethodSpec;
import io.jbock.javapoet.ParameterSpec;
import io.jbock.simple.Inject;
import net.jbock.annotated.Option;
import net.jbock.convert.Mapping;
import net.jbock.parse.OptionState;
import net.jbock.parse.OptionStateModeFlag;
import net.jbock.parse.OptionStateNonRepeatable;
import net.jbock.parse.OptionStateRepeatable;

import java.util.EnumMap;
import java.util.Map;
import java.util.function.Supplier;

import static javax.lang.model.element.Modifier.PRIVATE;
import static net.jbock.common.Constants.mapOf;
import static net.jbock.common.Suppliers.memoize;

final class OptionStatesMethod extends HasCommandRepresentation {

    @Inject
    OptionStatesMethod(CommandRepresentation commandRepresentation) {
        super(commandRepresentation);
    }

    private final Supplier define = memoize(() -> {
        ParameterSpec result = ParameterSpec.builder(
                mapOf(optType(), ClassName.get(OptionState.class)), "result").build();
        CodeBlock.Builder code = CodeBlock.builder();
        if (namedOptions().isEmpty()) {
            code.addStatement("$T $N = $T.of()", result.type, result, Map.class);
        } else {
            code.addStatement("$T $N = new $T<>($T.class)", result.type, result, EnumMap.class, sourceElement().optionEnumType());
        }
        for (Mapping namedOption : namedOptions()) {
            code.addStatement("$N.put($T.$L, new $T())",
                    result, sourceElement().optionEnumType(),
                    namedOption.enumName(), optionParserType(namedOption));
        }
        code.addStatement("return $N", result);
        return MethodSpec.methodBuilder("optionStates")
                .addCode(code.build())
                .returns(result.type)
                .addModifiers(PRIVATE)
                .build();
    });

    MethodSpec get() {
        return define.get();
    }

    private ClassName optionParserType(Mapping param) {
        if (param.isRepeatable()) {
            return ClassName.get(OptionStateRepeatable.class);
        }
        if (param.isNullary()) {
            return ClassName.get(OptionStateModeFlag.class);
        }
        return ClassName.get(OptionStateNonRepeatable.class);
    }
}