From 921de028319d06253bc2a73302e8f4ee51bc7ca0 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 17 Jan 2016 18:09:32 +0000 Subject: Implement replaceExpressions using NodeTransformer --- src/main/java/org/zwobble/couscous/ast/ExpressionNode.java | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) (limited to 'src/main/java/org/zwobble/couscous/ast/ExpressionNode.java') diff --git a/src/main/java/org/zwobble/couscous/ast/ExpressionNode.java b/src/main/java/org/zwobble/couscous/ast/ExpressionNode.java index d07b485..79a0cd9 100644 --- a/src/main/java/org/zwobble/couscous/ast/ExpressionNode.java +++ b/src/main/java/org/zwobble/couscous/ast/ExpressionNode.java @@ -2,6 +2,7 @@ package org.zwobble.couscous.ast; import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper; import org.zwobble.couscous.ast.visitors.NodeMapper; +import org.zwobble.couscous.ast.visitors.NodeTransformer; import java.util.function.Function; @@ -64,5 +65,14 @@ public interface ExpressionNode extends Node { }); } - ExpressionNode replaceExpressions(Function replace); + ExpressionNode transform(NodeTransformer transformer); + + default ExpressionNode replaceExpressions(Function replace) { + return transform(new NodeTransformer() { + @Override + public ExpressionNode visit(ExpressionNode value) { + return replace.apply(value); + } + }); + } } -- cgit v1.2.3