Move childNodes onto Node interface

1 files changed, 6 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/InstanceReceiver.java b/src/main/java/org/zwobble/couscous/ast/InstanceReceiver.java
index cbd4b6e..292b026 100644
--- a/src/main/java/org/zwobble/couscous/ast/InstanceReceiver.java
+++ b/src/main/java/org/zwobble/couscous/ast/InstanceReceiver.java
@@ -2,6 +2,7 @@ package org.zwobble.couscous.ast;
 
 import org.zwobble.couscous.ast.visitors.NodeMapper;
 import org.zwobble.couscous.ast.visitors.NodeTransformer;
+import org.zwobble.couscous.util.ExtraIterables;
 
 public class InstanceReceiver implements Receiver {
     public static Receiver instanceReceiver(ExpressionNode expression) {
@@ -24,6 +25,11 @@ public class InstanceReceiver implements Receiver {
     }
 
     @Override
+    public Iterable<? extends Node> childNodes() {
+        return ExtraIterables.of(expression);
+    }
+
+    @Override
     public Receiver transform(NodeTransformer transformer) {
         return new InstanceReceiver(transformer.transformExpression(expression));
     }