Move childNodes onto Node interface

1 files changed, 6 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/LocalVariableDeclarationNode.java b/src/main/java/org/zwobble/couscous/ast/LocalVariableDeclarationNode.java
index 4078adb..54a6804 100644
--- a/src/main/java/org/zwobble/couscous/ast/LocalVariableDeclarationNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/LocalVariableDeclarationNode.java
@@ -4,6 +4,7 @@ import org.zwobble.couscous.ast.identifiers.Identifier;
 import org.zwobble.couscous.types.Type;
 import org.zwobble.couscous.ast.visitors.NodeTransformer;
 import org.zwobble.couscous.ast.visitors.StatementNodeMapper;
+import org.zwobble.couscous.util.ExtraIterables;
 
 import static org.zwobble.couscous.ast.TypeCoercionNode.coerce;
 import static org.zwobble.couscous.ast.VariableDeclaration.var;
@@ -55,6 +56,11 @@ public class LocalVariableDeclarationNode implements VariableNode, StatementNode
     }
 
     @Override
+    public Iterable<? extends Node> childNodes() {
+        return ExtraIterables.of(initialValue);
+    }
+
+    @Override
     public StatementNode transform(NodeTransformer transformer) {
         return new LocalVariableDeclarationNode(
             transformer.transform(declaration),