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

use crate::lib::{
    config::loader::load_cfg,
    fsutil::{
        copy::copy,
        paths::{get_root, get_to_make},
    },
    structs::config::KelpDotConfig,
    util::{os::get_host_os, scripts::run_script},
};
use anyhow::Context;
use kelpdot_macros::*;
use std::path::Path;
/// Backup dotfiles
pub fn save() -> anyhow::Result<()> {
    let root = get_root()?;
    cyan_print!("[INFO] Saving dotfiles {}...", root);
    debug_print!("Building OS list...");
    let os = get_host_os().with_context(|| red!("Unable to get host OS name!"))?; // Here we get guest os; If OS is unreconized, return a generic GNU / Linux System
    cyan_print!("[INFO] Found Os {}", os.prettyname);
    let config: KelpDotConfig = load_cfg(root.clone())?; // Load a KelpConfig struct wich is basically $DOTFILES_ROOT/kelp.yaml

    if let Some(files) = config.homefiles {
        // If config has "homefiles" keys, copy each $HOME/$FILE
        let home = std::env::var("HOME").with_context(|| red!("Unable to get env var $HOME!"))?; // Get $HOME path or crash
        debug_print!("Home: {}", home);

        // Make sur that $DOTFILES_ROOT/home doesn't exist
        // or doesn't contain files
        if Path::new(&format!("{}/home", root)).exists() {
            std::fs::remove_dir_all(&format!("{}/home", root))
                .with_context(|| red!("Unable to remove old home directory!"))?;
        }
        std::fs::create_dir(format!("{}/home", root))?;
        for f in files {
            green_print!("[SAVE] Copying file {}...", f);
            let path = format!("{}/{}", home, f.path);
            let file = Path::new(&path);
            // Make sur that file exists
            if file.exists() {
                // Get path to make
                // Example:
                // home/**.config/i3** directory
                let tomake = get_to_make(f.path)?;
                // Create the file
                std::fs::create_dir_all(format!("{}/home/{}", root, tomake))?;
                copy(
                    path.clone(),
                    format!(
                        "{}/home/{}/{}",
                        root,
                        tomake,
                        file.file_name().unwrap().to_str().unwrap().to_owned()
                    ),
                )?;
            } else {
                red_print!("[ERROR] File {} not found, skipping...", path);
            }
        }
        cyan_print!("[OK] Homefiles saved!");
    }
    // If config has "rootfiles" key, backup every file
    if let Some(files) = config.rootfiles {
        for f in files {
            green_print!("[SAVE] Copying file {}", f);
            // Get path to make:
            // Example:
            // $DOTFILES_ROOT/etc/portage/repos.conf
            let path = f.path.to_owned();
            let tomake = get_to_make(f.path)?;
            let file = Path::new(&path);
            if file.exists() {
                let file_name = file.file_name().unwrap().to_str().unwrap().to_owned();
                let dest = format!("{}/{}/{}", root, tomake, &file_name);
                if Path::new(&dest).exists() {
                    if Path::new(&dest).is_file() {
                        std::fs::remove_file(dest)?;
                    } else {
                        std::fs::remove_dir_all(dest)?;
                    }
                }
                std::fs::create_dir_all(format!("{}/{}", root, tomake))
                    .with_context(|| red!("Unable to create dir {}/{}", root, tomake))?;
                copy(path.clone(), format!("{}/{}/{}", root, tomake, file_name))?;
            }
        }
        cyan_print!("[OK] Rootfiles saved!");
    }
    if let Some(scripts) = config.postsave {
        for script in scripts {
            cyan_print!("[POSTSAVE] Running script {}", script.path);
            run_script(root.clone(), script)?;
        }
    }
    magenta_print!("[OK] All dotfiles saved!");
    Ok(())
}