Move childNodes onto Node interface

1 files changed, 6 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java b/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java
index 6c49d39..a2f9f2d 100644
--- a/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java
@@ -3,6 +3,7 @@ package org.zwobble.couscous.ast;
 import org.zwobble.couscous.types.Type;
 import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper;
 import org.zwobble.couscous.ast.visitors.NodeTransformer;
+import org.zwobble.couscous.util.ExtraIterables;
 
 import static org.zwobble.couscous.ast.ExpressionStatementNode.expressionStatement;
 import static org.zwobble.couscous.ast.VariableReferenceNode.reference;
@@ -46,6 +47,11 @@ public class AssignmentNode implements ExpressionNode {
     }
 
     @Override
+    public Iterable<Node> childNodes() {
+        return ExtraIterables.of(value, target);
+    }
+
+    @Override
     public ExpressionNode transform(NodeTransformer transformer) {
         return new AssignmentNode(
             (AssignableExpressionNode) transformer.transformExpression(target),