diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/lombok/eclipse/handlers/HandleGetter.java | 73 | ||||
-rw-r--r-- | src/lombok/eclipse/handlers/HandleSetter.java | 86 | ||||
-rw-r--r-- | src/lombok/eclipse/handlers/PKG.java | 36 | ||||
-rw-r--r-- | src/lombok/javac/handlers/PKG.java | 2 |
4 files changed, 149 insertions, 48 deletions
diff --git a/src/lombok/eclipse/handlers/HandleGetter.java b/src/lombok/eclipse/handlers/HandleGetter.java index 6be978c6..ca561a48 100644 --- a/src/lombok/eclipse/handlers/HandleGetter.java +++ b/src/lombok/eclipse/handlers/HandleGetter.java @@ -1,8 +1,7 @@ package lombok.eclipse.handlers; -import java.lang.reflect.Modifier; +import static lombok.eclipse.handlers.PKG.*; -import lombok.AccessLevel; import lombok.Getter; import lombok.core.AnnotationValues; import lombok.core.TransformationsUtil; @@ -25,10 +24,6 @@ import org.mangosdk.spi.ProviderFor; @ProviderFor(EclipseAnnotationHandler.class) public class HandleGetter implements EclipseAnnotationHandler<Getter> { - private void generateDuplicateGetterWarning(Node annotationNode, String methodName) { - annotationNode.addWarning(String.format("Not generating %s(): A method with that name already exists", methodName)); - } - @Override public boolean handle(AnnotationValues<Getter> annotation, Annotation ast, Node annotationNode) { if ( !(annotationNode.up().get() instanceof FieldDeclaration) ) return false; FieldDeclaration field = (FieldDeclaration) annotationNode.up().get(); @@ -39,17 +34,37 @@ public class HandleGetter implements EclipseAnnotationHandler<Getter> { TypeDeclaration parent = (TypeDeclaration) annotationNode.up().up().get(); if ( parent.methods != null ) for ( AbstractMethodDeclaration method : parent.methods ) { if ( method.selector != null && new String(method.selector).equals(getterName) ) { - generateDuplicateGetterWarning(annotationNode, getterName); + annotationNode.addWarning(String.format( + "Not generating %s(): A method with that name already exists", getterName)); return false; } } + int modifier = toModifier(annotation.getInstance().value()); + + MethodDeclaration method = generateGetter(parent, field, getterName, modifier, ast); + + if ( parent.methods == null ) { + parent.methods = new AbstractMethodDeclaration[1]; + parent.methods[0] = method; + } else { + AbstractMethodDeclaration[] newArray = new AbstractMethodDeclaration[parent.methods.length + 1]; + System.arraycopy(parent.methods, 0, newArray, 0, parent.methods.length); + newArray[parent.methods.length] = method; + parent.methods = newArray; + } + + return true; + } + + private MethodDeclaration generateGetter(TypeDeclaration parent, FieldDeclaration field, String name, + int modifier, Annotation ast) { MethodDeclaration method = new MethodDeclaration(parent.compilationResult); - method.modifiers = toModifier(annotation.getInstance().value()); + method.modifiers = modifier; method.returnType = field.type; method.annotations = null; method.arguments = null; - method.selector = getterName.toCharArray(); + method.selector = name.toCharArray(); method.binding = null; method.thrownExceptions = null; method.typeParameters = null; @@ -60,44 +75,6 @@ public class HandleGetter implements EclipseAnnotationHandler<Getter> { method.bodyStart = method.declarationSourceStart = method.sourceStart = ast.sourceStart; method.bodyEnd = method.declarationSourceEnd = method.sourceEnd = ast.sourceEnd; method.statements = new Statement[] { returnStatement }; - if ( parent.methods == null ) { - parent.methods = new AbstractMethodDeclaration[1]; - parent.methods[0] = method; - } else { - AbstractMethodDeclaration[] newArray = new AbstractMethodDeclaration[parent.methods.length + 1]; - System.arraycopy(parent.methods, 0, newArray, 0, parent.methods.length); - newArray[parent.methods.length] = method; - parent.methods = newArray; - } - - return true; + return method; } - - private int toModifier(AccessLevel value) { - switch ( value ) { - case MODULE: - case PACKAGE: - return 0; - default: - case PUBLIC: - return Modifier.PUBLIC; - case PROTECTED: - return Modifier.PROTECTED; - case PRIVATE: - return Modifier.PRIVATE; - } - } - - private boolean nameEquals(char[][] typeName, String string) { - StringBuilder sb = new StringBuilder(); - boolean first = true; - for ( char[] elem : typeName ) { - if ( first ) first = false; - else sb.append('.'); - sb.append(elem); - } - - return string.contentEquals(sb); - } - } diff --git a/src/lombok/eclipse/handlers/HandleSetter.java b/src/lombok/eclipse/handlers/HandleSetter.java new file mode 100644 index 00000000..692061e4 --- /dev/null +++ b/src/lombok/eclipse/handlers/HandleSetter.java @@ -0,0 +1,86 @@ +package lombok.eclipse.handlers; + +import static lombok.eclipse.handlers.PKG.*; + +import org.eclipse.jdt.internal.compiler.ast.ASTNode; +import org.eclipse.jdt.internal.compiler.ast.AbstractMethodDeclaration; +import org.eclipse.jdt.internal.compiler.ast.Annotation; +import org.eclipse.jdt.internal.compiler.ast.Argument; +import org.eclipse.jdt.internal.compiler.ast.Assignment; +import org.eclipse.jdt.internal.compiler.ast.FieldDeclaration; +import org.eclipse.jdt.internal.compiler.ast.FieldReference; +import org.eclipse.jdt.internal.compiler.ast.MethodDeclaration; +import org.eclipse.jdt.internal.compiler.ast.SingleNameReference; +import org.eclipse.jdt.internal.compiler.ast.Statement; +import org.eclipse.jdt.internal.compiler.ast.ThisReference; +import org.eclipse.jdt.internal.compiler.ast.TypeDeclaration; +import org.eclipse.jdt.internal.compiler.ast.TypeReference; +import org.eclipse.jdt.internal.compiler.lookup.MethodScope; +import org.eclipse.jdt.internal.compiler.lookup.TypeIds; +import org.mangosdk.spi.ProviderFor; + +import lombok.Setter; +import lombok.core.AnnotationValues; +import lombok.core.TransformationsUtil; +import lombok.eclipse.EclipseAnnotationHandler; +import lombok.eclipse.EclipseAST.Node; + +@ProviderFor(EclipseAnnotationHandler.class) +public class HandleSetter implements EclipseAnnotationHandler<Setter> { + @Override public boolean handle(AnnotationValues<Setter> annotation, Annotation ast, Node annotationNode) { + if ( !(annotationNode.up().get() instanceof FieldDeclaration) ) return false; + FieldDeclaration field = (FieldDeclaration) annotationNode.up().get(); + String setterName = TransformationsUtil.toSetterName(new String(field.name)); + + TypeDeclaration parent = (TypeDeclaration) annotationNode.up().up().get(); + if ( parent.methods != null ) for ( AbstractMethodDeclaration method : parent.methods ) { + if ( method.selector != null && new String(method.selector).equals(setterName) ) { + annotationNode.addWarning(String.format( + "Not generating %s(%s %s): A method with that name already exists", + setterName, field.type, new String(field.name))); + return false; + } + } + + int modifier = toModifier(annotation.getInstance().value()); + + MethodDeclaration method = generateSetter(parent, field, setterName, modifier, ast); + + if ( parent.methods == null ) { + parent.methods = new AbstractMethodDeclaration[1]; + parent.methods[0] = method; + } else { + AbstractMethodDeclaration[] newArray = new AbstractMethodDeclaration[parent.methods.length + 1]; + System.arraycopy(parent.methods, 0, newArray, 0, parent.methods.length); + newArray[parent.methods.length] = method; + parent.methods = newArray; + } + + return true; + } + + private MethodDeclaration generateSetter(TypeDeclaration parent, FieldDeclaration field, String name, + int modifier, Annotation ast) { + long pos = (((long)ast.sourceStart) << 32) | ast.sourceEnd; + MethodDeclaration method = new MethodDeclaration(parent.compilationResult); + method.modifiers = modifier; + method.returnType = TypeReference.baseTypeReference(TypeIds.T_void, 0); + method.annotations = null; + Argument param = new Argument(field.name, pos, field.type, 0); + method.arguments = new Argument[] { param }; + method.selector = name.toCharArray(); + method.binding = null; + method.thrownExceptions = null; + method.typeParameters = null; + method.scope = parent.scope == null ? null : new MethodScope(parent.scope, method, false); + method.bits |= ASTNode.Bit24; + FieldReference thisX = new FieldReference(("this." + new String(field.name)).toCharArray(), pos); + thisX.receiver = new ThisReference(ast.sourceStart, ast.sourceEnd); + thisX.token = field.name; + Assignment assignment = new Assignment(thisX, new SingleNameReference(field.name, pos), (int)pos); + method.bodyStart = method.declarationSourceStart = method.sourceStart = ast.sourceStart; + method.bodyEnd = method.declarationSourceEnd = method.sourceEnd = ast.sourceEnd; + method.statements = new Statement[] { assignment }; + return method; + } +} diff --git a/src/lombok/eclipse/handlers/PKG.java b/src/lombok/eclipse/handlers/PKG.java new file mode 100644 index 00000000..61f71140 --- /dev/null +++ b/src/lombok/eclipse/handlers/PKG.java @@ -0,0 +1,36 @@ +package lombok.eclipse.handlers; + +import java.lang.reflect.Modifier; + +import lombok.AccessLevel; + +class PKG { + private PKG() {} + + static int toModifier(AccessLevel value) { + switch ( value ) { + case MODULE: + case PACKAGE: + return 0; + default: + case PUBLIC: + return Modifier.PUBLIC; + case PROTECTED: + return Modifier.PROTECTED; + case PRIVATE: + return Modifier.PRIVATE; + } + } + + static boolean nameEquals(char[][] typeName, String string) { + StringBuilder sb = new StringBuilder(); + boolean first = true; + for ( char[] elem : typeName ) { + if ( first ) first = false; + else sb.append('.'); + sb.append(elem); + } + + return string.contentEquals(sb); + } +} diff --git a/src/lombok/javac/handlers/PKG.java b/src/lombok/javac/handlers/PKG.java index 161248c8..2c038b2d 100644 --- a/src/lombok/javac/handlers/PKG.java +++ b/src/lombok/javac/handlers/PKG.java @@ -12,6 +12,8 @@ import lombok.core.TransformationsUtil; import lombok.javac.JavacAST; class PKG { + private PKG() {} + static String toGetterName(JCVariableDecl field) { CharSequence fieldName = field.name; |