From f2ed6b591da6149319c149d1c78694bfa9641a73 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 24 Jan 2016 16:01:06 +0000
Subject: Implement visitors on OperationNode instead of desugarring

---
 src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java')

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 2fd444d..a4cedca 100644
--- a/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
+++ b/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
@@ -28,6 +28,7 @@ public class NodeStructure {
     }
 
     public static Stream<? extends Node> childNodes(Node node) {
+        // TODO: is this just duplicating the logic already in node.transform?
         return node.accept(new NodeMapper<Stream<? extends Node>>() {
             @Override
             public Stream<Node> visit(LiteralNode literal) {
@@ -69,6 +70,11 @@ public class NodeStructure {
                 return call.getArguments().stream();
             }
 
+            @Override
+            public Stream<? extends Node> visit(OperationNode operation) {
+                return operation.getArguments().stream();
+            }
+
             @Override
             public Stream<Node> visit(FieldAccessNode fieldAccess) {
                 return Stream.of(fieldAccess.getLeft());
-- 
cgit v1.2.3