1 /*
2  * Copyright (C) 2020 The Dagger Authors.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 package dagger.hilt.processor.internal.aliasof;
18 
19 import static net.ltgt.gradle.incap.IncrementalAnnotationProcessorType.ISOLATING;
20 
21 import com.google.auto.service.AutoService;
22 import com.google.common.collect.ImmutableSet;
23 import dagger.hilt.processor.internal.BaseProcessor;
24 import dagger.hilt.processor.internal.ClassNames;
25 import dagger.hilt.processor.internal.ProcessorErrors;
26 import dagger.hilt.processor.internal.Processors;
27 import java.util.Set;
28 import javax.annotation.processing.Processor;
29 import javax.lang.model.element.AnnotationMirror;
30 import javax.lang.model.element.Element;
31 import javax.lang.model.element.TypeElement;
32 import net.ltgt.gradle.incap.IncrementalAnnotationProcessor;
33 
34 /** Processes the annotations annotated with {@link dagger.hilt.migration.AliasOf} */
35 @IncrementalAnnotationProcessor(ISOLATING)
36 @AutoService(Processor.class)
37 public final class AliasOfProcessor extends BaseProcessor {
38   @Override
getSupportedAnnotationTypes()39   public Set<String> getSupportedAnnotationTypes() {
40     return ImmutableSet.of(ClassNames.ALIAS_OF.toString());
41   }
42 
43   @Override
processEach(TypeElement annotation, Element element)44   public void processEach(TypeElement annotation, Element element) throws Exception {
45     ProcessorErrors.checkState(
46         Processors.hasAnnotation(element, ClassNames.SCOPE),
47         element,
48         "%s should only be used on scopes." + " However, it was found annotating %s",
49         annotation,
50         element);
51     AnnotationMirror annotationMirror =
52         Processors.getAnnotationMirror(element, ClassNames.ALIAS_OF);
53 
54     Element defineComponentScope =
55         Processors.getAnnotationClassValue(getElementUtils(), annotationMirror, "value");
56 
57     new AliasOfPropagatedDataGenerator(getProcessingEnv(), element, defineComponentScope)
58         .generate();
59   }
60 }
61