mirror of https://github.com/sharkdp/fd.git
53 lines
1.6 KiB
Rust
53 lines
1.6 KiB
Rust
use std::io;
|
|
use std::io::Write;
|
|
use std::process::Command;
|
|
use std::sync::Mutex;
|
|
|
|
use crate::error::print_error;
|
|
use crate::exit_codes::ExitCode;
|
|
|
|
/// Executes a command.
|
|
pub fn execute_command(
|
|
mut cmd: Command,
|
|
out_perm: &Mutex<()>,
|
|
enable_output_buffering: bool,
|
|
) -> ExitCode {
|
|
// Spawn the supplied command.
|
|
let output = if enable_output_buffering {
|
|
cmd.output()
|
|
} else {
|
|
// If running on only one thread, don't buffer output
|
|
// Allows for viewing and interacting with intermediate command output
|
|
cmd.spawn().and_then(|c| c.wait_with_output())
|
|
};
|
|
|
|
// Then wait for the command to exit, if it was spawned.
|
|
match output {
|
|
Ok(output) => {
|
|
// While this lock is active, this thread will be the only thread allowed
|
|
// to write its outputs.
|
|
let _lock = out_perm.lock().unwrap();
|
|
|
|
let stdout = io::stdout();
|
|
let stderr = io::stderr();
|
|
|
|
let _ = stdout.lock().write_all(&output.stdout);
|
|
let _ = stderr.lock().write_all(&output.stderr);
|
|
|
|
if output.status.code() == Some(0) {
|
|
ExitCode::Success
|
|
} else {
|
|
ExitCode::GeneralError
|
|
}
|
|
}
|
|
Err(ref why) if why.kind() == io::ErrorKind::NotFound => {
|
|
print_error(format!("Command not found: {:?}", cmd));
|
|
ExitCode::GeneralError
|
|
}
|
|
Err(why) => {
|
|
print_error(format!("Problem while executing command: {}", why));
|
|
ExitCode::GeneralError
|
|
}
|
|
}
|
|
}
|