From 722dc4b477b202cb143b70d821618753a9a7a491 Mon Sep 17 00:00:00 2001
From: Kasey White <kwhitemsg@gmail.com>
Date: Sat, 1 Oct 2022 20:36:16 -0400
Subject: [PATCH] work on adding when clause

---
 crates/lang/src/expr.rs         |  4 +--
 crates/lang/src/parser.rs       | 46 ++++++++++++++++++++++++++-------
 crates/lang/src/tests/parser.rs | 10 ++++---
 3 files changed, 44 insertions(+), 16 deletions(-)

diff --git a/crates/lang/src/expr.rs b/crates/lang/src/expr.rs
index 1fb24b8d..fbfa66b7 100644
--- a/crates/lang/src/expr.rs
+++ b/crates/lang/src/expr.rs
@@ -227,7 +227,7 @@ pub enum UntypedExpr {
         annotation: Option<Annotation>,
     },
 
-    Case {
+    When {
         location: Span,
         subjects: Vec<Self>,
         clauses: Vec<Clause<Self, (), (), ()>>,
@@ -327,7 +327,7 @@ impl UntypedExpr {
             | Self::Var { location, .. }
             | Self::Int { location, .. }
             | Self::Todo { location, .. }
-            | Self::Case { location, .. }
+            | Self::When { location, .. }
             | Self::Call { location, .. }
             | Self::List { location, .. }
             | Self::Float { location, .. }
diff --git a/crates/lang/src/parser.rs b/crates/lang/src/parser.rs
index 2e3e8ce2..9a9e664e 100644
--- a/crates/lang/src/parser.rs
+++ b/crates/lang/src/parser.rs
@@ -323,17 +323,43 @@ pub fn expr_parser() -> impl Parser<Token, expr::UntypedExpr, Error = ParseError
                 },
             );
 
-        let expr_block_parser = just(Token::LeftBrace)
-            .ignore_then(r.clone().then(r.repeated()))
-            .then_ignore(just(Token::RightBrace))
-            .map_with_span(|(a, b), span| {
-                let mut expressions = vec![a];
-                expressions.extend(b);
+        let when_clause_guard_parser = just(Token::If);
 
-                expr::UntypedExpr::Sequence {
+        let when_clause_parser = pattern_parser()
+            .separated_by(just(Token::Comma))
+            .then(
+                pattern_parser()
+                    .separated_by(just(Token::Comma))
+                    .separated_by(just(Token::Vbar))
+                    .allow_leading()
+                    .or_not(),
+            )
+            .then(when_clause_guard_parser)
+            // TODO: add hint "Did you mean to wrap a multi line clause in curly braces?"
+            .then_ignore(just(Token::RArrow))
+            .then(r.clone())
+            .map_with_span(
+                |(((patterns, alternative_patterns_opt), guard), then), span| ast::UntypedClause {
                     location: span,
-                    expressions,
-                }
+                    pattern: patterns,
+                    alternative_patterns: alternative_patterns_opt.unwrap_or_default(),
+                    guard,
+                    then,
+                },
+            );
+
+        let when_parser = just(Token::When)
+            // TODO: If subject is empty we should return ParseErrorType::ExpectedExpr,
+            .ignore_then(r.clone().separated_by(just(Token::Comma)))
+            .then_ignore(just(Token::Is))
+            .then_ignore(just(Token::LeftBrace))
+            // TODO: If clauses are empty we should return ParseErrorType::NoCaseClause
+            .then(when_clause_parser.repeated())
+            .then_ignore(just(Token::RightBrace))
+            .map_with_span(|(subjects, clauses), span| expr::UntypedExpr::When {
+                location: span,
+                subjects,
+                clauses,
             });
 
         let expr_unit_parser = choice((
@@ -343,7 +369,7 @@ pub fn expr_parser() -> impl Parser<Token, expr::UntypedExpr, Error = ParseError
             todo_parser,
             list_parser,
             assignment_parser,
-            expr_block_parser,
+            when_parser,
         ))
         .boxed();
 
diff --git a/crates/lang/src/tests/parser.rs b/crates/lang/src/tests/parser.rs
index c8ad1fd8..45fa655d 100644
--- a/crates/lang/src/tests/parser.rs
+++ b/crates/lang/src/tests/parser.rs
@@ -52,10 +52,12 @@ fn module() {
             }
 
             pub fn wow2(a: Int){
-                let x = 1 + {
-                    2 + 7 * 8
-                } / 2
-                x
+               when a is {
+                0 -> 3
+                1 | 2 -> 5
+                3 if a == 3 -> 9
+                _ -> 4
+               }
             }
         "#;
     let len = code.chars().count();