From 0e98e8252a823eb025c24857c9c47fb917f4c220 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 10 Apr 2016 20:19:48 +0100
Subject: Read values in enum

---
 src/main/java/org/zwobble/couscous/frontends/java/JavaReader.java | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'src/main/java/org/zwobble/couscous/frontends')

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) {
-- 
cgit v1.2.3