aboutsummaryrefslogtreecommitdiff
path: root/src/core/lombok
diff options
context:
space:
mode:
authorReinier Zwitserloot <reinier@zwitserloot.com>2010-07-19 21:18:16 +0200
committerReinier Zwitserloot <reinier@zwitserloot.com>2010-07-19 21:18:16 +0200
commit29cd0ac3e7571b37478bf831a1f4cb8e2653ce1a (patch)
treeb2990a5b498656a4e1d0ec074e8b7bf9b8312922 /src/core/lombok
parentb47e87f8735f2e498beb9372bbd6fb4347e27e34 (diff)
downloadlombok-29cd0ac3e7571b37478bf831a1f4cb8e2653ce1a.tar.gz
lombok-29cd0ac3e7571b37478bf831a1f4cb8e2653ce1a.tar.bz2
lombok-29cd0ac3e7571b37478bf831a1f4cb8e2653ce1a.zip
Added support for @AllArgsConstructor, @NoArgsConstructor, and @RequiredArgsConstructor for javac.
Diffstat (limited to 'src/core/lombok')
-rw-r--r--src/core/lombok/AllArgsConstructor.java38
-rw-r--r--src/core/lombok/NoArgsConstructor.java41
-rw-r--r--src/core/lombok/RequiredArgsConstructor.java38
-rw-r--r--src/core/lombok/javac/handlers/HandleConstructor.java223
-rw-r--r--src/core/lombok/javac/handlers/HandleData.java113
5 files changed, 343 insertions, 110 deletions
diff --git a/src/core/lombok/AllArgsConstructor.java b/src/core/lombok/AllArgsConstructor.java
new file mode 100644
index 00000000..605dc0f2
--- /dev/null
+++ b/src/core/lombok/AllArgsConstructor.java
@@ -0,0 +1,38 @@
+/*
+ * Copyright © 2010 Reinier Zwitserloot and Roel Spilker.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ * THE SOFTWARE.
+ */
+package lombok;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * Generates an all-args constructor. An all-args constructor requires one argument
+ * for every field in the class.
+ */
+@Target(ElementType.TYPE)
+@Retention(RetentionPolicy.SOURCE)
+public @interface AllArgsConstructor {
+ String staticName() default "";
+ AccessLevel access() default lombok.AccessLevel.PUBLIC;
+}
diff --git a/src/core/lombok/NoArgsConstructor.java b/src/core/lombok/NoArgsConstructor.java
new file mode 100644
index 00000000..4d0424ec
--- /dev/null
+++ b/src/core/lombok/NoArgsConstructor.java
@@ -0,0 +1,41 @@
+/*
+ * Copyright © 2010 Reinier Zwitserloot and Roel Spilker.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ * THE SOFTWARE.
+ */
+package lombok;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * Generates a no-args constructor. Will generate an error message if such
+ * a constructor cannot be written due to the existence of final fields.
+ *
+ * NB: Fields with constraints such as {@code @NonNull} will <em>NOT</em> be checked
+ * in such a constructor, of course!
+ */
+@Target(ElementType.TYPE)
+@Retention(RetentionPolicy.SOURCE)
+public @interface NoArgsConstructor {
+ String staticName() default "";
+ AccessLevel access() default lombok.AccessLevel.PUBLIC;
+}
diff --git a/src/core/lombok/RequiredArgsConstructor.java b/src/core/lombok/RequiredArgsConstructor.java
new file mode 100644
index 00000000..c2b2605c
--- /dev/null
+++ b/src/core/lombok/RequiredArgsConstructor.java
@@ -0,0 +1,38 @@
+/*
+ * Copyright © 2010 Reinier Zwitserloot and Roel Spilker.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ * THE SOFTWARE.
+ */
+package lombok;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * Generates a constructor with required arguments.
+ * Required arguments are final fields and fields with constraints such as {@code @NonNull}.
+ */
+@Target(ElementType.TYPE)
+@Retention(RetentionPolicy.SOURCE)
+public @interface RequiredArgsConstructor {
+ String staticName() default "";
+ AccessLevel access() default lombok.AccessLevel.PUBLIC;
+}
diff --git a/src/core/lombok/javac/handlers/HandleConstructor.java b/src/core/lombok/javac/handlers/HandleConstructor.java
new file mode 100644
index 00000000..2f4a8474
--- /dev/null
+++ b/src/core/lombok/javac/handlers/HandleConstructor.java
@@ -0,0 +1,223 @@
+/*
+ * Copyright © 2010 Reinier Zwitserloot, Roel Spilker and Robbert Jan Grootjans.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+ * THE SOFTWARE.
+ */
+package lombok.javac.handlers;
+
+import static lombok.javac.handlers.JavacHandlerUtil.*;
+import lombok.AccessLevel;
+import lombok.AllArgsConstructor;
+import lombok.NoArgsConstructor;
+import lombok.RequiredArgsConstructor;
+import lombok.core.AST.Kind;
+import lombok.core.AnnotationValues;
+import lombok.core.handlers.TransformationsUtil;
+import lombok.javac.Javac;
+import lombok.javac.JavacAnnotationHandler;
+import lombok.javac.JavacNode;
+
+import org.mangosdk.spi.ProviderFor;
+
+import com.sun.tools.javac.code.Flags;
+import com.sun.tools.javac.tree.JCTree.JCAnnotation;
+import com.sun.tools.javac.tree.JCTree.JCAssign;
+import com.sun.tools.javac.tree.JCTree.JCBlock;
+import com.sun.tools.javac.tree.JCTree.JCClassDecl;
+import com.sun.tools.javac.tree.JCTree.JCExpression;
+import com.sun.tools.javac.tree.JCTree.JCFieldAccess;
+import com.sun.tools.javac.tree.JCTree.JCIdent;
+import com.sun.tools.javac.tree.JCTree.JCMethodDecl;
+import com.sun.tools.javac.tree.JCTree.JCModifiers;
+import com.sun.tools.javac.tree.JCTree.JCReturn;
+import com.sun.tools.javac.tree.JCTree.JCStatement;
+import com.sun.tools.javac.tree.JCTree.JCTypeApply;
+import com.sun.tools.javac.tree.JCTree.JCTypeParameter;
+import com.sun.tools.javac.tree.JCTree.JCVariableDecl;
+import com.sun.tools.javac.tree.TreeMaker;
+import com.sun.tools.javac.util.List;
+
+public class HandleConstructor {
+ @ProviderFor(JavacAnnotationHandler.class)
+ public static class HandleNoArgsConstructor implements JavacAnnotationHandler<NoArgsConstructor> {
+ @Override public boolean handle(AnnotationValues<NoArgsConstructor> annotation, JCAnnotation ast, JavacNode annotationNode) {
+ markAnnotationAsProcessed(annotationNode, NoArgsConstructor.class);
+ JavacNode typeNode = annotationNode.up();
+ List<JavacNode> fields = List.nil();
+ NoArgsConstructor ann = annotation.getInstance();
+ AccessLevel level = ann.access();
+ String staticName = ann.staticName();
+ new HandleConstructor().generateConstructor(level, typeNode, fields, staticName, false);
+ return true;
+ }
+ }
+
+ @ProviderFor(JavacAnnotationHandler.class)
+ public static class HandleRequiredArgsConstructor implements JavacAnnotationHandler<RequiredArgsConstructor> {
+ @Override public boolean handle(AnnotationValues<RequiredArgsConstructor> annotation, JCAnnotation ast, JavacNode annotationNode) {
+ markAnnotationAsProcessed(annotationNode, RequiredArgsConstructor.class);
+ JavacNode typeNode = annotationNode.up();
+ List<JavacNode> fields = List.nil();
+ for (JavacNode child : typeNode.down()) {
+ if (child.getKind() != Kind.FIELD) continue;
+ JCVariableDecl fieldDecl = (JCVariableDecl) child.get();
+ //Skip fields that start with $
+ if (fieldDecl.name.toString().startsWith("$")) continue;
+ long fieldFlags = fieldDecl.mods.flags;
+ //Skip static fields.
+ if ((fieldFlags & Flags.STATIC) != 0) continue;
+ boolean isFinal = (fieldFlags & Flags.FINAL) != 0;
+ boolean isNonNull = !findAnnotations(child, TransformationsUtil.NON_NULL_PATTERN).isEmpty();
+ if ((isFinal || isNonNull) && fieldDecl.init == null) fields = fields.append(child);
+ }
+ RequiredArgsConstructor ann = annotation.getInstance();
+ AccessLevel level = ann.access();
+ String staticName = ann.staticName();
+ new HandleConstructor().generateConstructor(level, typeNode, fields, staticName, false);
+ return true;
+ }
+ }
+
+ @ProviderFor(JavacAnnotationHandler.class)
+ public static class HandleAllArgsConstructor implements JavacAnnotationHandler<AllArgsConstructor> {
+ @Override public boolean handle(AnnotationValues<AllArgsConstructor> annotation, JCAnnotation ast, JavacNode annotationNode) {
+ markAnnotationAsProcessed(annotationNode, AllArgsConstructor.class);
+ JavacNode typeNode = annotationNode.up();
+ List<JavacNode> fields = List.nil();
+ for (JavacNode child : typeNode.down()) {
+ if (child.getKind() != Kind.FIELD) continue;
+ JCVariableDecl fieldDecl = (JCVariableDecl) child.get();
+ //Skip fields that start with $
+ if (fieldDecl.name.toString().startsWith("$")) continue;
+ long fieldFlags = fieldDecl.mods.flags;
+ //Skip static fields.
+ if ((fieldFlags & Flags.STATIC) != 0) continue;
+ fields = fields.append(child);
+ }
+ AllArgsConstructor ann = annotation.getInstance();
+ AccessLevel level = ann.access();
+ String staticName = ann.staticName();
+ new HandleConstructor().generateConstructor(level, typeNode, fields, staticName, false);
+ return true;
+ }
+ }
+
+ public void generateConstructor(AccessLevel level, JavacNode typeNode, List<JavacNode> fields, String staticName, boolean skipIfConstructorExists) {
+ if (skipIfConstructorExists && constructorExists(typeNode) != MemberExistsResult.NOT_EXISTS) return;
+ if (skipIfConstructorExists) {
+ for (JavacNode child : typeNode.down()) {
+ if (child.getKind() == Kind.ANNOTATION) {
+ if (Javac.annotationTypeMatches(NoArgsConstructor.class, child) ||
+ Javac.annotationTypeMatches(AllArgsConstructor.class, child) ||
+ Javac.annotationTypeMatches(RequiredArgsConstructor.class, child))
+ return;
+ }
+ }
+ }
+
+ boolean staticConstrRequired = staticName != null && !staticName.equals("");
+
+ JCMethodDecl constr = createConstructor(staticConstrRequired ? AccessLevel.PRIVATE : level, typeNode, fields);
+ injectMethod(typeNode, constr);
+ if (staticConstrRequired) {
+ JCMethodDecl staticConstr = createStaticConstructor(staticName, level, typeNode, fields);
+ injectMethod(typeNode, staticConstr);
+ }
+ }
+
+ private JCMethodDecl createConstructor(AccessLevel level, JavacNode typeNode, List<JavacNode> fields) {
+ TreeMaker maker = typeNode.getTreeMaker();
+ JCClassDecl type = (JCClassDecl) typeNode.get();
+
+ List<JCStatement> nullChecks = List.nil();
+ List<JCStatement> assigns = List.nil();
+ List<JCVariableDecl> params = List.nil();
+
+ for (JavacNode fieldNode : fields) {
+ JCVariableDecl field = (JCVariableDecl) fieldNode.get();
+ List<JCAnnotation> nonNulls = findAnnotations(fieldNode, TransformationsUtil.NON_NULL_PATTERN);
+ List<JCAnnotation> nullables = findAnnotations(fieldNode, TransformationsUtil.NULLABLE_PATTERN);
+ JCVariableDecl param = maker.VarDef(maker.Modifiers(Flags.FINAL, nonNulls.appendList(nullables)), field.name, field.vartype, null);
+ params = params.append(param);
+ JCFieldAccess thisX = maker.Select(maker.Ident(fieldNode.toName("this")), field.name);
+ JCAssign assign = maker.Assign(thisX, maker.Ident(field.name));
+ assigns = assigns.append(maker.Exec(assign));
+
+ if (!nonNulls.isEmpty()) {
+ JCStatement nullCheck = generateNullCheck(maker, fieldNode);
+ if (nullCheck != null) nullChecks = nullChecks.append(nullCheck);
+ }
+ }
+
+ JCModifiers mods = maker.Modifiers(toJavacModifier(level));
+ return maker.MethodDef(mods, typeNode.toName("<init>"),
+ null, type.typarams, params, List.<JCExpression>nil(), maker.Block(0L, nullChecks.appendList(assigns)), null);
+ }
+
+ private JCMethodDecl createStaticConstructor(String name, AccessLevel level, JavacNode typeNode, List<JavacNode> fields) {
+ TreeMaker maker = typeNode.getTreeMaker();
+ JCClassDecl type = (JCClassDecl) typeNode.get();
+
+ JCModifiers mods = maker.Modifiers(Flags.STATIC | toJavacModifier(level));
+
+ JCExpression returnType, constructorType;
+
+ List<JCTypeParameter> typeParams = List.nil();
+ List<JCVariableDecl> params = List.nil();
+ List<JCExpression> typeArgs1 = List.nil();
+ List<JCExpression> typeArgs2 = List.nil();
+ List<JCExpression> args = List.nil();
+
+ if (!type.typarams.isEmpty()) {
+ for (JCTypeParameter param : type.typarams) {
+ typeArgs1 = typeArgs1.append(maker.Ident(param.name));
+ typeArgs2 = typeArgs2.append(maker.Ident(param.name));
+ typeParams = typeParams.append(maker.TypeParameter(param.name, param.bounds));
+ }
+ returnType = maker.TypeApply(maker.Ident(type.name), typeArgs1);
+ constructorType = maker.TypeApply(maker.Ident(type.name), typeArgs2);
+ } else {
+ returnType = maker.Ident(type.name);
+ constructorType = maker.Ident(type.name);
+ }
+
+ for (JavacNode fieldNode : fields) {
+ JCVariableDecl field = (JCVariableDecl) fieldNode.get();
+ JCExpression pType;
+ if (field.vartype instanceof JCIdent) pType = maker.Ident(((JCIdent)field.vartype).name);
+ else if (field.vartype instanceof JCTypeApply) {
+ JCTypeApply typeApply = (JCTypeApply) field.vartype;
+ List<JCExpression> tArgs = List.nil();
+ for (JCExpression arg : typeApply.arguments) tArgs = tArgs.append(arg);
+ pType = maker.TypeApply(typeApply.clazz, tArgs);
+ } else {
+ pType = field.vartype;
+ }
+ List<JCAnnotation> nonNulls = findAnnotations(fieldNode, TransformationsUtil.NON_NULL_PATTERN);
+ List<JCAnnotation> nullables = findAnnotations(fieldNode, TransformationsUtil.NULLABLE_PATTERN);
+ JCVariableDecl param = maker.VarDef(maker.Modifiers(Flags.FINAL, nonNulls.appendList(nullables)), field.name, pType, null);
+ params = params.append(param);
+ args = args.append(maker.Ident(field.name));
+ }
+ JCReturn returnStatement = maker.Return(maker.NewClass(null, List.<JCExpression>nil(), constructorType, args, null));
+ JCBlock body = maker.Block(0, List.<JCStatement>of(returnStatement));
+
+ return maker.MethodDef(mods, typeNode.toName(name), returnType, typeParams, params, List.<JCExpression>nil(), body, null);
+ }
+}
diff --git a/src/core/lombok/javac/handlers/HandleData.java b/src/core/lombok/javac/handlers/HandleData.java
index efd86bba..3bdd2dce 100644
--- a/src/core/lombok/javac/handlers/HandleData.java
+++ b/src/core/lombok/javac/handlers/HandleData.java
@@ -21,43 +21,24 @@
*/
package lombok.javac.handlers;
-import static lombok.javac.handlers.JavacHandlerUtil.constructorExists;
-import static lombok.javac.handlers.JavacHandlerUtil.findAnnotations;
-import static lombok.javac.handlers.JavacHandlerUtil.generateNullCheck;
-import static lombok.javac.handlers.JavacHandlerUtil.injectMethod;
-import static lombok.javac.handlers.JavacHandlerUtil.markAnnotationAsProcessed;
-import static lombok.javac.handlers.JavacHandlerUtil.methodExists;
+import static lombok.javac.handlers.JavacHandlerUtil.*;
-import java.lang.reflect.Modifier;
import java.util.LinkedHashMap;
import java.util.Map;
import lombok.AccessLevel;
import lombok.Data;
-import lombok.core.AnnotationValues;
import lombok.core.AST.Kind;
+import lombok.core.AnnotationValues;
import lombok.core.handlers.TransformationsUtil;
import lombok.javac.JavacAnnotationHandler;
import lombok.javac.JavacNode;
-import lombok.javac.handlers.JavacHandlerUtil.MemberExistsResult;
import org.mangosdk.spi.ProviderFor;
import com.sun.tools.javac.code.Flags;
-import com.sun.tools.javac.tree.TreeMaker;
import com.sun.tools.javac.tree.JCTree.JCAnnotation;
-import com.sun.tools.javac.tree.JCTree.JCAssign;
-import com.sun.tools.javac.tree.JCTree.JCBlock;
import com.sun.tools.javac.tree.JCTree.JCClassDecl;
-import com.sun.tools.javac.tree.JCTree.JCExpression;
-import com.sun.tools.javac.tree.JCTree.JCFieldAccess;
-import com.sun.tools.javac.tree.JCTree.JCIdent;
-import com.sun.tools.javac.tree.JCTree.JCMethodDecl;
-import com.sun.tools.javac.tree.JCTree.JCModifiers;
-import com.sun.tools.javac.tree.JCTree.JCReturn;
-import com.sun.tools.javac.tree.JCTree.JCStatement;
-import com.sun.tools.javac.tree.JCTree.JCTypeApply;
-import com.sun.tools.javac.tree.JCTree.JCTypeParameter;
import com.sun.tools.javac.tree.JCTree.JCVariableDecl;
import com.sun.tools.javac.util.List;
@@ -97,15 +78,7 @@ public class HandleData implements JavacAnnotationHandler<Data> {
String staticConstructorName = annotation.getInstance().staticConstructor();
- if (constructorExists(typeNode) == MemberExistsResult.NOT_EXISTS) {
- JCMethodDecl constructor = createConstructor(staticConstructorName.equals(""), typeNode, nodesForConstructor);
- injectMethod(typeNode, constructor);
- }
-
- if (!staticConstructorName.isEmpty() && methodExists("of", typeNode, false) == MemberExistsResult.NOT_EXISTS) {
- JCMethodDecl staticConstructor = createStaticConstructor(staticConstructorName, typeNode, nodesForConstructor);
- injectMethod(typeNode, staticConstructor);
- }
+ new HandleConstructor().generateConstructor(AccessLevel.PUBLIC, typeNode, nodesForConstructor, staticConstructorName, true);
for (Map.Entry<JavacNode, Boolean> field : gettersAndSetters.entrySet()) {
new HandleGetter().generateGetterForField(field.getKey(), annotationNode.get(), AccessLevel.PUBLIC, true);
@@ -117,84 +90,4 @@ public class HandleData implements JavacAnnotationHandler<Data> {
return true;
}
-
- private JCMethodDecl createConstructor(boolean isPublic, JavacNode typeNode, List<JavacNode> fields) {
- TreeMaker maker = typeNode.getTreeMaker();
- JCClassDecl type = (JCClassDecl) typeNode.get();
-
- List<JCStatement> nullChecks = List.nil();
- List<JCStatement> assigns = List.nil();
- List<JCVariableDecl> params = List.nil();
-
- for (JavacNode fieldNode : fields) {
- JCVariableDecl field = (JCVariableDecl) fieldNode.get();
- List<JCAnnotation> nonNulls = findAnnotations(fieldNode, TransformationsUtil.NON_NULL_PATTERN);
- List<JCAnnotation> nullables = findAnnotations(fieldNode, TransformationsUtil.NULLABLE_PATTERN);
- JCVariableDecl param = maker.VarDef(maker.Modifiers(Flags.FINAL, nonNulls.appendList(nullables)), field.name, field.vartype, null);
- params = params.append(param);
- JCFieldAccess thisX = maker.Select(maker.Ident(fieldNode.toName("this")), field.name);
- JCAssign assign = maker.Assign(thisX, maker.Ident(field.name));
- assigns = assigns.append(maker.Exec(assign));
-
- if (!nonNulls.isEmpty()) {
- JCStatement nullCheck = generateNullCheck(maker, fieldNode);
- if (nullCheck != null) nullChecks = nullChecks.append(nullCheck);
- }
- }
-
- JCModifiers mods = maker.Modifiers(isPublic ? Modifier.PUBLIC : Modifier.PRIVATE);
- return maker.MethodDef(mods, typeNode.toName("<init>"),
- null, type.typarams, params, List.<JCExpression>nil(), maker.Block(0L, nullChecks.appendList(assigns)), null);
- }
-
- private JCMethodDecl createStaticConstructor(String name, JavacNode typeNode, List<JavacNode> fields) {
- TreeMaker maker = typeNode.getTreeMaker();
- JCClassDecl type = (JCClassDecl) typeNode.get();
-
- JCModifiers mods = maker.Modifiers(Flags.STATIC | Flags.PUBLIC);
-
- JCExpression returnType, constructorType;
-
- List<JCTypeParameter> typeParams = List.nil();
- List<JCVariableDecl> params = List.nil();
- List<JCExpression> typeArgs1 = List.nil();
- List<JCExpression> typeArgs2 = List.nil();
- List<JCExpression> args = List.nil();
-
- if (!type.typarams.isEmpty()) {
- for (JCTypeParameter param : type.typarams) {
- typeArgs1 = typeArgs1.append(maker.Ident(param.name));
- typeArgs2 = typeArgs2.append(maker.Ident(param.name));
- typeParams = typeParams.append(maker.TypeParameter(param.name, param.bounds));
- }
- returnType = maker.TypeApply(maker.Ident(type.name), typeArgs1);
- constructorType = maker.TypeApply(maker.Ident(type.name), typeArgs2);
- } else {
- returnType = maker.Ident(type.name);
- constructorType = maker.Ident(type.name);
- }
-
- for (JavacNode fieldNode : fields) {
- JCVariableDecl field = (JCVariableDecl) fieldNode.get();
- JCExpression pType;
- if (field.vartype instanceof JCIdent) pType = maker.Ident(((JCIdent)field.vartype).name);
- else if (field.vartype instanceof JCTypeApply) {
- JCTypeApply typeApply = (JCTypeApply) field.vartype;
- List<JCExpression> tArgs = List.nil();
- for (JCExpression arg : typeApply.arguments) tArgs = tArgs.append(arg);
- pType = maker.TypeApply(typeApply.clazz, tArgs);
- } else {
- pType = field.vartype;
- }
- List<JCAnnotation> nonNulls = findAnnotations(fieldNode, TransformationsUtil.NON_NULL_PATTERN);
- List<JCAnnotation> nullables = findAnnotations(fieldNode, TransformationsUtil.NULLABLE_PATTERN);
- JCVariableDecl param = maker.VarDef(maker.Modifiers(Flags.FINAL, nonNulls.appendList(nullables)), field.name, pType, null);
- params = params.append(param);
- args = args.append(maker.Ident(field.name));
- }
- JCReturn returnStatement = maker.Return(maker.NewClass(null, List.<JCExpression>nil(), constructorType, args, null));
- JCBlock body = maker.Block(0, List.<JCStatement>of(returnStatement));
-
- return maker.MethodDef(mods, typeNode.toName(name), returnType, typeParams, params, List.<JCExpression>nil(), body, null);
- }
}