| /* |
| * Copyright (C) 2018 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 com.android.dialer.rootcomponentgenerator; |
| |
| import static javax.tools.Diagnostic.Kind.ERROR; |
| |
| import com.android.dialer.inject.IncludeInDialerRoot; |
| import com.android.dialer.inject.InstallIn; |
| import com.android.dialer.inject.RootComponentGeneratorMetadata; |
| import com.google.auto.common.BasicAnnotationProcessor.ProcessingStep; |
| import com.google.auto.common.MoreElements; |
| import com.google.common.collect.ImmutableSet; |
| import com.google.common.collect.SetMultimap; |
| import com.squareup.javapoet.AnnotationSpec; |
| import com.squareup.javapoet.TypeSpec; |
| import java.lang.annotation.Annotation; |
| import java.util.Collections; |
| import java.util.Set; |
| import javax.annotation.processing.ProcessingEnvironment; |
| import javax.lang.model.element.Element; |
| import javax.lang.model.element.TypeElement; |
| |
| /** |
| * Genereates metadata for every type annotated by {@link InstallIn} and {@link |
| * IncludeInDialerRoot}. |
| * |
| * <p>The metadata has the information where the annotated types are and it is used by annotation |
| * processor when the processor tries to generate root component. |
| */ |
| final class MetadataGeneratingStep implements ProcessingStep { |
| |
| private final ProcessingEnvironment processingEnv; |
| |
| MetadataGeneratingStep(ProcessingEnvironment processingEnv) { |
| this.processingEnv = processingEnv; |
| } |
| |
| @Override |
| public Set<? extends Class<? extends Annotation>> annotations() { |
| return ImmutableSet.of(IncludeInDialerRoot.class, InstallIn.class); |
| } |
| |
| @Override |
| public Set<? extends Element> process( |
| SetMultimap<Class<? extends Annotation>, Element> elementsByAnnotation) { |
| |
| for (Element element : elementsByAnnotation.get(IncludeInDialerRoot.class)) { |
| generateMetadataFor(IncludeInDialerRoot.class, MoreElements.asType(element)); |
| } |
| for (Element element : elementsByAnnotation.get(InstallIn.class)) { |
| if (element.getAnnotation(InstallIn.class).variants().length == 0) { |
| processingEnv |
| .getMessager() |
| .printMessage( |
| ERROR, String.format("@InstallIn %s must have at least one variant", element)); |
| continue; |
| } |
| generateMetadataFor(InstallIn.class, MoreElements.asType(element)); |
| } |
| |
| return Collections.emptySet(); |
| } |
| |
| private void generateMetadataFor( |
| Class<? extends Annotation> annotation, TypeElement annotatedElement) { |
| TypeSpec.Builder metadataClassBuilder = |
| TypeSpec.classBuilder( |
| annotatedElement.getQualifiedName().toString().replace('.', '_') + "Metadata"); |
| metadataClassBuilder.addAnnotation( |
| AnnotationSpec.builder(RootComponentGeneratorMetadata.class) |
| .addMember("tag", "$S", annotation.getSimpleName()) |
| .addMember("annotatedClass", "$T.class", annotatedElement.asType()) |
| .build()); |
| TypeSpec metadataClass = metadataClassBuilder.build(); |
| RootComponentUtils.writeJavaFile( |
| processingEnv, RootComponentUtils.METADATA_PACKAGE_NAME, metadataClass); |
| } |
| } |