From c7b0410af60f4267264a875a80cfda2d45381bb3 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 31 Jul 2016 19:33:48 +0100
Subject: Extract common code for building desugar statement transformers

---
 .../couscous/transforms/DesugarStatement.java      | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 src/main/java/org/zwobble/couscous/transforms/DesugarStatement.java

(limited to 'src/main/java/org/zwobble/couscous/transforms/DesugarStatement.java')

diff --git a/src/main/java/org/zwobble/couscous/transforms/DesugarStatement.java b/src/main/java/org/zwobble/couscous/transforms/DesugarStatement.java
new file mode 100644
index 0000000..700b31b
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/transforms/DesugarStatement.java
@@ -0,0 +1,22 @@
+package org.zwobble.couscous.transforms;
+
+import org.zwobble.couscous.ast.StatementNode;
+import org.zwobble.couscous.ast.visitors.NodeTransformer;
+
+import java.util.List;
+import java.util.Optional;
+import java.util.function.Function;
+
+class DesugarStatement {
+    static <T extends StatementNode> NodeTransformer transformer(int nodeType, Function<T, List<StatementNode>> desugar) {
+        return NodeTransformer.builder()
+            .transformStatement(statement -> {
+                if (statement.type() == nodeType) {
+                    return Optional.of(desugar.apply((T) statement));
+                } else {
+                    return Optional.empty();
+                }
+            })
+            .build();
+    }
+}
-- 
cgit v1.2.3