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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

use core::{
    fmt,
    future::Future,
    ops::{Deref, DerefMut},
    pin::Pin,
};
use std::{collections::HashMap, sync::Arc, time::SystemTime};

//
#[derive(Debug, Clone)]
#[non_exhaustive]
pub struct CallbackInfo {
    pub time: SystemTime,
}

impl Default for CallbackInfo {
    fn default() -> Self {
        Self {
            time: SystemTime::now(),
        }
    }
}

impl CallbackInfo {
    pub fn new() -> Self {
        Self::default()
    }

    pub fn time(&self) -> &SystemTime {
        &self.time
    }

    pub fn time_now() -> SystemTime {
        SystemTime::now()
    }
}

//
#[derive(Clone)]
pub enum Callback {
    Sync(Arc<dyn Fn(CallbackInfo) + Send + Sync + 'static>),
    #[allow(clippy::type_complexity)]
    Async(
        Arc<
            dyn Fn(CallbackInfo) -> Pin<Box<dyn Future<Output = ()> + Send + 'static>>
                + Send
                + Sync
                + 'static,
        >,
    ),
}

impl fmt::Debug for Callback {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::Sync(_) => write!(f, "Callback::Sync"),
            Self::Async(_) => write!(f, "Callback::Async"),
        }
    }
}

impl Callback {
    pub fn with_sync<F>(cb: F) -> Self
    where
        F: Fn(CallbackInfo) + Send + Sync + 'static,
    {
        Self::Sync(Arc::new(cb))
    }

    pub fn with_async<F>(cb: F) -> Self
    where
        F: Fn(CallbackInfo) -> Pin<Box<dyn Future<Output = ()> + Send + 'static>>
            + Send
            + Sync
            + 'static,
    {
        Self::Async(Arc::new(cb))
    }
}

//
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum CallbackType {
    Initialized,
    ReloadConfig,
    WaitForStop,
    PrintStats,
}

//
#[derive(Debug, Clone, Default)]
pub struct Callbacks(HashMap<CallbackType, Callback>);

impl Deref for Callbacks {
    type Target = HashMap<CallbackType, Callback>;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

impl DerefMut for Callbacks {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.0
    }
}

impl Callbacks {
    pub fn new() -> Self {
        Self::default()
    }

    pub fn into_inner(self) -> HashMap<CallbackType, Callback> {
        self.0
    }

    pub fn has_async(&self) -> bool {
        self.iter().any(|(_, cb)| matches!(cb, Callback::Async(_)))
    }
}