1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use std::io;
use std::path::Path;

use flate2::read::GzDecoder;
use lzf;

use crate::error::{CliResult, Error};

pub fn is_compressed(path: &Path) -> bool {
    let extension = path.extension().unwrap().to_str().unwrap();
    extension.eq("gz") || extension.eq("lzf")
}

pub fn decompress(path: &Path, reader: Box<dyn io::Read>) -> CliResult<Box<dyn io::Read>> {
    match path.extension().unwrap().to_str().unwrap() {
        "gz" => {
            let decoder = GzDecoder::new(reader);
            Ok(Box::new(decoder))
        },
        "lzf" => {
            let mut file = reader;
            let mut buf = Vec::new();
            file.read_to_end(&mut buf)?;
            let decompressed = lzf::decompress(&buf[..], buf.len() * 100).unwrap();
            let cursor = io::Cursor::new(decompressed);
            Ok(Box::new(cursor))
        },
        _ => Err(Error::from(format!("Cannot decompress file - {:?}", path)))
    }
}