rusty-tasks/src/config/mod.rs

extern crate serde;
extern crate serde_json;

use figment::providers::{Env, Format, Json, Serialized};
use figment::Figment;
use serde::{Deserialize, Serialize};
use std::env::var;
use std::fs::File;
use std::io::Write;
use std::path::PathBuf;

#[derive(Deserialize, Serialize, Debug)]
pub struct Config {
    pub editor: String,
    pub sections: Vec<String>,
    pub notes_dir: String,
}

impl Default for Config {
    fn default() -> Self {
        Config {
            editor: "nano".into(),
            sections: vec!["Daily".into(), "Weekly".into(), "Monthly".into()],
            notes_dir: "./Notes".into(),
        }
    }
}

#[derive(Debug)]
pub enum ConfigError {
    IOError(&'static str),
    ParseError(&'static str),
    EnvError(&'static str),
}

impl Config {
    pub fn load(cfg_file: &str) -> Result<Self, ConfigError> {
        Figment::from(Serialized::defaults(Config::default()))
            .merge(Env::raw().only(&["EDITOR"]))
            .merge(Json::file(cfg_file))
            .extract()
            .or(Err(ConfigError::IOError("Could not load config")))
    }

    pub fn write_default(cfg_file: &str) -> Result<(), ConfigError> {
        let buf = serde_json::to_string_pretty(&Self::default()).or_else(|_| {
            Err(ConfigError::ParseError(
                "could not serialize default config",
            ))
        })?;

        let mut f = File::create(cfg_file)
            .or_else(|_| Err(ConfigError::IOError("Could not open config file")))?;
        f.write_all(&buf.as_bytes()).or_else(|_| {
            Err(ConfigError::IOError(
                "could not write default config to file",
            ))
        })?;

        Ok(())
    }

    pub fn expected_locations() -> Result<Vec<PathBuf>, ConfigError> {
        let cfg_name = "rusty_task.json";
        let home = var("HOME").or(Err(ConfigError::EnvError(
            "$HOME environment variable not set",
        )))?;
        let pwd = var("PWD").or(Err(ConfigError::EnvError(
            "$PWD environment variable not set",
        )))?;

        let mut home_config_cfg = PathBuf::from(home.clone());
        home_config_cfg.push(".config");
        home_config_cfg.push(cfg_name);

        let mut home_cfg = PathBuf::from(home.clone());
        home_cfg.push(format!(".{}", cfg_name));

        let mut pwd_cfg = PathBuf::from(pwd.clone());
        pwd_cfg.push(format!(".{}", cfg_name));

        Ok(vec![home_config_cfg, home_cfg, pwd_cfg])
    }
}