zsync-rs 0.1.3

Efficient file transfer using rsync algorithm over HTTP
Documentation 
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

use std::fs::File;
use std::io::BufWriter;
use std::path::PathBuf;

use clap::Parser;
use zsync_rs::ControlFile;

#[derive(Parser)]
#[command(name = "zsyncmake")]
#[command(about = "Generate zsync control files for efficient delta transfers")]
struct Cli {
    /// Input file to generate zsync data for
    #[arg(value_name = "FILE")]
    file: PathBuf,

    /// URL for the file (default: filename)
    #[arg(short = 'u', value_name = "URL")]
    url: Option<String>,

    /// Output .zsync file (default: <input>.zsync)
    #[arg(short = 'o', value_name = "FILE")]
    output: Option<PathBuf>,

    /// Block size (default: auto-calculated)
    #[arg(short = 'b', value_name = "BLOCKSIZE")]
    blocksize: Option<usize>,
}

fn main() {
    let cli = Cli::parse();

    let filename = cli
        .file
        .file_name()
        .unwrap_or_default()
        .to_string_lossy()
        .to_string();

    let url = cli.url.unwrap_or_else(|| filename.clone());
    let output_path = cli
        .output
        .unwrap_or_else(|| PathBuf::from(format!("{filename}.zsync")));

    let mut input = match File::open(&cli.file) {
        Ok(f) => f,
        Err(e) => {
            eprintln!("Error opening {}: {e}", cli.file.display());
            std::process::exit(1);
        }
    };

    eprintln!("Reading {}...", cli.file.display());

    let control = match ControlFile::generate(&mut input, &filename, &url, cli.blocksize) {
        Ok(cf) => cf,
        Err(e) => {
            eprintln!("Error generating control file: {e}");
            std::process::exit(1);
        }
    };

    let output = match File::create(&output_path) {
        Ok(f) => f,
        Err(e) => {
            eprintln!("Error creating {}: {e}", output_path.display());
            std::process::exit(1);
        }
    };

    if let Err(e) = control.write(&mut BufWriter::new(output)) {
        eprintln!("Error writing control file: {e}");
        std::process::exit(1);
    }

    eprintln!(
        "Wrote {} ({} blocks, blocksize {})",
        output_path.display(),
        control.num_blocks(),
        control.blocksize
    );
}