aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java/org/zwobble/couscous/ast/AssignmentNode.java')
-rw-r--r--src/main/java/org/zwobble/couscous/ast/AssignmentNode.java6
1 files changed, 6 insertions, 0 deletions
diff --git a/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java b/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java
index 6c49d39..a2f9f2d 100644
--- a/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java
@@ -3,6 +3,7 @@ package org.zwobble.couscous.ast;
import org.zwobble.couscous.types.Type;
import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper;
import org.zwobble.couscous.ast.visitors.NodeTransformer;
+import org.zwobble.couscous.util.ExtraIterables;
import static org.zwobble.couscous.ast.ExpressionStatementNode.expressionStatement;
import static org.zwobble.couscous.ast.VariableReferenceNode.reference;
@@ -46,6 +47,11 @@ public class AssignmentNode implements ExpressionNode {
}
@Override
+ public Iterable<Node> childNodes() {
+ return ExtraIterables.of(value, target);
+ }
+
+ @Override
public ExpressionNode transform(NodeTransformer transformer) {
return new AssignmentNode(
(AssignableExpressionNode) transformer.transformExpression(target),