use crate::{Error, Result};
use figment::{providers::Serialized, value, Figment, Metadata, Profile, Provider};
use serde::{Deserialize, Serialize};
use std::{fs::File, path::PathBuf, sync::Arc};
use tracing_subscriber::{Layer, Registry};

#[derive(Debug, PartialEq, Eq, Serialize, Deserialize)]
#[serde(default)]
pub struct Config {
    #[serde(flatten)]
    pub paths: crate::paths::Config,
    pub finder: crate::finder::Config,
    pub log_enabled: bool,
    pub log_file: PathBuf,
}

impl Config {
    pub fn extract<T: Provider>(provider: T) -> figment::error::Result<Config> {
        Figment::from(provider).extract()
    }

    pub fn as_layer(&self) -> Result<Vec<Box<dyn Layer<Registry> + Send + Sync>>> {
        let mut layers = Vec::new();

        if self.log_enabled {
            let file = File::create(&self.log_file)?;
            let log_layer = tracing_subscriber::fmt::layer()
                .with_writer(Arc::new(file))
                .boxed();
            layers.push(log_layer);
        };

        Ok(layers)
    }
}

impl Default for Config {
    fn default() -> Self {
        Self {
            paths: Default::default(),
            finder: Default::default(),
            log_enabled: Default::default(),
            log_file: dirs::cache_dir()
                .map(|p| p.join("tmuxr"))
                .unwrap_or_default()
                .join("tmuxr.log"),
        }
    }
}

impl Provider for Config {
    fn metadata(&self) -> Metadata {
        Metadata::named("Tmuxr directory config")
    }

    fn data(&self) -> figment::error::Result<value::Map<Profile, value::Dict>> {
        Serialized::defaults(Self::default()).data()
    }
}

impl TryFrom<Figment> for Config {
    type Error = Error;

    fn try_from(value: Figment) -> Result<Self> {
        value.extract().map_err(Into::into)
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::paths::PathEntry;
    use figment::providers::{Format, Serialized, Toml};

    #[test]
    fn test_extract() {
        figment::Jail::expect_with(|jail| {
            jail.create_file(
                "tmuxr.toml",
                r#"
                log_enabled = true
                log_file = "/path/to/log_file"
                paths = [
                    "/path/to/projects",
                    { path = "/path/to/other_projects", recurse = 1, hidden = true },
                    { path = "/path/to/another_project", recurse = 0 },
                ]

                [finder]
                program = "fzf"
                args = ["-0", "-1", "--preview='cat'"]
                "#,
            )?;

            let config: Config = Figment::from(Serialized::defaults(Config::default()))
                .merge(Toml::file("tmuxr.toml"))
                .extract()?;

            assert_eq!(
                config,
                Config {
                    paths: crate::paths::Config {
                        paths: Vec::from([
                            PathEntry {
                                path: "/path/to/projects".into(),
                                hidden: false,
                                recurse: None,
                            },
                            PathEntry {
                                path: "/path/to/other_projects".into(),
                                hidden: true,
                                recurse: Some(1),
                            },
                            PathEntry {
                                path: "/path/to/another_project".into(),
                                hidden: false,
                                recurse: Some(0),
                            },
                        ]),
                    },
                    finder: crate::finder::Config {
                        program: "fzf".into(),
                        args: vec!["-0".into(), "-1".into(), "--preview='cat'".into()],
                    },
                    log_enabled: true,
                    log_file: "/path/to/log_file".into()
                }
            );

            Ok(())
        });
    }
}