PkmnLib_rs/src/static_data/libraries/item_library.rs

use crate::static_data::items::item::Item;
use crate::static_data::libraries::data_library::DataLibrary;
use std::collections::HashMap;

#[derive(Debug)]
pub struct ItemLibrary {
    map: HashMap<String, Item>,
    list: Vec<String>,
}

impl ItemLibrary {
    pub fn new(capacity: usize) -> ItemLibrary {
        ItemLibrary {
            map: HashMap::with_capacity(capacity),
            list: Vec::with_capacity(capacity),
        }
    }
}

impl DataLibrary<'_, Item> for ItemLibrary {
    fn map(&self) -> &HashMap<String, Item> {
        &self.map
    }

    fn list_values(&self) -> &Vec<String> {
        &self.list
    }

    fn get_modify(&mut self) -> (&mut HashMap<String, Item>, &mut Vec<String>) {
        (&mut self.map, &mut self.list)
    }
}

#[cfg(test)]
pub mod tests {
    use crate::static_data::items::item::Item;
    use crate::static_data::items::item_category::{BattleItemCategory, ItemCategory};
    use crate::static_data::libraries::data_library::DataLibrary;
    use crate::static_data::libraries::item_library::ItemLibrary;
    use std::collections::HashSet;

    fn build_item() -> Item {
        Item::new(
            "foo",
            ItemCategory::MiscItem,
            BattleItemCategory::MiscBattleItem,
            100,
            HashSet::new(),
        )
    }

    pub fn build() -> ItemLibrary {
        let mut lib = ItemLibrary::new(1);
        let m = build_item();
        // Borrow as mut so we can insert
        let w = &mut lib;
        w.add("foo", m);
        // Drops borrow as mut

        lib
    }
}