use anyhow::{anyhow, Result};
use std::{
    collections::HashMap,
    fs::read_to_string,
    ops::Index,
    path::Path,
    sync::{LazyLock, Mutex},
};

pub struct Strings(HashMap<String, String>);
impl Index<&'static str> for Strings {
    type Output = str;
    fn index(&self, index: &'static str) -> &Self::Output {
        self.0.get(index).map(|s| s.as_str()).unwrap_or(index)
    }
}

impl Strings {
    pub fn load(path: &Path) -> Result<Self> {
        Ok(Self(
            read_to_string(path)?
                .lines()
                .skip(1)
                .map(|l| {
                    let (k, v) = l.split_once("=").ok_or(anyhow!("'=' missing"))?;
                    Ok::<_, anyhow::Error>((
                        k.trim_end().to_owned(),
                        v.trim_start().replace("%n", "\n"),
                    ))
                })
                .try_collect()?,
        ))
    }
}

static TR_LOAD: Mutex<Option<Strings>> = Mutex::new(None);
static TR: LazyLock<Strings> = LazyLock::new(|| TR_LOAD.lock().unwrap().take().unwrap());

pub fn tr<'a>(s: &'static str) -> &'a str {
    &TR[s]
}
pub fn set_language(lang: &str) {
    *TR_LOAD.lock().unwrap() =
        Some(Strings::load(Path::new(&format!("locale/{lang}.ini"))).unwrap());
}