io.micronaut.annotation.processing.visitor.LoadedVisitor Maven / Gradle / Ivy
/*
* Copyright 2017-2020 original authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.micronaut.annotation.processing.visitor;
import io.micronaut.annotation.processing.GenericUtils;
import io.micronaut.core.annotation.AnnotationMetadata;
import io.micronaut.core.annotation.Internal;
import io.micronaut.core.order.Ordered;
import io.micronaut.core.reflect.GenericTypeUtils;
import io.micronaut.inject.visitor.TypeElementVisitor;
import io.micronaut.core.annotation.Nullable;
import javax.annotation.processing.ProcessingEnvironment;
import javax.lang.model.element.*;
import javax.lang.model.type.TypeMirror;
import java.util.List;
/**
* Used to store a reference to an underlying {@link TypeElementVisitor} and
* optionally invoke the visit methods on the visitor if it matches the
* element being visited by the annotation processor.
*
* @author James Kleeh
* @since 1.0
*/
@Internal
public class LoadedVisitor implements Ordered {
private static final String OBJECT_CLASS = Object.class.getName();
private final TypeElementVisitor visitor;
private final String classAnnotation;
private final String elementAnnotation;
private final JavaVisitorContext visitorContext;
private final JavaElementFactory elementFactory;
private JavaClassElement rootClassElement;
/**
* @param visitor The {@link TypeElementVisitor}
* @param visitorContext The visitor context
* @param genericUtils The generic utils
* @param processingEnvironment The {@link ProcessingEnvironment}
*/
public LoadedVisitor(TypeElementVisitor visitor,
JavaVisitorContext visitorContext,
GenericUtils genericUtils,
ProcessingEnvironment processingEnvironment) {
this.visitorContext = visitorContext;
this.elementFactory = visitorContext.getElementFactory();
this.visitor = visitor;
Class aClass = visitor.getClass();
TypeElement typeElement = processingEnvironment.getElementUtils().getTypeElement(aClass.getName());
if (typeElement != null) {
List generics = genericUtils.interfaceGenericTypesFor(typeElement, TypeElementVisitor.class.getName());
String typeName = generics.get(0).toString();
if (typeName.equals(OBJECT_CLASS)) {
classAnnotation = visitor.getClassType();
} else {
classAnnotation = typeName;
}
String elementName = generics.get(1).toString();
if (elementName.equals(OBJECT_CLASS)) {
elementAnnotation = visitor.getElementType();
} else {
elementAnnotation = elementName;
}
} else {
Class[] classes = GenericTypeUtils.resolveInterfaceTypeArguments(aClass, TypeElementVisitor.class);
if (classes != null && classes.length == 2) {
Class classGeneric = classes[0];
if (classGeneric == Object.class) {
classAnnotation = visitor.getClassType();
} else {
classAnnotation = classGeneric.getName();
}
Class elementGeneric = classes[1];
if (elementGeneric == Object.class) {
elementAnnotation = visitor.getElementType();
} else {
elementAnnotation = elementGeneric.getName();
}
} else {
classAnnotation = Object.class.getName();
elementAnnotation = Object.class.getName();
}
}
}
@Override
public int getOrder() {
return visitor.getOrder();
}
/**
* @return The visitor
*/
public TypeElementVisitor getVisitor() {
return visitor;
}
/**
* @param typeElement The class element
* @return True if the class element should be visited
*/
public boolean matches(TypeElement typeElement) {
if (classAnnotation.equals("java.lang.Object")) {
return true;
}
AnnotationMetadata annotationMetadata = visitorContext.getAnnotationUtils().getAnnotationMetadata(typeElement);
return annotationMetadata.hasStereotype(classAnnotation);
}
/**
* @param annotationMetadata The annotation data
* @return True if the element should be visited
*/
public boolean matches(AnnotationMetadata annotationMetadata) {
if (elementAnnotation.equals("java.lang.Object")) {
return true;
}
return annotationMetadata.hasStereotype(elementAnnotation);
}
/**
* Invoke the underlying visitor for the given element.
*
* @param element The element to visit
* @param annotationMetadata The annotation data for the node
*
* @return The element if one was created or null
*/
public @Nullable io.micronaut.inject.ast.Element visit(
Element element, AnnotationMetadata annotationMetadata) {
if (element instanceof VariableElement) {
if (element.getKind() == ElementKind.ENUM_CONSTANT) {
final JavaEnumConstantElement e = elementFactory.newEnumConstantElement(rootClassElement, (VariableElement) element, annotationMetadata);
visitor.visitEnumConstant(
e,
visitorContext
);
return e;
} else {
final JavaFieldElement e = elementFactory.newFieldElement(rootClassElement, (VariableElement) element, annotationMetadata);
visitor.visitField(
e,
visitorContext
);
return e;
}
} else if (element instanceof ExecutableElement) {
ExecutableElement executableElement = (ExecutableElement) element;
if (rootClassElement != null) {
if (executableElement.getSimpleName().toString().equals("")) {
final JavaConstructorElement e = elementFactory.newConstructorElement(rootClassElement, executableElement, annotationMetadata);
visitor.visitConstructor(
e,
visitorContext
);
return e;
} else {
final JavaMethodElement e = elementFactory.newSourceMethodElement(rootClassElement, executableElement, annotationMetadata);
visitor.visitMethod(
e,
visitorContext
);
return e;
}
}
} else if (element instanceof TypeElement) {
TypeElement typeElement = (TypeElement) element;
this.rootClassElement = elementFactory.newSourceClassElement(typeElement, annotationMetadata);
visitor.visitClass(
rootClassElement,
visitorContext
);
return rootClassElement;
}
return null;
}
@Override
public String toString() {
return visitor.toString();
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy