Introduce SwitchCaseNode to simplify types

2 files changed, 36 insertions, 3 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/IfStatementNode.java b/src/main/java/org/zwobble/couscous/ast/IfStatementNode.java
index cb5eee8..8b4709a 100644
--- a/src/main/java/org/zwobble/couscous/ast/IfStatementNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/IfStatementNode.java
@@ -1,14 +1,14 @@
 package org.zwobble.couscous.ast;
 
+import org.zwobble.couscous.ast.visitors.StatementNodeMapper;
+
 import java.util.List;
 import java.util.function.Function;
 
-import org.zwobble.couscous.ast.visitors.StatementNodeMapper;
-
 import static org.zwobble.couscous.util.ExtraLists.eagerMap;
 
 public class IfStatementNode implements StatementNode {
-    public static IfStatementNode ifStatement(
+    public static StatementNode ifStatement(
             ExpressionNode condition,
             List<StatementNode> trueBranch,
             List<StatementNode> falseBranch) {
diff --git a/src/main/java/org/zwobble/couscous/ast/sugar/SwitchCaseNode.java b/src/main/java/org/zwobble/couscous/ast/sugar/SwitchCaseNode.java
new file mode 100644
index 0000000..534f244
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/ast/sugar/SwitchCaseNode.java
@@ -0,0 +1,33 @@
+package org.zwobble.couscous.ast.sugar;
+
+import org.zwobble.couscous.ast.ExpressionNode;
+import org.zwobble.couscous.ast.StatementNode;
+
+import java.util.List;
+import java.util.Optional;
+
+public class SwitchCaseNode {
+    public static SwitchCaseNode switchCase(Optional<ExpressionNode> value, List<StatementNode> statements) {
+        return new SwitchCaseNode(value, statements);
+    }
+
+    private final Optional<ExpressionNode> value;
+    private final List<StatementNode> statements;
+
+    public SwitchCaseNode(Optional<ExpressionNode> value, List<StatementNode> statements) {
+        this.value = value;
+        this.statements = statements;
+    }
+
+    public boolean isDefault() {
+        return !value.isPresent();
+    }
+
+    public Optional<ExpressionNode> getValue() {
+        return value;
+    }
+
+    public List<StatementNode> getStatements() {
+        return statements;
+    }
+}