use crate::static_data::libraries::data_library::DataLibrary;
use crate::static_data::species_data::ability::Ability;
use crate::StringKey;
use hashbrown::HashMap;

#[derive(Debug)]
pub struct AbilityLibrary {
    map: HashMap<StringKey, Box<Ability>>,
    list: Vec<StringKey>,
}

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

impl DataLibrary<'_, Box<Ability>> for AbilityLibrary {
    fn map(&self) -> &HashMap<StringKey, Box<Ability>> {
        &self.map
    }

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

    fn get_modify(&mut self) -> (&mut HashMap<StringKey, Box<Ability>>, &mut Vec<StringKey>) {
        (&mut self.map, &mut self.list)
    }
}

#[cfg(test)]
pub mod tests {
    use crate::static_data::libraries::ability_library::AbilityLibrary;
    use crate::static_data::libraries::data_library::DataLibrary;
    use crate::static_data::species_data::ability::Ability;
    use crate::StringKey;

    pub fn build() -> AbilityLibrary {
        let mut lib = AbilityLibrary::new(1);
        lib.add(
            &StringKey::new("test_ability"),
            Box::new(Ability::new(
                &"test_ability".into(),
                &"test_ability".into(),
                Vec::new(),
            )),
        );
        // Drops borrow as mut

        lib
    }
}