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

---
 .../java/org/zwobble/couscous/ast/StaticMethodCallNode.java   | 11 +++++++++++
 1 file changed, 11 insertions(+)

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

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