use crate::dynamic_asset::DynamicAssets;
use bevy_app::App;
use bevy_asset::UntypedHandle;
use bevy_ecs::{resource::Resource, world::World};
pub use bevy_asset_loader_derive::AssetCollection;
#[diagnostic::on_unimplemented(
message = "`{Self}` is not an `AssetCollection`",
label = "invalid `AssetCollection`",
note = "consider annotating `{Self}` with `#[derive(AssetCollection)]`"
)]
pub trait AssetCollection: Resource {
fn create(world: &mut World) -> Self;
fn load(world: &mut World) -> Vec<UntypedHandle>;
}
pub trait AssetCollectionApp {
fn init_collection<A: AssetCollection>(&mut self) -> &mut Self;
}
impl AssetCollectionApp for App {
fn init_collection<Collection>(&mut self) -> &mut Self
where
Collection: AssetCollection,
{
if !self.world().contains_resource::<Collection>() {
self.init_resource::<DynamicAssets>();
let _ = Collection::load(self.world_mut());
let resource = Collection::create(self.world_mut());
self.insert_resource(resource);
}
self
}
}
pub trait AssetCollectionWorld {
fn init_collection<A: AssetCollection>(&mut self);
}
impl AssetCollectionWorld for World {
fn init_collection<A: AssetCollection>(&mut self) {
if self.get_resource::<A>().is_none() {
self.init_resource::<DynamicAssets>();
let collection = A::create(self);
self.insert_resource(collection);
}
}
}