move patterns, clauses into AST

src/base.rs

@@ -4,7 +4,7 @@ use ran::ran_f64;
 use std::rc::Rc;
 
 #[derive(Clone, Debug)]
-pub enum Base<'src> {
+pub enum BaseFn<'src> {
     Nullary(fn() -> Value<'src>),
     Unary(fn(&Value<'src>) -> Value<'src>),
     Binary(fn(&Value<'src>, &Value<'src>) -> Value<'src>),
@@ -572,59 +572,59 @@ pub fn r#mod<'src>(x: &Value, y: &Value) -> Value<'src> {
 
 pub fn base<'src>() -> Vec<(String, Value<'src>)> {
     let members = vec![
-        ("add", Value::Base(Base::Binary(add))),
-        ("and", Value::Base(Base::Binary(and))),
-        ("append", Value::Base(Base::Binary(append))),
-        ("assoc", Value::Base(Base::Ternary(assoc))),
-        ("at", Value::Base(Base::Binary(at))),
-        ("atan_2", Value::Base(Base::Binary(atan_2))),
-        ("bool", Value::Base(Base::Unary(r#bool))),
-        ("ceil", Value::Base(Base::Unary(ceil))),
-        ("chars", Value::Base(Base::Unary(chars))),
-        ("concat", Value::Base(Base::Binary(concat))),
-        ("cos", Value::Base(Base::Unary(cos))),
-        ("count", Value::Base(Base::Unary(count))),
-        ("dec", Value::Base(Base::Unary(dec))),
-        ("dissoc", Value::Base(Base::Binary(dissoc))),
-        ("div", Value::Base(Base::Binary(div))),
-        ("doc!", Value::Base(Base::Unary(doc))),
-        ("downcase", Value::Base(Base::Unary(downcase))),
-        ("eq?", Value::Base(Base::Binary(eq))),
-        ("first", Value::Base(Base::Unary(first))),
-        ("floor", Value::Base(Base::Unary(floor))),
-        ("get", Value::Base(Base::Binary(get))),
-        ("gt?", Value::Base(Base::Binary(gt))),
-        ("gte?", Value::Base(Base::Binary(gte))),
-        ("inc", Value::Base(Base::Unary(inc))),
-        ("last", Value::Base(Base::Unary(last))),
-        ("list", Value::Base(Base::Unary(list))),
-        ("lt?", Value::Base(Base::Binary(lt))),
-        ("lte?", Value::Base(Base::Binary(lte))),
-        ("mod", Value::Base(Base::Binary(r#mod))),
-        ("mult", Value::Base(Base::Binary(mult))),
-        ("number", Value::Base(Base::Unary(number))),
-        ("or", Value::Base(Base::Binary(or))),
+        ("add", Value::Base(BaseFn::Binary(add))),
+        ("and", Value::Base(BaseFn::Binary(and))),
+        ("append", Value::Base(BaseFn::Binary(append))),
+        ("assoc", Value::Base(BaseFn::Ternary(assoc))),
+        ("at", Value::Base(BaseFn::Binary(at))),
+        ("atan_2", Value::Base(BaseFn::Binary(atan_2))),
+        ("bool", Value::Base(BaseFn::Unary(r#bool))),
+        ("ceil", Value::Base(BaseFn::Unary(ceil))),
+        ("chars", Value::Base(BaseFn::Unary(chars))),
+        ("concat", Value::Base(BaseFn::Binary(concat))),
+        ("cos", Value::Base(BaseFn::Unary(cos))),
+        ("count", Value::Base(BaseFn::Unary(count))),
+        ("dec", Value::Base(BaseFn::Unary(dec))),
+        ("dissoc", Value::Base(BaseFn::Binary(dissoc))),
+        ("div", Value::Base(BaseFn::Binary(div))),
+        ("doc!", Value::Base(BaseFn::Unary(doc))),
+        ("downcase", Value::Base(BaseFn::Unary(downcase))),
+        ("eq?", Value::Base(BaseFn::Binary(eq))),
+        ("first", Value::Base(BaseFn::Unary(first))),
+        ("floor", Value::Base(BaseFn::Unary(floor))),
+        ("get", Value::Base(BaseFn::Binary(get))),
+        ("gt?", Value::Base(BaseFn::Binary(gt))),
+        ("gte?", Value::Base(BaseFn::Binary(gte))),
+        ("inc", Value::Base(BaseFn::Unary(inc))),
+        ("last", Value::Base(BaseFn::Unary(last))),
+        ("list", Value::Base(BaseFn::Unary(list))),
+        ("lt?", Value::Base(BaseFn::Binary(lt))),
+        ("lte?", Value::Base(BaseFn::Binary(lte))),
+        ("mod", Value::Base(BaseFn::Binary(r#mod))),
+        ("mult", Value::Base(BaseFn::Binary(mult))),
+        ("number", Value::Base(BaseFn::Unary(number))),
+        ("or", Value::Base(BaseFn::Binary(or))),
         ("pi", Value::Number(std::f64::consts::PI)),
-        ("print!", Value::Base(Base::Unary(print))),
-        ("random", Value::Base(Base::Nullary(random))),
-        ("range", Value::Base(Base::Binary(range))),
-        ("rest", Value::Base(Base::Unary(rest))),
-        ("round", Value::Base(Base::Unary(round))),
-        ("show", Value::Base(Base::Unary(show))),
-        ("sin", Value::Base(Base::Unary(sin))),
-        ("slice", Value::Base(Base::Ternary(slice))),
-        ("split", Value::Base(Base::Binary(split))),
-        ("sqrt", Value::Base(Base::Unary(sqrt))),
+        ("print!", Value::Base(BaseFn::Unary(print))),
+        ("random", Value::Base(BaseFn::Nullary(random))),
+        ("range", Value::Base(BaseFn::Binary(range))),
+        ("rest", Value::Base(BaseFn::Unary(rest))),
+        ("round", Value::Base(BaseFn::Unary(round))),
+        ("show", Value::Base(BaseFn::Unary(show))),
+        ("sin", Value::Base(BaseFn::Unary(sin))),
+        ("slice", Value::Base(BaseFn::Ternary(slice))),
+        ("split", Value::Base(BaseFn::Binary(split))),
+        ("sqrt", Value::Base(BaseFn::Unary(sqrt))),
         ("sqrt_2", Value::Number(std::f64::consts::SQRT_2)),
-        ("store!", Value::Base(Base::Binary(store))),
-        ("sub", Value::Base(Base::Binary(sub))),
-        ("tan", Value::Base(Base::Unary(tan))),
-        ("trim", Value::Base(Base::Unary(trim))),
-        ("triml", Value::Base(Base::Unary(triml))),
-        ("trimr", Value::Base(Base::Unary(trimr))),
-        ("type", Value::Base(Base::Unary(r#type))),
-        ("unbox", Value::Base(Base::Unary(unbox))),
-        ("upcase", Value::Base(Base::Unary(upcase))),
+        ("store!", Value::Base(BaseFn::Binary(store))),
+        ("sub", Value::Base(BaseFn::Binary(sub))),
+        ("tan", Value::Base(BaseFn::Unary(tan))),
+        ("trim", Value::Base(BaseFn::Unary(trim))),
+        ("triml", Value::Base(BaseFn::Unary(triml))),
+        ("trimr", Value::Base(BaseFn::Unary(trimr))),
+        ("type", Value::Base(BaseFn::Unary(r#type))),
+        ("unbox", Value::Base(BaseFn::Unary(unbox))),
+        ("upcase", Value::Base(BaseFn::Unary(upcase))),
     ];
     let pkg = Value::Dict(HashMap::from(members));
     vec![("base".to_string(), pkg)]

src/context.rs

@@ -1,5 +1,6 @@
 use crate::base::*;
 use crate::parser::*;
+use crate::spans::*;
 use crate::value::{Fn, Value};
 use imbl::HashMap;
 use imbl::Vector;
@@ -71,16 +72,17 @@ impl<'src> Context<'src> {
         }
     }
 
-    pub fn match_pattern(&mut self, patt: &Pattern, val: &Value<'src>) -> Option<&Context<'src>> {
+    pub fn match_pattern(&mut self, patt: &Ast, val: &Value<'src>) -> Option<&Context<'src>> {
+        use Ast::*;
         match (patt, val) {
-            (Pattern::Nil, Value::Nil) => Some(self),
-            (Pattern::Placeholder, _) => Some(self),
-            (Pattern::Number(x), Value::Number(y)) => self.match_eq(x, y),
-            (Pattern::Boolean(x), Value::Boolean(y)) => self.match_eq(x, y),
-            (Pattern::Keyword(x), Value::Keyword(y)) => self.match_eq(x, y),
-            (Pattern::String(x), Value::InternedString(y)) => self.match_eq(x, y),
-            (Pattern::String(x), Value::AllocatedString(y)) => self.match_eq(&x.to_string(), y),
-            (Pattern::Interpolated(_, StringMatcher(matcher)), Value::InternedString(y)) => {
+            (NilPattern, Value::Nil) => Some(self),
+            (PlaceholderPattern, _) => Some(self),
+            (NumberPattern(x), Value::Number(y)) => self.match_eq(x, y),
+            (BooleanPattern(x), Value::Boolean(y)) => self.match_eq(x, y),
+            (KeywordPattern(x), Value::Keyword(y)) => self.match_eq(x, y),
+            (StringPattern(x), Value::InternedString(y)) => self.match_eq(x, y),
+            (StringPattern(x), Value::AllocatedString(y)) => self.match_eq(&x.to_string(), y),
+            (InterpolatedPattern(_, StringMatcher(matcher)), Value::InternedString(y)) => {
                 match matcher(y.to_string()) {
                     Some(matches) => {
                         let mut matches = matches
@@ -98,11 +100,11 @@ impl<'src> Context<'src> {
                     None => None,
                 }
             }
-            (Pattern::Word(w), val) => {
+            (WordPattern(w), val) => {
                 self.bind(w.to_string(), &val);
                 Some(self)
             }
-            (Pattern::As(word, type_str), value) => {
+            (AsPattern(word, type_str), value) => {
                 let ludus_type = r#type(value);
                 let type_kw = Value::Keyword(type_str);
                 if type_kw == ludus_type {
@@ -112,16 +114,14 @@ impl<'src> Context<'src> {
                     None
                 }
             }
-            (Pattern::Tuple(x), Value::Tuple(y)) => {
-                let has_splat = x
-                    .iter()
-                    .any(|patt| matches!(patt, (Pattern::Splattern(_), _)));
+            (TuplePattern(x), Value::Tuple(y)) => {
+                let has_splat = x.iter().any(|patt| matches!(patt, (Splattern(_), _)));
                 if x.len() > y.len() || (!has_splat && x.len() != y.len()) {
                     return None;
                 };
                 let to = self.locals.len();
                 for i in 0..x.len() {
-                    if let Pattern::Splattern(patt) = &x[i].0 {
+                    if let Splattern(patt) = &x[i].0 {
                         let mut list = Vector::new();
                         for i in i..y.len() {
                             list.push_back(y[i].clone())
@@ -135,16 +135,14 @@ impl<'src> Context<'src> {
                 }
                 Some(self)
             }
-            (Pattern::List(x), Value::List(y)) => {
-                let has_splat = x
-                    .iter()
-                    .any(|patt| matches!(patt, (Pattern::Splattern(_), _)));
+            (ListPattern(x), Value::List(y)) => {
+                let has_splat = x.iter().any(|patt| matches!(patt, (Splattern(_), _)));
                 if x.len() > y.len() || (!has_splat && x.len() != y.len()) {
                     return None;
                 };
                 let to = self.locals.len();
                 for (i, (patt, _)) in x.iter().enumerate() {
-                    if let Pattern::Splattern(patt) = &patt {
+                    if let Splattern(patt) = &patt {
                         let list = Value::List(y.skip(i));
                         self.match_pattern(&patt.0, &list);
                     } else if self.match_pattern(patt, y.get(i).unwrap()).is_none() {
@@ -157,10 +155,8 @@ impl<'src> Context<'src> {
             // TODO: optimize this on several levels
             // - [ ] opportunistic mutation
             // - [ ] get rid of all the pointer indirection in word splats
-            (Pattern::Dict(x), Value::Dict(y)) => {
-                let has_splat = x
-                    .iter()
-                    .any(|patt| matches!(patt, (Pattern::Splattern(_), _)));
+            (DictPattern(x), Value::Dict(y)) => {
+                let has_splat = x.iter().any(|patt| matches!(patt, (Splattern(_), _)));
                 if x.len() > y.len() || (!has_splat && x.len() != y.len()) {
                     return None;
                 };
@@ -168,7 +164,7 @@ impl<'src> Context<'src> {
                 let mut matched = vec![];
                 for (pattern, _) in x {
                     match pattern {
-                        Pattern::Pair(key, patt) => {
+                        PairPattern(key, patt) => {
                             if let Some(val) = y.get(key) {
                                 if self.match_pattern(&patt.0, val).is_none() {
                                     self.pop_to(to);
@@ -180,8 +176,8 @@ impl<'src> Context<'src> {
                                 return None;
                             };
                         }
-                        Pattern::Splattern(pattern) => match pattern.0 {
-                            Pattern::Word(w) => {
+                        Splattern(pattern) => match pattern.0 {
+                            WordPattern(w) => {
                                 // TODO: find a way to take ownership
                                 // this will ALWAYS make structural changes, because of this clone
                                 // we want opportunistic mutation if possible
@@ -191,7 +187,7 @@ impl<'src> Context<'src> {
                                 }
                                 self.bind(w.to_string(), &Value::Dict(unmatched));
                             }
-                            Pattern::Placeholder => (),
+                            PlaceholderPattern => (),
                             _ => unreachable!(),
                         },
                         _ => unreachable!(),
@@ -206,14 +202,15 @@ impl<'src> Context<'src> {
     pub fn match_clauses(
         &mut self,
         value: &Value<'src>,
-        clauses: &'src [MatchClause],
+        clauses: &'src [Spanned<Ast>],
     ) -> LResult<'src> {
         {
             let parent = self.ast;
             let to = self.locals.len();
-            for MatchClause { patt, body, guard } in clauses.iter() {
+            let mut clauses = clauses.iter();
+            while let Some((Ast::MatchClause(patt, guard, body), _)) = clauses.next() {
                 if let Some(_) = self.match_pattern(&patt.0, value) {
-                    let pass_guard = match guard {
+                    let pass_guard = match guard.as_ref() {
                         None => true,
                         Some((ast, _)) => {
                             self.ast = ast;
@@ -240,30 +237,31 @@ impl<'src> Context<'src> {
     }
 
     pub fn apply(&mut self, callee: Value<'src>, caller: Value<'src>) -> LResult<'src> {
+        use Value::*;
         match (callee, caller) {
-            (Value::Keyword(kw), Value::Dict(dict)) => {
+            (Keyword(kw), Dict(dict)) => {
                 if let Some(val) = dict.get(kw) {
                     Ok(val.clone())
                 } else {
-                    Ok(Value::Nil)
+                    Ok(Nil)
                 }
             }
-            (Value::Dict(dict), Value::Keyword(kw)) => {
+            (Dict(dict), Keyword(kw)) => {
                 if let Some(val) = dict.get(kw) {
                     Ok(val.clone())
                 } else {
-                    Ok(Value::Nil)
+                    Ok(Nil)
                 }
             }
-            (Value::Fn(f), Value::Tuple(args)) => {
-                let args = Value::Tuple(args);
+            (Fn(f), Tuple(args)) => {
+                let args = Tuple(args);
                 self.match_clauses(&args, f.body)
             }
-            (Value::Fn(_f), Value::Args(_args)) => todo!(),
-            (_, Value::Keyword(_)) => Ok(Value::Nil),
-            (_, Value::Args(_)) => Err(LErr::new(format!("you may only call a function"))),
-            (Value::Base(f), Value::Tuple(args)) => match f {
-                Base::Nullary(f) => {
+            (Fn(_f), Args(_args)) => todo!(),
+            (_, Keyword(_)) => Ok(Nil),
+            (_, Args(_)) => Err(LErr::new(format!("you may only call a function"))),
+            (Base(f), Tuple(args)) => match f {
+                BaseFn::Nullary(f) => {
                     let num_args = args.len();
                     if num_args != 0 {
                         Err(LErr::new(format!(
@@ -273,7 +271,7 @@ impl<'src> Context<'src> {
                         Ok(f())
                     }
                 }
-                Base::Unary(f) => {
+                BaseFn::Unary(f) => {
                     let num_args = args.len();
                     if num_args != 1 {
                         Err(LErr::new(format!(
@@ -283,7 +281,7 @@ impl<'src> Context<'src> {
                         Ok(f(&args[0]))
                     }
                 }
-                Base::Binary(r#fn) => {
+                BaseFn::Binary(r#fn) => {
                     let num_args = args.len();
                     if num_args != 2 {
                         Err(LErr::new(format!(
@@ -293,7 +291,7 @@ impl<'src> Context<'src> {
                         Ok(r#fn(&args[0], &args[1]))
                     }
                 }
-                Base::Ternary(f) => {
+                BaseFn::Ternary(f) => {
                     let num_args = args.len();
                     if num_args != 3 {
                         Err(LErr::new(format!(
@@ -309,15 +307,16 @@ impl<'src> Context<'src> {
     }
 
     pub fn eval(&mut self) -> LResult<'src> {
+        use Ast::*;
         let root = self.ast;
         let result = match self.ast {
-            Ast::Nil => Ok(Value::Nil),
-            Ast::Boolean(b) => Ok(Value::Boolean(*b)),
-            Ast::Number(n) => Ok(Value::Number(*n)),
-            Ast::Keyword(k) => Ok(Value::Keyword(k)),
-            Ast::String(s) => Ok(Value::InternedString(s)),
-            Ast::Interpolated(parts) => {
-                let mut interpolated = String::new();
+            Nil => Ok(Value::Nil),
+            Boolean(b) => Ok(Value::Boolean(*b)),
+            Number(n) => Ok(Value::Number(*n)),
+            Keyword(k) => Ok(Value::Keyword(k)),
+            String(s) => Ok(Value::InternedString(s)),
+            Interpolated(parts) => {
+                let mut interpolated = std::string::String::new();
                 for part in parts {
                     match &part.0 {
                         StringPart::Data(s) => interpolated.push_str(s.as_str()),
@@ -330,7 +329,7 @@ impl<'src> Context<'src> {
                 }
                 Ok(Value::AllocatedString(Rc::new(interpolated)))
             }
-            Ast::Block(exprs) => {
+            Block(exprs) => {
                 let parent = self.ast;
                 let to = self.locals.len();
                 let mut result = Value::Nil;
@@ -342,7 +341,7 @@ impl<'src> Context<'src> {
                 self.ast = parent;
                 Ok(result)
             }
-            Ast::If(cond, if_true, if_false) => {
+            If(cond, if_true, if_false) => {
                 let parent = self.ast;
                 self.ast = &cond.0;
                 let truthy = self.eval()?.bool();
@@ -351,7 +350,7 @@ impl<'src> Context<'src> {
                 self.ast = parent;
                 result
             }
-            Ast::List(members) => {
+            List(members) => {
                 let parent = self.ast;
                 let mut vect = Vector::new();
                 for member in members {
@@ -373,7 +372,7 @@ impl<'src> Context<'src> {
                 self.ast = parent;
                 Ok(Value::List(vect))
             }
-            Ast::Tuple(members) => {
+            Tuple(members) => {
                 let parent = self.ast;
                 let mut vect = Vec::new();
                 for member in members {
@@ -383,11 +382,11 @@ impl<'src> Context<'src> {
                 self.ast = parent;
                 Ok(Value::Tuple(Rc::new(vect)))
             }
-            Ast::Word(w) | Ast::Splat(w) => {
+            Word(w) | Ast::Splat(w) => {
                 let val = self.resolve(&w.to_string())?;
                 Ok(val)
             }
-            Ast::Let(patt, expr) => {
+            Let(patt, expr) => {
                 let parent = self.ast;
                 self.ast = &expr.0;
                 let val = self.eval()?;
@@ -398,9 +397,8 @@ impl<'src> Context<'src> {
                 self.ast = parent;
                 result
             }
-            Ast::Placeholder => Ok(Value::Placeholder),
-            Ast::Error => unreachable!(),
-            Ast::Arguments(a) => {
+            Placeholder => Ok(Value::Placeholder),
+            Arguments(a) => {
                 let parent = self.ast;
                 let mut args = vec![];
                 for (arg, _) in a.iter() {
@@ -416,7 +414,7 @@ impl<'src> Context<'src> {
                 self.ast = parent;
                 result
             }
-            Ast::Dict(terms) => {
+            Dict(terms) => {
                 let parent = self.ast;
                 let mut dict = HashMap::new();
                 for term in terms {
@@ -441,7 +439,7 @@ impl<'src> Context<'src> {
                 self.ast = parent;
                 Ok(Value::Dict(dict))
             }
-            Ast::Box(name, expr) => {
+            LBox(name, expr) => {
                 let parent = self.ast;
                 self.ast = &expr.0;
                 let val = self.eval()?;
@@ -450,7 +448,7 @@ impl<'src> Context<'src> {
                 self.ast = parent;
                 Ok(boxed)
             }
-            Ast::Synthetic(root, first, rest) => {
+            Synthetic(root, first, rest) => {
                 let parent = self.ast;
                 self.ast = &root.0;
                 let root = self.eval()?;
@@ -465,10 +463,12 @@ impl<'src> Context<'src> {
                 self.ast = parent;
                 Ok(curr)
             }
-            Ast::When(clauses) => {
+            When(clauses) => {
                 let parent = self.ast;
                 for clause in clauses.iter() {
-                    let WhenClause { cond, body } = &clause.0;
+                    let WhenClause(cond, body) = &clause.0 else {
+                        unreachable!()
+                    };
                     self.ast = &cond.0;
                     if self.eval()?.bool() {
                         self.ast = &body.0;
@@ -479,7 +479,7 @@ impl<'src> Context<'src> {
                 }
                 Err(LErr::new(format!("no match")))
             }
-            Ast::Match(value, clauses) => {
+            Match(value, clauses) => {
                 let parent = self.ast;
                 self.ast = &value.0;
                 let value = self.eval()?;
@@ -487,9 +487,9 @@ impl<'src> Context<'src> {
                 self.ast = parent;
                 result
             }
-            Ast::Fn(name, clauses, doc) => {
+            Fn(name, clauses, doc) => {
                 let doc = doc.map(|s| s.to_string());
-                let the_fn = Value::Fn::<'src>(Rc::new(Fn::<'src> {
+                let the_fn = Value::Fn::<'src>(Rc::new(crate::value::Fn::<'src> {
                     name: name.to_string(),
                     body: clauses,
                     doc,
@@ -497,13 +497,13 @@ impl<'src> Context<'src> {
                 self.bind(name.to_string(), &the_fn);
                 Ok(the_fn)
             }
-            Ast::FnDeclaration(_name) => Ok(Value::Nil),
-            Ast::Panic(msg) => {
+            FnDeclaration(_name) => Ok(Value::Nil),
+            Panic(msg) => {
                 self.ast = &msg.0;
                 let msg = self.eval()?;
                 Err(LErr::new(format!("{msg}")))
             }
-            Ast::Repeat(times, body) => {
+            Repeat(times, body) => {
                 let parent = self.ast;
                 self.ast = &times.0;
                 let times_num = match self.eval() {
@@ -517,7 +517,7 @@ impl<'src> Context<'src> {
                 self.ast = parent;
                 Ok(Value::Nil)
             }
-            Ast::Do(terms) => {
+            Do(terms) => {
                 let parent = self.ast;
                 self.ast = &terms[0].0;
                 let mut result = self.eval()?;
@@ -530,10 +530,7 @@ impl<'src> Context<'src> {
                 self.ast = parent;
                 Ok(result)
             }
-            Ast::Pair(..) => {
-                unreachable!()
-            }
-            Ast::Loop(init, clauses) => {
+            Loop(init, clauses) => {
                 let parent = self.ast;
                 self.ast = &init.0;
                 let mut args = self.eval()?;
@@ -547,7 +544,7 @@ impl<'src> Context<'src> {
                     }
                 }
             }
-            Ast::Recur(args) => {
+            Recur(args) => {
                 let parent = self.ast;
                 let mut vect = Vec::new();
                 for arg in args {
@@ -557,6 +554,7 @@ impl<'src> Context<'src> {
                 self.ast = parent;
                 Ok(Value::Recur(vect))
             }
+            _ => unreachable!(),
         };
         self.ast = root;
         result

src/main.rs

@@ -150,26 +150,25 @@ pub fn run(src: &'static str) {
 
     let mut valxor = validator::Validator::new(&ast, span, &dummy_prelude);
 
-    valxor.validate();
+    // valxor.validate();
 
-    dbg!(valxor);
+    // dbg!(valxor);
 
-    // let mut ctx = prelude();
-    // ctx.ast = *
+    let mut ctx = prelude();
+    ctx.ast = *
 
-    // let result = ctx.eval();
+    let result = ctx.eval();
 
-    // match result {
-    //     Ok(result) => println!("{}", result),
-    //     Err(LErr { msg, .. }) => println!("Errors!\n{}", msg),
-    // }
+    match result {
+        Ok(result) => println!("{}", result),
+        Err(LErr { msg, .. }) => println!("Errors!\n{}", msg),
+    }
 }
 
 pub fn main() {
     let src = "
-loop () with {
-    () -> :baz
-}
+let #{:a (x, y), :b [1, 2, (a, b)]} = #{:a (1, 2), :b [1, 2, (7, 8)]}
+(x, y, a, b)
     ";
     run(src);
     // struct_scalpel::print_dissection_info::<value::Value>()

src/parser.rs

@@ -4,34 +4,34 @@ use chumsky::{input::ValueInput, prelude::*, recursive::Recursive};
 use std::fmt;
 use struct_scalpel::Dissectible;
 
-#[derive(Clone, Debug, PartialEq)]
-pub struct WhenClause {
-    pub cond: Spanned<Ast>,
-    pub body: Spanned<Ast>,
-}
+// #[derive(Clone, Debug, PartialEq)]
+// pub struct WhenClause {
+//     pub cond: Spanned<Ast>,
+//     pub body: Spanned<Ast>,
+// }
 
-impl fmt::Display for WhenClause {
-    fn fmt(self: &WhenClause, f: &mut fmt::Formatter) -> fmt::Result {
-        write!(f, "cond: {}, body: {}", self.cond.0, self.body.0)
-    }
-}
+// impl fmt::Display for WhenClause {
+//     fn fmt(self: &WhenClause, f: &mut fmt::Formatter) -> fmt::Result {
+//         write!(f, "cond: {}, body: {}", self.cond.0, self.body.0)
+//     }
+// }
 
-#[derive(Clone, Debug, PartialEq)]
-pub struct MatchClause {
-    pub patt: Spanned<Pattern>,
-    pub guard: Option<Spanned<Ast>>,
-    pub body: Spanned<Ast>,
-}
+// #[derive(Clone, Debug, PartialEq)]
+// pub struct MatchClause {
+//     pub patt: Spanned<Pattern>,
+//     pub guard: Option<Spanned<Ast>>,
+//     pub body: Spanned<Ast>,
+// }
 
-impl fmt::Display for MatchClause {
-    fn fmt(self: &MatchClause, f: &mut fmt::Formatter) -> fmt::Result {
-        write!(
-            f,
-            "pattern: {}, guard: {:?} body: {}",
-            self.patt.0, self.guard, self.body.0
-        )
-    }
-}
+// impl fmt::Display for MatchClause {
+//     fn fmt(self: &MatchClause, f: &mut fmt::Formatter) -> fmt::Result {
+//         write!(
+//             f,
+//             "pattern: {}, guard: {:?} body: {}",
+//             self.patt.0, self.guard, self.body.0
+//         )
+//     }
+// }
 
 #[derive(Clone, Debug, PartialEq, Eq)]
 pub enum StringPart {
@@ -53,7 +53,11 @@ impl fmt::Display for StringPart {
 
 #[derive(Clone, Debug, PartialEq, Dissectible)]
 pub enum Ast {
+    // a special Error node
+    // may come in handy?
     Error,
+
+    // expression nodes
     Placeholder,
     Nil,
     Boolean(bool),
@@ -68,29 +72,52 @@ pub enum Ast {
     Arguments(Vec<Spanned<Self>>),
     List(Vec<Spanned<Self>>),
     Dict(Vec<Spanned<Self>>),
-    Let(Box<Spanned<Pattern>>, Box<Spanned<Self>>),
-    Box(&'static str, Box<Spanned<Self>>),
+    Let(Box<Spanned<Self>>, Box<Spanned<Self>>),
+    LBox(&'static str, Box<Spanned<Self>>),
     Synthetic(Box<Spanned<Self>>, Box<Spanned<Self>>, Vec<Spanned<Self>>),
-    When(Vec<Spanned<WhenClause>>),
-    Match(Box<Spanned<Self>>, Vec<MatchClause>),
-    Fn(&'static str, Vec<MatchClause>, Option<&'static str>),
+    When(Vec<Spanned<Self>>),
+    WhenClause(Box<Spanned<Self>>, Box<Spanned<Self>>),
+    Match(Box<Spanned<Self>>, Vec<Spanned<Self>>),
+    MatchClause(
+        Box<Spanned<Self>>,
+        Box<Option<Spanned<Self>>>,
+        Box<Spanned<Self>>,
+    ),
+    Fn(&'static str, Vec<Spanned<Self>>, Option<&'static str>),
     FnDeclaration(&'static str),
     Panic(Box<Spanned<Self>>),
     Do(Vec<Spanned<Self>>),
     Repeat(Box<Spanned<Self>>, Box<Spanned<Self>>),
     Splat(&'static str),
     Pair(&'static str, Box<Spanned<Self>>),
-    Loop(Box<Spanned<Self>>, Vec<MatchClause>),
+    Loop(Box<Spanned<Self>>, Vec<Spanned<Self>>),
     Recur(Vec<Spanned<Self>>),
+
+    // pattern nodes
+    NilPattern,
+    BooleanPattern(bool),
+    NumberPattern(f64),
+    StringPattern(&'static str),
+    InterpolatedPattern(Vec<Spanned<StringPart>>, StringMatcher),
+    KeywordPattern(&'static str),
+    WordPattern(&'static str),
+    AsPattern(&'static str, &'static str),
+    Splattern(Box<Spanned<Self>>),
+    PlaceholderPattern,
+    TuplePattern(Vec<Spanned<Self>>),
+    ListPattern(Vec<Spanned<Self>>),
+    PairPattern(&'static str, Box<Spanned<Self>>),
+    DictPattern(Vec<Spanned<Self>>),
 }
 
 impl fmt::Display for Ast {
     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        use Ast::*;
         match self {
-            Ast::Error => write!(f, "Error"),
-            Ast::Nil => write!(f, "nil"),
-            Ast::String(s) => write!(f, "String: \"{}\"", s),
-            Ast::Interpolated(strs) => {
+            Error => write!(f, "Error"),
+            Nil => write!(f, "nil"),
+            String(s) => write!(f, "String: \"{}\"", s),
+            Interpolated(strs) => {
                 write!(
                     f,
                     "Interpolated: \"{}\"",
@@ -100,11 +127,11 @@ impl fmt::Display for Ast {
                         .join("")
                 )
             }
-            Ast::Boolean(b) => write!(f, "Boolean: {}", b),
-            Ast::Number(n) => write!(f, "Number: {}", n),
-            Ast::Keyword(k) => write!(f, "Keyword: :{}", k),
-            Ast::Word(w) => write!(f, "Word: {}", w),
-            Ast::Block(b) => write!(
+            Boolean(b) => write!(f, "Boolean: {}", b),
+            Number(n) => write!(f, "Number: {}", n),
+            Keyword(k) => write!(f, "Keyword: :{}", k),
+            Word(w) => write!(f, "Word: {}", w),
+            Block(b) => write!(
                 f,
                 "Block: <{}>",
                 b.iter()
@@ -112,15 +139,15 @@ impl fmt::Display for Ast {
                     .collect::<Vec<_>>()
                     .join("\n")
             ),
-            Ast::If(cond, then_branch, else_branch) => write!(
+            If(cond, then_branch, else_branch) => write!(
                 f,
                 "If: {} Then: {} Else: {}",
                 cond.0, then_branch.0, else_branch.0
             ),
-            Ast::Let(pattern, expression) => {
+            Let(pattern, expression) => {
                 write!(f, "Let: {} = {}", pattern.0, expression.0)
             }
-            Ast::Dict(entries) => write!(
+            Dict(entries) => write!(
                 f,
                 "#{{{}}}",
                 entries
@@ -129,7 +156,7 @@ impl fmt::Display for Ast {
                     .collect::<Vec<_>>()
                     .join(", ")
             ),
-            Ast::List(l) => write!(
+            List(l) => write!(
                 f,
                 "List: [{}]",
                 l.iter()
@@ -137,7 +164,7 @@ impl fmt::Display for Ast {
                     .collect::<Vec<_>>()
                     .join("\n")
             ),
-            Ast::Tuple(t) | Ast::Arguments(t) => write!(
+            Tuple(t) | Ast::Arguments(t) => write!(
                 f,
                 "Tuple: ({})",
                 t.iter()
@@ -145,7 +172,7 @@ impl fmt::Display for Ast {
                     .collect::<Vec<_>>()
                     .join("\n")
             ),
-            Ast::Synthetic(root, first, rest) => write!(
+            Synthetic(root, first, rest) => write!(
                 f,
                 "Synth: [{}, {}, {}]",
                 root.0,
@@ -155,7 +182,7 @@ impl fmt::Display for Ast {
                     .collect::<Vec<_>>()
                     .join("\n")
             ),
-            Ast::When(clauses) => write!(
+            When(clauses) => write!(
                 f,
                 "When: [{}]",
                 clauses
@@ -164,35 +191,35 @@ impl fmt::Display for Ast {
                     .collect::<Vec<_>>()
                     .join("\n")
             ),
-            Ast::Placeholder => todo!(),
-            Ast::Box(_name, _rhs) => todo!(),
-            Ast::Match(value, clauses) => {
+            Placeholder => todo!(),
+            LBox(_name, _rhs) => todo!(),
+            Match(value, clauses) => {
                 write!(
                     f,
                     "match: {} with {}",
                     &value.0.to_string(),
                     clauses
                         .iter()
-                        .map(|clause| clause.to_string())
+                        .map(|clause| clause.0.to_string())
                         .collect::<Vec<_>>()
                         .join("\n")
                 )
             }
-            Ast::Fn(name, clauses, _) => {
+            Fn(name, clauses, _) => {
                 write!(
                     f,
                     "fn: {}\n{}",
                     name,
                     clauses
                         .iter()
-                        .map(|clause| clause.to_string())
+                        .map(|clause| clause.0.to_string())
                         .collect::<Vec<_>>()
                         .join("\n")
                 )
             }
-            Ast::FnDeclaration(_name) => todo!(),
-            Ast::Panic(_expr) => todo!(),
-            Ast::Do(terms) => {
+            FnDeclaration(_name) => todo!(),
+            Panic(_expr) => todo!(),
+            Do(terms) => {
                 write!(
                     f,
                     "do: {}",
@@ -203,25 +230,25 @@ impl fmt::Display for Ast {
                         .join(" > ")
                 )
             }
-            Ast::Repeat(_times, _body) => todo!(),
-            Ast::Splat(word) => {
+            Repeat(_times, _body) => todo!(),
+            Splat(word) => {
                 write!(f, "splat: {}", word)
             }
-            Ast::Pair(k, v) => {
+            Pair(k, v) => {
                 write!(f, "pair: {} {}", k, v.0)
             }
-            Ast::Loop(init, body) => {
+            Loop(init, body) => {
                 write!(
                     f,
                     "loop: {} with {}",
                     init.0,
                     body.iter()
-                        .map(|clause| clause.to_string())
+                        .map(|clause| clause.0.to_string())
                         .collect::<Vec<_>>()
                         .join("\n")
                 )
             }
-            Ast::Recur(args) => {
+            Recur(args) => {
                 write!(
                     f,
                     "recur: {}",
@@ -231,19 +258,62 @@ impl fmt::Display for Ast {
                         .join(", ")
                 )
             }
+            MatchClause(pattern, guard, body) => {
+                write!(
+                    f,
+                    "match clause: {} if {:?} -> {}",
+                    pattern.0, guard, body.0
+                )
             }
+            WhenClause(cond, body) => {
+                write!(f, "when clause: {} -> {}", cond.0, body.0)
             }
-}
 
-#[derive(Clone, Debug, PartialEq)]
-pub struct PairPattern {
-    pub key: &'static str,
-    pub patt: Spanned<Pattern>,
-}
-
-impl fmt::Display for PairPattern {
-    fn fmt(self: &PairPattern, f: &mut fmt::Formatter) -> fmt::Result {
-        write!(f, "pair pattern: {}: {}", self.key, self.patt.0)
+            NilPattern => write!(f, "nil"),
+            BooleanPattern(b) => write!(f, "{}", b),
+            NumberPattern(n) => write!(f, "{}", n),
+            StringPattern(s) => write!(f, "{}", s),
+            KeywordPattern(k) => write!(f, ":{}", k),
+            WordPattern(w) => write!(f, "{}", w),
+            AsPattern(w, t) => write!(f, "{} as {}", w, t),
+            Splattern(p) => write!(f, "...{}", p.0),
+            PlaceholderPattern => write!(f, "_"),
+            TuplePattern(t) => write!(
+                f,
+                "({})",
+                t.iter()
+                    .map(|x| x.0.to_string())
+                    .collect::<Vec<_>>()
+                    .join(", ")
+            ),
+            ListPattern(l) => write!(
+                f,
+                "({})",
+                l.iter()
+                    .map(|x| x.0.to_string())
+                    .collect::<Vec<_>>()
+                    .join(", ")
+            ),
+            DictPattern(entries) => write!(
+                f,
+                "#{{{}}}",
+                entries
+                    .iter()
+                    .map(|(pair, _)| pair.to_string())
+                    .collect::<Vec<_>>()
+                    .join(", ")
+            ),
+            PairPattern(key, value) => write!(f, ":{} {}", key, value.0),
+            InterpolatedPattern(strprts, _) => write!(
+                f,
+                "interpolated: \"{}\"",
+                strprts
+                    .iter()
+                    .map(|part| part.0.to_string())
+                    .collect::<Vec<_>>()
+                    .join("")
+            ),
+        }
     }
 }
 
@@ -273,74 +343,74 @@ impl fmt::Debug for StringMatcher {
     }
 }
 
-#[derive(Clone, Debug, PartialEq)]
-pub enum Pattern {
-    Nil,
-    Boolean(bool),
-    Number(f64),
-    String(&'static str),
-    Interpolated(Vec<Spanned<StringPart>>, StringMatcher),
-    Keyword(&'static str),
-    Word(&'static str),
-    As(&'static str, &'static str),
-    Splattern(Box<Spanned<Self>>),
-    Placeholder,
-    Tuple(Vec<Spanned<Self>>),
-    List(Vec<Spanned<Self>>),
-    Pair(&'static str, Box<Spanned<Self>>),
-    Dict(Vec<Spanned<Self>>),
-}
+// #[derive(Clone, Debug, PartialEq)]
+// pub enum Pattern {
+//     Nil,
+//     Boolean(bool),
+//     Number(f64),
+//     String(&'static str),
+//     Interpolated(Vec<Spanned<StringPart>>, StringMatcher),
+//     Keyword(&'static str),
+//     Word(&'static str),
+//     As(&'static str, &'static str),
+//     Splattern(Box<Spanned<Self>>),
+//     Placeholder,
+//     Tuple(Vec<Spanned<Self>>),
+//     List(Vec<Spanned<Self>>),
+//     Pair(&'static str, Box<Spanned<Self>>),
+//     Dict(Vec<Spanned<Self>>),
+// }
 
-impl fmt::Display for Pattern {
-    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
-        match self {
-            Pattern::Nil => write!(f, "nil"),
-            Pattern::Boolean(b) => write!(f, "{}", b),
-            Pattern::Number(n) => write!(f, "{}", n),
-            Pattern::String(s) => write!(f, "{}", s),
-            Pattern::Keyword(k) => write!(f, ":{}", k),
-            Pattern::Word(w) => write!(f, "{}", w),
-            Pattern::As(w, t) => write!(f, "{} as {}", w, t),
-            Pattern::Splattern(p) => write!(f, "...{}", p.0),
-            Pattern::Placeholder => write!(f, "_"),
-            Pattern::Tuple(t) => write!(
-                f,
-                "({})",
-                t.iter()
-                    .map(|x| x.0.to_string())
-                    .collect::<Vec<_>>()
-                    .join(", ")
-            ),
-            Pattern::List(l) => write!(
-                f,
-                "({})",
-                l.iter()
-                    .map(|x| x.0.to_string())
-                    .collect::<Vec<_>>()
-                    .join(", ")
-            ),
-            Pattern::Dict(entries) => write!(
-                f,
-                "#{{{}}}",
-                entries
-                    .iter()
-                    .map(|(pair, _)| pair.to_string())
-                    .collect::<Vec<_>>()
-                    .join(", ")
-            ),
-            Pattern::Pair(key, value) => write!(f, ":{} {}", key, value.0),
-            Pattern::Interpolated(strprts, _) => write!(
-                f,
-                "interpolated: \"{}\"",
-                strprts
-                    .iter()
-                    .map(|part| part.0.to_string())
-                    .collect::<Vec<_>>()
-                    .join("")
-            ),
-        }
-    }
-}
+// impl fmt::Display for Pattern {
+//     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+//         match self {
+//             Pattern::Nil => write!(f, "nil"),
+//             Pattern::Boolean(b) => write!(f, "{}", b),
+//             Pattern::Number(n) => write!(f, "{}", n),
+//             Pattern::String(s) => write!(f, "{}", s),
+//             Pattern::Keyword(k) => write!(f, ":{}", k),
+//             Pattern::Word(w) => write!(f, "{}", w),
+//             Pattern::As(w, t) => write!(f, "{} as {}", w, t),
+//             Pattern::Splattern(p) => write!(f, "...{}", p.0),
+//             Pattern::Placeholder => write!(f, "_"),
+//             Pattern::Tuple(t) => write!(
+//                 f,
+//                 "({})",
+//                 t.iter()
+//                     .map(|x| x.0.to_string())
+//                     .collect::<Vec<_>>()
+//                     .join(", ")
+//             ),
+//             Pattern::List(l) => write!(
+//                 f,
+//                 "({})",
+//                 l.iter()
+//                     .map(|x| x.0.to_string())
+//                     .collect::<Vec<_>>()
+//                     .join(", ")
+//             ),
+//             Pattern::Dict(entries) => write!(
+//                 f,
+//                 "#{{{}}}",
+//                 entries
+//                     .iter()
+//                     .map(|(pair, _)| pair.to_string())
+//                     .collect::<Vec<_>>()
+//                     .join(", ")
+//             ),
+//             Pattern::Pair(key, value) => write!(f, ":{} {}", key, value.0),
+//             Pattern::Interpolated(strprts, _) => write!(
+//                 f,
+//                 "interpolated: \"{}\"",
+//                 strprts
+//                     .iter()
+//                     .map(|part| part.0.to_string())
+//                     .collect::<Vec<_>>()
+//                     .join("")
+//             ),
+//         }
+//     }
+// }
 
 fn is_word_char(c: char) -> bool {
     if c.is_ascii_alphanumeric() {
@@ -476,6 +546,8 @@ pub fn parser<I>(
 where
     I: ValueInput<'static, Token = Token, Span = Span>,
 {
+    use Ast::*;
+
     let mut expr = Recursive::declare();
 
     let mut pattern = Recursive::declare();
@@ -497,16 +569,15 @@ where
     });
 
     let placeholder_pattern =
-        select! {Token::Punctuation("_") => Pattern::Placeholder}.map_with(|p, e| (p, e.span()));
+        select! {Token::Punctuation("_") => PlaceholderPattern}.map_with(|p, e| (p, e.span()));
 
-    let word_pattern =
-        select! { Token::Word(w) => Pattern::Word(w) }.map_with(|w, e| (w, e.span()));
+    let word_pattern = select! { Token::Word(w) => WordPattern(w) }.map_with(|w, e| (w, e.span()));
 
     let atom_pattern = select! {
-        Token::Nil => Pattern::Nil,
-        Token::Boolean(b) => Pattern::Boolean(b),
-        Token::Number(n) => Pattern::Number(n),
-        Token::Keyword(k) => Pattern::Keyword(k),
+        Token::Nil => NilPattern,
+        Token::Boolean(b) => BooleanPattern(b),
+        Token::Number(n) => NumberPattern(n),
+        Token::Keyword(k) => KeywordPattern(k),
     }
     .map_with(|a, e| (a, e.span()));
 
@@ -514,9 +585,9 @@ where
         let parsed = parse_string(s, e.span());
         match parsed {
             Ok(parts) => match parts[0] {
-                (StringPart::Inline(_), _) => Ok((Pattern::String(s), e.span())),
+                (StringPart::Inline(_), _) => Ok((StringPattern(s), e.span())),
                 _ => Ok((
-                    Pattern::Interpolated(parts.clone(), compile_string_pattern(parts)),
+                    InterpolatedPattern(parts.clone(), compile_string_pattern(parts)),
                     e.span(),
                 )),
             },
@@ -526,7 +597,7 @@ where
 
     let bare_splat = just(Token::Punctuation("...")).map_with(|_, e| {
         (
-            Pattern::Splattern(Box::new((Pattern::Placeholder, e.span()))),
+            Splattern(Box::new((PlaceholderPattern, e.span()))),
             e.span(),
         )
     });
@@ -535,7 +606,7 @@ where
 
     let patt_splat = just(Token::Punctuation("..."))
         .ignore_then(splattable)
-        .map_with(|x, e| (Pattern::Splattern(Box::new(x)), e.span()));
+        .map_with(|x, e| (Splattern(Box::new(x)), e.span()));
 
     let splattern = patt_splat.or(bare_splat);
 
@@ -547,7 +618,7 @@ where
         .allow_trailing()
         .collect()
         .delimited_by(just(Token::Punctuation("(")), just(Token::Punctuation(")")))
-        .map_with(|tuple, e| (Pattern::Tuple(tuple), e.span()))
+        .map_with(|tuple, e| (TuplePattern(tuple), e.span()))
         .labelled("tuple pattern");
 
     let list_pattern = pattern
@@ -558,15 +629,15 @@ where
         .allow_trailing()
         .collect()
         .delimited_by(just(Token::Punctuation("[")), just(Token::Punctuation("]")))
-        .map_with(|list, e| (Pattern::List(list), e.span()));
+        .map_with(|list, e| (ListPattern(list), e.span()));
 
     let pair_pattern = select! {Token::Keyword(k) => k}
         .then(pattern.clone())
-        .map_with(|(key, patt), e| (Pattern::Pair(key, Box::new(patt)), e.span()));
+        .map_with(|(key, patt), e| (PairPattern(key, Box::new(patt)), e.span()));
 
     let shorthand_pattern = select! {Token::Word(w) => w}.map_with(|w, e| {
         (
-            Pattern::Pair(w, Box::new((Pattern::Word(w), e.span()))),
+            PairPattern(w, Box::new((WordPattern(w), e.span()))),
             e.span(),
         )
     });
@@ -582,14 +653,14 @@ where
             just(Token::Punctuation("#{")),
             just(Token::Punctuation("}")),
         )
-        .map_with(|dict, e| (Pattern::Dict(dict), e.span()));
+        .map_with(|dict, e| (DictPattern(dict), e.span()));
 
-    let keyword = select! {Token::Keyword(k) => Ast::Keyword(k),}.map_with(|k, e| (k, e.span()));
+    let keyword = select! {Token::Keyword(k) => Keyword(k),}.map_with(|k, e| (k, e.span()));
 
     let as_pattern = select! {Token::Word(w) => w}
         .then_ignore(just(Token::Reserved("as")))
         .then(select! {Token::Keyword(k) => k})
-        .map_with(|(w, t), e| (Pattern::As(w, t), e.span()));
+        .map_with(|(w, t), e| (AsPattern(w, t), e.span()));
 
     pattern.define(
         atom_pattern
@@ -604,16 +675,16 @@ where
     );
 
     let placeholder =
-        select! {Token::Punctuation("_") => Ast::Placeholder}.map_with(|p, e| (p, e.span()));
+        select! {Token::Punctuation("_") => Placeholder}.map_with(|p, e| (p, e.span()));
 
-    let word = select! { Token::Word(w) => Ast::Word(w) }
+    let word = select! { Token::Word(w) => Word(w) }
         .map_with(|w, e| (w, e.span()))
         .labelled("word");
 
     let value = select! {
-        Token::Nil => Ast::Nil,
-        Token::Boolean(b) => Ast::Boolean(b),
-        Token::Number(n) => Ast::Number(n),
+        Token::Nil => Nil,
+        Token::Boolean(b) => Boolean(b),
+        Token::Number(n) => Number(n),
     }
     .map_with(|v, e| (v, e.span()));
 
@@ -621,8 +692,8 @@ where
         let parsed = parse_string(s, e.span());
         match parsed {
             Ok(parts) => match parts[0] {
-                (StringPart::Inline(_), _) => Ok((Ast::String(s), e.span())),
-                _ => Ok((Ast::Interpolated(parts), e.span())),
+                (StringPart::Inline(_), _) => Ok((String(s), e.span())),
+                _ => Ok((Interpolated(parts), e.span())),
             },
             Err(msg) => Err(Rich::custom(e.span(), msg)),
         }
@@ -635,7 +706,7 @@ where
         .allow_trailing()
         .collect()
         .delimited_by(just(Token::Punctuation("(")), just(Token::Punctuation(")")))
-        .map_with(|tuple, e| (Ast::Tuple(tuple), e.span()));
+        .map_with(|tuple, e| (Tuple(tuple), e.span()));
 
     let args = simple
         .clone()
@@ -645,7 +716,7 @@ where
         .allow_trailing()
         .collect()
         .delimited_by(just(Token::Punctuation("(")), just(Token::Punctuation(")")))
-        .map_with(|args, e| (Ast::Arguments(args), e.span()));
+        .map_with(|args, e| (Arguments(args), e.span()));
 
     let synth_root = word.or(keyword);
 
@@ -655,21 +726,14 @@ where
         .then(synth_term.clone())
         .then(synth_term.clone().repeated().collect())
         .map_with(|((root, first), rest), e| {
-            (
-                Ast::Synthetic(Box::new(root), Box::new(first), rest),
-                e.span(),
-            )
+            (Synthetic(Box::new(root), Box::new(first), rest), e.span())
         });
 
     let splat = just(Token::Punctuation("..."))
         .ignore_then(word)
         .map_with(|(w, _), e| {
             (
-                Ast::Splat(if let Ast::Word(w) = w {
-                    w
-                } else {
-                    unreachable!()
-                }),
+                Splat(if let Word(w) = w { w } else { unreachable!() }),
                 e.span(),
             )
         });
@@ -682,14 +746,14 @@ where
         .allow_trailing()
         .collect()
         .delimited_by(just(Token::Punctuation("[")), just(Token::Punctuation("]")))
-        .map_with(|list, e| (Ast::List(list), e.span()));
+        .map_with(|list, e| (List(list), e.span()));
 
     let pair = select! {Token::Keyword(k) => k}
         .then(simple.clone())
-        .map_with(|(key, value), e| (Ast::Pair(key, Box::new(value)), e.span()));
+        .map_with(|(key, value), e| (Pair(key, Box::new(value)), e.span()));
 
     let shorthand = select! {Token::Word(w) => w}
-        .map_with(|w, e| (Ast::Pair(w, Box::new((Ast::Word(w), e.span()))), e.span()));
+        .map_with(|w, e| (Pair(w, Box::new((Word(w), e.span()))), e.span()));
 
     let dict = pair
         .or(shorthand)
@@ -702,15 +766,15 @@ where
             just(Token::Punctuation("#{")),
             just(Token::Punctuation("}")),
         )
-        .map_with(|dict, e| (Ast::Dict(dict), e.span()));
+        .map_with(|dict, e| (Dict(dict), e.span()));
 
     let recur = just(Token::Reserved("recur"))
         .ignore_then(tuple.clone())
         .map_with(|args, e| {
-            let (Ast::Tuple(args), _) = args else {
+            let (Tuple(args), _) = args else {
                 unreachable!()
             };
-            (Ast::Recur(args), e.span())
+            (Recur(args), e.span())
         });
 
     let block = expr
@@ -720,7 +784,7 @@ where
         .allow_trailing()
         .collect()
         .delimited_by(just(Token::Punctuation("{")), just(Token::Punctuation("}")))
-        .map_with(|block, e| (Ast::Block(block), e.span()))
+        .map_with(|block, e| (Block(block), e.span()))
         .recover_with(via_parser(nested_delimiters(
             Token::Punctuation("{"),
             Token::Punctuation("}"),
@@ -728,7 +792,7 @@ where
                 (Token::Punctuation("("), Token::Punctuation(")")),
                 (Token::Punctuation("["), Token::Punctuation("]")),
             ],
-            |span| (Ast::Error, span),
+            |span| (Error, span),
         )));
 
     let if_ = just(Token::Reserved("if"))
@@ -741,7 +805,7 @@ where
         .then(expr.clone())
         .map_with(|((condition, then_branch), else_branch), e| {
             (
-                Ast::If(
+                If(
                     Box::new(condition),
                     Box::new(then_branch),
                     Box::new(else_branch),
@@ -754,7 +818,7 @@ where
         .clone()
         .then_ignore(just(Token::Punctuation("->")))
         .then(expr.clone())
-        .map_with(|(cond, body), e| (WhenClause { cond, body }, e.span()));
+        .map_with(|(cond, body), e| (WhenClause(Box::new(cond), Box::new(body)), e.span()));
 
     let when = just(Token::Reserved("when"))
         .ignore_then(
@@ -765,7 +829,7 @@ where
                 .collect()
                 .delimited_by(just(Token::Punctuation("{")), just(Token::Punctuation("}"))),
         )
-        .map_with(|clauses, e| (Ast::When(clauses), e.span()));
+        .map_with(|clauses, e| (When(clauses), e.span()));
 
     let guarded_clause = pattern
         .clone()
@@ -773,23 +837,25 @@ where
         .then(simple.clone())
         .then_ignore(just(Token::Punctuation("->")))
         .then(expr.clone())
-        .map_with(|((patt, guard), body), _| MatchClause {
-            patt,
-            guard: Some(guard),
-            body,
+        .map_with(|((patt, guard), body), e| {
+            (
+                MatchClause(Box::new(patt), Box::new(Some(guard)), Box::new(body)),
+                e.span(),
+            )
         });
 
     let match_clause = pattern
         .clone()
         .then_ignore(just(Token::Punctuation("->")))
         .then(expr.clone())
-        .map_with(|(patt, body), _| MatchClause {
-            patt,
-            guard: None,
-            body,
+        .map_with(|(patt, body), e| {
+            (
+                MatchClause(Box::new(patt), Box::new(None), Box::new(body)),
+                e.span(),
+            )
         });
 
-    let match_ = just(Token::Reserved("match"))
+    let r#match = just(Token::Reserved("match"))
         .ignore_then(simple.clone())
         .then_ignore(just(Token::Reserved("with")))
         .then(
@@ -802,13 +868,13 @@ where
                 .collect()
                 .delimited_by(just(Token::Punctuation("{")), just(Token::Punctuation("}"))),
         )
-        .map_with(|(expr, clauses), e| (Ast::Match(Box::new(expr), clauses), e.span()));
+        .map_with(|(expr, clauses), e| (Match(Box::new(expr), clauses), e.span()));
 
-    let conditional = when.or(if_).or(match_);
+    let conditional = when.or(if_).or(r#match);
 
     let panic = just(Token::Reserved("panic!"))
         .ignore_then(nonbinding.clone())
-        .map_with(|expr, e| (Ast::Panic(Box::new(expr)), e.span()));
+        .map_with(|expr, e| (Panic(Box::new(expr)), e.span()));
 
     let do_ = just(Token::Reserved("do"))
         .ignore_then(
@@ -819,12 +885,12 @@ where
                 )
                 .collect(),
         )
-        .map_with(|exprs, e| (Ast::Do(exprs), e.span()));
+        .map_with(|exprs, e| (Do(exprs), e.span()));
 
     let repeat = just(Token::Reserved("repeat"))
         .ignore_then(simple.clone())
         .then(block.clone())
-        .map_with(|(count, body), e| (Ast::Repeat(Box::new(count), Box::new(body)), e.span()));
+        .map_with(|(count, body), e| (Repeat(Box::new(count), Box::new(body)), e.span()));
 
     let fn_guarded = tuple_pattern
         .clone()
@@ -832,10 +898,11 @@ where
         .then(simple.clone())
         .then_ignore(just(Token::Punctuation("->")))
         .then(nonbinding.clone())
-        .map_with(|((patt, guard), body), _| MatchClause {
-            patt,
-            body,
-            guard: Some(guard),
+        .map_with(|((patt, guard), body), e| {
+            (
+                MatchClause(Box::new(patt), Box::new(Some(guard)), Box::new(body)),
+                e.span(),
+            )
         })
         .labelled("function clause");
 
@@ -843,10 +910,11 @@ where
         .clone()
         .then_ignore(just(Token::Punctuation("->")))
         .then(nonbinding.clone())
-        .map_with(|(patt, body), _| MatchClause {
-            patt,
-            body,
-            guard: None,
+        .map_with(|(patt, body), e| {
+            (
+                MatchClause(Box::new(patt), Box::new(None), Box::new(body)),
+                e.span(),
+            )
         })
         .labelled("function clause");
 
@@ -854,7 +922,7 @@ where
 
     let lambda = just(Token::Reserved("fn"))
         .ignore_then(fn_unguarded.clone())
-        .map_with(|clause, e| (Ast::Fn("anonymous", vec![clause], None), e.span()));
+        .map_with(|clause, e| (Fn("anonymous", vec![clause], None), e.span()));
 
     let fn_clauses = fn_clause
         .clone()
@@ -873,7 +941,7 @@ where
         .ignore_then(tuple.clone())
         .then_ignore(just(Token::Reserved("with")))
         .then(loop_multiclause.clone().or(fn_single_clause.clone()))
-        .map_with(|(init, body), e| (Ast::Loop(Box::new(init), body), e.span()));
+        .map_with(|(init, body), e| (Loop(Box::new(init), body), e.span()));
 
     simple.define(
         synthetic
@@ -906,7 +974,7 @@ where
         .then_ignore(just(Token::Punctuation("=")))
         .then(nonbinding.clone())
         .map_with(|(pattern, expression), e| {
-            (Ast::Let(Box::new(pattern), Box::new(expression)), e.span())
+            (Let(Box::new(pattern), Box::new(expression)), e.span())
         });
 
     let box_ = just(Token::Reserved("box"))
@@ -914,35 +982,35 @@ where
         .then_ignore(just(Token::Punctuation("=")))
         .then(nonbinding.clone())
         .map_with(|(word, expr), e| {
-            let name = if let Ast::Word(w) = word.0 {
+            let name = if let Word(w) = word.0 {
                 w
             } else {
                 unreachable!()
             };
-            (Ast::Box(name, Box::new(expr)), e.span())
+            (LBox(name, Box::new(expr)), e.span())
         });
 
     let fn_decl = just(Token::Reserved("fn"))
         .ignore_then(word)
         .map_with(|(word, _), e| {
-            let name = if let Ast::Word(w) = word {
+            let name = if let Word(w) = word {
                 w
             } else {
                 unreachable!()
             };
-            (Ast::FnDeclaration(name), e.span())
+            (FnDeclaration(name), e.span())
         });
 
     let fn_named = just(Token::Reserved("fn"))
         .ignore_then(word)
         .then(fn_unguarded.clone())
         .map_with(|(word, clause), e| {
-            let name = if let Ast::Word(word) = word.0 {
+            let name = if let Word(word) = word.0 {
                 word
             } else {
                 unreachable!()
             };
-            (Ast::Fn(name, vec![clause], None), e.span())
+            (Fn(name, vec![clause], None), e.span())
         });
 
     let docstr = select! {Token::String(s) => s};
@@ -959,12 +1027,12 @@ where
         .ignore_then(word)
         .then(fn_multiclause)
         .map_with(|(word, (docstr, clauses, _)), e| {
-            let name = if let Ast::Word(word) = word.0 {
+            let name = if let Word(word) = word.0 {
                 word
             } else {
                 unreachable!()
             };
-            (Ast::Fn(name, clauses, docstr), e.span())
+            (Fn(name, clauses, docstr), e.span())
         });
 
     let fn_ = fn_named.or(fn_compound).or(fn_decl);
@@ -978,7 +1046,7 @@ where
         .allow_trailing()
         .allow_leading()
         .collect()
-        .map_with(|exprs, e| (Ast::Block(exprs), e.span()));
+        .map_with(|exprs, e| (Block(exprs), e.span()));
 
     script
 }

src/validator.rs

@@ -315,7 +315,7 @@ impl<'a> Validator<'a> {
 
             // binding forms
             // TODO: set up errors to include original binding
-            Ast::Box(name, boxed) => {
+            Ast::LBox(name, boxed) => {
                 if self.bound(name).is_some() {
                     self.err(format!("box name `{name}` is already bound"));
                 } else {
@@ -470,6 +470,7 @@ impl<'a> Validator<'a> {
             }
             // terminals can never be invalid
             Ast::Nil | Ast::Boolean(_) | Ast::Number(_) | Ast::Keyword(_) | Ast::String(_) => (),
+            _ => todo!(),
         };
         self.ast = root;
     }

src/value.rs

@@ -1,5 +1,6 @@
 use crate::base::*;
 use crate::parser::*;
+use crate::spans::*;
 use imbl::*;
 use std::cell::RefCell;
 use std::fmt;
@@ -9,7 +10,7 @@ use struct_scalpel::Dissectible;
 #[derive(Clone, Debug)]
 pub struct Fn<'src> {
     pub name: String,
-    pub body: &'src Vec<MatchClause>,
+    pub body: &'src Vec<Spanned<Ast>>,
     pub doc: Option<String>,
 }
 
@@ -29,7 +30,7 @@ pub enum Value<'src> {
     Dict(HashMap<&'static str, Self>),
     Box(&'static str, Rc<RefCell<Self>>),
     Fn(Rc<Fn<'src>>),
-    Base(Base<'src>),
+    Base(BaseFn<'src>),
     Recur(Vec<Self>),
     // Set(HashSet<Self>),
     // Sets are hard