blob: 62dfeb35e618d382e06d6583062bfa4e462ba908 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
/*
* Copyright 2014 Google LLC
*
* 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.google.auto.value.processor;
import static com.google.auto.value.processor.AutoValueOrOneOfProcessor.hasAnnotationMirror;
import static com.google.auto.value.processor.ClassNames.AUTO_VALUE_BUILDER_NAME;
import static com.google.auto.value.processor.ClassNames.AUTO_VALUE_NAME;
import com.google.auto.common.SuperficialValidation;
import com.google.auto.service.AutoService;
import java.util.Set;
import javax.annotation.processing.AbstractProcessor;
import javax.annotation.processing.Processor;
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.TypeElement;
import javax.tools.Diagnostic;
import net.ltgt.gradle.incap.IncrementalAnnotationProcessor;
import net.ltgt.gradle.incap.IncrementalAnnotationProcessorType;
/**
* Annotation processor that checks that the type that {@code AutoValue.Builder} is applied to is
* nested inside an {@code @AutoValue} class. The actual code generation for builders is done in
* {@link AutoValueProcessor}.
*
* @author Éamonn McManus
*/
@AutoService(Processor.class)
@IncrementalAnnotationProcessor(IncrementalAnnotationProcessorType.ISOLATING)
@SupportedAnnotationTypes(AUTO_VALUE_BUILDER_NAME)
public class AutoValueBuilderProcessor extends AbstractProcessor {
@Override
public SourceVersion getSupportedSourceVersion() {
return SourceVersion.latest();
}
@Override
public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
TypeElement autoValueBuilder =
processingEnv.getElementUtils().getTypeElement(AUTO_VALUE_BUILDER_NAME);
Set<? extends Element> builderTypes = roundEnv.getElementsAnnotatedWith(autoValueBuilder);
if (!SuperficialValidation.validateElements(builderTypes)) {
return false;
}
for (Element annotatedType : builderTypes) {
// Double-check that the annotation is there. Sometimes the compiler gets confused in case of
// erroneous source code. SuperficialValidation should protect us against this but it doesn't
// cost anything to check again.
if (hasAnnotationMirror(annotatedType, AUTO_VALUE_BUILDER_NAME)) {
validate(
annotatedType,
"@AutoValue.Builder can only be applied to a class or interface inside an"
+ " @AutoValue class");
}
}
return false;
}
private void validate(Element annotatedType, String errorMessage) {
Element container = annotatedType.getEnclosingElement();
if (!hasAnnotationMirror(container, AUTO_VALUE_NAME)) {
processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR, errorMessage, annotatedType);
}
}
}
|
