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 {
command = command.current_dir(cwd);
command.current_dir(cwd);
}
if let Some(env) = options.env {
command = command.envs(env);
command.envs(env);
} else {
command = command.env_clear();
command.env_clear();
}
let encoding = match options.encoding {
Option::None => EncodingWrapper::Text(None),

@ -13,6 +13,8 @@
use std::{
collections::HashMap,
ffi::OsStr,
path::Path,
sync::{Arc, Mutex},
};
@ -45,7 +47,7 @@ pub struct Shell<R: Runtime> {
impl<R: Runtime> Shell<R> {
/// 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)
}
@ -53,7 +55,7 @@ impl<R: Runtime> Shell<R> {
///
/// 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).
pub fn sidecar(&self, program: impl Into<String>) -> Result<Command> {
pub fn sidecar(&self, program: impl AsRef<Path>) -> Result<Command> {
Command::new_sidecar(program)
}

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

@ -237,13 +237,15 @@ impl Scope {
.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()))?
} else {
Command::new(command_s)
};
Ok(command.args(args))
command.args(args);
Ok(command)
}
/// Open a path in the default (or specified) browser.

Loading…
Cancel
Save