Replace captured variable references in method bodies

1 files changed, 11 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/StaticMethodCallNode.java b/src/main/java/org/zwobble/couscous/ast/StaticMethodCallNode.java
index b33e94c..ccb9854 100644
--- a/src/main/java/org/zwobble/couscous/ast/StaticMethodCallNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/StaticMethodCallNode.java
@@ -1,6 +1,7 @@
 package org.zwobble.couscous.ast;
 
 import java.util.List;
+import java.util.function.Function;
 
 import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper;
 import org.zwobble.couscous.values.BooleanValue;
@@ -9,6 +10,7 @@ import org.zwobble.couscous.values.InternalCouscousValue;
 import org.zwobble.couscous.values.ObjectValues;
 
 import static java.util.Arrays.asList;
+import static org.zwobble.couscous.util.ExtraLists.eagerMap;
 
 public class StaticMethodCallNode implements ExpressionNode {
     public static ExpressionNode same(ExpressionNode left, ExpressionNode right) {
@@ -101,6 +103,15 @@ public class StaticMethodCallNode implements ExpressionNode {
     }
 
     @Override
+    public ExpressionNode replaceExpressions(Function<ExpressionNode, ExpressionNode> replace) {
+        return new StaticMethodCallNode(
+            className,
+            methodName,
+            eagerMap(arguments, replace::apply),
+            type);
+    }
+
+    @Override
     public String toString() {
         return "StaticMethodCallNode(className=" + className + ", methodName="
                + methodName + ", arguments=" + arguments + ", type=" + type