From 20fb078ad287fdb32accba0eaaae1df4e9011b92 Mon Sep 17 00:00:00 2001
From: Danila Fedorin <danila.fedorin@gmail.com>
Date: Fri, 10 Aug 2018 19:56:46 -0700
Subject: [PATCH] Change assignment behavior to update existing table entries.

---
 src/reserved.c | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/reserved.c b/src/reserved.c
index 3cedc33..a506665 100644
--- a/src/reserved.c
+++ b/src/reserved.c
@@ -4,6 +4,18 @@
 #include "value.h"
 #include "libabacus.h"
 
+libab_result _update_entry(libab_table* table, const char* name, libab_ref* value) {
+    libab_result result = LIBAB_SUCCESS;
+    libab_table_entry* value_entry = libab_table_search_entry_value(table, name);
+    if(value_entry) {
+        libab_ref_free(&value_entry->data_u.value);
+        libab_ref_copy(value, &value_entry->data_u.value);
+    } else {
+        result = libab_put_table_value(table, name, value);
+    }
+    return result;
+}
+
 libab_result _behavior_assign(libab* ab, libab_ref* scope, 
                               libab_tree* left, libab_tree* right,
                               libab_ref* into) {
@@ -12,7 +24,7 @@ libab_result _behavior_assign(libab* ab, libab_ref* scope,
     if(left->variant == TREE_ID) {
         result = libab_run_tree_scoped(ab, right, scope, into);
         if(result == LIBAB_SUCCESS) {
-            result = libab_put_table_value(libab_ref_get(scope), left->string_value, into);
+            result = _update_entry(libab_ref_get(scope), left->string_value, into);
         }
 
         if(result != LIBAB_SUCCESS) {