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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
/*
* Copyright (C) 2020 The Dagger 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
*
* 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 dagger.hilt.android.processor.internal.customtestapplication;
import com.squareup.javapoet.FieldSpec;
import com.squareup.javapoet.JavaFile;
import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.ParameterSpec;
import com.squareup.javapoet.ParameterizedTypeName;
import com.squareup.javapoet.TypeName;
import com.squareup.javapoet.TypeSpec;
import dagger.hilt.processor.internal.ClassNames;
import dagger.hilt.processor.internal.Processors;
import java.io.IOException;
import javax.annotation.processing.ProcessingEnvironment;
import javax.lang.model.element.Modifier;
/**
* Generates an Android Application that holds the Singleton component.
*/
final class CustomTestApplicationGenerator {
private static final ParameterSpec COMPONENT_MANAGER =
ParameterSpec.builder(ClassNames.TEST_APPLICATION_COMPONENT_MANAGER, "componentManager")
.build();
private final ProcessingEnvironment processingEnv;
private final CustomTestApplicationMetadata metadata;
public CustomTestApplicationGenerator(
ProcessingEnvironment processingEnv, CustomTestApplicationMetadata metadata) {
this.processingEnv = processingEnv;
this.metadata = metadata;
}
public void generate() throws IOException {
TypeSpec.Builder generator =
TypeSpec.classBuilder(metadata.appName())
.addOriginatingElement(metadata.element())
.addModifiers(Modifier.PUBLIC, Modifier.FINAL)
.superclass(metadata.baseAppName())
.addSuperinterface(
ParameterizedTypeName.get(ClassNames.GENERATED_COMPONENT_MANAGER, TypeName.OBJECT))
.addSuperinterface(ClassNames.TEST_APPLICATION_COMPONENT_MANAGER_HOLDER)
.addField(getComponentManagerField())
.addMethod(getAttachBaseContextMethod())
.addMethod(getComponentManagerMethod())
.addMethod(getComponentMethod());
Processors.addGeneratedAnnotation(
generator, processingEnv, CustomTestApplicationProcessor.class);
JavaFile.builder(metadata.appName().packageName(), generator.build())
.build()
.writeTo(processingEnv.getFiler());
}
// Initialize this in attachBaseContext to not pull it into the main dex.
/** private TestApplicationComponentManager componentManager; */
private static FieldSpec getComponentManagerField() {
return FieldSpec.builder(COMPONENT_MANAGER.type, COMPONENT_MANAGER.name, Modifier.PRIVATE)
.build();
}
/**
* Initializes application fields. These fields are initialized in attachBaseContext to avoid
* potential multidexing issues.
*
* <pre><code>
* {@literal @Override} protected void attachBaseContext(Context base) {
* super.attachBaseContext(base);
* componentManager = new TestApplicationComponentManager(this);
* }
* </code></pre>
*/
private static MethodSpec getAttachBaseContextMethod() {
return MethodSpec.methodBuilder("attachBaseContext")
.addAnnotation(Override.class)
.addModifiers(Modifier.PROTECTED, Modifier.FINAL)
.addParameter(ClassNames.CONTEXT, "base")
.addStatement("super.attachBaseContext(base)")
.addStatement("$N = new $T(this)", COMPONENT_MANAGER, COMPONENT_MANAGER.type)
.build();
}
private static MethodSpec getComponentMethod() {
return MethodSpec.methodBuilder("generatedComponent")
.addAnnotation(Override.class)
.addModifiers(Modifier.PUBLIC, Modifier.FINAL)
.returns(TypeName.OBJECT)
.addStatement("return $N.generatedComponent()", COMPONENT_MANAGER)
.build();
}
private static MethodSpec getComponentManagerMethod() {
return MethodSpec.methodBuilder("componentManager")
.addAnnotation(Override.class)
.addModifiers(Modifier.PUBLIC, Modifier.FINAL)
.returns(TypeName.OBJECT)
.addStatement("return $N", COMPONENT_MANAGER)
.build();
}
}
|
