From 24a703199b06e483fd28823be64b87ec2ca44645 Mon Sep 17 00:00:00 2001
From: Bruce Hill <bruce@bruce-hill.com>
Date: Sun, 18 Feb 2024 00:12:23 -0500
Subject: Compile tables

---
 compile.c | 32 +++++++++++++++++++++++++++++---
 1 file changed, 29 insertions(+), 3 deletions(-)

(limited to 'compile.c')

diff --git a/compile.c b/compile.c
index 133cff0a..704898c7 100644
--- a/compile.c
+++ b/compile.c
@@ -325,19 +325,45 @@ CORD compile(env_t *env, ast_t *ast)
         return CORD_asprintf("max(%r, %r)", compile(env, Match(ast, Max)->lhs), compile(env, Match(ast, Max)->rhs));
     }
     // Min, Max,
-    // Array, Table, TableEntry,
     case Array: {
         auto array = Match(ast, Array);
         if (!array->items)
             return "(array_t){}";
            
-        CORD code = "$array(";
+        CORD code = "$Array(";
         for (ast_list_t *item = array->items; item; item = item->next) {
             code = CORD_cat(code, compile(env, item->ast));
             if (item->next) code = CORD_cat(code, ", ");
         }
-        return CORD_cat_char(code, ')');
+        return CORD_cat(code, ")");
+    }
+    case Table: {
+        auto table = Match(ast, Table);
+        if (!table->entries) {
+            CORD code = "(table_t){";
+            if (table->fallback)
+                code = CORD_all(code, ".fallback=", compile(env, table->fallback),",");
+            if (table->default_value)
+                code = CORD_all(code, ".default_value=", compile(env, table->default_value),",");
+            return CORD_cat(code, "}");
+        }
+           
+        type_t *table_t = get_type(env, ast);
+        // TODO: figure out a clever way to optimize table literals:
+        CORD code = CORD_all("({ table_t $table = {}; TypeInfo *$table_type = ", compile_type_info(env, table_t), ";");
+        for (ast_list_t *entry = table->entries; entry; entry = entry->next) {
+            auto entry = Match(entry->ast, TableEntry);
+            code = CORD_all(code, " Table_set(&$table, $stack(",
+                            compile(env, entry->key), "), $stack(", compile(env, entry->value), "), $table_type);");
+        }
+        if (table->fallback)
+            code = CORD_all(code, " $table.fallback = $heap(", compile(env, table->fallback), ");");
+        if (table->default_value)
+            code = CORD_all(code, " $table.default_value = $heap(", compile(env, table->default_value), ");");
+        return CORD_cat(code, " $table; })");
+
     }
+    // Table, TableEntry,
     case FunctionDef: {
         auto fndef = Match(ast, FunctionDef);
         CORD name = compile(env, fndef->name);
-- 
cgit v1.2.3