| /* |
| * Copyright 2019 The Android Open Source Project |
| * |
| * 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 |
| * |
| * http://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 android.processor.view.inspector; |
| |
| import static javax.tools.Diagnostic.Kind.ERROR; |
| |
| import com.squareup.javapoet.ClassName; |
| |
| import java.io.IOException; |
| import java.util.HashMap; |
| import java.util.Map; |
| import java.util.Optional; |
| import java.util.Set; |
| |
| import javax.annotation.processing.AbstractProcessor; |
| import javax.annotation.processing.RoundEnvironment; |
| import javax.annotation.processing.SupportedAnnotationTypes; |
| import javax.lang.model.SourceVersion; |
| import javax.lang.model.element.Element; |
| import javax.lang.model.element.ElementKind; |
| import javax.lang.model.element.TypeElement; |
| |
| |
| /** |
| * An annotation processor for the platform inspectable annotations. |
| * |
| * It mostly delegates to {@link ModelProcessor} and {@link InspectionCompanionGenerator}. This |
| * modular architecture allows the core generation code to be reused for comparable annotations |
| * outside the platform, such as in AndroidX. |
| * |
| * @see android.view.inspector.InspectableNodeName |
| * @see android.view.inspector.InspectableProperty |
| */ |
| @SupportedAnnotationTypes({ |
| PlatformInspectableProcessor.NODE_NAME_QUALIFIED_NAME, |
| PlatformInspectableProcessor.PROPERTY_QUALIFIED_NAME |
| }) |
| public final class PlatformInspectableProcessor extends AbstractProcessor { |
| static final String NODE_NAME_QUALIFIED_NAME = |
| "android.view.inspector.InspectableNodeName"; |
| static final String PROPERTY_QUALIFIED_NAME = |
| "android.view.inspector.InspectableProperty"; |
| |
| @Override |
| public SourceVersion getSupportedSourceVersion() { |
| return SourceVersion.latest(); |
| } |
| |
| @Override |
| public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) { |
| final Map<String, InspectableClassModel> modelMap = new HashMap<>(); |
| |
| for (TypeElement annotation : annotations) { |
| if (annotation.getQualifiedName().contentEquals(NODE_NAME_QUALIFIED_NAME)) { |
| runModelProcessor( |
| roundEnv.getElementsAnnotatedWith(annotation), |
| new InspectableNodeNameProcessor(NODE_NAME_QUALIFIED_NAME, processingEnv), |
| modelMap); |
| |
| } else if (annotation.getQualifiedName().contentEquals(PROPERTY_QUALIFIED_NAME)) { |
| runModelProcessor( |
| roundEnv.getElementsAnnotatedWith(annotation), |
| new InspectablePropertyProcessor(PROPERTY_QUALIFIED_NAME, processingEnv), |
| modelMap); |
| |
| } else { |
| fail("Unexpected annotation type", annotation); |
| } |
| } |
| |
| final InspectionCompanionGenerator generator = |
| new InspectionCompanionGenerator(processingEnv.getFiler(), getClass()); |
| |
| for (InspectableClassModel model : modelMap.values()) { |
| try { |
| generator.generate(model); |
| } catch (IOException ioException) { |
| fail(String.format( |
| "Unable to generate inspection companion for %s due to %s", |
| model.getClassName().toString(), |
| ioException.getMessage())); |
| } |
| } |
| |
| return true; |
| } |
| |
| /** |
| * Run a {@link ModelProcessor} for a set of elements |
| * |
| * @param elements Elements to process, should be annotated correctly |
| * @param processor The processor to use |
| * @param modelMap A map of qualified class names to models |
| */ |
| private void runModelProcessor( |
| Set<? extends Element> elements, |
| ModelProcessor processor, |
| Map<String, InspectableClassModel> modelMap) { |
| for (Element element : elements) { |
| final Optional<TypeElement> classElement = enclosingClassElement(element); |
| |
| if (!classElement.isPresent()) { |
| fail("Element not contained in a class", element); |
| break; |
| } |
| |
| final InspectableClassModel model = modelMap.computeIfAbsent( |
| classElement.get().getQualifiedName().toString(), |
| k -> new InspectableClassModel(ClassName.get(classElement.get()))); |
| |
| processor.process(element, model); |
| } |
| } |
| |
| /** |
| * Get the nearest enclosing class if there is one. |
| * |
| * If {@param element} represents a class, it will be returned wrapped in an optional. |
| * |
| * @param element An element to search from |
| * @return A TypeElement of the nearest enclosing class or an empty optional |
| */ |
| private static Optional<TypeElement> enclosingClassElement(Element element) { |
| Element cursor = element; |
| |
| while (cursor != null) { |
| if (cursor.getKind() == ElementKind.CLASS) { |
| return Optional.of((TypeElement) cursor); |
| } |
| |
| cursor = cursor.getEnclosingElement(); |
| } |
| |
| return Optional.empty(); |
| } |
| |
| /** |
| * Print message and fail the build. |
| * |
| * @param message Message to print |
| */ |
| private void fail(String message) { |
| processingEnv.getMessager().printMessage(ERROR, message); |
| } |
| |
| /** |
| * Print message and fail the build. |
| * |
| * @param message Message to print |
| * @param element The element that failed |
| */ |
| private void fail(String message, Element element) { |
| processingEnv.getMessager().printMessage(ERROR, message, element); |
| } |
| } |