use std::any::{Any, TypeId};
use std::collections::HashMap;
#[derive(Debug)]
pub struct Resources {
resources: HashMap<TypeId, Box<dyn Any>>,
}
impl Resources {
pub fn new() -> Self {
Self {
resources: HashMap::new(),
}
}
pub fn insert<T: 'static>(&mut self, resource: T) {
self.resources.insert(TypeId::of::<T>(), Box::new(resource));
}
pub fn get<T: 'static>(&self) -> Option<&T> {
self.resources
.get(&TypeId::of::<T>())
.and_then(|resource| resource.downcast_ref::<T>())
}
pub fn get_mut<T: 'static>(&mut self) -> Option<&mut T> {
self.resources
.get_mut(&TypeId::of::<T>())
.and_then(|resource| resource.downcast_mut::<T>())
}
pub fn remove<T: 'static>(&mut self) -> Option<T> {
self.resources
.remove(&TypeId::of::<T>())
.and_then(|resource| resource.downcast::<T>().ok())
.map(|boxed| *boxed)
}
pub fn contains<T: 'static>(&self) -> bool {
self.resources.contains_key(&TypeId::of::<T>())
}
}
impl Default for Resources {
fn default() -> Self {
Self::new()
}
}