Replace captured variable references in method bodies

1 files changed, 7 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/LocalVariableDeclarationNode.java b/src/main/java/org/zwobble/couscous/ast/LocalVariableDeclarationNode.java
index bb272fc..3499e26 100644
--- a/src/main/java/org/zwobble/couscous/ast/LocalVariableDeclarationNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/LocalVariableDeclarationNode.java
@@ -2,6 +2,8 @@ package org.zwobble.couscous.ast;
 
 import org.zwobble.couscous.ast.visitors.StatementNodeMapper;
 
+import java.util.function.Function;
+
 import static org.zwobble.couscous.ast.VariableDeclaration.var;
 
 public class LocalVariableDeclarationNode implements VariableNode, StatementNode {
@@ -51,6 +53,11 @@ public class LocalVariableDeclarationNode implements VariableNode, StatementNode
     }
 
     @Override
+    public StatementNode replaceExpressions(Function<ExpressionNode, ExpressionNode> replace) {
+        return new LocalVariableDeclarationNode(declaration, replace.apply(initialValue));
+    }
+
+    @Override
     public String toString() {
         return "LocalVariableDeclarationNode(declaration=" + declaration
                + ", initialValue=" + initialValue + ")";