Add reading of empty enums

4 files changed, 16 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/visitors/NodeMapper.java b/src/main/java/org/zwobble/couscous/ast/visitors/NodeMapper.java
index 8b7cb3b..3edece1 100644
--- a/src/main/java/org/zwobble/couscous/ast/visitors/NodeMapper.java
+++ b/src/main/java/org/zwobble/couscous/ast/visitors/NodeMapper.java
@@ -34,4 +34,5 @@ public interface NodeMapper<T> {
     T visit(FieldDeclarationNode declaration);
     T visit(ClassNode classNode);
     T visit(InterfaceNode interfaceNode);
+    T visit(EnumNode enumNode);
 }
diff --git a/src/main/java/org/zwobble/couscous/ast/visitors/NodeMapperWithDefault.java b/src/main/java/org/zwobble/couscous/ast/visitors/NodeMapperWithDefault.java
index 6ff3c06..c67116d 100644
--- a/src/main/java/org/zwobble/couscous/ast/visitors/NodeMapperWithDefault.java
+++ b/src/main/java/org/zwobble/couscous/ast/visitors/NodeMapperWithDefault.java
@@ -135,6 +135,11 @@ public class NodeMapperWithDefault<T> implements NodeMapper<T> {
     }
 
     @Override
+    public T visit(EnumNode enumNode) {
+        return defaultValue;
+    }
+
+    @Override
     public T visit(MethodNode methodNode) {
         return defaultValue;
     }
diff --git a/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java b/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
index e86e2ee..73d0919 100644
--- a/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
+++ b/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
@@ -164,6 +164,11 @@ public class NodeTransformer {
             public Node visit(InterfaceNode interfaceNode) {
                 return transformInterface(interfaceNode);
             }
+
+            @Override
+            public Node visit(EnumNode enumNode) {
+                return transformEnum(enumNode);
+            }
         });
     }
 
@@ -283,6 +288,10 @@ public class NodeTransformer {
         return interfaceNode.transform(this);
     }
 
+    private Node transformEnum(EnumNode enumNode) {
+        return enumNode.transform(this);
+    }
+
     public List<AnnotationNode> transformAnnotations(List<AnnotationNode> annotations) {
         return transformList(annotations, this::transformAnnotation);
     }
diff --git a/src/main/java/org/zwobble/couscous/ast/visitors/NodeVisitor.java b/src/main/java/org/zwobble/couscous/ast/visitors/NodeVisitor.java
index 336d016..167e89b 100644
--- a/src/main/java/org/zwobble/couscous/ast/visitors/NodeVisitor.java
+++ b/src/main/java/org/zwobble/couscous/ast/visitors/NodeVisitor.java
@@ -33,4 +33,5 @@ public interface NodeVisitor {
     void visit(FieldDeclarationNode declaration);
     void visit(ClassNode classNode);
     void visit(InterfaceNode interfaceNode);
+    void visit(EnumNode enumNode);
 }