nu-system 0.113.0

Nushell system querying
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

use std::io;
use std::process::Command as CommandSys;

/// Tries to forcefully kill a process by its PID
pub fn kill_by_pid(pid: i64) -> Result<(), KillByPidError> {
    let mut cmd = build_kill_command(true, std::iter::once(pid), None);

    let output = cmd.output().map_err(KillByPidError::Output)?;

    match output.status.success() {
        true => Ok(()),
        false => Err(KillByPidError::KillProcess),
    }
}

/// Error while killing a process forcefully by its PID.
pub enum KillByPidError {
    /// I/O error while capturing the output of the process.
    Output(io::Error),

    /// Killing the process failed.
    KillProcess,
}

/// Create a `std::process::Command` for the current target platform, for killing
/// the processes with the given PIDs
pub fn build_kill_command(
    force: bool,
    pids: impl Iterator<Item = i64>,
    signal: Option<u32>,
) -> CommandSys {
    if cfg!(windows) {
        let mut cmd = CommandSys::new("taskkill");

        if force {
            cmd.arg("/F");
        }

        // each pid must written as `/PID 0` otherwise
        // taskkill will act as `killall` unix command
        for id in pids {
            cmd.arg("/PID");
            cmd.arg(id.to_string());
        }

        cmd
    } else {
        let mut cmd = CommandSys::new("kill");
        if let Some(signal_value) = signal {
            cmd.arg(format!("-{signal_value}"));
        } else if force {
            cmd.arg("-9");
        }

        cmd.arg("--");
        cmd.args(pids.map(|id| id.to_string()));

        cmd
    }
}

#[cfg(all(test, unix))]
mod tests {
    use super::*;

    #[test]
    fn build_kill_command_separates_options_from_negative_pids() {
        let cmd = build_kill_command(false, std::iter::once(-9), None);
        let args = cmd
            .get_args()
            .map(|arg| arg.to_string_lossy().into_owned())
            .collect::<Vec<_>>();

        assert_eq!(args, ["--", "-9"]);
    }

    #[test]
    fn build_kill_command_keeps_signal_and_pid_separator() {
        let cmd = build_kill_command(false, std::iter::once(123), Some(9));
        let args = cmd
            .get_args()
            .map(|arg| arg.to_string_lossy().into_owned())
            .collect::<Vec<_>>();

        assert_eq!(args, ["-9", "--", "123"]);
    }
}