From 1619e765e3ae2a88765c94a01ec731e8cabd0ff1 Mon Sep 17 00:00:00 2001
From: WormHeamer
Date: Tue, 21 Oct 2025 14:50:53 -0400
Subject: save & load scratch in parse_stmt instead of reset in parse_proc

---
 main.c | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'main.c')

diff --git a/main.c b/main.c
index 9b22751..25469bc 100644
--- a/main.c
+++ b/main.c
@@ -347,8 +347,8 @@ void parse_if(Lexer *l, Proc *p) {
 }
 
 void parse_stmt(Lexer *l, Proc *p) {
-	/* TODO */
-	(void)l;
+	ArenaMark m;
+	arena_save(&p->scratch, &m);
 	switch (l->tok) {
 	case TOK_RETURN:
 		parse_return(l, p);
@@ -369,6 +369,7 @@ void parse_stmt(Lexer *l, Proc *p) {
 		lex_expected(l, TM_RBRACE);
 		break;
 	}
+	arena_load(&p->scratch, &m);
 }
 
 void parse_args_list(Lexer *l, Proc *proc) {
@@ -504,7 +505,6 @@ Proc *parse_proc(Lexer *l, Unit *u) {
 	while (l->tok != TOK_RBRACE) {
 		lex_expected_not(l, TM_EOF);
 		parse_stmt(l, proc);
-		arena_reset(&proc->scratch);
 	}
 	scope_pop(&proc->scope, &proc->graph);
 	lex_expected(l, TM_RBRACE);
-- 
cgit v1.2.3