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

use crate::{
    modules::selector::selector,
    utils::{
        get_config::get_config,
        snapshots_selector::snapshots_selector,
        tools::{clear, pause},
    },
};
use anyhow::Result;
use cmd_lib::run_cmd;
use color_print::{cformat, cprintln};
use dialoguer::{theme::ColorfulTheme, Confirm, Select};
use std::env;

fn do_restore(
    backend: &str,
    repository: &str,
    restore_folder: &str,
    restore_snapshot: &str,
    user: &str,
) -> Result<()> {
    if run_cmd!(
        restic -r $backend:$repository --verbose --verbose restore $restore_snapshot --target $restore_folder;
    )
    .is_err()
    {
        cprintln!("<r>Failed to restore snapshot: <c>{restore_snapshot}</c> into: <c>{restore_folder}</c></r>");
    }
    if run_cmd!(chown -R $user:$user $restore_folder).is_err() {
        cprintln!("<r>Failed to change ownership of: <c>{restore_folder}</c></r>");
    }

    Ok(())
}

pub fn restore(noconfirm: bool) -> Result<()> {
    let settings = get_config()?;
    clear()?;
    cprintln!("<c,u,s>RESTORE");
    println!();
    let selection = if settings.len() > 1 {
        let selections: Vec<String> = settings.iter().map(|x| x.name.to_owned()).collect();
        Select::with_theme(&ColorfulTheme::default())
            .with_prompt(cformat!("<y>Where do you want to restore from?"))
            .default(0)
            .max_length(10)
            .items(&selections[..])
            .interact()?
    } else {
        0
    };

    env::set_var("USER", &settings[selection].user);
    env::set_var("RESTIC_PASSWORD", &settings[selection].restic_password);
    for env in &settings[selection].env {
        for (key, value) in env {
            env::set_var(key, value);
        }
    }

    let backend = &settings[selection].backend;
    let repository = &settings[selection].repository;
    let restore_folder = &settings[selection].restore_folder;
    let restore_snapshot = snapshots_selector(backend, repository)?;
    let user = &settings[selection].user;

    if Confirm::with_theme(&ColorfulTheme::default())
        .with_prompt(cformat!(
            "<y>Do you want to restore the snapshot with ID {restore_snapshot}? (Y/n): "
        ))
        .default(true)
        .interact()?
    {
        do_restore(backend, repository, restore_folder, &restore_snapshot, user)?;
        pause()?;
    }
    if !noconfirm {
        selector()?;
    }
    Ok(())
}