From 842ef24a60ca0e1fe76a38cee783c98659471cba Mon Sep 17 00:00:00 2001
From: MeexReay <meexreay@gmail.com>
Date: Sat, 27 Jul 2024 21:22:28 +0300
Subject: [PATCH] tabs and

---
 src/lib.rs  | 40 ++++++++++++++++++++++++----------------
 src/main.rs | 12 ++++++++++--
 test.sus    | 23 ++++++++++++-----------
 3 files changed, 46 insertions(+), 29 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 506cc9e..cd4e8c5 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -956,7 +956,13 @@ fn prepare_script(text: String) -> Vec<String> {
             Some(s) => s.0,
             None => s,
         })
-        .map(|s| s.trim_end_matches(" ").to_string())
+        .map(|s| {
+            s.trim_end_matches(" ")
+                .trim_end_matches("\t")
+                .trim_start_matches(" ")
+                .trim_start_matches("\t")
+                .to_string()
+        })
         .collect()
 }
 
@@ -1226,6 +1232,8 @@ impl RunningScript {
         let mut var: Option<&mut Variable> = None;
         let parts: Vec<&str> = name.split('.').collect();
 
+        let global = global || (self.variables.contains_key(&name) && !locals.contains_key(&name));
+
         if parts.len() == 1 {
             if global {
                 self.variables.insert(name, value);
@@ -1457,6 +1465,21 @@ impl RunningScript {
                 CommandType::Return => {
                     return Ok(());
                 }
+                CommandType::For => {
+                    let func_name = command.args[0].clone();
+                    let start_index = self.get_var(command.args[1].clone(), locals)?.as_int()?;
+                    let end_index = self.get_var(command.args[2].clone(), locals)?.as_int()?;
+
+                    let func = self.get_function(func_name).unwrap();
+
+                    for index in start_index..=end_index {
+                        self.exec_function(
+                            func.clone(),
+                            "null".to_string(),
+                            vec![Variable::from_int(Some(index))],
+                        )?;
+                    }
+                }
                 CommandType::ToString => {}
                 CommandType::ToBytes => {}
                 CommandType::ToInteger => {}
@@ -1473,21 +1496,6 @@ impl RunningScript {
                 CommandType::ReadAll => {}
                 CommandType::ReadStr => {}
                 CommandType::ReadStrAll => {}
-                CommandType::For => {
-                    let func_name = command.args[0].clone();
-                    let start_index = self.get_var(command.args[1].clone(), locals)?.as_int()?;
-                    let end_index = self.get_var(command.args[2].clone(), locals)?.as_int()?;
-
-                    let func = self.get_function(func_name).unwrap();
-
-                    for index in start_index..=end_index {
-                        self.exec_function(
-                            func.clone(),
-                            "null".to_string(),
-                            vec![Variable::from_int(Some(index))],
-                        )?;
-                    }
-                }
                 CommandType::ForMap => {}
                 CommandType::ForList => {}
                 CommandType::While => {}
diff --git a/src/main.rs b/src/main.rs
index a22328e..f38638b 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -7,8 +7,16 @@ use std::{
 use sustlang::{RunningScript, Script};
 
 fn main() {
-    let script = Script::parse(fs::read_to_string("test.sus").unwrap()).unwrap();
+    let args: Vec<String> = args().collect();
+
+    let filename = args[1].clone();
+    let args = args[1..].to_vec();
+
+    let script = Script::parse(fs::read_to_string(filename).unwrap()).unwrap();
+
     let mut running_script = RunningScript::new(script);
-    running_script.set_standard_vars(args().collect(), Box::new(stdout()), Box::new(stdin()));
+    running_script
+        .set_standard_vars(args, Box::new(stdout()), Box::new(stdin()))
+        .unwrap();
     running_script.run().unwrap();
 }
diff --git a/test.sus b/test.sus
index 657845d..377679f 100644
--- a/test.sus
+++ b/test.sus
@@ -1,20 +1,21 @@
 FUNC null println text string # println command
-
-TEMP_VAR char br 10           # line break var
-ADD_STR text br               # add line break to text var
-WRITE text cout               # write text var to console
-
+    TEMP_VAR char br 10       # line break var
+    ADD_STR text br           # add line break to text var
+    WRITE text cout           # write text var to console
 FUNC_END                      # end println command
 
+FUNC null hello_world index integer
+    USE_FUNC println null text
+FUNC_END
+
+
+
 INIT_VAR string text
 SET_VAR text Hello World!
 
-FUNC null hello_world index integer
-
-USE_FUNC println null text
-
-FUNC_END
-
 TEMP_VAR integer start_index 0
 TEMP_VAR integer end_index 9
+
 FOR hello_world start_index end_index
+
+DROP_VAR text