Add an update method for changing environment values

1 files changed, 14 insertions, 0 deletions

diff --git a/src/bootstrap/environment.c b/src/bootstrap/environment.c
index 0a6a866..d4e9f86 100644
--- a/src/bootstrap/environment.c
+++ b/src/bootstrap/environment.c
@@ -60,4 +60,18 @@ env_lookup(Environment *env, Object *symbol) {
     return obj_err;
 }
 
+void
+env_update_symbol(Environment *env, Object *symbol, Object *value) {
+    // Try to find an existing symbol in the current environment.
+    for (size_t i = 0; i < env->size; i++) {
+        EnvEntry entry = env->buf[i];
+        if (obj_eq(symbol, entry.symbol)) {
+            env->buf[i].value = value;
+            return;
+        }
+    }
+    // If we don't find the symbol, add a new entry.
+    env_add_symbol(env, symbol, value);
+}
+
 // TODO: Free env function.