Replace captured variable references in method bodies

1 files changed, 4 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/StatementNode.java b/src/main/java/org/zwobble/couscous/ast/StatementNode.java
index a052917..4b9c9de 100644
--- a/src/main/java/org/zwobble/couscous/ast/StatementNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/StatementNode.java
@@ -3,6 +3,8 @@ package org.zwobble.couscous.ast;
 import org.zwobble.couscous.ast.visitors.NodeMapper;
 import org.zwobble.couscous.ast.visitors.StatementNodeMapper;
 
+import java.util.function.Function;
+
 public interface StatementNode extends Node {
     <T> T accept(StatementNodeMapper<T> visitor);
     
@@ -35,4 +37,6 @@ public interface StatementNode extends Node {
             }
         });
     }
+
+    StatementNode replaceExpressions(Function<ExpressionNode, ExpressionNode> replace);
 }