Load a list of clients

src/data/entity_manager.rs

@@ -3,14 +3,16 @@ use std::slice::Iter;
 
 use crate::utils::err::Res;
 
+enum FileFormat { Json, Yaml }
+
 pub struct EntityManager<E> {
     file_path: PathBuf,
     list: Vec<E>,
 }
 
 impl<E> EntityManager<E>
-where
-    E: serde::Serialize + serde::de::DeserializeOwned + Eq + Clone,
+    where
+        E: serde::Serialize + serde::de::DeserializeOwned + Eq + Clone,
 {
     /// Open entity
     pub fn open_or_create<A: AsRef<Path>>(path: A) -> Res<Self> {
@@ -23,12 +25,34 @@ where
         }
 
         log::info!("Open existing entity file {:?}", path.as_ref());
+        let file_content = std::fs::read_to_string(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())?)?,
+            list: match Self::file_format(path.as_ref()) {
+                FileFormat::Json => serde_json::from_str(&file_content)?,
+                FileFormat::Yaml => serde_yaml::from_str(&file_content)?
+            },
         })
     }
 
+    /// Save the list
+    fn save(&self) -> Res {
+        Ok(std::fs::write(
+            &self.file_path,
+            match Self::file_format(self.file_path.as_ref()) {
+                FileFormat::Json => serde_json::to_string(&self.list)?,
+                FileFormat::Yaml => serde_yaml::to_string(&self.list)?,
+            },
+        )?)
+    }
+
+    fn file_format(p: &Path) -> FileFormat {
+        match p.to_string_lossy().ends_with(".json") {
+            true => FileFormat::Json,
+            false => FileFormat::Yaml
+        }
+    }
+
     /// Get the number of entries in the list
     pub fn len(&self) -> usize {
         self.list.len()
@@ -38,14 +62,6 @@ where
         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);
@@ -54,8 +70,8 @@ where
 
     /// Replace entries in the list that matches a criteria
     pub fn replace_entries<F>(&mut self, filter: F, el: &E) -> Res
-    where
-        F: Fn(&E) -> bool,
+        where
+            F: Fn(&E) -> bool,
     {
         for i in 0..self.list.len() {
             if filter(&self.list[i]) {
@@ -70,4 +86,9 @@ where
     pub fn iter(&self) -> Iter<'_, E> {
         self.list.iter()
     }
+
+    /// Get a cloned list of entries
+    pub fn cloned(&self) -> Vec<E> {
+        self.list.clone()
+    }
 }