From 2368822a1d143e9fd677f02b0b944bbace0c911d Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 17 Jan 2016 18:23:09 +0000 Subject: Implement StatementNode.replaceExpressions using node transformer --- src/main/java/org/zwobble/couscous/ast/StatementNode.java | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) (limited to 'src/main/java/org/zwobble/couscous/ast/StatementNode.java') diff --git a/src/main/java/org/zwobble/couscous/ast/StatementNode.java b/src/main/java/org/zwobble/couscous/ast/StatementNode.java index 4b9c9de..87b1319 100644 --- a/src/main/java/org/zwobble/couscous/ast/StatementNode.java +++ b/src/main/java/org/zwobble/couscous/ast/StatementNode.java @@ -1,6 +1,7 @@ package org.zwobble.couscous.ast; import org.zwobble.couscous.ast.visitors.NodeMapper; +import org.zwobble.couscous.ast.visitors.NodeTransformer; import org.zwobble.couscous.ast.visitors.StatementNodeMapper; import java.util.function.Function; @@ -38,5 +39,13 @@ public interface StatementNode extends Node { }); } - StatementNode replaceExpressions(Function replace); + StatementNode transform(NodeTransformer transformer); + default StatementNode replaceExpressions(Function replace) { + return transform(new NodeTransformer() { + @Override + public ExpressionNode visit(ExpressionNode value) { + return replace.apply(value); + } + }); + } } -- cgit v1.2.3