From 38c95d6f76efde8b14a93e3a31d28c5d56bfb638 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 10 Jul 2016 18:10:01 +0100
Subject: Move childNodes onto Node interface

---
 src/main/java/org/zwobble/couscous/ast/MethodNode.java | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'src/main/java/org/zwobble/couscous/ast/MethodNode.java')

diff --git a/src/main/java/org/zwobble/couscous/ast/MethodNode.java b/src/main/java/org/zwobble/couscous/ast/MethodNode.java
index 1cb3766..2c1c099 100644
--- a/src/main/java/org/zwobble/couscous/ast/MethodNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/MethodNode.java
@@ -1,6 +1,7 @@
 package org.zwobble.couscous.ast;
 
 import com.google.common.collect.ImmutableList;
+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.Type;
@@ -193,6 +194,15 @@ public class MethodNode implements Node {
         return visitor.visit(this);
     }
 
+    @Override
+    public Iterable<? extends Node> childNodes() {
+        return Iterables.concat(
+            annotations,
+            arguments,
+            body.orElse(ImmutableList.of())
+        );
+    }
+
     public MethodNode transform(NodeTransformer transformer) {
         return new MethodNode(
             transformer.transformAnnotations(annotations),
-- 
cgit v1.2.3