forked from androidannotations/androidannotations
/
ExtraHandler.java
115 lines (96 loc) · 4.68 KB
/
ExtraHandler.java
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
/**
* Copyright (C) 2010-2015 eBusiness Information, Excilys Group
*
* 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 org.androidannotations.handler;
import static com.sun.codemodel.JExpr.invoke;
import static com.sun.codemodel.JExpr.lit;
import static com.sun.codemodel.JMod.FINAL;
import static com.sun.codemodel.JMod.PUBLIC;
import static com.sun.codemodel.JMod.STATIC;
import javax.annotation.processing.ProcessingEnvironment;
import javax.lang.model.element.Element;
import javax.lang.model.type.TypeMirror;
import org.androidannotations.annotations.Extra;
import org.androidannotations.helper.APTCodeModelHelper;
import org.androidannotations.helper.AnnotationHelper;
import org.androidannotations.helper.BundleHelper;
import org.androidannotations.helper.CaseHelper;
import org.androidannotations.holder.HasExtras;
import org.androidannotations.holder.HasIntentBuilder;
import org.androidannotations.model.AnnotationElements;
import org.androidannotations.process.IsValid;
import com.sun.codemodel.JBlock;
import com.sun.codemodel.JClass;
import com.sun.codemodel.JExpr;
import com.sun.codemodel.JExpression;
import com.sun.codemodel.JFieldRef;
import com.sun.codemodel.JFieldVar;
import com.sun.codemodel.JMethod;
import com.sun.codemodel.JVar;
public class ExtraHandler extends BaseAnnotationHandler<HasExtras> {
private final AnnotationHelper annotationHelper;
private final APTCodeModelHelper codeModelHelper = new APTCodeModelHelper();
public ExtraHandler(ProcessingEnvironment processingEnvironment) {
super(Extra.class, processingEnvironment);
annotationHelper = new AnnotationHelper(processingEnv);
}
@Override
public void validate(Element element, AnnotationElements validatedElements, IsValid valid) {
/*
* TODO since we override setIntent(), we should check that the
* setIntent() method can be overridden
*/
validatorHelper.enclosingElementHasEActivity(element, validatedElements, valid);
validatorHelper.isNotPrivate(element, valid);
}
@Override
public void process(Element element, HasExtras holder) {
Extra annotation = element.getAnnotation(Extra.class);
String extraKey = annotation.value();
String fieldName = element.getSimpleName().toString();
if (extraKey.isEmpty()) {
extraKey = fieldName;
}
JFieldVar extraKeyStaticField = createStaticExtraField(holder, extraKey, fieldName);
injectExtraInComponent(element, holder, extraKeyStaticField, fieldName);
if (holder instanceof HasIntentBuilder) {
createIntentInjectionMethod(element, (HasIntentBuilder) holder, extraKeyStaticField, fieldName);
}
}
private JFieldVar createStaticExtraField(HasExtras holder, String extraKey, String fieldName) {
String staticFieldName = CaseHelper.camelCaseToUpperSnakeCase(null, fieldName, "Extra");
JFieldVar staticExtraField = holder.getGeneratedClass().fields().get(staticFieldName);
if (staticExtraField == null) {
staticExtraField = holder.getGeneratedClass().field(PUBLIC | STATIC | FINAL, classes().STRING, staticFieldName, lit(extraKey));
}
return staticExtraField;
}
private void injectExtraInComponent(Element element, HasExtras hasExtras, JFieldVar extraKeyStaticField, String fieldName) {
JMethod injectExtrasMethod = hasExtras.getInjectExtrasMethod();
JVar extras = hasExtras.getInjectExtras();
JBlock injectExtrasBlock = hasExtras.getInjectExtrasBlock();
TypeMirror type = codeModelHelper.getActualType(element, hasExtras);
JClass elementClass = codeModelHelper.typeMirrorToJClass(element.asType(), hasExtras);
BundleHelper bundleHelper = new BundleHelper(annotationHelper, type);
JFieldRef extraField = JExpr.ref(fieldName);
JExpression intent = invoke("getIntent");
JBlock ifContainsKey = injectExtrasBlock._if(JExpr.invoke(extras, "containsKey").arg(extraKeyStaticField))._then();
JExpression restoreMethodCall = bundleHelper.getExpressionToRestoreFromIntentOrBundle(elementClass, intent, extras, extraKeyStaticField, injectExtrasMethod, hasExtras);
ifContainsKey.assign(extraField, restoreMethodCall);
}
private void createIntentInjectionMethod(Element element, HasIntentBuilder holder, JFieldVar extraKeyStaticField, String fieldName) {
holder.getIntentBuilder().getPutExtraMethod(element.asType(), fieldName, extraKeyStaticField);
}
}