cli-app-template 0.2.4

Starter template for CLI applications
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

use std::{
    fs,
    path::{Path, PathBuf},
};

use config::{File, FileFormat};
use serde::Deserialize;

use crate::error::Result;

const CONFIG_FILE_NAME: &str = "config.toml";

const DEFAULT_CONFIG_FILE_CONTENT: &str = include_str!("default.toml");

#[derive(Deserialize)]
pub struct Config {
    pub data_dir: PathBuf,

    #[serde(skip)]
    pub dir: PathBuf,
}

impl Config {
    pub fn load(dir: &Path) -> Result<Config> {
        let file = dir.join(CONFIG_FILE_NAME);
        if !file.exists() {
            fs::create_dir_all(dir)?;
            fs::write(&file, DEFAULT_CONFIG_FILE_CONTENT)?;
        }
        let config = config::Config::builder()
            .add_source(File::from_str(
                DEFAULT_CONFIG_FILE_CONTENT,
                FileFormat::Toml,
            ))
            .add_source(File::from(file).format(FileFormat::Toml))
            .build()?;
        let mut config: Config = config.try_deserialize()?;
        config.dir = dir.into();
        Ok(config)
    }
}