Add reading of empty enums

7 files changed, 104 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/EnumNode.java b/src/main/java/org/zwobble/couscous/ast/EnumNode.java
new file mode 100644
index 0000000..9e7a0b7
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/ast/EnumNode.java
@@ -0,0 +1,77 @@
+package org.zwobble.couscous.ast;
+
+import org.zwobble.couscous.ast.visitors.NodeMapper;
+import org.zwobble.couscous.ast.visitors.NodeTransformer;
+import org.zwobble.couscous.types.ScalarType;
+import org.zwobble.couscous.types.Type;
+
+import java.util.List;
+import java.util.Set;
+
+import static org.zwobble.couscous.util.ExtraLists.list;
+import static org.zwobble.couscous.util.ExtraSets.set;
+
+public class EnumNode implements TypeNode {
+    public static EnumNode declareEnum(ScalarType name) {
+        return new EnumNode(name);
+    }
+
+    private final ScalarType name;
+
+    public EnumNode(ScalarType name) {
+        this.name = name;
+    }
+
+    @Override
+    public ScalarType getName() {
+        return name;
+    }
+
+    @Override
+    public List<FormalTypeParameterNode> getTypeParameters() {
+        return list();
+    }
+
+    @Override
+    public Set<Type> getSuperTypes() {
+        return set();
+    }
+
+    @Override
+    public List<MethodNode> getMethods() {
+        return list();
+    }
+
+    @Override
+    public TypeNode transform(NodeTransformer transformer) {
+        return new EnumNode(transformer.transform(name));
+    }
+
+    @Override
+    public String toString() {
+        return "EnumNode(" +
+            "name=" + name +
+            ')';
+    }
+
+    @Override
+    public <T> T accept(NodeMapper<T> visitor) {
+        return visitor.visit(this);
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+
+        EnumNode enumNode = (EnumNode) o;
+
+        return name.equals(enumNode.name);
+
+    }
+
+    @Override
+    public int hashCode() {
+        return name.hashCode();
+    }
+}
diff --git a/src/main/java/org/zwobble/couscous/ast/Node.java b/src/main/java/org/zwobble/couscous/ast/Node.java
index 89d9b75..629b93f 100644
--- a/src/main/java/org/zwobble/couscous/ast/Node.java
+++ b/src/main/java/org/zwobble/couscous/ast/Node.java
@@ -167,6 +167,12 @@ public interface Node {
                 visitor.visit(interfaceNode);
                 return null;
             }
+
+            @Override
+            public Void visit(EnumNode enumNode) {
+                visitor.visit(enumNode);
+                return null;
+            }
         });
     }
     <T> T accept(NodeMapper<T> visitor);
diff --git a/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java b/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
index 3e19037..550082c 100644
--- a/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
+++ b/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
@@ -165,6 +165,11 @@ public class NodeStructure {
             }
 
             @Override
+            public Stream<? extends Node> visit(EnumNode enumNode) {
+                return Stream.empty();
+            }
+
+            @Override
             public Stream<Node> visit(MethodNode methodNode) {
                 return concatStreams(
                     methodNode.getAnnotations().stream(),
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);
 }