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

use crate::emitters::Emitters;
use crate::resources::Resources;
use crate::states::States;
use async_trait::async_trait;
use core::future::Future;
use std::fmt::Debug;

pub type SystemResult = miette::Result<()>;

#[async_trait]
pub trait System: Debug + Send + Sync {
    async fn run(&self, states: States, resources: Resources, emitters: Emitters) -> SystemResult;
}

pub type BoxedSystem = Box<dyn System>;

#[async_trait]
pub trait SystemFunc: Send + Sync {
    async fn call(&self, states: States, resources: Resources, emitters: Emitters) -> SystemResult;
}

#[async_trait]
impl<T: Send + Sync, F> SystemFunc for T
where
    T: Fn(States, Resources, Emitters) -> F,
    F: Future<Output = SystemResult> + Send + 'static,
{
    async fn call(&self, states: States, resources: Resources, emitters: Emitters) -> SystemResult {
        self(states, resources, emitters).await
    }
}

pub struct CallbackSystem {
    func: Box<dyn SystemFunc>,
}

impl CallbackSystem {
    pub fn new<F: SystemFunc + 'static>(func: F) -> Self {
        Self {
            func: Box::new(func),
        }
    }
}

#[async_trait]
impl System for CallbackSystem {
    async fn run(&self, states: States, resources: Resources, emitters: Emitters) -> SystemResult {
        self.func.call(states, resources, emitters).await
    }
}

impl Debug for CallbackSystem {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("CallbackSystem").finish()
    }
}