Port of lombok-pg's @ExtensionMethod support for eclipse. Doesn't work yet.

2 files changed, 61 insertions, 0 deletions

diff --git a/src/core/lombok/eclipse/handlers/HandleExtensionMethod.java b/src/core/lombok/eclipse/handlers/HandleExtensionMethod.java
new file mode 100644
index 00000000..30408118
--- /dev/null
+++ b/src/core/lombok/eclipse/handlers/HandleExtensionMethod.java
@@ -0,0 +1,57 @@
+/*
+ * Copyright (C) 2012 The Project Lombok Authors.
+ * 
+ * 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.eclipse.handlers;
+
+import java.util.List;
+
+import org.eclipse.jdt.internal.compiler.ast.Annotation;
+import org.eclipse.jdt.internal.compiler.ast.TypeDeclaration;
+import org.eclipse.jdt.internal.compiler.classfmt.ClassFileConstants;
+
+import lombok.core.AnnotationValues;
+import lombok.eclipse.EclipseAnnotationHandler;
+import lombok.eclipse.EclipseNode;
+import lombok.experimental.ExtensionMethod;
+
+// This handler just does some additional error checking; the real work is done in the agent.
+public class HandleExtensionMethod extends EclipseAnnotationHandler<ExtensionMethod> {
+	@Override public void handle(AnnotationValues<ExtensionMethod> annotation, Annotation ast, EclipseNode annotationNode) {
+		TypeDeclaration typeDecl = null;
+		EclipseNode owner = annotationNode.up();
+		if (owner.get() instanceof TypeDeclaration) typeDecl = (TypeDeclaration) owner.get();
+		int modifiers = typeDecl == null ? 0 : typeDecl.modifiers;
+		
+		boolean notAClass = (modifiers &
+				(ClassFileConstants.AccInterface | ClassFileConstants.AccAnnotation)) != 0;
+		
+		if (typeDecl == null || notAClass) {
+			annotationNode.addError("@ExtensionMethod is legal only on classes and enums.");
+			return;
+		}
+		
+		List<Object> listenerInterfaces = annotation.getActualExpressions("value");
+		if (listenerInterfaces.isEmpty()) {
+			annotationNode.addWarning(String.format("@ExtensionMethod has no effect since no extension types were specified."));
+			return;
+		}
+	}
+}
diff --git a/src/core/lombok/experimental/ExtensionMethod.java b/src/core/lombok/experimental/ExtensionMethod.java
index b25e535e..a9eaa6b7 100644
--- a/src/core/lombok/experimental/ExtensionMethod.java
+++ b/src/core/lombok/experimental/ExtensionMethod.java
@@ -60,5 +60,9 @@ public @interface ExtensionMethod {
 	/** All types whose static methods will be exposed as extension methods. */
 	Class<?>[] value();
 
+	/**
+	 * If {@code true}, an applicable extension method is used (if found) even if the method call already was compilable (this is the default).
+	 * If {@code false}, an extension method is only used if the method call is not also defined by the type itself.
+	 */
 	boolean suppressBaseMethods() default true;
 }