2022-08-20 11:17:20 +00:00
|
|
|
use std::sync::Arc;
|
|
|
|
|
2022-07-01 15:07:22 +00:00
|
|
|
use indexmap::IndexMap;
|
2022-06-19 19:34:08 +00:00
|
|
|
|
2022-06-11 18:51:37 +00:00
|
|
|
use crate::static_data::Ability;
|
|
|
|
use crate::static_data::DataLibrary;
|
2022-10-08 11:15:04 +00:00
|
|
|
use crate::{StringKey, ValueIdentifiable, ValueIdentifier};
|
2022-06-11 15:22:46 +00:00
|
|
|
|
2022-07-01 15:07:22 +00:00
|
|
|
/// A storage for all abilities that can be used in this data library.
|
2022-06-11 15:22:46 +00:00
|
|
|
#[derive(Debug)]
|
|
|
|
pub struct AbilityLibrary {
|
2022-10-08 11:15:04 +00:00
|
|
|
/// A unique identifier so we know what value this is.
|
|
|
|
identifier: ValueIdentifier,
|
2022-07-01 15:07:22 +00:00
|
|
|
/// The underlying map for the library.
|
2022-08-20 11:17:20 +00:00
|
|
|
map: IndexMap<StringKey, Arc<Ability>>,
|
2022-06-11 15:22:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
impl AbilityLibrary {
|
2022-07-01 15:07:22 +00:00
|
|
|
/// Instantiates a new ability library.
|
2022-06-11 15:22:46 +00:00
|
|
|
pub fn new(capacity: usize) -> AbilityLibrary {
|
|
|
|
AbilityLibrary {
|
2022-10-08 11:15:04 +00:00
|
|
|
identifier: Default::default(),
|
2022-07-01 15:07:22 +00:00
|
|
|
map: IndexMap::with_capacity(capacity),
|
2022-06-11 15:22:46 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-08-20 11:17:20 +00:00
|
|
|
impl DataLibrary<'_, Ability> for AbilityLibrary {
|
|
|
|
fn map(&self) -> &IndexMap<StringKey, Arc<Ability>> {
|
2022-06-11 15:22:46 +00:00
|
|
|
&self.map
|
|
|
|
}
|
2022-08-20 11:17:20 +00:00
|
|
|
fn get_modify(&mut self) -> &mut IndexMap<StringKey, Arc<Ability>> {
|
2022-07-01 15:07:22 +00:00
|
|
|
&mut self.map
|
2022-06-11 15:22:46 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-10-08 11:15:04 +00:00
|
|
|
impl ValueIdentifiable for AbilityLibrary {
|
|
|
|
fn value_identifier(&self) -> ValueIdentifier {
|
|
|
|
self.identifier
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-06-11 15:22:46 +00:00
|
|
|
#[cfg(test)]
|
|
|
|
pub mod tests {
|
2022-06-19 19:34:08 +00:00
|
|
|
use crate::static_data::Ability;
|
|
|
|
use crate::static_data::AbilityLibrary;
|
|
|
|
use crate::static_data::DataLibrary;
|
2022-06-11 15:22:46 +00:00
|
|
|
use crate::StringKey;
|
2022-10-08 11:15:04 +00:00
|
|
|
use std::sync::Arc;
|
2022-06-11 15:22:46 +00:00
|
|
|
|
|
|
|
pub fn build() -> AbilityLibrary {
|
|
|
|
let mut lib = AbilityLibrary::new(1);
|
|
|
|
lib.add(
|
2022-10-14 14:53:30 +00:00
|
|
|
&StringKey::new("test_ability"),
|
2022-10-08 11:15:04 +00:00
|
|
|
Arc::new(Ability::new(&"test_ability".into(), &"test_ability".into(), Vec::new())),
|
2022-06-11 15:22:46 +00:00
|
|
|
);
|
|
|
|
lib
|
|
|
|
}
|
2022-06-11 18:51:37 +00:00
|
|
|
|
|
|
|
#[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();
|
2022-07-01 15:07:22 +00:00
|
|
|
let ability = lib.get(&"test_ability".into());
|
2022-06-11 18:51:37 +00:00
|
|
|
assert!(ability.is_some());
|
|
|
|
assert_eq!(ability.unwrap().name(), &"test_ability".into());
|
|
|
|
}
|
2022-06-11 15:22:46 +00:00
|
|
|
}
|