1 files changed, 36 insertions, 52 deletions

diff --git a/src/lisp/env.rs b/src/lisp/env.rs
index ad7cc2b..94c0e05 100644
--- a/src/lisp/env.rs
+++ b/src/lisp/env.rs
@@ -1,8 +1,11 @@
 use crate::{
     app::AppState,
     lisp::{error::LispError, expr::LispExpr, number::LispNumber, Environment},
+    primitive,
 };
 
+use log::warn;
+
 pub fn is_bound<S: AsRef<str>>(env: &mut Environment, name: S) -> bool {
     env.get(name.as_ref()).is_some()
 }
@@ -13,21 +16,39 @@ pub fn new_binding<S: AsRef<str>>(env: &mut Environment, name: S, value: LispExp
 
 pub fn with_prelude() -> Environment {
     let mut env = Environment::new();
-    new_binding(
-        &mut env,
-        "+",
-        LispExpr::PrimitiveFunc(|args, _| {
-            if args.iter().any(|arg| !matches!(arg, LispExpr::Number(_))) {
-                Err(LispError::EvalError)
-            } else {
-                let result = args
-                    .iter()
-                    .map(|n| unwrap_number(n))
-                    .fold(LispNumber::Integer(0), |acc, x| acc + *x);
-                Ok(LispExpr::Number(result))
+    primitive!(env, Some(2), "+", |args, _| {
+        if args.iter().any(|arg| !matches!(arg, LispExpr::Number(_))) {
+            Err(LispError::EvalError)
+        } else {
+            Ok(LispExpr::Number(
+                args.iter()
+                    .map(|arg| unwrap_number(arg))
+                    .fold(LispNumber::Integer(0), |acc, x| acc + *x),
+            ))
+        }
+    });
+    primitive!(env, Some(2), "sub", |args, _| {
+        if args.iter().any(|arg| !matches!(arg, LispExpr::Number(_))) {
+            Err(LispError::EvalError)
+        } else {
+            let mut acc = unwrap_number(&args[0]).clone();
+            for arg in args.into_iter().skip(1) {
+                acc = acc - *unwrap_number(&arg);
             }
-        }),
-    );
+            Ok(LispExpr::Number(acc))
+        }
+    });
+    primitive!(env, Some(2), "*", |args, _| {
+        if args.iter().any(|arg| !matches!(arg, LispExpr::Number(_))) {
+            Err(LispError::EvalError)
+        } else {
+            Ok(LispExpr::Number(
+                args.iter()
+                    .map(|arg| unwrap_number(arg))
+                    .fold(LispNumber::Integer(1), |acc, x| acc * *x),
+            ))
+        }
+    });
     env
 }
 
@@ -56,7 +77,7 @@ pub fn eval(expr: &LispExpr, app: &mut AppState) -> Result<LispExpr, LispError>
                             for item in li[1..].iter() {
                                 args.push(eval(item, app)?);
                             }
-                            (f)(&args, None)
+                            f.call(&args, app)
                         }
                         _ => Err(LispError::EvalError),
                     }
@@ -78,40 +99,3 @@ pub fn unwrap_number(n: &LispExpr) -> &LispNumber {
         _ => panic!("unwrap_number expected number"),
     }
 }
-
-#[cfg(test)]
-mod tests {
-    use super::*;
-
-    #[test]
-    fn eval_primitive_call() {
-        let mut env = Environment::new();
-        new_binding(&mut env, "age", LispExpr::Number(LispNumber::Float(1.4)));
-        new_binding(
-            &mut env,
-            "+",
-            LispExpr::PrimitiveFunc(|args, _| {
-                if args.iter().any(|arg| !matches!(arg, LispExpr::Number(_))) {
-                    Err(LispError::EvalError)
-                } else {
-                    let result = args
-                        .iter()
-                        .map(|n| unwrap_number(n))
-                        .fold(LispNumber::Integer(0), |acc, x| acc + *x);
-                    Ok(LispExpr::Number(result))
-                }
-            }),
-        );
-        let mut numbers = (1..=3)
-            .map(LispNumber::Integer)
-            .map(LispExpr::Number)
-            .collect::<Vec<_>>();
-        let mut expr = Vec::new();
-        expr.push(LispExpr::Ident("+".into()));
-        expr.append(&mut numbers);
-        // assert!(matches!(
-        //     eval(&LispExpr::List(expr), ).unwrap(),
-        //     LispExpr::Number(LispNumber::Integer(6))
-        // ));
-    }
-}