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
use std::io::{self, Read};
use std::process;
pub struct Command {
program: Option<String>,
args: Vec<String>,
dir: Option<String>,
}
pub struct CommandRun {
pub status: process::ExitStatus,
pub stdout: String,
pub stderr: String,
}
impl Command {
pub fn exists_program(program: &str) -> bool {
Command::new().program(program).spawn().is_ok()
}
pub fn new() -> Command {
Command {
program: None,
args: vec![],
dir: None,
}
}
pub fn program(&mut self, program: &str) -> &mut Self {
self.program = Some(program.to_owned());
self
}
pub fn arg(&mut self, arg: &str) -> &mut Self {
self.args.push(arg.to_owned());
self
}
pub fn arg_from_parts(&mut self, parts: Vec<&str>) -> &mut Self {
let arg = parts.join("");
self.args.push(arg);
self
}
pub fn current_dir(&mut self, dir: &str) -> &mut Self {
self.dir = Some(dir.to_owned());
self
}
pub fn spawn(&mut self) -> io::Result<CommandRun> {
match &self.program {
None => Err(io::Error::new(io::ErrorKind::InvalidInput, "")),
Some(program) => {
let mut command = process::Command::new(program);
command
.args(&self.args)
.stdout(process::Stdio::piped())
.stderr(process::Stdio::piped());
if let Some(dir) = &self.dir {
command.current_dir(dir);
}
let mut process = command.spawn()?;
let status = process.wait()?;
let mut stdout = String::new();
process.stdout.unwrap().read_to_string(&mut stdout)?;
let mut stderr = String::new();
process.stderr.unwrap().read_to_string(&mut stderr)?;
Ok(CommandRun {
status,
stdout,
stderr,
})
}
}
}
}
impl Default for Command {
fn default() -> Self {
Self::new()
}
}