os.args and type()

Author: cyteon

lang/src/cli/repl.rs

@@ -54,10 +54,6 @@ impl Highlighter for Syntax {
             format!("{}{}", caps[1].blue().to_string(), &caps[2])
         }).to_string();
 
-        result = self.string_re.replace_all(&result, |caps: &regex::Captures| {
-            caps[0].green().to_string()
-        }).to_string();
-
         result = self.comment_re.replace_all(&result, |caps: &regex::Captures| {
             caps[0].dimmed().to_string()
         }).to_string();
@@ -70,6 +66,10 @@ impl Highlighter for Syntax {
             caps[0].cyan().to_string()
         }).to_string();
 
+        result = self.string_re.replace_all(&result, |caps: &regex::Captures| {
+            caps[0].green().to_string()
+        }).to_string();
+
         std::borrow::Cow::Owned(result)
     }
 

lang/src/functions.rs

@@ -22,6 +22,7 @@ pub fn get_functions() -> Vec<BuiltinFn> {
         builtin("float", float),
         builtin("str", str),
         builtin("bool", bool),
+        builtin("type", r#type),
         builtin("exit", exit),
         builtin("error", error),
     ]
@@ -127,6 +128,14 @@ fn bool(args: Vec<Value>) -> Result<Value, String> {
     Ok(Value::Bool(args[0].truthy()))
 }
 
+fn r#type(args: Vec<Value>) -> Result<Value, String> {
+    if args.len() != 1 {
+        return Err(format!("type() takes exactly one argument ({} given)", args.len()));
+    }
+
+    Ok(Value::String(args[0].type_name().to_string()))
+}
+
 fn exit(args: Vec<Value>) -> Result<Value, String> {
     if args.len() == 0 {
         std::process::exit(0);

lang/src/stdlib/os.rs

@@ -14,6 +14,8 @@ pub fn object() -> Value {
     methods.insert("getenv".to_string(), Value::BuiltinFn(BuiltinFn::new("getenv", getenv)));
     methods.insert("setenv".to_string(), Value::BuiltinFn(BuiltinFn::new("setenv", setenv)));
     methods.insert("unsetenv".to_string(), Value::BuiltinFn(BuiltinFn::new("unsetenv", unsetenv)));
+    methods.insert("args".to_string(), Value::BuiltinFn(BuiltinFn::new("args", args)));
+
     methods.insert("name".to_string(), Value::String({
         if cfg!(target_os = "windows") {
             "windows"
@@ -156,4 +158,9 @@ fn unsetenv(args: Vec<Value>) -> Result<Value, String> {
     }
 
     Ok(Value::Null)
+}
+
+fn args(_args: Vec<Value>) -> Result<Value, String> {
+    let args: Vec<Value> = std::env::args().map(|s| Value::String(s)).collect();
+    Ok(Value::Array(args))
 }

lang/src/vm/vm.rs

@@ -241,6 +241,7 @@ impl VM {
                             match (func.func)(args) {
                                 Ok(result) => self.stack.push(result),
                                 Err(e) => {
+                                    // this isnt great lmfao but it works ig
                                     if func.name == "error" && !e.contains("error() takes exactly one argument") {
                                         return Err(self.runtime_error(format!("{}", e), span));
                                     }