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

//! This module contains all event related types.

use std::{boxed::Box, future::Future, pin::Pin};

#[cfg(not(feature = "threadsafe"))]
type EventHandler<'a, A> = Box<dyn FnMut(&A) -> Pin<Box<dyn Future<Output = ()> + 'a>> + 'a>;
#[cfg(feature = "threadsafe")]
type EventHandler<'a, A> = Box<dyn FnMut(&A) -> Pin<Box<dyn Future<Output = ()> + 'a>> + Send + 'a>;

pub struct Event<'a, A> {
	handlers: Vec<EventHandler<'a, A>>,
}

impl<'a, A> Event<'a, A> {
	/// Invokes the event, which calls all handlers that have been registered to
	/// this event.
	pub(crate) fn invoke(&mut self, args: &A) {
		for h in self.handlers.iter_mut() {
			h(args);
		}
	}

	/// Register a closure to be invoked for this event.
	#[cfg(not(feature = "threadsafe"))]
	pub fn register<H>(&mut self, mut handler: H)
	where
		H: FnMut(&A) + 'a,
	{
		self.handlers.push(Box::new(move |args| {
			handler(args);
			Box::pin(async {})
		}));
	}

	/// Register a closure to be invoked for this event.
	#[cfg(feature = "threadsafe")]
	pub fn register<H>(&mut self, mut handler: H)
	where
		H: FnMut(&A) + Send + 'a,
	{
		self.handlers.push(Box::new(move |args| {
			handler(args);
			Box::pin(async {})
		}));
	}

	/// Register an 'async closure' to be invoked for this event.
	///
	/// # Example
	/// ```ignore
	/// my_event.register_async(|args| async move {
	///     // Do something ...
	/// });
	/// ```
	#[cfg(not(feature = "threadsafe"))]
	pub fn register_async<H, F>(&mut self, mut handler: H)
	where
		H: FnMut(&A) -> F + 'a,
		F: Future<Output = ()> + 'a,
	{
		self.handlers
			.push(Box::new(move |args| Box::pin(handler(args))));
	}

	/// Register an 'async closure' to be invoked for this event.
	///
	/// # Example
	/// ```ignore
	/// my_event.register_async(|args| async move {
	///     // Do something ...
	/// });
	/// ```
	#[cfg(feature = "threadsafe")]
	pub fn register_async<H, F>(&mut self, mut handler: H)
	where
		H: FnMut(&A) -> F + Send + 'a,
		F: Future<Output = ()> + 'a,
	{
		self.handlers
			.push(Box::new(move |args| Box::pin(handler(args))));
	}
}

impl<'a, A> Default for Event<'a, A> {
	fn default() -> Self {
		Self {
			handlers: Vec::new(),
		}
	}
}