Replace captured variable references in method bodies

1 files changed, 8 insertions, 0 deletions

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,
@@ -36,6 +39,11 @@ public class ConstructorCallNode implements ExpressionNode {
     }
 
     @Override
+    public ExpressionNode replaceExpressions(Function<ExpressionNode, ExpressionNode> replace) {
+        return new ConstructorCallNode(type, eagerMap(arguments, replace::apply));
+    }
+
+    @Override
     public String toString() {
         return "ConstructorCallNode(type=" + type + ", arguments=" + arguments
                + ")";