rustywatch 0.2.11

use std::{
    io::{BufRead, BufReader, Error},
    process::{Child, Command, Stdio},
};
use tokio::task;

// @TODO
// execute command with blocking mode.
pub async fn exec(cmd: String) -> Result<Child, Error> {
    task::spawn_blocking(move || match cfg!(windows) {
        true => Command::new("cmd")
            .arg("/C")
            .arg(cmd)
            .stdout(Stdio::piped())
            .stderr(Stdio::piped())
            .spawn(),
        false => Command::new("sh")
            .arg("-c")
            .arg(cmd)
            .stdout(Stdio::piped())
            .stderr(Stdio::piped())
            .spawn(),
    })
    .await?
}

// @TODO
// macro for printout lines.
macro_rules! process_output {
    ($reader:expr, $print_fn:ident) => {
        for line in $reader.lines() {
            $print_fn!("{}", line.unwrap());
        }
    };
}

// @TODO
// buf_reader for printout result as a buffer
pub fn buf_reader(child: Child) {
    let stdout = child.stdout.unwrap();
    let stderr = child.stderr.unwrap();
    let stdout_reader = BufReader::new(stdout);
    let stderr_reader = BufReader::new(stderr);

    process_output!(stdout_reader, println);
    process_output!(stderr_reader, eprintln);
}

#[cfg(test)]
mod tests {
    use super::*;

    #[tokio::test]
    #[cfg(not(windows))]
    async fn test_exec_unix() {
        let result = exec("echo 'Hello, World!'".to_string()).await;
        assert!(result.is_ok());

        let child = result.unwrap();
        let output = child.wait_with_output().unwrap();

        assert!(output.status.success());
        assert_eq!(
            String::from_utf8_lossy(&output.stdout).trim(),
            "Hello, World!"
        );
    }

    #[tokio::test]
    #[cfg(windows)]
    async fn test_exec_windows() {
        let result = exec("echo Hello, World!".to_string()).await;
        assert!(result.is_ok());

        let child = result.unwrap();
        let output = child.wait_with_output().unwrap();

        assert!(output.status.success());
        assert_eq!(
            String::from_utf8_lossy(&output.stdout).trim(),
            "Hello, World!"
        );
    }
}