net.jbock.context.OptEnum Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jbock-compiler Show documentation

jbock annotation processor

There is a newer version: 5.18

package net.jbock.context;

import com.squareup.javapoet.TypeSpec;
import net.jbock.annotated.AnnotatedOption;
import net.jbock.common.EnumName;
import net.jbock.convert.Mapping;
import net.jbock.processor.SourceElement;

import javax.inject.Inject;
import java.util.List;

import static javax.lang.model.element.Modifier.PRIVATE;

/**
 * Defines the inner class *Parser.Opt.
 *
 * @see GeneratedClass
 */
@ContextScope
public class OptEnum {

    private final List> options;
    private final SourceElement sourceElement;

    @Inject
    OptEnum(List> options,
            SourceElement sourceElement) {
        this.options = options;
        this.sourceElement = sourceElement;
    }

    TypeSpec define() {
        TypeSpec.Builder spec = TypeSpec.enumBuilder(sourceElement.optionEnumType());
        for (Mapping option : options) {
            EnumName enumName = option.enumName();
            String enumConstant = enumName.enumConstant();
            spec.addEnumConstant(enumConstant);
        }
        return spec.addModifiers(PRIVATE)
                .build();
    }
}