use hashbrown::HashMap;

use crate::static_data::Ability;
use crate::static_data::DataLibrary;
use crate::StringKey;

#[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::Ability;
    use crate::static_data::AbilityLibrary;
    use crate::static_data::DataLibrary;
    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
    }

    #[test]
    fn get_ability_library_access() {
        let lib = build();
        let ability = lib.get(&"test_ability".into());
        assert!(ability.is_some());
        assert_eq!(ability.unwrap().name(), &"test_ability".into());
    }

    #[test]
    fn get_ability_library_direct_map_access() {
        let lib = build();
        let map = lib.map();
        let ability = map.get(&"test_ability".into());
        assert!(ability.is_some());
        assert_eq!(ability.unwrap().name(), &"test_ability".into());
    }

    #[test]
    fn get_ability_library_direct_list_access() {
        let lib = build();
        let list = lib.list_values();
        assert_eq!(list.len(), 1);
        assert!(list.contains(&StringKey::new("test_ability")));
    }
}