use directories::{ProjectDirs, UserDirs};
use serde::{Deserialize, Serialize};

#[derive(Serialize, Deserialize)]
pub struct Config {
    pub music_roots: Vec<String>,
    pub audio_exts: Vec<String>,
    pub skip_window_secs: u64,
    pub learning_rate: f32,
    pub decay_rate: f32,
    pub min_score_thresh: f32,
    pub temperature: f32,
}

pub fn default_state_path() -> String {
    ProjectDirs::from("qualifier", "organisation", "mood")
        .and_then(|pd| {
            pd.config_dir()
                .join("mood.ron")
                .to_str()
                .map(|x| x.to_string())
        })
        .unwrap_or("~/.config/mood/mood.ron".to_string())
}

impl Default for Config {
    fn default() -> Config {
        Config {
            music_roots: vec![
                UserDirs::new()
                    .and_then(|u| {
                        u.audio_dir()
                            .and_then(|p| p.to_str().map(|x| x.to_string()))
                    })
                    .unwrap_or("~/Music".to_string()),
            ],
            audio_exts: [".mp3", ".flac", ".wav", ".m4a", ".ogg", ".vorbis"]
                .into_iter()
                .map(String::from)
                .collect(),
            skip_window_secs: 60,
            learning_rate: 0.1,
            decay_rate: 0.001,
            min_score_thresh: 1e-5,
            temperature: 0.6,
        }
    }
}