Add initial local definition/access

1 files changed, 35 insertions, 4 deletions

diff --git a/src/compiler.h b/src/compiler.h
index fe36ecf..023a039 100644
--- a/src/compiler.h
+++ b/src/compiler.h
@@ -23,6 +23,7 @@ do { \
     array_insert(current_context, buf, n_chars); \
 } while(false);
 
+static Environment *current_env = NULL;
 
 // TODO: Separate c/h files
 // TODO: Create a "driver.c" file with the (display) function for external
@@ -51,7 +52,7 @@ do { \
 void compile_object(Object *obj);
 void compile_fixnum(Object *obj);
 void compile_proc_call(Object *obj);
-void compile(Root *roots);
+void compile(Program program);
 
 char *
 generate_label(char *prefix) {
@@ -506,6 +507,25 @@ compile_lambda(Object *obj) {
 }
 
 void
+compile_def(Object *obj) {
+    context_printf("    ;; --> compile_def\n");
+    compile_object(obj->var_expr);
+    ssize_t idx = find_local_index(current_env->locals, obj->var_name);
+    context_printf("    pop     rax\n");
+    context_printf("    mov     [rbp + %ld], rax\n", 8 * (idx + 1));
+    context_printf("    ;; <-- compile_def\n");
+}
+
+void
+compile_symbol(Object *obj) {
+    context_printf("    ;; --> compile_symbol\n");
+    ssize_t idx = find_local_index(current_env->locals, obj);
+    context_printf("    mov     rax, [rbp + %ld]\n", 8 * (idx + 1));
+    context_printf("    push    rax\n");
+    context_printf("    ;; <-- compile_symbol\n");
+}
+
+void
 compile_object(Object *obj) {
     switch (obj->type) {
         case OBJ_TYPE_NIL: { compile_nil(); } break;
@@ -516,6 +536,8 @@ compile_object(Object *obj) {
         case OBJ_TYPE_STRING: { compile_string(obj); } break;
         case OBJ_TYPE_IF: { compile_if(obj); } break;
         case OBJ_TYPE_LAMBDA: { compile_lambda(obj); } break;
+        case OBJ_TYPE_DEF: { compile_def(obj); } break;
+        case OBJ_TYPE_SYMBOL: { compile_symbol(obj); } break;
         default: break;
     }
 }
@@ -555,16 +577,17 @@ emit_data_section(void) {
 }
 
 void
-compile(Root *roots) {
+compile(Program program) {
     // Prepare compilation variables.
     array_init(constants, 0);
     array_init(labels, 0);
     array_init(procedures, 0);
     array_init(current_context, 0);
+    current_env = program.env;
 
     // Compile program.
-    for (size_t i = 0; i < array_size(roots); i++) {
-        Object *root = roots[i];
+    for (size_t i = 0; i < array_size(program.roots); i++) {
+        Object *root = program.roots[i];
         compile_object(root);
     }
 
@@ -604,7 +627,15 @@ compile(Root *roots) {
     printf("alignb  8\n");
     printf("global _start\n");
     printf("_start:\n");
+
+    // Initialize heap pointer.
     printf("    mov     r15, bdl_heap\n");
+
+    // Initialize main locals.
+    printf("    sub     rsp, %zu\n", 8 * array_size(program.env->locals));
+    printf("    mov     rbp, rsp\n");
+
+    // Keep the bottom stack value set as NIL_VAL for a default return value.
     printf("    push    NIL_VAL\n");
     for (size_t i = 0; i < array_size(current_context); i++) {
         putchar(current_context[i]);