Fix lambda and closures

1 files changed, 24 insertions, 31 deletions

diff --git a/src/bootstrap/primitives.c b/src/bootstrap/primitives.c
index 461799e..4ef56d5 100644
--- a/src/bootstrap/primitives.c
+++ b/src/bootstrap/primitives.c
@@ -1,6 +1,7 @@
+#define DEBUG_OBJ(MSG,OBJ) printf((MSG)); display(OBJ); printf("\n");
+
 Object *
 eval(Environment* env, Object *root) {
-tailcall:
     switch (root->type) {
         case OBJ_TYPE_ERR:
         case OBJ_TYPE_PROCEDURE:
@@ -35,17 +36,26 @@ tailcall:
                 if (val->type == OBJ_TYPE_PROCEDURE) {
                     return val->proc(env, root->cdr);
                 }
+                if (val->type == OBJ_TYPE_LAMBDA) {
+                    goto eval_lambda;
+                }
                 error_push((Error){
                     .type = ERR_TYPE_RUNTIME,
                     .value = ERR_OBJ_NOT_CALLABLE,
                 });
                 return obj_err;
             }
-            Object* lambda = eval(env, root->car);
+            Object* lambda;
+eval_lambda:
+            lambda = eval(env, root->car);
+            if (lambda == obj_err) {
+                return obj_err;
+            }
             if (lambda->type == OBJ_TYPE_LAMBDA) {
                 Object *fun = lambda;
                 Object *args = root->cdr;
                 Object *params = fun->args;
+                env = env_extend(fun->env, env);
                 while (params != obj_nil) {
                     if (args == obj_nil) {
                         error_push((Error){
@@ -66,12 +76,7 @@ tailcall:
                         });
                         return obj_err;
                     }
-                    ssize_t index = env_symbol_index_in_current_env(fun->env, symbol);
-                    if (index == -1) {
-                        env_add_symbol(fun->env, obj_duplicate(symbol), obj_duplicate(value));
-                    } else {
-                        fun->env->buf[index].value = obj_duplicate(value);
-                    }
+                    env_add_or_update_current(env, symbol, value);
                     args = args->cdr;
                     params = params->cdr;
                 }
@@ -82,10 +87,14 @@ tailcall:
                     });
                     return obj_err;
                 }
-
-                env = fun->env;
                 root = fun->body;
-                goto tailcall;
+                while (root->cdr != obj_nil) {
+                    if (eval(env, root->car) == obj_err) {
+                        return obj_err;
+                    };
+                    root = root->cdr;
+                }
+                return eval(env, root->car);
             }
         } break;
     }
@@ -693,13 +702,7 @@ proc_define(Environment *env, Object *obj) {
         return obj_err;
     }
 
-    // Make a copy of the symbol and to make them permanent in the environment.
-    ssize_t index = env_symbol_index_in_current_env(env, symbol);
-    if (index == -1) {
-        env_add_symbol(env, obj_duplicate(symbol), obj_duplicate(value));
-    } else {
-        env->buf[index].value = obj_duplicate(value);
-    }
+    env_add_or_update_current(env, symbol, value);
     return obj_nil;
 }
 
@@ -727,17 +730,7 @@ proc_set(Environment *env, Object *obj) {
         return obj_err;
     }
 
-    ssize_t index = env_symbol_index_in_current_env(env, symbol);
-    if (index == -1) {
-        error_push((Error){
-            .type = ERR_TYPE_RUNTIME,
-            .value = ERR_SYMBOL_NOT_FOUND,
-        });
-        return obj_err;
-    }
-
-    env->buf[index].value = obj_duplicate(value);
-    return obj_nil;
+    return env_update(env, symbol, value);
 }
 
 Object *
@@ -757,11 +750,11 @@ proc_lambda(Environment *env, Object *obj) {
         });
         return obj_err;
     }
-    Object *body = obj->cdr->car;
+    Object *body = obj->cdr;
     Object *fun = alloc_object(OBJ_TYPE_LAMBDA);
     fun->args = obj_duplicate(args);
     fun->body = obj_duplicate(body);
-    fun->env = env_create(env);
+    fun->env = env;
     return fun;
 }