wrestic 1.3.0

Restic wrapper built in Rust
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

use crate::{
    modules::selector::selector,
    utils::{
        get_current_shell::get_current_shell,
        macros::error,
        tools::{clear, pause},
    },
};
use anyhow::{Context, Result};
use cmd_lib::run_cmd;
use color_print::cprintln;
use flate2::read::GzDecoder;
use indicatif::ProgressBar;
use std::{
    env::current_exe,
    fs::{remove_file, File},
    io::BufReader,
    path::Path,
    process::Command,
    time::Duration,
};
use tar::Archive;

fn get_current_version() -> Result<String> {
    let version = env!("CARGO_PKG_VERSION").to_string();
    Ok(version)
}
fn get_latest_version(url: &str) -> Result<String> {
    let shell = get_current_shell()?;
    let output = Command::new(shell)
        .arg("-c")
        .arg(format!(
            r#"curl -s "{url}" | grep tag_name | grep -Eo '[0-9.]+'"#
        ))
        .output()
        .context(error!("Failed fetching the latest version of wrestic."))?;
    let version_string = String::from_utf8_lossy(&output.stdout);
    let version = version_string.trim().to_string();
    if version.is_empty() {
        Err(error!(
            "Failed fetching the latest version of wrestic. Try again later."
        ))?;
    }
    Ok(version)
}

fn extract_wrestic(file_path: &str, extract_path: &str) -> Result<()> {
    let file = File::open(file_path)?;
    let gz = GzDecoder::new(file);
    let tar = BufReader::new(gz);
    let mut archive = Archive::new(tar);
    let extract_path_parent = Path::new(extract_path).parent().unwrap().to_owned();
    archive.unpack(extract_path_parent)?;
    Ok(())
}

pub fn update(noconfirm: bool) -> Result<()> {
    clear()?;
    cprintln!("<c,u,s>UPDATER");
    println!();

    let current_executable = &current_exe()?;
    let bin_path = current_executable.to_str().unwrap();
    let tmp_path = "/tmp/wrestic.tar.gz";
    let url = "https://api.github.com/repos/alvaro17f/wrestic/releases/latest";

    let command = format!(
        r#"curl -sL $(curl -s "{url}" | grep browser_download_url | cut -d '"' -f 4) -o {tmp_path}"#
    );

    if get_current_version()? >= get_latest_version(url)? {
        cprintln!("<g,u>Wrestic is already up to date!\n");
        pause()?
    } else {
        cprintln!(
            "<y>Wrestic is outdated!\n<r>current: <k>{}\n<g>latest: <k>{}\n",
            get_current_version()?,
            get_latest_version(url)?
        );

        let pb = ProgressBar::new_spinner();
        pb.enable_steady_tick(Duration::from_millis(120));
        pb.set_message("Updating wrestic...");

        let shell = get_current_shell()?;

        if run_cmd!(
            $shell -c $command;
        )
        .is_err()
        {
            pb.finish_and_clear();
            Err(error!("Failed downloading the latest version of wrestic"))?;
        }

        if remove_file(bin_path).is_err() {
            pb.finish_and_clear();
            Err(error!("Failed removing the old wrestic version"))?;
        }

        if extract_wrestic(tmp_path, bin_path).is_err() {
            pb.finish_and_clear();
            Err(error!(format!("Failed extracting wrestic into {bin_path}")))?;
        };

        if remove_file(tmp_path).is_err() {
            pb.finish_and_clear();
            Err(error!("Failed removing tmp files"))?;
        } else {
            pb.finish_and_clear();
            cprintln!("<g,u>Wrestic was successfully updated\n");
        }

        pause()?;
    }

    if !noconfirm {
        selector()?;
    }
    Ok(())
}