rura 1.3.0

Interactive TUI pipeline editor built for rapid iteration
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

use anyhow::{Result, anyhow};
use log::info;
use std::io::Write;
use std::process::{Command, Stdio};
use std::thread;

pub struct CmdRunner {
    shell: String,
}

impl CmdRunner {
    pub fn new(shell: &str) -> Self {
        Self {
            shell: shell.into(),
        }
    }

    pub fn run(&self, command: &str, stdin: &str) -> Result<Output> {
        info!("executing command: '{command}'");

        let mut cmd = Command::new("/usr/bin/env");
        cmd.args([&self.shell, "-c", &command]);

        let mut child = cmd
            .stdin(Stdio::piped())
            .stdout(Stdio::piped())
            .stderr(Stdio::piped())
            .spawn()
            .map_err(|e| anyhow!("Failed to spawn command [{cmd:?}]: {e}"))?;

        let mut child_stdin = child
            .stdin
            .take()
            .ok_or(anyhow!("Failed to take stdin handle"))?;

        let owned_stdin = stdin.to_owned();

        thread::spawn(move || {
            let _ = child_stdin.write_all(owned_stdin.as_bytes());
        });

        if let Ok(output) = child.wait_with_output() {
            if output.status.success() {
                let stdout = output.stdout.as_slice();
                let str = String::from_utf8_lossy(stdout);
                Ok(Output::ok_command(&command, &str))
            } else {
                let stderr = output.stderr.as_slice();
                let str = String::from_utf8_lossy(stderr);
                Ok(Output::err_command(&command, &str, output.status.code()))
            }
        } else {
            Ok(Output::err_command(
                &command,
                "Failed to execute command",
                None,
            ))
        }
    }
}

#[derive(Clone, PartialEq, Eq)]
pub struct Output {
    pub command: Option<String>,
    pub lines: Vec<String>,
    pub status_code: Option<i32>,
    pub ok: bool,
}

impl Output {
    pub fn ok_command(command: &str, str: &str) -> Self {
        Self {
            command: Some(command.into()),
            lines: Self::lines(str),
            status_code: Some(0),
            ok: true,
        }
    }

    pub fn err_command(command: &str, str: &str, status_code: Option<i32>) -> Self {
        Self {
            command: Some(command.into()),
            lines: Self::lines(str),
            status_code,
            ok: false,
        }
    }

    pub fn ok_stdin(str: &str) -> Self {
        Self {
            command: None,
            lines: Self::lines(str),
            status_code: Some(0),
            ok: true,
        }
    }

    pub fn err_stdin(str: &str) -> Self {
        Self {
            command: None,
            lines: Self::lines(str),
            status_code: None,
            ok: false,
        }
    }

    pub fn len(&self) -> usize {
        self.lines.len()
    }

    fn lines(input: &str) -> Vec<String> {
        input.lines().map(|a| a.into()).collect()
    }
}