sproto 0.1.0

Rust client for the Synology Drive sync protocol
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

#[path = "support/mod.rs"]
mod support;

use anyhow::{Context as _, Result, bail};
use clap::Parser;
use indicatif::{ProgressBar, ProgressStyle};
use std::path::PathBuf;

use sproto::actions::list::FileType;

#[derive(Parser)]
#[command(
    name = "sproto-dl",
    about = "Download a single file from Synology Drive"
)]
struct Args {
    /// Remote file path (e.g. /Documents/report.pdf)
    remote_path: String,

    /// Local destination path
    local_path: PathBuf,

    #[command(flatten)]
    conn: support::ConnectionArgs,
}

#[tokio::main]
async fn main() -> Result<()> {
    support::init_tracing();
    let args = Args::parse();

    let client = args.conn.connect().await?;
    let view_id = args.conn.resolve_view(&client).await?;

    let (parent, filename) = match args.remote_path.rsplit_once('/') {
        Some(("", name)) => ("/", name),
        Some((parent, name)) => (parent, name),
        None => bail!("remote path must be absolute (start with /)"),
    };

    let nodes = client
        .list_dir(view_id, parent)
        .await
        .with_context(|| format!("failed to list '{parent}'"))?;

    let file = nodes
        .iter()
        .find(|n| n.name == filename && n.file_type == FileType::File)
        .with_context(|| format!("file '{filename}' not found in '{parent}'"))?;

    let pb = ProgressBar::new(file.file_size);
    pb.set_style(
        #[allow(
            clippy::literal_string_with_formatting_args,
            reason = "indicatif template syntax"
        )]
        ProgressStyle::default_bar()
            .template("{msg} [{bar:40}] {bytes}/{total_bytes} ({bytes_per_sec})")
            .unwrap()
            .progress_chars("=> "),
    );
    pb.set_message(filename.to_string());

    let mut writer = support::ProgressWriter::new(args.local_path.clone(), pb.clone());
    client
        .download_to(&file.file_id, &mut writer)
        .await
        .context("download failed")?;

    pb.finish_with_message(format!("Saved to {}", args.local_path.display()));

    Ok(())
}