feat: add zstd support

AOSC-Dev · Jul 24, 2024 · b1e9b90 · b1e9b90
1 parent 3ee44c9
commit b1e9b90
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -46,6 +46,7 @@ flate2 = "1.0"
 tabwriter = { version = "^1", features = ["ansi_formatting"] }
 unsquashfs-wrapper = "0.2"
 inotify = "0.10"
+zstd = "0.13.2"
 
 [build-dependencies]
 clap = { version = "^4", features = ["string", "env"] }

diff --git a/src/repo/scan.rs b/src/repo/scan.rs
@@ -19,6 +19,7 @@ use xz2::read::XzDecoder;
 enum TarFormat {
     Xzip,
     Gzip,
+    Zstd,
 }
 
 fn collect_control<R: Read>(reader: R) -> Result<Vec<u8>> {
@@ -40,6 +41,7 @@ fn open_compressed_control<R: Read>(reader: R, format: &TarFormat) -> Result<Vec
     match format {
         TarFormat::Xzip => collect_control(XzDecoder::new(reader)),
         TarFormat::Gzip => collect_control(GzDecoder::new(reader)),
+        TarFormat::Zstd => collect_control(zstd::stream::read::Decoder::new(reader)?),
     }
 }
 
@@ -48,6 +50,8 @@ fn determine_format(format: &[u8]) -> Result<TarFormat> {
         Ok(TarFormat::Xzip)
     } else if format.ends_with(b".gz") {
         Ok(TarFormat::Gzip)
+    } else if format.ends_with(b".zst") {
+        Ok(TarFormat::Zstd)
     } else {
         Err(anyhow!("Unknown format: {:?}", format))
     }