kutil 0.0.6

Kutil utilities collection
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

use std::{process::*, sync::*};

/// Call a hook on exit.
///
/// Hooks will be called in the reverse order in which they are added.
pub fn on_exit<HookT>(hook: HookT)
where
    HookT: FnMut() + Send + 'static,
{
    let hook = Box::new(hook);

    let mut hooks = EXIT_HOOKS.lock().expect("EXIT_HOOKS");
    match hooks.as_mut() {
        Some(hooks) => {
            hooks.push(hook);
        }

        None => {
            let mut new_hooks = Vec::<ExitHook>::default();
            new_hooks.push(hook);
            *hooks = Some(new_hooks);
            set_ctrlc_handler();
        }
    }
}

//
// ExitHook
//

type ExitHook = Box<dyn FnMut() + Send>;

type Static<StaticT> = LazyLock<Mutex<Option<StaticT>>>;

static EXIT_HOOKS: Static<Vec<ExitHook>> = LazyLock::new(|| Default::default());

// ctrlc

// Can only be called once!
fn set_ctrlc_handler() {
    ctrlc::set_handler(ctrlc_handler).expect("ctrlc::set_handler");
}

fn ctrlc_handler() {
    if let Some(hooks) = EXIT_HOOKS.lock().expect("EXIT_HOOKS").take() {
        for mut hook in hooks.into_iter().rev() {
            hook();
        }
    }

    exit(130);
}