dmp-core/src/music_controller/config.rs

use std::fs;
use std::fs::read_to_string;
use std::path::PathBuf;

use serde::{Deserialize, Serialize};

use crate::music_tracker::music_tracker::{DiscordRPCConfig, LastFMConfig, ListenBrainzConfig};

#[derive(Serialize, Deserialize, PartialEq, Eq)]
pub struct Config {
    pub db_path: Box<PathBuf>,
    pub lastfm: Option<LastFMConfig>,
    pub discord: Option<DiscordRPCConfig>,
    pub listenbrainz: Option<ListenBrainzConfig>,
}

impl Default for Config {
    fn default() -> Self {
        let path = PathBuf::from("./music_database");

        Config {
            db_path: Box::new(path),

            lastfm: None,

            discord: Some(DiscordRPCConfig {
                enabled: true,
                dango_client_id: 1144475145864499240,
                dango_icon: String::from("flat"),
            }),

            listenbrainz: Some(ListenBrainzConfig {
                enabled: false,
                api_url: String::from("https://api.listenbrainz.org"),
                auth_token: String::from(""),
            }),
        }
    }
}

impl Config {
    /// Creates and saves a new config with default values
    pub fn new(config_file: &PathBuf) -> std::io::Result<Config> {
        let config = Config::default();
        config.save(config_file)?;

        Ok(config)
    }

    /// Loads config from given file path
    pub fn from(config_file: &PathBuf) -> std::result::Result<Config, toml::de::Error> {
        toml::from_str(
            &read_to_string(config_file)
                .expect("Failed to initalize music config: File not found!"),
        )
    }

    /// Saves config to given path
    /// Saves -> temp file, if successful, removes old config, and renames temp to given path
    pub fn save(&self, config_file: &PathBuf) -> std::io::Result<()> {
        let toml = toml::to_string_pretty(self).unwrap();

        let mut temp_file = config_file.clone();
        temp_file.set_extension("tomltemp");

        fs::write(&temp_file, toml)?;

        // If configuration file already exists, delete it
        match fs::metadata(config_file) {
            Ok(_) => fs::remove_file(config_file)?,
            Err(_) => {}
        }

        fs::rename(temp_file, config_file)?;
        Ok(())
    }
}