beet_flow 0.0.8

An ECS control flow library
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

//! Actions related to the loading and handling of bevy assets.

use std::marker::PhantomData;

use crate::prelude::*;
use beet_core::prelude::*;

/// A plugin that registers the [`RunOnAssetReady`] system.
pub struct RunOnAssetReadyPlugin<A: Asset>(PhantomData<A>);

impl<A: Asset> Default for RunOnAssetReadyPlugin<A> {
	fn default() -> Self { Self(PhantomData) }
}

impl<A: Asset> Plugin for RunOnAssetReadyPlugin<A> {
	fn build(&self, app: &mut App) {
		app.add_systems(Update, run_on_asset_ready::<A>);
	}
}

/// An action that will trigger [`Run`] when an asset with
/// the provided handle is loaded.
/// ## Warning
/// The [`RunOnAssetReadyPlugin`] must be registered with matching
/// generic parameters for this action to work.
// TODO refactor to ready_on_asset_load
#[derive(Debug, Component)]
pub struct RunOnAssetReady<A: Asset> {
	/// The handle of the asset to wait for.
	pub handle: Handle<A>,
}

impl<A: Asset> RunOnAssetReady<A> {
	/// Create a new [`RunOnAssetReady`] action with the provided handle.
	pub fn new(handle: Handle<A>) -> Self { Self { handle } }
}

fn run_on_asset_ready<A: Asset>(
	mut asset_events: MessageReader<AssetEvent<A>>,
	mut commands: Commands,
	query: Query<(Entity, &RunOnAssetReady<A>)>,
) {
	for ev in asset_events.read() {
		match ev {
			AssetEvent::LoadedWithDependencies { id } => {
				for (entity, run_on_ready) in query.iter() {
					if run_on_ready.handle.id() == *id {
						commands
							.entity(entity)
							.remove::<RunOnAssetReady<A>>()
							.trigger_target(GetOutcome);
					}
				}
			}
			_ => {}
		}
	}
}