iforgor 0.3.2

The CLI tool for all those commands you forget about
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

use std::sync::{Mutex, OnceLock};

#[derive(Clone, Copy)]
pub enum Mode {
    Ignore,
    Kill,
}

pub fn set_mode(mode: Mode) {
    static MODE: OnceLock<Mutex<Mode>> = OnceLock::new();
    let mutex = MODE.get_or_init(|| {
        // One first change we setup or custom Ctrl+C handler as it should
        // be setuped once.
        ctrlc::set_handler(|| {
            match *MODE
                .get()
                .expect("static MODE to be initialized")
                .lock()
                .unwrap_or_else(|e| e.into_inner())
            {
                Mode::Ignore => (),
                Mode::Kill => std::process::exit(1),
            }
        })
        .expect("Error setting Ctrl-C handler");

        Mutex::new(mode)
    });

    *mutex.lock().unwrap_or_else(|e| e.into_inner()) = mode;
}