nedots 0.1.3

A tool to manage configuration files/(ne)dots.
use crate::paths::{ResolvePath, StripPrefix};
use anyhow::Context;
use std::{
    collections::HashMap,
    path::{Path, PathBuf},
};

#[derive(Debug, Default, serde::Serialize, serde::Deserialize)]
pub(crate) struct Config {
    pub(crate) root: PathBuf,
    pub(crate) dots_dir: PathBuf,
    pub(crate) backup_dir: PathBuf,
    pub(crate) remote: String,
    pub(crate) sources: Vec<PathBuf>,
    pub(crate) git_repos: Vec<GitRepo>,
    pub(crate) systemd_services: Vec<String>,
}

#[derive(Debug, Default, serde::Serialize, serde::Deserialize)]
pub(crate) struct GitRepo {
    pub(crate) id: String,
    pub(crate) remote: String,
    pub(crate) path: PathBuf,
}

pub(crate) fn read(path: &Path) -> anyhow::Result<Config> {
    log::trace!("Resolving path: {}", path.display());
    let path = path.resolve_path()?;

    log::trace!("Reading file: {}", path.display());
    let file = std::fs::read_to_string(&path)
        .with_context(|| format!("failed to read file: {}", path.display()))?;

    log::trace!("Deserializing yaml...");
    let config: Config = serde_yaml::from_str(&file)
        .with_context(|| format!("failed to deserialize str: {:#?}", &file))?;

    Ok(config)
}

impl Config {
    pub(crate) fn resolve_paths(mut self) -> Config {
        match self.root.resolve_path() {
            Ok(path) => {
                log::trace!("Resolved {}", path.display());
                self.root = path;
            }
            Err(err) => log::error!("❌ {}: {}", self.root.display(), err),
        }

        self.dots_dir = self.root.join(self.dots_dir);
        match self.dots_dir.resolve_path() {
            Ok(path) => {
                log::trace!("Resolved {}", path.display());
                self.dots_dir = path;
            }
            Err(err) => log::error!("❌ {}: {}", self.dots_dir.display(), err),
        }

        self.backup_dir = self.root.join(self.backup_dir);
        match self.backup_dir.resolve_path() {
            Ok(path) => {
                log::trace!("Resolved {}", path.display());
                self.backup_dir = path;
            }
            Err(err) => log::error!("❌ {}: {}", self.backup_dir.display(), err),
        }

        self.sources = self
            .sources
            .into_iter()
            .map(|s| match s.resolve_path() {
                Ok(path) => {
                    log::trace!("Resolved {}", path.display());
                    path
                }
                Err(err) => {
                    log::error!("❌ {}: {}", s.display(), err);
                    Path::new("").to_path_buf()
                }
            })
            .filter(|s| s.ne(Path::new("")))
            .collect();

        self.git_repos = self
            .git_repos
            .into_iter()
            .map(|mut gr| match gr.path.resolve_path() {
                Ok(path) => {
                    log::trace!("Resolved {}", path.display());
                    gr.path = path;
                    gr
                }
                Err(err) => {
                    log::error!("❌ {}: {}", gr.path.display(), err);
                    gr.path = Path::new("").to_path_buf();
                    gr
                }
            })
            .filter(|gr| gr.path.ne(Path::new("")))
            .collect();

        self
    }

    pub(crate) fn get_dst_from_src(&self, src: &Path) -> anyhow::Result<PathBuf> {
        let stripped = StripPrefix::strip_prefix(src, "/")?;
        Ok(self.dots_dir.join(stripped))
    }

    pub(crate) fn get_sources_as_hashmap(&self) -> HashMap<&str, PathBuf> {
        let mut all_parts: Vec<&str> = Vec::new();
        for pb in &self.sources {
            let _: Vec<&str> = pb
                .to_str()
                .unwrap_or("")
                .split('/')
                .map(|part| {
                    all_parts.push(part);
                    part
                })
                .collect();
        }

        all_parts.sort();
        all_parts.dedup_by(|a, b| a.eq(&b));

        let mut hash_map: HashMap<&str, PathBuf> = HashMap::new();
        for part in &all_parts {
            if !part.is_empty() {
                hash_map.insert(
                    part,
                    self.sources
                        .iter()
                        .find(|s| s.to_str().unwrap_or("").contains(part))
                        .unwrap()
                        .to_path_buf(),
                );
            }
        }

        log::debug!("Keyed sources: {:#?}", &hash_map);
        hash_map
    }
}