bevy-hara-utils 0.1.7

Opinionated set of utilities
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 bevy::{
    ecs::{schedule::ScheduleLabel, system::ScheduleSystem},
    prelude::*,
};

pub trait AppStateExt {
    fn add_on_enter_state_systems<S: States, M>(
        &mut self,

        systems: impl IntoScheduleConfigs<ScheduleSystem, M>,
        state: S,
    ) -> &mut Self;

    fn add_on_exit_state_systems<S: States, M>(
        &mut self,

        systems: impl IntoScheduleConfigs<ScheduleSystem, M>,
        state: S,
    ) -> &mut Self;

    fn add_in_state_systems<S: States, M>(
        &mut self,
        schedule: impl ScheduleLabel,
        systems: impl IntoScheduleConfigs<ScheduleSystem, M>,
        state: S,
    ) -> &mut Self;
}

impl AppStateExt for App {
    fn add_on_enter_state_systems<S: States, M>(
        &mut self,

        systems: impl IntoScheduleConfigs<ScheduleSystem, M>,
        state: S,
    ) -> &mut Self {
        self.add_systems(OnEnter(state), systems)
    }

    fn add_on_exit_state_systems<S: States, M>(
        &mut self,

        systems: impl IntoScheduleConfigs<ScheduleSystem, M>,
        state: S,
    ) -> &mut Self {
        self.add_systems(OnExit(state), systems)
    }

    fn add_in_state_systems<S: States, M>(
        &mut self,
        schedule: impl ScheduleLabel,
        systems: impl IntoScheduleConfigs<ScheduleSystem, M>,
        state: S,
    ) -> &mut Self {
        self.add_systems(schedule, systems.run_if(in_state(state)))
    }
}