From 90352dd264fe022287f1e9eea60ef5bc7b41d0b4 Mon Sep 17 00:00:00 2001
From: Jeff <dev@jeffa.io>
Date: Fri, 6 Oct 2023 21:00:31 -0400
Subject: [PATCH] Implement function calls

---
 src/abstract_tree/function_call.rs | 6 ++++++
 src/evaluator.rs                   | 2 --
 src/value/variable_map.rs          | 7 +++++++
 3 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/abstract_tree/function_call.rs b/src/abstract_tree/function_call.rs
index aae7faa..27bd125 100644
--- a/src/abstract_tree/function_call.rs
+++ b/src/abstract_tree/function_call.rs
@@ -61,6 +61,12 @@ impl AbstractTree for FunctionCall {
             results.push(result);
         }
 
+        for identifier in definition.identifiers() {
+            let key = identifier.inner();
+
+            context.remove(&key);
+        }
+
         Ok(Value::List(results))
     }
 }
diff --git a/src/evaluator.rs b/src/evaluator.rs
index 6465c0a..fb79cf8 100644
--- a/src/evaluator.rs
+++ b/src/evaluator.rs
@@ -86,8 +86,6 @@ impl<'context, 'code> Evaluator<'context, 'code> {
         let item_count = root_node.child_count();
         let mut results = Vec::with_capacity(item_count);
 
-        println!("{}", root_node.to_sexp());
-
         for item_node in root_node.children(&mut cursor) {
             let item_result = Item::from_syntax_node(item_node, self.source);
 
diff --git a/src/value/variable_map.rs b/src/value/variable_map.rs
index fa114ae..d589c3d 100644
--- a/src/value/variable_map.rs
+++ b/src/value/variable_map.rs
@@ -108,6 +108,13 @@ impl VariableMap {
         }
     }
 
+    /// Removes and assignmed variable.
+    ///
+    /// TODO: Support dot notation.
+    pub fn remove(&mut self, key: &str) -> Option<Value> {
+        self.variables.remove(key)
+    }
+
     /// Returns a reference to the inner BTreeMap.
     pub fn inner(&self) -> &BTreeMap<String, Value> {
         &self.variables