PkmnLib_rs/src/static_data/libraries/ability_library.rs

use std::sync::Arc;

use indexmap::IndexMap;

use crate::static_data::Ability;
use crate::static_data::DataLibrary;
use crate::{StringKey, ValueIdentifiable, ValueIdentifier};

/// A storage for all abilities that can be used in this data library.
#[derive(Debug)]
pub struct AbilityLibrary {
    /// A unique identifier so we know what value this is.
    identifier: ValueIdentifier,
    /// The underlying map for the library.
    map: IndexMap<StringKey, Arc<dyn Ability>>,
}

impl AbilityLibrary {
    /// Instantiates a new ability library.
    pub fn new(capacity: usize) -> AbilityLibrary {
        AbilityLibrary {
            identifier: Default::default(),
            map: IndexMap::with_capacity(capacity),
        }
    }
}

impl DataLibrary<dyn Ability> for AbilityLibrary {
    fn map(&self) -> &IndexMap<StringKey, Arc<dyn Ability>> {
        &self.map
    }
    fn get_modify(&mut self) -> &mut IndexMap<StringKey, Arc<dyn Ability>> {
        &mut self.map
    }
}

impl ValueIdentifiable for AbilityLibrary {
    fn value_identifier(&self) -> ValueIdentifier {
        self.identifier
    }
}

#[cfg(test)]
pub mod tests {
    use crate::static_data::AbilityImpl;
    use crate::static_data::AbilityLibrary;
    use crate::static_data::DataLibrary;
    use crate::StringKey;
    use std::sync::Arc;

    pub fn build() -> AbilityLibrary {
        let mut lib = AbilityLibrary::new(1);
        lib.add(
            &StringKey::new("test_ability"),
            Arc::new(AbilityImpl::new(
                &"test_ability".into(),
                &"test_ability".into(),
                Vec::new(),
            )),
        );
        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 ability = lib.get(&"test_ability".into());
        assert!(ability.is_some());
        assert_eq!(ability.unwrap().name(), &"test_ability".into());
    }
}