From c7b0410af60f4267264a875a80cfda2d45381bb3 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sun, 31 Jul 2016 19:33:48 +0100 Subject: Extract common code for building desugar statement transformers --- .../zwobble/couscous/transforms/DesugarSwitchToIfElse.java | 11 +---------- 1 file changed, 1 insertion(+), 10 deletions(-) (limited to 'src/main/java/org/zwobble/couscous/transforms/DesugarSwitchToIfElse.java') diff --git a/src/main/java/org/zwobble/couscous/transforms/DesugarSwitchToIfElse.java b/src/main/java/org/zwobble/couscous/transforms/DesugarSwitchToIfElse.java index 5ecd25d..006a7c9 100644 --- a/src/main/java/org/zwobble/couscous/transforms/DesugarSwitchToIfElse.java +++ b/src/main/java/org/zwobble/couscous/transforms/DesugarSwitchToIfElse.java @@ -11,7 +11,6 @@ import org.zwobble.couscous.types.Types; import org.zwobble.couscous.util.ExtraLists; import java.util.List; -import java.util.Optional; import static com.google.common.collect.Iterables.tryFind; import static org.zwobble.couscous.ast.IfStatementNode.ifStatement; @@ -30,15 +29,7 @@ public class DesugarSwitchToIfElse { public static NodeTransformer transformer() { Scope scope = Scope.create().temporaryPrefix("_couscous_desugar_switch_to_if"); - return NodeTransformer.builder() - .transformStatement(statement -> { - if (statement.type() == NodeTypes.SWITCH) { - return Optional.of(desugar(scope, ((SwitchNode) statement))); - } else { - return Optional.empty(); - } - }) - .build(); + return DesugarStatement.transformer(NodeTypes.SWITCH, node -> desugar(scope, node)); } public static List desugar(Scope scope, SwitchNode switchNode) { -- cgit v1.2.3