Read values in enum

1 files changed, 3 insertions, 1 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 a83be40..fe7646b 100644
--- a/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
+++ b/src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java
@@ -104,7 +104,9 @@ public class JavaReader {
 
     private TypeNode readEnumDeclaration(EnumDeclaration type) {
         ScalarType name = (ScalarType) typeOf(type.resolveBinding());
-        return new EnumNode(name);
+        @SuppressWarnings("unchecked")
+        List<EnumConstantDeclaration> values = type.enumConstants();
+        return new EnumNode(name, eagerMap(values, declaration -> declaration.getName().getIdentifier()));
     }
 
     private TypeNode readTypeDeclaration(TypeDeclaration type) {