doclib 0.1.0

Tag-indexed CLI catalog for books and articles, with rclone-backed storage
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

//! Everything that knows rclone exists lives here. Swapping in a native S3
//! client later means rewriting this file and nothing else.

use anyhow::{Context, Result, bail};
use std::path::Path;
use std::process::{Command, Stdio};

use crate::config::Config;

/// Verify both halves of the setup before any work begins: the binary is
/// installed, and the configured remote means what the user thinks it means.
pub fn check_ready(cfg: &Config) -> Result<()> {
    check_available()?;
    cfg.validate_remote()
}

/// Fail loudly at startup rather than halfway through a 400-file import.
pub fn check_available() -> Result<()> {
    let status = Command::new("rclone")
        .arg("version")
        .stdout(Stdio::null())
        .stderr(Stdio::null())
        .status();

    match status {
        Ok(s) if s.success() => Ok(()),
        Ok(s) => bail!("`rclone version` exited with {s}"),
        Err(_) => bail!(
            "rclone not found on PATH.\n\
             install it (https://rclone.org/install/) and configure a remote with `rclone config`."
        ),
    }
}

fn remote_url(cfg: &Config, remote_rel: &str) -> String {
    format!("{}/{}", cfg.remote.trim_end_matches('/'), remote_rel)
}

fn run(args: &[&str]) -> Result<std::process::Output> {
    let out = Command::new("rclone")
        .args(args)
        .output()
        .context("spawning rclone")?;
    Ok(out)
}

pub fn upload(cfg: &Config, local: &Path, remote_rel: &str) -> Result<()> {
    let dest = remote_url(cfg, remote_rel);
    let local = local.to_string_lossy();
    let out = run(&["copyto", &local, &dest])?;
    if !out.status.success() {
        bail!(
            "rclone copyto {local} -> {dest} failed: {}",
            String::from_utf8_lossy(&out.stderr).trim()
        );
    }
    Ok(())
}

pub fn download(cfg: &Config, remote_rel: &str, dest: &Path) -> Result<()> {
    if let Some(parent) = dest.parent() {
        std::fs::create_dir_all(parent)?;
    }
    let src = remote_url(cfg, remote_rel);
    let dest_s = dest.to_string_lossy();
    let out = run(&["copyto", &src, &dest_s])?;
    if !out.status.success() {
        bail!(
            "rclone copyto {src} -> {dest_s} failed: {}",
            String::from_utf8_lossy(&out.stderr).trim()
        );
    }
    Ok(())
}

/// Every stored file, as (path relative to the remote root, size in bytes).
pub fn list(cfg: &Config) -> Result<Vec<(String, u64)>> {
    let root = cfg.remote.trim_end_matches('/').to_string();
    let out = run(&[
        "lsf",
        "-R",
        "--files-only",
        "--format",
        "ps",
        "--separator",
        "\t",
        &root,
    ])?;
    if !out.status.success() {
        bail!(
            "rclone lsf {root} failed: {}",
            String::from_utf8_lossy(&out.stderr).trim()
        );
    }

    let text = String::from_utf8_lossy(&out.stdout);
    Ok(text
        .lines()
        .filter_map(|line| {
            let (path, size) = line.split_once('\t')?;
            Some((path.to_string(), size.trim().parse().unwrap_or(0)))
        })
        .collect())
}

pub fn delete(cfg: &Config, remote_rel: &str) -> Result<()> {
    let target = remote_url(cfg, remote_rel);
    let out = run(&["deletefile", &target])?;
    if !out.status.success() {
        bail!(
            "rclone deletefile {target} failed: {}",
            String::from_utf8_lossy(&out.stderr).trim()
        );
    }
    Ok(())
}

pub fn exists(cfg: &Config, remote_rel: &str) -> Result<bool> {
    let target = remote_url(cfg, remote_rel);
    let out = run(&["lsjson", "--stat", &target])?;
    Ok(out.status.success())
}