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

pub mod components;
pub mod resources;
pub mod systems;

pub mod prelude {
    pub use crate::{components::*, resources::*, systems::*, AiSystemInstallerSetup};
}

pub use emergent;

use crate::{
    components::AiInstance,
    resources::AiBehaviors,
    systems::{ai_system, AiSystemResources},
};
use oxygengine_core::{
    app::AppBuilder,
    ecs::{
        pipeline::{PipelineBuilder, PipelineBuilderError},
        Component,
    },
    prefab::PrefabManager,
};

pub struct AiSystemInstallerSetup<'a, C>
where
    C: Component,
{
    pub postfix: &'a str,
    pub behaviors: AiBehaviors<C>,
}

pub fn ai_system_installer<PB, C>(
    builder: &mut AppBuilder<PB>,
    setup: AiSystemInstallerSetup<C>,
) -> Result<(), PipelineBuilderError>
where
    PB: PipelineBuilder,
    C: Component,
{
    builder.install_resource(setup.behaviors);
    builder.install_system::<AiSystemResources<C>>(
        &format!("ai-system-{}", setup.postfix),
        ai_system::<C>,
        &[],
    )?;
    Ok(())
}

pub fn ai_prefab_installer<C>(postfix: &str, prefabs: &mut PrefabManager)
where
    C: Component + Default,
{
    prefabs.register_component_factory::<AiInstance<C>>(&format!("AiInstance-{}", postfix));
}