You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.6 KiB
Rust
55 lines
1.6 KiB
Rust
use crate::data::{get_version_specific_file, ENCHANTMENTS_FILE};
|
|
use crate::models::enchantment::Enchantment;
|
|
use crate::models::version::Version;
|
|
use crate::{DataError, DataResult};
|
|
use itertools::*;
|
|
use std::collections::HashMap;
|
|
use std::sync::Arc;
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub struct Enchantments {
|
|
version: Arc<Version>,
|
|
}
|
|
|
|
impl Enchantments {
|
|
pub fn new(version: Arc<Version>) -> Self {
|
|
Self { version }
|
|
}
|
|
|
|
/// Returns a list of enchantments
|
|
pub fn enchantments_array(&self) -> DataResult<Vec<Enchantment>> {
|
|
let content = get_version_specific_file(&self.version, ENCHANTMENTS_FILE)?;
|
|
|
|
serde_json::from_str::<Vec<Enchantment>>(&content).map_err(DataError::from)
|
|
}
|
|
|
|
/// Returns a map of enchantments indexed by ID
|
|
pub fn enchantments(&self) -> DataResult<HashMap<u32, Enchantment>> {
|
|
Ok(self
|
|
.enchantments_array()?
|
|
.into_iter()
|
|
.map(|e| (e.id, e))
|
|
.collect())
|
|
}
|
|
|
|
/// Returns a map of enchantments indexed by Name
|
|
pub fn enchantments_by_name(&self) -> DataResult<HashMap<String, Enchantment>> {
|
|
Ok(self
|
|
.enchantments_array()?
|
|
.into_iter()
|
|
.map(|e| (e.name.clone(), e))
|
|
.collect())
|
|
}
|
|
|
|
/// Returns enchantments grouped by category
|
|
pub fn enchantments_by_category(&self) -> DataResult<HashMap<String, Vec<Enchantment>>> {
|
|
Ok(self
|
|
.enchantments_array()?
|
|
.into_iter()
|
|
.group_by(|e| e.category.clone())
|
|
.into_iter()
|
|
.map(|(key, group)| (key, group.collect()))
|
|
.collect())
|
|
}
|
|
}
|