feat(std): more commands

Cargo.toml

@@ -14,7 +14,9 @@ itertools = "0.11.0"
 clap = { version = "4.4.18", features = ["derive"] }
 tempfile = "3.10.1"
 
+# test dependencies
 [dev-dependencies]
+tiny_http = "0.12.0"
 assert_cmd = "2.0.14"
 predicates = "3.1.0"
 

src/std/main.ab

@@ -128,3 +128,72 @@ pub fun includes(arr, value) {
     }
     return false
 }
+
+pub fun is_command(command: Text): Bool {
+    if (unsafe $command -v "{command}" > /dev/null$) {
+        return true
+    }
+    return false
+}
+
+pub fun create_symbolic_link(origin: Text, destination: Text): Bool {
+    if file_exist(origin) {
+        unsafe $ln -s "{origin}" "{destination}"$
+        return true
+    }
+
+    echo "The file {origin} doesn't exist!"
+    return false
+}
+
+pub fun create_dir(path: Text): Null {
+    if not dir_exist(path) {
+        unsafe $mkdir -p "{path}"$
+    }
+}
+
+pub fun make_executable(path: Text): Bool {
+    if file_exist(path) {
+        unsafe $chmod +x "{path}"$
+        return true
+    }
+
+    echo "The file {path} doesn't exist!"
+    return false
+}
+
+pub fun switch_user_permission(user: Text, path: Text): Bool {
+    if file_exist(path) or dir_exist(path) {
+        unsafe $chown -R "{user}" "{path}"$
+        return true
+    }
+
+    return false
+}
+
+pub fun download(url: Text, path: Text): Bool {
+    if {
+        is_command("curl") {
+            unsafe $curl -o "{path}" "{url}"$
+        }
+        is_command("wget") {
+            unsafe $wget "{url}" -P "{path}"$
+        }
+        is_command("aria2c") {
+            unsafe $aria2c "{url}" -d "{path}"$
+        }
+        else {
+            return false
+        }
+    }
+
+    return true
+}
+
+pub fun is_root(): Bool {
+    if (unsafe $id -u$ == "0") {
+        return true
+    }
+
+    return false
+}

src/tests/stdlib.rs

@@ -5,6 +5,7 @@ use std::fs;
 use std::io::Read;
 use std::io::Write;
 use std::path::PathBuf;
+use std::time::Duration;
 use tempfile::tempdir;
 use tempfile::TempDir;
 use std::process::{Command, Stdio};
@@ -22,6 +23,16 @@ fn mkfile() -> (PathBuf, TempDir) {
     (file_path, temp_dir)
 }
 
+fn http_server() {
+    use tiny_http::{Server, Response};
+
+    let server = Server::http("127.0.0.1:8081").expect("Can't bind to 127.0.0.1:8081");
+    for req in server.incoming_requests() {
+        req.respond(Response::from_string("ok")).expect("Can't respond");
+        break;
+    }
+}
+
 #[test]
 fn input() {
     let prompt_message = "Please enter your name:";
@@ -430,3 +441,116 @@ fn lines() {
     ";
     test_amber!(code, "line: hello\nline: world")
 }
+
+#[test]
+fn is_command() {
+    let code = "
+        import { is_command } from \"std\"
+        main {
+            if is_command(\"cat\") {
+                echo \"exist\"
+            }
+        }
+    ";
+    test_amber!(code, "exist")
+}
+
+#[test]
+fn create_symbolic_link() {
+    let code = "
+        import { create_symbolic_link } from \"std\"
+        main {
+            unsafe $touch /tmp/amber-symbolic$
+            if create_symbolic_link(\"/tmp/amber-symbolic\", \"/tmp/amber-symbolic-link\") {
+                echo \"created\"
+            } else {
+                echo \"failed\"
+            }
+            unsafe $rm /tmp/amber-symbolic$
+            unsafe $rm /tmp/amber-symbolic-link$
+        }
+    ";
+    test_amber!(code, "created")
+}
+
+#[test]
+fn create_dir() {
+    let code = "
+        import { create_dir, dir_exist } from \"std\"
+        main {
+            create_dir(\"/tmp/amber-test\")
+            if dir_exist(\"/tmp/amber-test\") {
+                unsafe $rm /tmp/amber-test$
+                echo \"created\"
+            }
+        }
+    ";
+    test_amber!(code, "created")
+}
+
+#[test]
+fn make_executable() {
+    let code = "
+        import { make_executable } from \"std\"
+        main {
+            unsafe $touch /tmp/amber-symbolic$
+            if make_executable(\"/tmp/amber-symbolic\") {
+                echo \"created\"
+            }
+            unsafe $rm /tmp/amber-symbolic$
+        }
+    ";
+    test_amber!(code, "created")
+}
+
+#[test]
+fn switch_user_permission() {
+    // We use `whoami` to get the running user to assign again the same user as permission
+    let code = "
+        import { switch_user_permission } from \"std\"
+        main {
+            unsafe $touch /tmp/amber-symbolic$
+            if switch_user_permission(unsafe $whoami$,\"/tmp/amber-symbolic\") {
+                echo \"done\"
+            }
+            unsafe $rm /tmp/amber-symbolic$
+        }
+    ";
+    test_amber!(code, "done")
+}
+
+#[test]
+fn download() {
+    let server = std::thread::spawn(http_server);
+
+    let code = "
+        import { download, is_command, exit } from \"std\"
+        main {
+            let tempfile = unsafe $mktemp$
+            if download(\"http://127.0.0.1:8081/\", tempfile) {
+                $cat {tempfile}$ failed {
+                    echo \"{tempfile} does not exist!!\"
+                }
+                unsafe $rm -f {tempfile}$
+            }
+        }
+    ";
+
+    test_amber!(code, "ok");
+
+    std::thread::sleep(Duration::from_millis(150));
+    assert!(server.is_finished(), "Server has not stopped!");
+}
+
+#[test]
+fn is_root() {
+    let code = "
+        import { is_root } from \"std\"
+        main {
+            if not is_root() {
+                echo \"no\"
+            }
+        }
+    ";
+    test_amber!(code, "no")
+}