org.mapstruct.ap.internal.prism.ValueMappingsPrism Maven / Gradle / Ivy
package org.mapstruct.ap.internal.prism;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import javax.lang.model.element.AnnotationMirror;
import javax.lang.model.element.Element;
import javax.lang.model.element.VariableElement;
import javax.lang.model.element.AnnotationValue;
import javax.lang.model.type.TypeMirror;
import net.java.dev.hickory.prism.internal.*;
import java.util.HashMap;
import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.TypeElement;
import javax.lang.model.util.ElementFilter;
/** A Prism representing an {@code @org.mapstruct.ValueMappings} annotation.
*/
public class ValueMappingsPrism {
/** store prism value of value */
private List _value;
/**
* An instance of the Values inner class whose
* methods return the AnnotationValues used to build this prism.
* Primarily intended to support using Messager.
*/
public final Values values;
/** Return a prism representing the {@code @org.mapstruct.ValueMappings} annotation on 'e'.
* similar to {@code e.getAnnotation(org.mapstruct.ValueMappings.class)} except that
* an instance of this class rather than an instance of {@code org.mapstruct.ValueMappings}
* is returned.
*/
public static ValueMappingsPrism getInstanceOn(Element e) {
AnnotationMirror m = getMirror("org.mapstruct.ValueMappings",e);
if(m == null) return null;
return getInstance(m);
}
/** Return a prism of the {@code @org.mapstruct.ValueMappings} annotation whose mirror is mirror.
*/
public static ValueMappingsPrism getInstance(AnnotationMirror mirror) {
return new ValueMappingsPrism(mirror);
}
private ValueMappingsPrism(AnnotationMirror mirror) {
for(ExecutableElement key : mirror.getElementValues().keySet()) {
memberValues.put(key.getSimpleName().toString(),mirror.getElementValues().get(key));
}
for(ExecutableElement member : ElementFilter.methodsIn(mirror.getAnnotationType().asElement().getEnclosedElements())) {
defaults.put(member.getSimpleName().toString(),member.getDefaultValue());
}
List valueMirrors = getArrayValues("value",AnnotationMirror.class);
_value = new ArrayList(valueMirrors.size());
for(AnnotationMirror valueMirror : valueMirrors) {
_value.add(ValueMappingPrism.getInstance(valueMirror));
}
this.values = new Values(memberValues);
this.mirror = mirror;
this.isValid = valid;
}
/**
* Returns a List representing the value of the {@code value()} member of the Annotation.
* @see org.mapstruct.ValueMappings#value()
*/
public List value() { return _value; }
/**
* Determine whether the underlying AnnotationMirror has no errors.
* True if the underlying AnnotationMirror has no errors.
* When true is returned, none of the methods will return null.
* When false is returned, a least one member will either return null, or another
* prism that is not valid.
*/
public final boolean isValid;
/**
* The underlying AnnotationMirror of the annotation
* represented by this Prism.
* Primarily intended to support using Messager.
*/
public final AnnotationMirror mirror;
/**
* A class whose members corespond to those of org.mapstruct.ValueMappings
* but which each return the AnnotationValue corresponding to
* that member in the model of the annotations. Returns null for
* defaulted members. Used for Messager, so default values are not useful.
*/
public static class Values {
private Map values;
private Values(Map values) {
this.values = values;
}
/** Return the AnnotationValue corresponding to the value()
* member of the annotation, or null when the default value is implied.
*/
public AnnotationValue value(){ return values.get("value");}
}
private Map defaults = new HashMap(10);
private Map memberValues = new HashMap(10);
private boolean valid = true;
private T getValue(String name, Class clazz) {
T result = ValueMappingsPrism.getValue(memberValues,defaults,name,clazz);
if(result == null) valid = false;
return result;
}
private List getArrayValues(String name, final Class clazz) {
List result = ValueMappingsPrism.getArrayValues(memberValues,defaults,name,clazz);
if(result == null) valid = false;
return result;
}
private static AnnotationMirror getMirror(String fqn, Element target) {
for (AnnotationMirror m :target.getAnnotationMirrors()) {
CharSequence mfqn = ((TypeElement)m.getAnnotationType().asElement()).getQualifiedName();
if(fqn.contentEquals(mfqn)) return m;
}
return null;
}
private static T getValue(Map memberValues, Map defaults, String name, Class clazz) {
AnnotationValue av = memberValues.get(name);
if(av == null) av = defaults.get(name);
if(av == null) {
return null;
}
if(clazz.isInstance(av.getValue())) return clazz.cast(av.getValue());
return null;
}
private static List getArrayValues(Map memberValues, Map defaults, String name, final Class clazz) {
AnnotationValue av = memberValues.get(name);
if(av == null) av = defaults.get(name);
if(av == null) {
return null;
}
if(av.getValue() instanceof List) {
List result = new ArrayList();
for(AnnotationValue v : getValueAsList(av)) {
if(clazz.isInstance(v.getValue())) {
result.add(clazz.cast(v.getValue()));
} else{
return null;
}
}
return result;
} else {
return null;
}
}
@SuppressWarnings("unchecked")
private static List getValueAsList(AnnotationValue av) {
return (List)av.getValue();
}
}