From 5835e5d19fd4b22c88fbb8d304858786b060fe71 Mon Sep 17 00:00:00 2001
From: Marc Guenther <marcguenther@me.com>
Date: Wed, 14 Sep 2022 22:15:01 +0200
Subject: fixes ...: allow @ExtensionMethod on interfaces

---
 src/core/lombok/javac/handlers/HandleExtensionMethod.java | 6 +++---
 src/core/lombok/javac/handlers/JavacHandlerUtil.java      | 4 ++++
 2 files changed, 7 insertions(+), 3 deletions(-)

(limited to 'src/core/lombok/javac/handlers')

diff --git a/src/core/lombok/javac/handlers/HandleExtensionMethod.java b/src/core/lombok/javac/handlers/HandleExtensionMethod.java
index dd565f72..98b000c2 100644
--- a/src/core/lombok/javac/handlers/HandleExtensionMethod.java
+++ b/src/core/lombok/javac/handlers/HandleExtensionMethod.java
@@ -73,10 +73,10 @@ public class HandleExtensionMethod extends JavacAnnotationHandler<ExtensionMetho
 		
 		deleteAnnotationIfNeccessary(annotationNode, ExtensionMethod.class);
 		JavacNode typeNode = annotationNode.up();
-		boolean isClassOrEnum = isClassOrEnum(typeNode);
+		boolean isClassOrEnumOrInterface = isClassOrEnumOrInterface(typeNode);
 		
-		if (!isClassOrEnum) {
-			annotationNode.addError("@ExtensionMethod can only be used on a class or an enum");
+		if (!isClassOrEnumOrInterface) {
+			annotationNode.addError("@ExtensionMethod can only be used on a class or an enum or an interface");
 			return;
 		}
 		
diff --git a/src/core/lombok/javac/handlers/JavacHandlerUtil.java b/src/core/lombok/javac/handlers/JavacHandlerUtil.java
index b17e34d8..9d153a72 100644
--- a/src/core/lombok/javac/handlers/JavacHandlerUtil.java
+++ b/src/core/lombok/javac/handlers/JavacHandlerUtil.java
@@ -2069,6 +2069,10 @@ public class JavacHandlerUtil {
 		return isClassAndDoesNotHaveFlags(typeNode, Flags.INTERFACE | Flags.ANNOTATION | RECORD);
 	}
 	
+	public static boolean isClassOrEnumOrInterface(JavacNode typeNode) {
+		return isClassAndDoesNotHaveFlags(typeNode, Flags.ANNOTATION | RECORD);
+	}
+	
 	/**
 	 * Returns {@code true} if the provided node is an actual class, an enum or a record and not some other type declaration (so, not an annotation definition or interface).
 	 */
-- 
cgit