From 51e8ff721cc7079212119ba44fcc4bf0f9818d1e Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 6 Dec 2015 11:20:51 +0000
Subject: Replace captured variable references in method bodies

---
 src/main/java/org/zwobble/couscous/ast/MethodCallNode.java | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

(limited to 'src/main/java/org/zwobble/couscous/ast/MethodCallNode.java')

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) {
@@ -116,6 +117,15 @@ public class MethodCallNode implements ExpressionNode {
         return visitor.visit(this);
     }
 
+    @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="
-- 
cgit v1.2.3