Replace captured variable references in method bodies

1 files changed, 11 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/MethodCallNode.java b/src/main/java/org/zwobble/couscous/ast/MethodCallNode.java
index cce140f..f65a9ae 100644
--- a/src/main/java/org/zwobble/couscous/ast/MethodCallNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/MethodCallNode.java
@@ -1,14 +1,15 @@
 package org.zwobble.couscous.ast;
 
-import com.sun.org.apache.xpath.internal.operations.Bool;
 import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper;
 import org.zwobble.couscous.values.BooleanValue;
 import org.zwobble.couscous.values.IntegerValue;
 
 import java.util.Collections;
 import java.util.List;
+import java.util.function.Function;
 
 import static java.util.Arrays.asList;
+import static org.zwobble.couscous.util.ExtraLists.eagerMap;
 
 public class MethodCallNode implements ExpressionNode {
     public static ExpressionNode not(ExpressionNode value) {
@@ -117,6 +118,15 @@ public class MethodCallNode implements ExpressionNode {
     }
 
     @Override
+    public ExpressionNode replaceExpressions(Function<ExpressionNode, ExpressionNode> replace) {
+        return new MethodCallNode(
+            replace.apply(receiver),
+            methodName,
+            eagerMap(arguments, replace::apply),
+            type);
+    }
+
+    @Override
     public String toString() {
         return "MethodCallNode(receiver=" + receiver + ", methodName="
                + methodName + ", arguments=" + arguments + ", type=" + type