From 51e8ff721cc7079212119ba44fcc4bf0f9818d1e Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 6 Dec 2015 11:20:51 +0000 Subject: Replace captured variable references in method bodies --- src/main/java/org/zwobble/couscous/ast/AssignmentNode.java | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'src/main/java/org/zwobble/couscous/ast/AssignmentNode.java') diff --git a/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java b/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java index b6c8f9c..15ecf8f 100644 --- a/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java +++ b/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java @@ -2,6 +2,8 @@ package org.zwobble.couscous.ast; import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper; +import java.util.function.Function; + import static org.zwobble.couscous.ast.ExpressionStatementNode.expressionStatement; import static org.zwobble.couscous.ast.VariableReferenceNode.reference; @@ -43,6 +45,13 @@ public class AssignmentNode implements ExpressionNode { return visitor.visit(this); } + @Override + public ExpressionNode replaceExpressions(Function replace) { + return new AssignmentNode( + (AssignableExpressionNode) replace.apply(target), + replace.apply(value)); + } + @Override public TypeName getType() { return value.getType(); -- cgit v1.2.3