Read array creation nodes

1 files changed, 13 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/frontends/java/JavaExpressionReader.java b/src/main/java/org/zwobble/couscous/frontends/java/JavaExpressionReader.java
index 4f08151..fcce1e1 100644
--- a/src/main/java/org/zwobble/couscous/frontends/java/JavaExpressionReader.java
+++ b/src/main/java/org/zwobble/couscous/frontends/java/JavaExpressionReader.java
@@ -73,6 +73,9 @@ public class JavaExpressionReader {
                 case ASTNode.TYPE_LITERAL:
                     return readTypeLiteral((TypeLiteral)expression);
 
+                case ASTNode.ARRAY_CREATION:
+                    return readArrayCreation((ArrayCreation)expression);
+
                 case ASTNode.PARENTHESIZED_EXPRESSION:
                     return readParenthesizedExpression((ParenthesizedExpression)expression);
 
@@ -158,6 +161,16 @@ public class JavaExpressionReader {
         return literal(erasure(typeOf(expression.getType())));
     }
 
+    private ExpressionNode readArrayCreation(ArrayCreation expression) {
+        if (expression.getType().getDimensions() != 1) {
+            throw new UnsupportedOperationException();
+        }
+        Type elementType = typeOf(expression.getType().getElementType());
+        @SuppressWarnings("unchecked")
+        List<Expression> expressions = expression.getInitializer().expressions();
+        return array(elementType, eagerMap(expressions, element -> readExpression(elementType, element)));
+    }
+
     private ExpressionNode readParenthesizedExpression(ParenthesizedExpression expression) {
         return readExpressionWithoutBoxing(expression.getExpression());
     }