rura 1.6.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

use crate::shell::output::Output;
use anyhow::anyhow;
use std::io::Write;
use std::process::{Command, Stdio};
use std::sync::Arc;
use std::thread;

pub trait Exec {
    fn exec(&self, command: Command, stdin: Arc<[u8]>) -> anyhow::Result<Output>;
}

pub struct SystemExec;

impl Exec for SystemExec {
    fn exec(&self, mut command: Command, stdin: Arc<[u8]>) -> anyhow::Result<Output> {
        let mut child = command
            .stdin(Stdio::piped())
            .stdout(Stdio::piped())
            .stderr(Stdio::piped())
            .spawn()
            .map_err(|e| anyhow!("Failed to spawn command [{command:?}]: {e}"))?;

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

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

        match child.wait_with_output() {
            Ok(output) => {
                if output.status.success() {
                    Ok(Output::Ok(Arc::from(output.stdout)))
                } else {
                    // failed successfully!
                    Ok(Output::Err(Arc::from(output.stderr), output.status.code()))
                }
            }
            Err(e) => Err(anyhow!("Failed to execute command '{command:?}': {e}")),
        }
    }
}

#[cfg(test)]
pub struct MockExec {
    pub calls: std::rc::Rc<std::cell::RefCell<Vec<(String, String)>>>,
}

#[cfg(test)]
impl Exec for MockExec {
    fn exec(&self, command: Command, stdin: Arc<[u8]>) -> anyhow::Result<Output> {
        let program = command.get_program().to_string_lossy().into_owned();
        self.calls
            .borrow_mut()
            .push((program.clone(), String::from_utf8_lossy(&stdin).into()));
        if program.ends_with("err") {
            Ok(Output::Err(
                Arc::from(format!("{}-output", program).into_bytes()),
                Some(1),
            ))
        } else {
            Ok(Output::Ok(Arc::from(
                format!("{}-output", program).into_bytes(),
            )))
        }
    }
}