use bevy_app::{App, Plugin};
use bevy_asset::io::Reader;
use bevy_asset::{Asset, AssetApp, AssetLoader, LoadContext};
use bevy_reflect::TypePath;
use serde_ron::de::from_bytes;
use std::marker::PhantomData;
use thiserror::Error;
pub struct RonAssetPlugin<A> {
extensions: Vec<&'static str>,
_marker: PhantomData<A>,
}
impl<A> Plugin for RonAssetPlugin<A>
where
for<'de> A: serde::Deserialize<'de> + Asset,
{
fn build(&self, app: &mut App) {
app.init_asset::<A>()
.register_asset_loader(RonAssetLoader::<A> {
extensions: self.extensions.clone(),
_marker: PhantomData,
});
}
}
impl<A> RonAssetPlugin<A>
where
for<'de> A: serde::Deserialize<'de> + Asset,
{
pub fn new(extensions: &[&'static str]) -> Self {
Self {
extensions: extensions.to_owned(),
_marker: PhantomData,
}
}
}
#[derive(TypePath)]
pub struct RonAssetLoader<A> {
extensions: Vec<&'static str>,
_marker: PhantomData<A>,
}
#[non_exhaustive]
#[derive(Debug, Error)]
pub enum RonLoaderError {
#[error("Could not read the file: {0}")]
Io(#[from] std::io::Error),
#[error("Could not parse RON: {0}")]
RonError(#[from] serde_ron::error::SpannedError),
}
impl<A> AssetLoader for RonAssetLoader<A>
where
for<'de> A: serde::Deserialize<'de> + Asset,
{
type Asset = A;
type Settings = ();
type Error = RonLoaderError;
async fn load(
&self,
reader: &mut dyn Reader,
_settings: &(),
_load_context: &mut LoadContext<'_>,
) -> Result<Self::Asset, Self::Error> {
let mut bytes = Vec::new();
reader.read_to_end(&mut bytes).await?;
let asset = from_bytes::<A>(&bytes)?;
Ok(asset)
}
fn extensions(&self) -> &[&str] {
&self.extensions
}
}