From 394f720fb0bdcc9acfa3614a5ded650383ec8feb Mon Sep 17 00:00:00 2001
From: Bruce Hill <bruce@bruce-hill.com>
Date: Sun, 25 Feb 2024 12:24:38 -0500
Subject: Make stack references work with local vars

---
 compile.c | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/compile.c b/compile.c
index 6607c1b4..455f13f4 100644
--- a/compile.c
+++ b/compile.c
@@ -175,7 +175,12 @@ CORD compile(env_t *env, ast_t *ast)
     case Not: return CORD_asprintf("not(%r)", compile(env, Match(ast, Not)->value));
     case Negative: return CORD_asprintf("-(%r)", compile(env, Match(ast, Negative)->value));
     case HeapAllocate: return CORD_asprintf("$heap(%r)", compile(env, Match(ast, HeapAllocate)->value));
-    case StackReference: return CORD_asprintf("$stack(%r)", compile(env, Match(ast, StackReference)->value));
+    case StackReference: {
+        ast_t *value = Match(ast, StackReference)->value;
+        if (value->tag == Var)
+            return CORD_cat("&", compile(env, Match(ast, StackReference)->value));
+        return CORD_all("$stack(", compile(env, Match(ast, StackReference)->value), ")");
+    }
     case BinaryOp: {
         auto binop = Match(ast, BinaryOp);
         CORD lhs = compile(env, binop->lhs);
-- 
cgit v1.2.3