zlib_cli 0.1.0

A CLI tool for compress and decompress zlib file
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

use std::{fs::File, path::PathBuf};
use flate2::Compression;
use flate2::write::ZlibEncoder;
use flate2::read::ZlibDecoder;
use clap::{App, Arg};

#[allow(warnings)]
fn main() -> std::io::Result<()> {
    let matches = App::new("Zlib compress/decompress")
    .version("1.0")
    .author("Addrew Ryan <dnrops@outlook.com>")
    .about("A CLI tool for compress and decompress zlib file")
    .arg(
        Arg::with_name("compress")
            .short("c")
            .long("compress")
            .value_name("STRING")
            .help("compress a string in zlib file")
            .takes_value(true),
    )
    .arg(
        Arg::with_name("decompress")
            .short("d")
            .long("decompress")
            .value_name("STRING")
            .help("decompress a zlib file to data")
            .takes_value(true),
    )
    .get_matches();


    if let Some(input) = matches.value_of("compress") {
        compress(input.into())?;
    } else if let Some(input) = matches.value_of("decompress") {
        decompress(input.into())?;
    }

    fn compress(path:PathBuf)->std::io::Result<()>{
            // 压缩数据
            let mut input_file = File::open(path)?;
            let output_file = File::create("data.zlib")?;
            let mut encoder = ZlibEncoder::new(output_file, Compression::default());
            std::io::copy(&mut input_file, &mut encoder)?;
            encoder.finish()?;
            Ok(())
    }

    fn decompress(path:PathBuf)->std::io::Result<()>{
            // 解压缩数据
            let input_file = File::open(path)?;
            let mut output_file = File::create("data_new.txt")?;
            let mut decoder = ZlibDecoder::new(input_file);
            std::io::copy(&mut decoder, &mut output_file)?;
         Ok(())
    }
    Ok(())
}