1 files changed, 41 insertions, 0 deletions
diff --git a/vendor/ruzstd/src/bin/zstd_stream.rs b/vendor/ruzstd/src/bin/zstd_stream.rs
new file mode 100644
index 000000000..cb2ebc76b
--- /dev/null
+++ b/vendor/ruzstd/src/bin/zstd_stream.rs
@@ -0,0 +1,41 @@
+extern crate ruzstd;
+use std::fs::File;
+use std::io::{Read, Write};
+
+fn main() {
+    let mut file_paths: Vec<_> = std::env::args().filter(|f| !f.starts_with('-')).collect();
+    let flags: Vec<_> = std::env::args().filter(|f| f.starts_with('-')).collect();
+    file_paths.remove(0);
+
+    if !flags.contains(&"-d".to_owned()) {
+        eprintln!("This zstd implementation only supports decompression. Please add a \"-d\" flag");
+        return;
+    }
+
+    if !flags.contains(&"-c".to_owned()) {
+        eprintln!("This zstd implementation only supports output on the stdout. Please add a \"-c\" flag and pipe the output into a file");
+        return;
+    }
+
+    if flags.len() != 2 {
+        eprintln!(
+            "No flags other than -d and -c are currently implemented. Flags used: {:?}",
+            flags
+        );
+        return;
+    }
+
+    for path in file_paths {
+        eprintln!("File: {}", path);
+        let f = File::open(path).unwrap();
+        let mut buf_read = std::io::BufReader::new(f);
+
+        let mut decoder = ruzstd::StreamingDecoder::new(&mut buf_read).unwrap();
+        let mut buf = [0u8; 1024 * 1024];
+        let mut stdout = std::io::stdout();
+        while !decoder.decoder.is_finished() || decoder.decoder.can_collect() > 0 {
+            let bytes = decoder.read(&mut buf[..]).unwrap();
+            stdout.write_all(&buf[..bytes]).unwrap();
+        }
+    }
+}