From c0ee563b448974b3902031c26835e7e5c967cbf6 Mon Sep 17 00:00:00 2001
From: madomado <madonuko@outlook.com>
Date: Sun, 10 Dec 2023 16:20:51 +0800
Subject: [PATCH] feat(andax/fns/io): `sh_stream`

---
 andax/src/fns/io.rs      | 44 ++++++++++++++++++++++++++++++++++++++--
 andax/src/fns/tsunagu.rs |  6 ++++++
 2 files changed, 48 insertions(+), 2 deletions(-)

diff --git a/andax/src/fns/io.rs b/andax/src/fns/io.rs
index 90ce44c..f7d2a4e 100644
--- a/andax/src/fns/io.rs
+++ b/andax/src/fns/io.rs
@@ -9,14 +9,18 @@ use rhai::{
 use std::io::Write;
 use std::process::Command;
 use tracing::{debug, instrument};
+
 macro_rules! _sh_out {
     ($ctx:expr, $o:expr) => {
         Ok((
-            $o.status.code().ok_or::<Box<EvalAltResult>>("No exit code".into())?,
+            _sh_out!($o)?,
             String::from_utf8($o.stdout).ehdl($ctx)?,
             String::from_utf8($o.stderr).ehdl($ctx)?,
         ))
     };
+    ($o:expr) => {{
+        $o.status.code().ok_or::<Box<EvalAltResult>>("No exit code".into())
+    }};
 }
 macro_rules! _cmd {
     ($cmd:expr) => {{
@@ -32,7 +36,14 @@ macro_rules! _cmd {
     }};
 }
 
+macro_rules! _stream_cmd {
+    ($cmd:expr) => {{
+        _cmd!($cmd).stdout(Stdio::inherit()).stderr(Stdio::inherit())
+    }};
+}
+
 type T = Result<(i32, String, String), Box<EvalAltResult>>;
+type U = Result<i32, Box<EvalAltResult>>;
 
 /// for andax, shell():
 /// ```
@@ -51,7 +62,7 @@ pub mod ar {
     pub fn sh_rc(o: (i32, String, String)) -> i32 {
         o.0
     }
-    /// get stdout from the return value of `sh()` 
+    /// get stdout from the return value of `sh()`
     #[rhai_fn(global)]
     pub fn sh_stdout(o: (i32, String, String)) -> String {
         o.1
@@ -62,6 +73,35 @@ pub mod ar {
         o.2
     }
 
+    /// run a command using `cmd` on Windows and `sh` on other systems
+    #[instrument(skip(ctx))]
+    #[rhai_fn(return_raw, name = "sh_stream", global)]
+    pub fn shell_stream(ctx: NativeCallContext, cmd: &str) -> U {
+        debug!("Running in shell");
+        _sh_out!(_cmd!(cmd).output().ehdl(&ctx)?)
+    }
+    /// run a command using `cmd` on Windows and `sh` on other systems in working dir
+    #[instrument(skip(ctx))]
+    #[rhai_fn(return_raw, name = "sh_stream", global)]
+    pub fn shell_cwd_stream(ctx: NativeCallContext, cmd: &str, cwd: &str) -> U {
+        debug!("Running in shell");
+        _sh_out!(_cmd!(cmd).current_dir(cwd).output().ehdl(&ctx)?)
+    }
+    /// run an executable
+    #[instrument(skip(ctx))]
+    #[rhai_fn(return_raw, name = "sh_stream", global)]
+    pub fn sh_stream(ctx: NativeCallContext, cmd: Vec<&str>) -> U {
+        debug!("Running executable");
+        _sh_out!(Command::new(cmd[0]).args(&cmd[1..]).output().ehdl(&ctx)?)
+    }
+    /// run an executable in working directory
+    #[instrument(skip(ctx))]
+    #[rhai_fn(return_raw, name = "sh_stream", global)]
+    pub fn sh_cwd_stream(ctx: NativeCallContext, cmd: Vec<&str>, cwd: &str) -> U {
+        debug!("Running executable");
+        _sh_out!(Command::new(cmd[0]).args(&cmd[1..]).current_dir(cwd).output().ehdl(&ctx)?)
+    }
+
     /// run a command using `cmd` on Windows and `sh` on other systems
     #[instrument(skip(ctx))]
     #[rhai_fn(return_raw, name = "sh", global)]
diff --git a/andax/src/fns/tsunagu.rs b/andax/src/fns/tsunagu.rs
index 398be95..255d138 100644
--- a/andax/src/fns/tsunagu.rs
+++ b/andax/src/fns/tsunagu.rs
@@ -119,6 +119,12 @@ pub mod ar {
             Err(VarError::NotUnicode(o)) => Err(format!("env(`{key}`): invalid UTF: {o:?}").into()),
         }
     }
+
+    #[rhai_fn(return_raw, global)]
+    pub fn rust2rpm(name: &str) -> Res<()> {
+        let cmd = std::process::Command::new("rust2rpm").arg(name).output();
+        Ok(())
+    }
 }
 
 #[derive(Clone)]