diff options
Diffstat (limited to 'src/lombok/javac/handlers/HandleSetter.java')
-rw-r--r-- | src/lombok/javac/handlers/HandleSetter.java | 52 |
1 files changed, 39 insertions, 13 deletions
diff --git a/src/lombok/javac/handlers/HandleSetter.java b/src/lombok/javac/handlers/HandleSetter.java index da4e9ff0..f1e73489 100644 --- a/src/lombok/javac/handlers/HandleSetter.java +++ b/src/lombok/javac/handlers/HandleSetter.java @@ -1,15 +1,18 @@ package lombok.javac.handlers; import static lombok.javac.handlers.PKG.*; +import lombok.AccessLevel; import lombok.Setter; import lombok.core.AnnotationValues; import lombok.core.AST.Kind; +import lombok.javac.Javac; import lombok.javac.JavacAST; import lombok.javac.JavacAnnotationHandler; import lombok.javac.JavacAST.Node; 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; @@ -23,37 +26,60 @@ import com.sun.tools.javac.tree.JCTree.JCTypeParameter; import com.sun.tools.javac.tree.JCTree.JCVariableDecl; import com.sun.tools.javac.util.List; import com.sun.tools.javac.util.Name; +import com.sun.tools.javac.util.JCDiagnostic.DiagnosticPosition; @ProviderFor(JavacAnnotationHandler.class) public class HandleSetter implements JavacAnnotationHandler<Setter> { + public void generateSetterForField(Node fieldNode, DiagnosticPosition pos) { + AccessLevel level = Setter.DEFAULT_ACCESS_LEVEL; + Node errorNode = fieldNode; + + for ( Node child : fieldNode.down() ) { + if ( child.getKind() == Kind.ANNOTATION ) { + if ( Javac.annotationTypeMatches(Setter.class, child) ) { + level = Javac.createAnnotation(Setter.class, child).getInstance().value(); + errorNode = child; + pos = child.get(); + break; + } + } + } + + createSetterForField(level, fieldNode, errorNode, pos); + } + @Override public boolean handle(AnnotationValues<Setter> annotation, JCAnnotation ast, Node annotationNode) { - if ( annotationNode.up().getKind() != Kind.FIELD ) { - annotationNode.addError("@Setter is only supported on a field."); + Node fieldNode = annotationNode.up(); + AccessLevel level = annotation.getInstance().value(); + return createSetterForField(level, fieldNode, annotationNode, annotationNode.get()); + } + + private boolean createSetterForField(AccessLevel level, Node fieldNode, Node errorNode, DiagnosticPosition pos) { + if ( fieldNode.getKind() != Kind.FIELD ) { + fieldNode.addError("@Setter is only supported on a field."); return false; } - JCVariableDecl fieldNode = (JCVariableDecl) annotationNode.up().get(); - String methodName = toSetterName(fieldNode); + JCVariableDecl fieldDecl = (JCVariableDecl)fieldNode.get(); + String methodName = toSetterName(fieldDecl); - if ( methodExists(methodName, annotationNode.up()) ) { - annotationNode.addWarning( + if ( methodExists(methodName, fieldNode) ) { + errorNode.addWarning( String.format("Not generating %s(%s %s): A method with that name already exists", - methodName, fieldNode.vartype, fieldNode.name)); + methodName, fieldDecl.vartype, fieldDecl.name)); return false; } - Setter setter = annotation.getInstance(); - - JCClassDecl javacClassTree = (JCClassDecl) annotationNode.up().up().get(); + JCClassDecl javacClassTree = (JCClassDecl) fieldNode.up().get(); - int access = toJavacModifier(setter.value()); + long access = toJavacModifier(level) | (fieldDecl.mods.flags & Flags.STATIC); - JCMethodDecl setterMethod = createSetter(access, annotationNode.up(), annotationNode.getTreeMaker()); + JCMethodDecl setterMethod = createSetter(access, fieldNode, fieldNode.getTreeMaker()); javacClassTree.defs = javacClassTree.defs.append(setterMethod); return true; } - private JCMethodDecl createSetter(int access, JavacAST.Node field, TreeMaker treeMaker) { + private JCMethodDecl createSetter(long access, JavacAST.Node field, TreeMaker treeMaker) { JCVariableDecl fieldDecl = (JCVariableDecl) field.get(); JCFieldAccess thisX = treeMaker.Select(treeMaker.Ident(field.toName("this")), fieldDecl.name); |