Change table removal function to use values instead of references

3 files changed, 8 insertions, 2 deletions

diff --git a/builtins/table.h b/builtins/table.h
index e136f460..36f6d75e 100644
--- a/builtins/table.h
+++ b/builtins/table.h
@@ -38,6 +38,7 @@ void Table$set(table_t *t, const void *key, const void *value, const TypeInfo *t
                                                         Table$set(t, &k, &v, type); })
 #define Table$reserve_value(t, key_expr, type) ({ __typeof(key_expr) k = key_expr; Table$reserve(t, &k, NULL, type); })
 void Table$remove(table_t *t, const void *key, const TypeInfo *type);
+#define Table$remove_value(t, key_expr, type) ({ __typeof(key_expr) k = key_expr; Table$remove(t, &k, type); })
 void Table$clear(table_t *t);
 table_t Table$sorted(table_t t, const TypeInfo *type);
 void Table$mark_copy_on_write(table_t *t);
diff --git a/compile.c b/compile.c
index a141302b..3a632afe 100644
--- a/compile.c
+++ b/compile.c
@@ -1976,8 +1976,8 @@ CORD compile(env_t *env, ast_t *ast)
                                 compile_type_info(env, self_value_t), ")");
             } else if (streq(call->name, "remove")) {
                 CORD self = compile_to_pointer_depth(env, call->self, 1, false);
-                arg_t *arg_spec = new(arg_t, .name="key", .type=Type(PointerType, .pointed=table->key_type, .is_stack=true, .is_readonly=true));
-                return CORD_all("Table$remove(", self, ", ", compile_arguments(env, ast, arg_spec, call->args), ", ",
+                arg_t *arg_spec = new(arg_t, .name="key", .type=table->key_type);
+                return CORD_all("Table$remove_value(", self, ", ", compile_arguments(env, ast, arg_spec, call->args), ", ",
                                 compile_type_info(env, self_value_t), ")");
             } else if (streq(call->name, "clear")) {
                 CORD self = compile_to_pointer_depth(env, call->self, 1, false);
diff --git a/test/tables.tm b/test/tables.tm
index 4c61bb77..132e4301 100644
--- a/test/tables.tm
+++ b/test/tables.tm
@@ -56,3 +56,8 @@ func main():
 	= {1:10, 3:30, 5:50}
 	>> {x:10*x for x in y if x > 1 for y in [3, 4, 5] if y < 5}
 	= {2:20, 3:30, 4:40}
+
+	>> t3 := {1:10, 2:20, 3:30}
+	>> t3:remove(3)
+	>> t3
+	= {1:10, 2:20}