53 lines
1.4 KiB
Rust
53 lines
1.4 KiB
Rust
use std::path::{Path, PathBuf};
|
|
use std::slice::Iter;
|
|
|
|
use crate::utils::err::Res;
|
|
|
|
pub struct EntityManager<E> {
|
|
file_path: PathBuf,
|
|
list: Vec<E>,
|
|
}
|
|
|
|
impl<E> EntityManager<E> where E: serde::Serialize + serde::de::DeserializeOwned + Eq + Clone {
|
|
/// Open entity
|
|
pub fn open_or_create<A: AsRef<Path>>(path: A) -> Res<Self> {
|
|
if !path.as_ref().is_file() {
|
|
log::warn!("Entities at {:?} does not point to a file, creating a new empty entity container...", path.as_ref());
|
|
return Ok(Self {
|
|
file_path: path.as_ref().to_path_buf(),
|
|
list: vec![],
|
|
});
|
|
}
|
|
|
|
log::info!("Open existing entity file {:?}", path.as_ref());
|
|
Ok(Self {
|
|
file_path: path.as_ref().to_path_buf(),
|
|
list: serde_json::from_str(&std::fs::read_to_string(path.as_ref())?)?,
|
|
})
|
|
}
|
|
|
|
/// Get the number of entries in the list
|
|
pub fn len(&self) -> usize {
|
|
self.list.len()
|
|
}
|
|
|
|
pub fn is_empty(&self) -> bool {
|
|
self.len() == 0
|
|
}
|
|
|
|
/// Save the list
|
|
fn save(&self) -> Res {
|
|
Ok(std::fs::write(&self.file_path, serde_json::to_string(&self.list)?)?)
|
|
}
|
|
|
|
/// Insert a new element in the list
|
|
pub fn insert(&mut self, el: E) -> Res {
|
|
self.list.push(el);
|
|
self.save()
|
|
}
|
|
|
|
/// Iterate over the entries of this entity manager
|
|
pub fn iter(&self) -> Iter<'_, E> {
|
|
self.list.iter()
|
|
}
|
|
} |