Move childNodes onto Node interface

1 files changed, 13 insertions, 2 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/ClassNode.java b/src/main/java/org/zwobble/couscous/ast/ClassNode.java
index bcdd275..97f66f8 100644
--- a/src/main/java/org/zwobble/couscous/ast/ClassNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/ClassNode.java
@@ -1,9 +1,11 @@
 package org.zwobble.couscous.ast;
 
-import org.zwobble.couscous.types.ScalarType;
-import org.zwobble.couscous.types.Type;
+import com.google.common.collect.Iterables;
 import org.zwobble.couscous.ast.visitors.NodeMapper;
 import org.zwobble.couscous.ast.visitors.NodeTransformer;
+import org.zwobble.couscous.types.ScalarType;
+import org.zwobble.couscous.types.Type;
+import org.zwobble.couscous.util.ExtraIterables;
 
 import java.util.List;
 import java.util.Set;
@@ -96,6 +98,15 @@ public class ClassNode implements TypeNode {
     }
 
     @Override
+    public Iterable<? extends Node> childNodes() {
+        return Iterables.concat(
+            ExtraIterables.of(constructor),
+            fields,
+            methods
+        );
+    }
+
+    @Override
     public ClassNode transform(NodeTransformer transformer) {
         return new ClassNode(
             transformer.transform(name),