GROOVY-10148: Groovy should support sealed classes (additional edge c…

…ases)
jlxc106 · Aug 9, 2021 · 53c5681 · 53c5681
1 parent 4248678
commit 53c5681
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/main/java/org/codehaus/groovy/transform/NonSealedASTTransformation.java b/src/main/java/org/codehaus/groovy/transform/NonSealedASTTransformation.java
@@ -43,6 +43,14 @@ public void visit(ASTNode[] nodes, SourceUnit source) {
 
         if (parent instanceof ClassNode) {
             ClassNode cNode = (ClassNode) parent;
+            if (cNode.isEnum()) {
+                addError("@" + NON_SEALED_CLASS.getSimpleName() + " not allowed for enum", cNode);
+                return;
+            }
+            if (cNode.isAnnotationDefinition()) {
+                addError("@" + NON_SEALED_CLASS.getSimpleName() + " not allowed for annotation definition", cNode);
+                return;
+            }
             cNode.putNodeMetaData(NON_SEALED_CLASS, Boolean.TRUE);
         }
     }