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/ConstructorCallNode.java | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'src/main/java/org/zwobble/couscous/ast/ConstructorCallNode.java') diff --git a/src/main/java/org/zwobble/couscous/ast/ConstructorCallNode.java b/src/main/java/org/zwobble/couscous/ast/ConstructorCallNode.java index 6a4655a..ab4b1ea 100644 --- a/src/main/java/org/zwobble/couscous/ast/ConstructorCallNode.java +++ b/src/main/java/org/zwobble/couscous/ast/ConstructorCallNode.java @@ -1,9 +1,12 @@ package org.zwobble.couscous.ast; import java.util.List; +import java.util.function.Function; import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper; +import static org.zwobble.couscous.util.ExtraLists.eagerMap; + public class ConstructorCallNode implements ExpressionNode { public static ConstructorCallNode constructorCall( TypeName type, @@ -35,6 +38,11 @@ public class ConstructorCallNode implements ExpressionNode { return visitor.visit(this); } + @Override + public ExpressionNode replaceExpressions(Function replace) { + return new ConstructorCallNode(type, eagerMap(arguments, replace::apply)); + } + @Override public String toString() { return "ConstructorCallNode(type=" + type + ", arguments=" + arguments -- cgit v1.2.3