querydsl/querydsl-apt/src/main/java/com/mysema/query/apt/Configuration.java

/*
 * Copyright (c) 2009 Mysema Ltd.
 * All rights reserved.
 * 
 */
package com.mysema.query.apt;

import java.lang.annotation.Annotation;
import java.util.List;

import javax.lang.model.element.Element;
import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.Modifier;
import javax.lang.model.element.TypeElement;
import javax.lang.model.element.VariableElement;

import com.mysema.commons.lang.Assert;

/**
 * @author tiwe
 *
 */
public class Configuration {
    
    protected final Class<? extends Annotation> entityAnn, superTypeAnn, embeddableAnn, dtoAnn, skipAnn;
    
    private boolean useFields = true, useGetters = true;
    
    public Configuration(
            Class<? extends Annotation> entityAnn, 
            Class<? extends Annotation> superTypeAnn,
            Class<? extends Annotation> embeddableAnn,
            Class<? extends Annotation> dtoAnn,
            Class<? extends Annotation> skipAnn) {
        this.entityAnn = Assert.notNull(entityAnn);
        this.superTypeAnn = superTypeAnn;
        this.embeddableAnn = embeddableAnn;
        this.dtoAnn = dtoAnn;
        this.skipAnn = skipAnn;             
    }
    
    public VisitorConfig getConfig(TypeElement e, List<? extends Element> elements){
        return VisitorConfig.ALL;
    }
    
    public boolean isValidConstructor(ExecutableElement constructor) {
        return constructor.getModifiers().contains(Modifier.PUBLIC)
            && !constructor.getParameters().isEmpty();
    }
    
    public boolean isValidField(VariableElement field) {
        return useFields
            && field.getAnnotation(skipAnn) == null
            && !field.getModifiers().contains(Modifier.TRANSIENT) 
            && !field.getModifiers().contains(Modifier.STATIC);
    }

    public boolean isValidGetter(ExecutableElement getter){
        return useGetters
            && getter.getAnnotation(skipAnn) == null
            && !getter.getModifiers().contains(Modifier.STATIC);
    }

    public Class<? extends Annotation> getEntityAnn() {
        return entityAnn;
    }

    public Class<? extends Annotation> getSuperTypeAnn() {
        return superTypeAnn;
    }

    public Class<? extends Annotation> getEmbeddableAnn() {
        return embeddableAnn;
    }
    
    public Class<? extends Annotation> getDtoAnn() {
        return dtoAnn;
    }

    public Class<? extends Annotation> getSkipAnn() {
        return skipAnn;
    }

    public boolean isUseFields() {
        return useFields;
    }

    public boolean isUseGetters() {
        return useGetters;
    }

    public void setUseGetters(boolean b) {
        this.useGetters = b;        
    }
    
    public void setUseFields(boolean b){
        this.useFields = b;
    }
    

}