1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

use std::cell::RefCell;
use std::collections::HashMap;
use std::rc::Rc;

use eyre::Result;

use crate::errors::BbEcsError;

use super::resource::Resource;

pub struct ResourcesData {
    resources: HashMap<String, Rc<RefCell<Resource>>>,
}

impl ResourcesData {
    pub fn new() -> Self {
        Self::default()
    }

    /// Inserts a resource into storage, overwriting any resource that had the same name
    pub fn insert(&mut self, name: String, resource: Resource) {
        self.resources.insert(name, Rc::new(RefCell::new(resource)));
    }

    pub fn get(&self, name: &str) -> Result<&Rc<RefCell<Resource>>> {
        if let Some(resource) = self.resources.get(name) {
            Ok(resource)
        } else {
            Err(BbEcsError::ResourceNotFound(name.to_owned()).into())
        }
    }
}

impl Default for ResourcesData {
    fn default() -> Self {
        Self {
            resources: HashMap::new(),
        }
    }
}