Add reading of empty enums

1 files changed, 13 insertions, 2 deletions

diff --git a/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java b/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
index 44cbfab..a83be40 100644
--- a/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
+++ b/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
@@ -88,14 +88,25 @@ public class JavaReader {
     }
 
     private void readCompilationUnit(CompilationUnit ast) {
-        TypeDeclaration type = (TypeDeclaration)ast.types().get(0);
-        classes.add(readTypeDeclaration(type));
+        AbstractTypeDeclaration type = (AbstractTypeDeclaration) ast.types().get(0);
+        if (type instanceof TypeDeclaration) {
+            classes.add(readTypeDeclaration((TypeDeclaration) type));
+        } else if (type instanceof EnumDeclaration) {
+            classes.add(readEnumDeclaration((EnumDeclaration)type));
+        } else {
+            throw new UnsupportedOperationException();
+        }
     }
 
     private List<TypeNode> types() {
         return this.classes.build();
     }
 
+    private TypeNode readEnumDeclaration(EnumDeclaration type) {
+        ScalarType name = (ScalarType) typeOf(type.resolveBinding());
+        return new EnumNode(name);
+    }
+
     private TypeNode readTypeDeclaration(TypeDeclaration type) {
         ScalarType name = erasure(typeOf(type.resolveBinding()));
         Scope scope = topScope.enterClass(name);