From b68150b304daeb631eb4f975b94451a3f56bf9a1 Mon Sep 17 00:00:00 2001
From: Artemy <artegoser@gmail.com>
Date: Mon, 15 Aug 2022 18:16:32 +0300
Subject: [PATCH] fix: return function errors

---
 src/interpreter.rs | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/src/interpreter.rs b/src/interpreter.rs
index 8e95b0a..b39aa51 100644
--- a/src/interpreter.rs
+++ b/src/interpreter.rs
@@ -307,7 +307,6 @@ impl Interpreter {
                 let real_len = args.len();
                 let func_args = function.get("args").unwrap().as_array().unwrap();
                 if real_len == func_args.len() {
-                    self.enter_the_scope();
                     for i in 0..real_len {
                         let argname = &func_args[i];
                         match argname {
@@ -317,12 +316,11 @@ impl Interpreter {
                             _ => self.error("Argument name must be a string"),
                         }
                     }
+
                     let name = self.run_nodes(function.get("body").unwrap().as_array().unwrap());
                     match name {
                         Value::Object(name) => {
-                            let result = self.eval_node(&name.get("return").unwrap());
-                            self.exit_from_scope();
-                            return result;
+                            return name.get("return").unwrap().clone();
                         }
                         _ => return Value::Null,
                     }
@@ -336,7 +334,6 @@ impl Interpreter {
                 }
             }
         }
-        self.exit_from_scope();
         Value::Null
     }
     fn define_fn(&mut self, value: &Map<String, Value>) {
@@ -377,7 +374,11 @@ impl Interpreter {
     }
 
     fn if_node(&mut self, value: &Map<String, Value>) -> Value {
-        let condition = self.eval_node(&value["condition"]);
+        let condition = self.eval_node(
+            &value
+                .get("condition")
+                .expect("`if` must have a `condition` argument"),
+        );
         let nodes = &value.get("body");
         let else_nodes = &value.get("else");
 
@@ -411,7 +412,7 @@ impl Interpreter {
                 _ => {}
             },
             None => {
-                self.error("if must have a body");
+                self.error("`if` must have a body");
             }
         }
         Value::Null
@@ -443,13 +444,18 @@ impl Interpreter {
             match to_do {
                 Value::String(name) => {
                     if name == "break" || name == "continue" {
+                        self.exit_from_scope();
                         return Value::String(name);
                     }
                 }
                 Value::Object(name) => {
                     let check = name.get("return");
                     match check {
-                        Some(_) => return Value::Object(name),
+                        Some(check) => {
+                            let result = json!({"return": self.eval_node(check)});
+                            self.exit_from_scope();
+                            return result;
+                        }
                         None => {}
                     }
                 }