in.workarounds.bundler.compiler.model.ArgModel Maven / Gradle / Ivy

Go to download

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

Compiler for bundler library

The newest version!

package in.workarounds.bundler.compiler.model;

import com.squareup.javapoet.AnnotationSpec;
import com.squareup.javapoet.ClassName;
import com.squareup.javapoet.FieldSpec;
import com.squareup.javapoet.ParameterSpec;

import java.util.ArrayList;
import java.util.List;

import javax.lang.model.element.AnnotationMirror;
import javax.lang.model.element.Element;
import javax.lang.model.element.Modifier;

import in.workarounds.bundler.annotations.Arg;
import in.workarounds.bundler.annotations.Required;
import in.workarounds.bundler.compiler.Provider;
import in.workarounds.bundler.compiler.util.Utils;

/**
 * Created by madki on 30/10/15.
 */
public class ArgModel extends AnnotatedField {
    private Element element;
    private List supportAnnotations;
    private Required required;
    private boolean requireAll;

    public ArgModel(Element element, Provider provider, boolean requireAll, ClassName serializer, String keyValue) {
        super(element, provider, Arg.class, serializer, keyValue);
        this.element = element;
        this.requireAll = requireAll;

        supportAnnotations = new ArrayList<>();
        for(AnnotationMirror annotationMirror: element.getAnnotationMirrors()) {
           if(Utils.isSupportAnnotation(annotationMirror)) {
               supportAnnotations.add(AnnotationSpec.get(annotationMirror));
           }
        }

        required = element.getAnnotation(Required.class);
    }

    public boolean isRequired() {
        return required == null ? requireAll : required.value();
    }

    public List getSupportAnnotations() {
        return supportAnnotations;
    }

    public Element getElement() {
        return element;
    }

    @Override
    public ParameterSpec getAsParameter(Modifier... modifiers) {
        return super.getAsParameter(modifiers).toBuilder()
                .addAnnotations(getSupportAnnotations())
                .build();
    }

    @Override
    public FieldSpec getAsField(Modifier... modifiers) {
        return super.getAsField(modifiers).toBuilder()
                .addAnnotations(getSupportAnnotations())
                .build();
    }
}