xshell 0.3.0-pre.2

Utilities for quick shell scripting in Rust
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

use std::io::{self, Write};
use std::thread;
use std::time::Duration;

fn main() {
    if let Err(err) = try_main() {
        eprintln!("{err}");
        std::process::exit(1);
    }
}

fn try_main() -> io::Result<()> {
    let mut sleep_seconds = 0;
    let mut fail = false;
    let mut suicide = false;

    let mut args = std::env::args().skip(1).peekable();
    while let Some(arg) = args.peek() {
        match arg.as_str() {
            "-f" => fail = true,
            "-s" => suicide = true,
            _ => break,
        }
        args.next();
    }

    if let Some(arg) = args.next() {
        sleep_seconds = arg.parse().unwrap_or_else(|_| {
            eprintln!("error: invalid number of seconds");
            std::process::exit(1);
        });
    }

    thread::sleep(Duration::from_secs(sleep_seconds));

    if fail {
        return Err(io::ErrorKind::Other.into());
    }
    if suicide {
        #[cfg(unix)]
        unsafe {
            let pid = signals::getpid();
            if pid > 0 {
                signals::kill(pid, 9);
            }
        }
    }

    Ok(())
}

#[cfg(unix)]
mod signals {
    use std::os::raw::c_int;
    extern "C" {
        pub fn kill(pid: c_int, sig: c_int) -> c_int;
        pub fn getpid() -> c_int;
    }
}