natter/src/config/full.rs

use std::path::Path;
use std::path::PathBuf;
use tokio::fs::File;
use tokio::io::AsyncWriteExt;

use super::raw::RawConfig;

/// This is the config struct used by most of the code, which is an interpreted version of the RawConfig struct which is the raw disk-representation of the config.
pub(crate) struct Config {
    raw: RawConfig,
    root_dir: PathBuf,
}

impl Config {
    pub(crate) fn new<P: Into<PathBuf>>(root_dir: P) -> Result<Config, Box<dyn std::error::Error>> {
        Ok(Config {
            raw: RawConfig::default(),
            root_dir: root_dir.into(),
        })
    }

    pub(crate) async fn load_from_file<P: AsRef<Path>>(
        path: P,
    ) -> Result<Config, Box<dyn std::error::Error>> {
        async fn inner(path: &Path) -> Result<Config, Box<dyn std::error::Error>> {
            let contents = tokio::fs::read_to_string(path).await?;
            let parsed_contents: RawConfig = toml::from_str(contents.as_str())?;
            let root_directory = path
                .parent()
                .expect("Writer config must exist inside a directory.");
            Ok(Config {
                raw: parsed_contents,
                root_dir: root_directory.to_owned(),
            })
        }
        inner(path.as_ref()).await
    }

    pub(crate) async fn write_to_disk(&self) -> Result<(), Box<dyn std::error::Error>> {
        let mut config_file = File::create(self.get_path_to_config_file()).await?;
        config_file
            .write_all(toml::to_string(&self.raw)?.as_bytes())
            .await?;
        Ok(())
    }

    pub(crate) fn get_path_to_config_file(&self) -> PathBuf {
        self.root_dir.join("writer.toml")
    }
}