change: Switch from flate to zstd

pueue/Cargo.toml

@@ -36,7 +36,6 @@ color-eyre.workspace = true
 comfy-table = "7"
 command-group.workspace = true
 ctrlc = { version = "3", features = ["termination"] }
-flate2 = "1"
 handlebars.workspace = true
 interim = { version = "0.2.0", features = ["chrono_0_4"] }
 pest = "2.7"
@@ -58,6 +57,7 @@ tracing-subscriber = { version = "0.3.19", features = [
     "fmt",
     "local-time",
 ] }
+zstd = "0.13"
 
 [dev-dependencies]
 assert_cmd = "2"

pueue/src/daemon/internal_state/state.rs

@@ -1,13 +1,11 @@
 use std::{
     collections::BTreeMap,
     fs::{File, read_to_string},
-    io::{Read, Write},
     process::Child,
     sync::{Arc, Mutex, MutexGuard},
 };
 
 use chrono::Local;
-use flate2::Compression;
 use pueue_lib::{
     Group, GroupStatus, Settings, State, TaskResult,
     error::Error,
@@ -258,14 +256,11 @@ impl InternalState {
             temp = path.join("state.json.gz.partial");
             real = path.join("state.json.gz");
 
-            let file = if temp.exists() {
-                File::open(&temp)?
-            } else {
-                File::create(&temp)?
-            };
+            let compressed = zstd::stream::encode_all(serialized.as_bytes(), 3)
+                .context("failed to compress state")?;
 
-            let mut encoder = flate2::write::GzEncoder::new(file, Compression::default());
-            encoder.write_all(serialized.as_bytes())?;
+            std::fs::write(&temp, compressed)
+                .context("Failed to write compressed state to temp file while saving state.")?;
         } else {
             // Write to temporary log file first, to prevent loss due to crashes.
             std::fs::write(&temp, serialized)
@@ -303,12 +298,12 @@ impl InternalState {
 
         // Try to load the file.
         let data = if settings.daemon.compress_state_file {
-            let file = File::open(path)?;
-            let mut decoder = flate2::read::GzDecoder::new(file);
-            let mut data = String::new();
-            decoder.read_to_string(&mut data)?;
+            let f = File::open(&path).context("State restore: Failed to open file:\n\n{}")?;
+
+            let bytes =
+                zstd::stream::decode_all(f).context("State restore: Failed to decompress file")?;
 
-            data
+            String::from_utf8(bytes).context("State restore: Failed to parse state as UTF-8")?
         } else {
             read_to_string(&path).context("State restore: Failed to read file:\n\n{}")?
         };