Move childNodes onto Node interface

1 files changed, 6 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/WhileNode.java b/src/main/java/org/zwobble/couscous/ast/WhileNode.java
index 8ea693f..3dd6bce 100644
--- a/src/main/java/org/zwobble/couscous/ast/WhileNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/WhileNode.java
@@ -3,6 +3,7 @@ package org.zwobble.couscous.ast;
 import com.google.common.base.Objects;
 import org.zwobble.couscous.ast.visitors.NodeTransformer;
 import org.zwobble.couscous.ast.visitors.StatementNodeMapper;
+import org.zwobble.couscous.util.ExtraIterables;
 
 import java.util.List;
 
@@ -33,6 +34,11 @@ public class WhileNode implements StatementNode {
     }
 
     @Override
+    public Iterable<? extends Node> childNodes() {
+        return ExtraIterables.lazyCons(condition, body);
+    }
+
+    @Override
     public StatementNode transform(NodeTransformer transformer) {
         return new WhileNode(
             transformer.transformExpression(condition),