util/
command.rs

1use std::ffi::OsStr;
2#[cfg(not(target_os = "macos"))]
3use std::path::Path;
4
5#[cfg(target_os = "macos")]
6mod darwin;
7
8#[cfg(target_os = "macos")]
9pub use darwin::{Child, Command, Stdio};
10
11#[cfg(target_os = "windows")]
12const CREATE_NO_WINDOW: u32 = 0x0800_0000_u32;
13
14pub fn new_command(program: impl AsRef<OsStr>) -> Command {
15    Command::new(program)
16}
17
18#[cfg(target_os = "windows")]
19pub fn new_std_command(program: impl AsRef<OsStr>) -> std::process::Command {
20    use std::os::windows::process::CommandExt;
21
22    let mut command = std::process::Command::new(program);
23    command.creation_flags(CREATE_NO_WINDOW);
24    command
25}
26
27#[cfg(not(target_os = "windows"))]
28pub fn new_std_command(program: impl AsRef<OsStr>) -> std::process::Command {
29    std::process::Command::new(program)
30}
31
32#[cfg(not(target_os = "macos"))]
33pub type Child = smol::process::Child;
34
35#[cfg(not(target_os = "macos"))]
36pub use std::process::Stdio;
37
38#[cfg(not(target_os = "macos"))]
39#[derive(Debug)]
40pub struct Command(smol::process::Command);
41
42#[cfg(not(target_os = "macos"))]
43impl Command {
44    #[inline]
45    pub fn new(program: impl AsRef<OsStr>) -> Self {
46        #[cfg(target_os = "windows")]
47        {
48            use smol::process::windows::CommandExt;
49            let mut cmd = smol::process::Command::new(program);
50            cmd.creation_flags(CREATE_NO_WINDOW);
51            Self(cmd)
52        }
53        #[cfg(not(target_os = "windows"))]
54        Self(smol::process::Command::new(program))
55    }
56
57    pub fn arg(&mut self, arg: impl AsRef<OsStr>) -> &mut Self {
58        self.0.arg(arg);
59        self
60    }
61
62    pub fn args<I, S>(&mut self, args: I) -> &mut Self
63    where
64        I: IntoIterator<Item = S>,
65        S: AsRef<OsStr>,
66    {
67        self.0.args(args);
68        self
69    }
70
71    pub fn get_args(&self) -> impl Iterator<Item = &OsStr> {
72        self.0.get_args()
73    }
74
75    pub fn env(&mut self, key: impl AsRef<OsStr>, val: impl AsRef<OsStr>) -> &mut Self {
76        self.0.env(key, val);
77        self
78    }
79
80    pub fn envs<I, K, V>(&mut self, vars: I) -> &mut Self
81    where
82        I: IntoIterator<Item = (K, V)>,
83        K: AsRef<OsStr>,
84        V: AsRef<OsStr>,
85    {
86        self.0.envs(vars);
87        self
88    }
89
90    pub fn env_remove(&mut self, key: impl AsRef<OsStr>) -> &mut Self {
91        self.0.env_remove(key);
92        self
93    }
94
95    pub fn env_clear(&mut self) -> &mut Self {
96        self.0.env_clear();
97        self
98    }
99
100    pub fn current_dir(&mut self, dir: impl AsRef<Path>) -> &mut Self {
101        self.0.current_dir(dir);
102        self
103    }
104
105    pub fn stdin(&mut self, cfg: impl Into<Stdio>) -> &mut Self {
106        self.0.stdin(cfg.into());
107        self
108    }
109
110    pub fn stdout(&mut self, cfg: impl Into<Stdio>) -> &mut Self {
111        self.0.stdout(cfg.into());
112        self
113    }
114
115    pub fn stderr(&mut self, cfg: impl Into<Stdio>) -> &mut Self {
116        self.0.stderr(cfg.into());
117        self
118    }
119
120    pub fn kill_on_drop(&mut self, kill_on_drop: bool) -> &mut Self {
121        self.0.kill_on_drop(kill_on_drop);
122        self
123    }
124
125    pub fn spawn(&mut self) -> std::io::Result<Child> {
126        self.0.spawn()
127    }
128
129    pub async fn output(&mut self) -> std::io::Result<std::process::Output> {
130        self.0.output().await
131    }
132
133    pub async fn status(&mut self) -> std::io::Result<std::process::ExitStatus> {
134        self.0.status().await
135    }
136
137    pub fn get_program(&self) -> &OsStr {
138        self.0.get_program()
139    }
140}
util/command.rs

util/
command.rs
