Merge pull request #10 from cuklev/zstd

Support zstd compression

Author: desultory

pyproject.toml

@@ -17,7 +17,7 @@ classifiers = [
     "License :: OSI Approved :: GNU General Public License v2 (GPLv2)",
     "Operating System :: OS Independent",
 ]
-dependencies = ["zenlib >= 3.0.2"]
+dependencies = ["zenlib >= 3.0.2", "zstd >= 1.5.6.1"]
 
 [project.scripts]
 pycpio = "pycpio.main:main"

src/pycpio/writer/writer.py

@@ -55,6 +55,11 @@ def compress(self, data):
 
             self.logger.info("XZ compressing the CPIO data, original size: %.2f MiB" % (len(data) / (2**20)))
             data = lzma.compress(data, check=self.xz_crc)
+        elif self.compression == "zstd":
+            import zstd
+
+            self.logger.info("ZSTD compressing the CPIO data, original size: %.2f MiB" % (len(data) / (2**20)))
+            data = zstd.compress(data, 10)
         elif self.compression is not False:
             raise NotImplementedError("Compression type not supported: %s" % self.compression)
         return data