lyra-engine/lyra-ecs/src/resource.rs

use std::{any::{TypeId, Any}, cell::{RefCell, Ref, RefMut}};

/// Shorthand for `Send + Sync + 'static`, so it never needs to be implemented manually.
pub trait ResourceObject: 'static {}
impl<T: 'static> ResourceObject for T {}

/// A type erased storage for a Resource.
pub struct ResourceData {
    pub(crate) data: Box<RefCell<dyn Any>>,
    type_id: TypeId,
}

impl ResourceData {
    pub fn new<T: Any>(data: T) -> Self {

        Self {
            data: Box::new(RefCell::new(data)),
            type_id: TypeId::of::<T>(),
        }
    }

    /// Returns a boolean indicating whether or not `T`` is of the same type of the Resource
    pub fn is<T: 'static>(&self) -> bool {
        self.type_id == TypeId::of::<T>()
    }

    /// Borrow the data inside of the resource.
    /// 
    /// # Panics
    /// 
    /// * If the data is already borrowed mutably, this will panic.
    /// * If the type of `T` is not the same as the resource type.
    pub fn get<T: 'static>(&self) -> Ref<T> {
        Ref::map(self.data.borrow(), |a| a.downcast_ref().unwrap())
    }

    /// Mutably borrow the data inside of the resource.
    /// 
    /// # Panics
    /// 
    /// * If the data is already borrowed mutably, this will panic.
    /// * If the type of `T` is not the same as the resource type.
    pub fn get_mut<T: 'static>(&self) -> RefMut<T> {
        RefMut::map(self.data.borrow_mut(), |a| a.downcast_mut().unwrap())
    }

    /// Borrow the data inside of the resource.
    /// 
    /// # Panics
    /// 
    /// * If the type of `T` is not the same as the resource type.
    pub fn try_get<T: 'static>(&self) -> Option<Ref<T>> {
        
        self.data.try_borrow()
            .map(|r| Ref::map(r, |a| a.downcast_ref().unwrap()))
            .ok()
    }

    /// Mutably borrow the data inside of the resource.
    /// 
    /// # Panics
    /// 
    /// * If the type of `T` is not the same as the resource type.
    pub fn try_get_mut<T: 'static>(&self) -> Option<RefMut<T>> {
        self.data.try_borrow_mut()
            .map(|r| RefMut::map(r, |a| a.downcast_mut().unwrap()))
            .ok()
    }
}