Move childNodes onto Node interface

1 files changed, 6 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/ConstructorNode.java b/src/main/java/org/zwobble/couscous/ast/ConstructorNode.java
index 1bd33d0..ea83c93 100644
--- a/src/main/java/org/zwobble/couscous/ast/ConstructorNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/ConstructorNode.java
@@ -1,5 +1,6 @@
 package org.zwobble.couscous.ast;
 
+import com.google.common.collect.Iterables;
 import org.zwobble.couscous.ast.visitors.NodeMapper;
 import org.zwobble.couscous.ast.visitors.NodeTransformer;
 
@@ -38,6 +39,11 @@ public class ConstructorNode implements Node {
         return visitor.visit(this);
     }
 
+    @Override
+    public Iterable<? extends Node> childNodes() {
+        return Iterables.concat(arguments, body);
+    }
+
     public ConstructorNode transform(NodeTransformer transformer) {
         return new ConstructorNode(
             transformer.transformFormalArguments(arguments),