adding compression

1 files changed, 7 insertions, 3 deletions

diff --git a/src/state.rs b/src/state.rs
index a04412f..425e5d7 100644
--- a/src/state.rs
+++ b/src/state.rs
@@ -3,11 +3,12 @@ use directories::ProjectDirs;
 use expanduser::expanduser;
 use serde::{Deserialize, Serialize};
 use walkdir::WalkDir;
+use zstd;
 
 use crate::{config::Config, learner::Learner, trajectory::Trajectory};
 
 fn default_state_path() -> String {
-    let name = "mood.bin";
+    let name = "mood.bin.zstd";
     ProjectDirs::from("qualifier", "organisation", "mood")
         .and_then(|pd| pd.config_dir().join(name).to_str().map(|x| x.to_string()))
         .unwrap_or(format!("~/.config/mood/{name}"))
@@ -57,7 +58,8 @@ impl State {
         expanduser(default_state_path())
             .map_err(|e| e.to_string())
             .and_then(|p| fs::read(p).map_err(|e| e.to_string()))
-            .and_then(|contents| bincode::deserialize(&contents).map_err(|e| e.to_string()))
+            .and_then(|contents| zstd::decode_all(&contents[..]).map_err(|e| e.to_string()))
+            .and_then(|decompressed| bincode::deserialize(&decompressed).map_err(|e| e.to_string()))
     }
 
     pub fn try_save(&self) -> Result<(), String> {
@@ -66,7 +68,9 @@ impl State {
         _ = path.parent().map(std::fs::create_dir_all);
         let serialised =
             bincode::serialize(self).map_err(|e| format!("Failed to serialize state: {}", e))?;
-        fs::write(&path, serialised)
+        let compressed = zstd::encode_all(serialised.as_slice(), 3)
+            .map_err(|e| format!("Failed to compress state: {}", e))?;
+        fs::write(&path, compressed)
             .map_err(|e| format!("Failed to write state file at {}: {}", str_path, e))?;
         Ok(())
     }