85 lines
2.4 KiB
Rust
85 lines
2.4 KiB
Rust
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])
|
|
}
|
|
}
|