Extract common code for building desugar statement transformers

1 files changed, 22 insertions, 0 deletions

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();
+    }
+}