Add support for reading vararg method invocations

1 files changed, 5 insertions, 0 deletions

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 f1f37a6..ed5b474 100644
--- a/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
+++ b/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
@@ -47,6 +47,11 @@ public class NodeStructure {
             }
 
             @Override
+            public Stream<? extends Node> visit(ArrayNode array) {
+                return array.getElements().stream();
+            }
+
+            @Override
             public Stream<Node> visit(AssignmentNode assignment) {
                 return Stream.of(assignment.getValue(), assignment.getTarget());
             }