69 lines
2.1 KiB
Rust
69 lines
2.1 KiB
Rust
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()
|
|
}
|
|
} |