notan_app 0.9.1

#![allow(clippy::wrong_self_convention, non_snake_case, clippy::type_complexity)]

use super::manager::Assets;
use super::storage::AssetStorage;
use crate::app::App;
use crate::graphics::Graphics;
use crate::plugins::Plugins;

use std::any::TypeId;
use std::rc::Rc;

pub(crate) type LoaderParams<'a, S> = (&'a mut App, &'a mut Graphics, &'a mut Plugins, &'a mut S);

/// Defines how parse files once they are loaded
#[derive(Default, Clone)]
pub struct AssetLoader {
    extensions: Vec<String>,
    parser: Option<LoaderCallback>,
    type_id: Option<TypeId>,
}

impl AssetLoader {
    /// Creates a new instance
    pub fn new() -> Self {
        Default::default()
    }

    /// Set the file extension that will be parsed
    pub fn extension(mut self, ext: &str) -> Self {
        self.extensions.push(ext.to_string());
        self
    }

    /// Set the file extensions that will be parsed
    pub fn extensions(mut self, exts: &[&str]) -> Self {
        for ext in exts {
            self.extensions.push(ext.to_string());
        }
        self
    }

    /// Set the parser function
    pub fn use_parser<H, A, Params>(mut self, handler: H) -> Self
    where
        H: LoaderHandler<A, Params>,
        A: Send + Sync + 'static,
    {
        self.type_id = Some(TypeId::of::<A>());
        self.parser = Some(handler.callback());
        self
    }

    pub(crate) fn apply(self, manager: &mut Assets) -> Result<(), String> {
        let AssetLoader {
            extensions,
            parser,
            type_id,
        } = self;

        if extensions.is_empty() {
            return Err("Loader without extensions associated.".to_string());
        }

        let type_id =
            type_id.ok_or_else(|| "Loader without output type associated.".to_string())?;
        let mut parser = parser.ok_or_else(|| "Loader without parser associated.".to_string())?;
        parser.set_type_id(type_id);

        extensions.iter().for_each(|ext| {
            manager.loaders.insert(ext.to_string(), parser.clone());
        });

        Ok(())
    }
}

#[derive(Clone)]
pub enum LoaderCallback {
    Basic(
        Option<TypeId>,
        Rc<dyn Fn(&mut AssetStorage, &str, Vec<u8>) -> Result<(), String>>,
    ),
    A(
        Option<TypeId>,
        Rc<dyn Fn(&mut AssetStorage, &str, Vec<u8>, &mut App) -> Result<(), String>>,
    ),
    G(
        Option<TypeId>,
        Rc<dyn Fn(&mut AssetStorage, &str, Vec<u8>, &mut Graphics) -> Result<(), String>>,
    ),
    GP(
        Option<TypeId>,
        Rc<
            dyn Fn(
                &mut AssetStorage,
                &str,
                Vec<u8>,
                &mut Graphics,
                &mut Plugins,
            ) -> Result<(), String>,
        >,
    ),
    P(
        Option<TypeId>,
        Rc<dyn Fn(&mut AssetStorage, &str, Vec<u8>, &mut Plugins) -> Result<(), String>>,
    ),
}

pub trait LoaderHandler<A, Params>
where
    A: Send + Sync,
{
    fn callback(self) -> LoaderCallback;
}

macro_rules! loader_handler {
    ($variant:expr, $($param:ident),*) => {
        #[allow(unused_parens)]
        impl<A, F> LoaderHandler<A, (&str, Vec<u8>, $(&mut $param),*)> for F
        where
            F: Fn(&str, Vec<u8>, $(&mut $param),*) -> Result<A, String> + 'static,
            A: Send + Sync + 'static

        {
            fn callback(self) -> LoaderCallback {
                $variant(None, Rc::new(move |storage, id, bytes, $($param),*| {
                    let asset = self(id, bytes, $($param),*)?;
                    storage.parse::<A>(id, asset)
                }))
            }
        }
    }
}

loader_handler!(LoaderCallback::Basic,);
loader_handler!(LoaderCallback::A, App);
loader_handler!(LoaderCallback::G, Graphics);
loader_handler!(LoaderCallback::GP, Graphics, Plugins);
loader_handler!(LoaderCallback::P, Plugins);

impl LoaderCallback {
    pub(crate) fn exec<S>(
        &self,
        id: &str,
        data: Vec<u8>,
        storage: &mut AssetStorage,
        params: &mut LoaderParams<S>,
    ) -> Result<(), String> {
        use LoaderCallback::*;

        let (app, graphics, plugins, _state) = params;

        match self {
            Basic(_, cb) => cb(storage, id, data),
            A(_, cb) => cb(storage, id, data, app),
            G(_, cb) => cb(storage, id, data, graphics),
            GP(_, cb) => cb(storage, id, data, graphics, plugins),
            P(_, cb) => cb(storage, id, data, plugins),
        }
    }

    pub(crate) fn set_type_id(&mut self, type_id: TypeId) {
        use LoaderCallback::*;
        let ty = match self {
            Basic(ref mut ty, _) => ty,
            A(ref mut ty, _) => ty,
            G(ref mut ty, _) => ty,
            GP(ref mut ty, _) => ty,
            P(ref mut ty, _) => ty,
        };

        *ty = Some(type_id);
    }

    pub(crate) fn type_id(&self) -> Option<TypeId> {
        use LoaderCallback::*;
        match self {
            Basic(ty, _) => *ty,
            A(ty, _) => *ty,
            G(ty, _) => *ty,
            GP(ty, _) => *ty,
            P(ty, _) => *ty,
        }
    }
}