com.flextrade.builder.model.PropertyTranslator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pojobuilder-maven-plugin Show documentation

Maven Plugin which generates Builders and matchers

There is a newer version: 1.0.1

Show newest version

package com.flextrade.builder.model;

import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.lang.reflect.TypeVariable;
import java.lang.reflect.WildcardType;
import java.util.Arrays;
import java.util.Map;

import com.sun.codemodel.JClass;
import com.sun.codemodel.JCodeModel;
import com.sun.codemodel.JType;
import com.sun.codemodel.JTypeVar;

public class PropertyTranslator {

    private final JCodeModel codeModel;
    private final Map typeVars;

    public PropertyTranslator(JCodeModel codeModel, Map typeVars) {
        this.codeModel = codeModel;
        this.typeVars = typeVars;
    }

    public JType getJType(Type type) {

        if(type instanceof Class && ((Class)type).isPrimitive())
            return codeModel._ref((Class)type);

        return getJClass(type);
    }

    private JClass getJClass(Type type) {
        if(type instanceof Class)
            return codeModel.ref((Class) type);

        if(type instanceof ParameterizedType) {
            return new ParameterizedJTypeEncoder(type).getJClass();
        }

        if(type instanceof TypeVariable) {
            return typeVars.get(((TypeVariable) type).getName());
        }

        throw new IllegalArgumentException("Cannot handle non-class types " + type.getTypeName());
    }

    private class ParameterizedJTypeEncoder {

        JClass raw;

        public ParameterizedJTypeEncoder(Type type) {
            ParameterizedType param = (ParameterizedType) type;

            raw = codeModel.ref((Class) param.getRawType());
            Arrays.stream(param.getActualTypeArguments()).forEach(ParameterizedJTypeEncoder.this::narrow);
        }

        private void narrow(Type type) {
            if(type instanceof WildcardType) {
                if(((WildcardType) type).getUpperBounds().length > 0) {
                    JClass upperBound = PropertyTranslator.this.getJClass(((WildcardType) type).getUpperBounds()[0]);
                    upperBound = upperBound.wildcard();
                    raw = raw.narrow(upperBound);
                } else {
                    raw = raw.narrow(codeModel.wildcard());
                }
            } else {
                raw = raw.narrow(getJType(type));
            }
        }

        public JClass getJClass() {
            return raw;
        }
    }


}