From 38c95d6f76efde8b14a93e3a31d28c5d56bfb638 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 10 Jul 2016 18:10:01 +0100 Subject: Move childNodes onto Node interface --- src/main/java/org/zwobble/couscous/ast/ConstructorNode.java | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'src/main/java/org/zwobble/couscous/ast/ConstructorNode.java') 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 childNodes() { + return Iterables.concat(arguments, body); + } + public ConstructorNode transform(NodeTransformer transformer) { return new ConstructorNode( transformer.transformFormalArguments(arguments), -- cgit v1.2.3