refactor(shell): Command adhere to std::process::Command

pull/526/head
Lucas Nogueira 2 years ago
parent 6f01bc11ab
commit 0ae5de8e42
No known key found for this signature in database
GPG Key ID: 7C32FCA95C8C95D7

@ -0,0 +1,5 @@
---
"shell": "patch"
---
Refactor the `Command` API adhere to `std::process::Command`.

@ -133,12 +133,12 @@ pub fn execute<R: Runtime>(
} }
}; };
if let Some(cwd) = options.cwd { if let Some(cwd) = options.cwd {
command = command.current_dir(cwd); command.current_dir(cwd);
} }
if let Some(env) = options.env { if let Some(env) = options.env {
command = command.envs(env); command.envs(env);
} else { } else {
command = command.env_clear(); command.env_clear();
} }
let encoding = match options.encoding { let encoding = match options.encoding {
Option::None => EncodingWrapper::Text(None), Option::None => EncodingWrapper::Text(None),

@ -13,6 +13,8 @@
use std::{ use std::{
collections::HashMap, collections::HashMap,
ffi::OsStr,
path::Path,
sync::{Arc, Mutex}, sync::{Arc, Mutex},
}; };
@ -45,7 +47,7 @@ pub struct Shell<R: Runtime> {
impl<R: Runtime> Shell<R> { impl<R: Runtime> Shell<R> {
/// Creates a new Command for launching the given program. /// Creates a new Command for launching the given program.
pub fn command(&self, program: impl Into<String>) -> Command { pub fn command(&self, program: impl AsRef<OsStr>) -> Command {
Command::new(program) Command::new(program)
} }
@ -53,7 +55,7 @@ impl<R: Runtime> Shell<R> {
/// ///
/// A sidecar program is a embedded external binary in order to make your application work /// A sidecar program is a embedded external binary in order to make your application work
/// or to prevent users having to install additional dependencies (e.g. Node.js, Python, etc). /// or to prevent users having to install additional dependencies (e.g. Node.js, Python, etc).
pub fn sidecar(&self, program: impl Into<String>) -> Result<Command> { pub fn sidecar(&self, program: impl AsRef<Path>) -> Result<Command> {
Command::new_sidecar(program) Command::new_sidecar(program)
} }

@ -3,9 +3,10 @@
// SPDX-License-Identifier: MIT // SPDX-License-Identifier: MIT
use std::{ use std::{
collections::HashMap, ffi::OsStr,
io::{BufReader, Write}, io::{BufReader, Write},
path::PathBuf, ops::{Deref, DerefMut},
path::{Path, PathBuf},
process::{Command as StdCommand, Stdio}, process::{Command as StdCommand, Stdio},
sync::{Arc, RwLock}, sync::{Arc, RwLock},
thread::spawn, thread::spawn,
@ -53,12 +54,20 @@ pub enum CommandEvent {
/// The type to spawn commands. /// The type to spawn commands.
#[derive(Debug)] #[derive(Debug)]
pub struct Command { pub struct Command(StdCommand);
program: String,
args: Vec<String>, impl Deref for Command {
env_clear: bool, type Target = StdCommand;
env: HashMap<String, String>,
current_dir: Option<PathBuf>, fn deref(&self) -> &Self::Target {
&self.0
}
}
impl DerefMut for Command {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.0
}
} }
/// Spawned child process. /// Spawned child process.
@ -116,83 +125,37 @@ pub struct Output {
pub stderr: Vec<u8>, pub stderr: Vec<u8>,
} }
fn relative_command_path(command: String) -> crate::Result<String> { fn relative_command_path(command: &Path) -> crate::Result<PathBuf> {
match platform::current_exe()?.parent() { match platform::current_exe()?.parent() {
#[cfg(windows)] #[cfg(windows)]
Some(exe_dir) => Ok(format!("{}\\{command}.exe", exe_dir.display())), Some(exe_dir) => Ok(exe_dir.join(format!("{command}.exe"))),
#[cfg(not(windows))] #[cfg(not(windows))]
Some(exe_dir) => Ok(format!("{}/{command}", exe_dir.display())), Some(exe_dir) => Ok(exe_dir.join(command)),
None => Err(crate::Error::CurrentExeHasNoParent), None => Err(crate::Error::CurrentExeHasNoParent),
} }
} }
impl From<Command> for StdCommand { impl From<Command> for StdCommand {
fn from(cmd: Command) -> StdCommand { fn from(cmd: Command) -> StdCommand {
let mut command = StdCommand::new(cmd.program); cmd.0
command.args(cmd.args);
command.stdout(Stdio::piped());
command.stdin(Stdio::piped());
command.stderr(Stdio::piped());
if cmd.env_clear {
command.env_clear();
}
command.envs(cmd.env);
if let Some(current_dir) = cmd.current_dir {
command.current_dir(current_dir);
}
#[cfg(windows)]
command.creation_flags(CREATE_NO_WINDOW);
command
} }
} }
impl Command { impl Command {
pub(crate) fn new<S: Into<String>>(program: S) -> Self { pub(crate) fn new<S: AsRef<OsStr>>(program: S) -> Self {
Self { let mut command = StdCommand::new(program);
program: program.into(),
args: Default::default(),
env_clear: false,
env: Default::default(),
current_dir: None,
}
}
pub(crate) fn new_sidecar<S: Into<String>>(program: S) -> crate::Result<Self> {
Ok(Self::new(relative_command_path(program.into())?))
}
/// Appends arguments to the command.
#[must_use]
pub fn args<I, S>(mut self, args: I) -> Self
where
I: IntoIterator<Item = S>,
S: AsRef<str>,
{
for arg in args {
self.args.push(arg.as_ref().to_string());
}
self
}
/// Clears the entire environment map for the child process. command.stdout(Stdio::piped());
#[must_use] command.stdin(Stdio::piped());
pub fn env_clear(mut self) -> Self { command.stderr(Stdio::piped());
self.env_clear = true; #[cfg(windows)]
self command.creation_flags(CREATE_NO_WINDOW);
}
/// Adds or updates multiple environment variable mappings. Self(command)
#[must_use]
pub fn envs(mut self, env: HashMap<String, String>) -> Self {
self.env = env;
self
} }
/// Sets the working directory for the child process. pub(crate) fn new_sidecar<S: AsRef<Path>>(program: S) -> crate::Result<Self> {
#[must_use] Ok(Self::new(relative_command_path(program.as_ref())?))
pub fn current_dir(mut self, current_dir: PathBuf) -> Self {
self.current_dir.replace(current_dir);
self
} }
/// Spawns the command. /// Spawns the command.

@ -237,13 +237,15 @@ impl Scope {
.into_owned() .into_owned()
}) })
.unwrap_or_else(|| command.command.to_string_lossy().into_owned()); .unwrap_or_else(|| command.command.to_string_lossy().into_owned());
let command = if command.sidecar { let mut command = if command.sidecar {
Command::new_sidecar(command_s).map_err(|e| Error::Sidecar(e.to_string()))? Command::new_sidecar(command_s).map_err(|e| Error::Sidecar(e.to_string()))?
} else { } else {
Command::new(command_s) Command::new(command_s)
}; };
Ok(command.args(args)) command.args(args);
Ok(command)
} }
/// Open a path in the default (or specified) browser. /// Open a path in the default (or specified) browser.

Loading…
Cancel
Save