diff options
Diffstat (limited to 'src/main/java/org/zwobble/couscous/ast')
| -rw-r--r-- | src/main/java/org/zwobble/couscous/ast/ClassNode.java | 5 | ||||
| -rw-r--r-- | src/main/java/org/zwobble/couscous/ast/InterfaceNode.java | 9 |
2 files changed, 12 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 322a32a..0d37130 100644 --- a/src/main/java/org/zwobble/couscous/ast/ClassNode.java +++ b/src/main/java/org/zwobble/couscous/ast/ClassNode.java @@ -113,9 +113,12 @@ public class ClassNode implements TypeNode { @Override public Iterable<? extends Node> childNodes() { return Iterables.concat( + typeParameters, + staticConstructor, ExtraIterables.of(constructor), fields, - methods + methods, + innerTypes ); } diff --git a/src/main/java/org/zwobble/couscous/ast/InterfaceNode.java b/src/main/java/org/zwobble/couscous/ast/InterfaceNode.java index 3159d03..6a56a6b 100644 --- a/src/main/java/org/zwobble/couscous/ast/InterfaceNode.java +++ b/src/main/java/org/zwobble/couscous/ast/InterfaceNode.java @@ -1,5 +1,6 @@ package org.zwobble.couscous.ast; +import com.google.common.collect.Iterables; import org.zwobble.couscous.ast.visitors.NodeTransformer; import org.zwobble.couscous.types.ScalarType; import org.zwobble.couscous.types.Type; @@ -91,7 +92,13 @@ public class InterfaceNode implements TypeNode { @Override public Iterable<? extends Node> childNodes() { - return methods; + return Iterables.concat( + typeParameters, + staticConstructor, + fields, + methods, + innerTypes + ); } @Override |
