From bc06840e00425db342bc5a021c0900d9e63b71f0 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Fri, 15 Jul 2016 14:23:47 +0100
Subject: Split switch handling off to desugar step

---
 .../org/zwobble/couscous/ast/sugar/SwitchNode.java | 48 ++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 src/main/java/org/zwobble/couscous/ast/sugar/SwitchNode.java

(limited to 'src/main/java/org/zwobble/couscous/ast/sugar/SwitchNode.java')

diff --git a/src/main/java/org/zwobble/couscous/ast/sugar/SwitchNode.java b/src/main/java/org/zwobble/couscous/ast/sugar/SwitchNode.java
new file mode 100644
index 0000000..8e1e226
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/ast/sugar/SwitchNode.java
@@ -0,0 +1,48 @@
+package org.zwobble.couscous.ast.sugar;
+
+import org.zwobble.couscous.ast.ExpressionNode;
+import org.zwobble.couscous.ast.Node;
+import org.zwobble.couscous.ast.NodeTypes;
+import org.zwobble.couscous.ast.StatementNode;
+import org.zwobble.couscous.ast.visitors.NodeTransformer;
+
+import java.util.List;
+
+import static org.zwobble.couscous.util.ExtraIterables.lazyCons;
+import static org.zwobble.couscous.util.ExtraLists.eagerMap;
+
+public class SwitchNode implements StatementNode {
+    private final ExpressionNode value;
+    private final List<SwitchCaseNode> cases;
+
+    public SwitchNode(ExpressionNode value, List<SwitchCaseNode> cases) {
+        this.value = value;
+        this.cases = cases;
+    }
+
+    public ExpressionNode getValue() {
+        return value;
+    }
+
+    public List<SwitchCaseNode> getCases() {
+        return cases;
+    }
+
+    @Override
+    public int type() {
+        return NodeTypes.SWITCH;
+    }
+
+    @Override
+    public Iterable<? extends Node> childNodes() {
+        return lazyCons(value, cases);
+    }
+
+    @Override
+    public StatementNode transform(NodeTransformer transformer) {
+        return new SwitchNode(
+            transformer.transformExpression(value),
+            eagerMap(cases, caseNode -> caseNode.transform(transformer))
+        );
+    }
+}
-- 
cgit v1.2.3