mirror of https://github.com/sharkdp/fd.git
95 lines
2.4 KiB
Rust
95 lines
2.4 KiB
Rust
use std::process;
|
|
|
|
#[cfg(unix)]
|
|
use nix::sys::signal::{raise, signal, SigHandler, Signal};
|
|
|
|
#[derive(Debug, Clone, Copy, PartialEq)]
|
|
pub enum ExitCode {
|
|
Success,
|
|
HasResults(bool),
|
|
GeneralError,
|
|
KilledBySigint,
|
|
}
|
|
|
|
impl From<ExitCode> for i32 {
|
|
fn from(code: ExitCode) -> Self {
|
|
match code {
|
|
ExitCode::Success => 0,
|
|
ExitCode::HasResults(has_results) => !has_results as i32,
|
|
ExitCode::GeneralError => 1,
|
|
ExitCode::KilledBySigint => 130,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl ExitCode {
|
|
fn is_error(self) -> bool {
|
|
i32::from(self) != 0
|
|
}
|
|
|
|
/// Exit the process with the appropriate code.
|
|
pub fn exit(self) -> ! {
|
|
#[cfg(unix)]
|
|
if self == ExitCode::KilledBySigint {
|
|
// Get rid of the SIGINT handler, if present, and raise SIGINT
|
|
unsafe {
|
|
if signal(Signal::SIGINT, SigHandler::SigDfl).is_ok() {
|
|
let _ = raise(Signal::SIGINT);
|
|
}
|
|
}
|
|
}
|
|
|
|
process::exit(self.into())
|
|
}
|
|
}
|
|
|
|
pub fn merge_exitcodes(results: impl IntoIterator<Item = ExitCode>) -> ExitCode {
|
|
if results.into_iter().any(ExitCode::is_error) {
|
|
return ExitCode::GeneralError;
|
|
}
|
|
ExitCode::Success
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn success_when_no_results() {
|
|
assert_eq!(merge_exitcodes([]), ExitCode::Success);
|
|
}
|
|
|
|
#[test]
|
|
fn general_error_if_at_least_one_error() {
|
|
assert_eq!(
|
|
merge_exitcodes([ExitCode::GeneralError]),
|
|
ExitCode::GeneralError
|
|
);
|
|
assert_eq!(
|
|
merge_exitcodes([ExitCode::KilledBySigint]),
|
|
ExitCode::GeneralError
|
|
);
|
|
assert_eq!(
|
|
merge_exitcodes([ExitCode::KilledBySigint, ExitCode::Success]),
|
|
ExitCode::GeneralError
|
|
);
|
|
assert_eq!(
|
|
merge_exitcodes([ExitCode::Success, ExitCode::GeneralError]),
|
|
ExitCode::GeneralError
|
|
);
|
|
assert_eq!(
|
|
merge_exitcodes([ExitCode::GeneralError, ExitCode::KilledBySigint]),
|
|
ExitCode::GeneralError
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn success_if_no_error() {
|
|
assert_eq!(merge_exitcodes([ExitCode::Success]), ExitCode::Success);
|
|
assert_eq!(
|
|
merge_exitcodes([ExitCode::Success, ExitCode::Success]),
|
|
ExitCode::Success
|
|
);
|
|
}
|
|
}
|