use std::{
    collections::BTreeMap,
    sync::{Arc, Mutex},
};

use crate::{
    font::Font,
    texture_atlas::{Texture, TextureAtlas},
};

/// TODO: rename
#[derive(Copy, Clone, Debug, Eq, Ord, PartialEq, PartialOrd, Serialize)]
pub enum TextStyle {
    Body,
    Button,
    Heading,
    // Monospace,
}

pub struct Fonts {
    fonts: BTreeMap<TextStyle, Font>,
    texture: Texture,
}

impl Fonts {
    pub fn new() -> Fonts {
        let mut atlas = TextureAtlas::new(128, 8); // TODO: better default?

        // Make one white pixel for use for various stuff:
        let pos = atlas.allocate((1, 1));
        atlas.texture_mut()[pos] = 255;

        let atlas = Arc::new(Mutex::new(atlas));

        // TODO: figure out a way to make the wasm smaller despite including a font.
        // let typeface_data = include_bytes!("../fonts/ProggyClean.ttf"); // Use 13 for this. NOTHING ELSE.
        // let typeface_data = include_bytes!("../fonts/DejaVuSans.ttf");
        let typeface_data = include_bytes!("../fonts/Roboto-Regular.ttf");

        let mut fonts = BTreeMap::new();
        fonts.insert(
            TextStyle::Body,
            Font::new(atlas.clone(), typeface_data, 20.0),
        );
        fonts.insert(TextStyle::Button, fonts[&TextStyle::Body].clone());
        fonts.insert(
            TextStyle::Heading,
            Font::new(atlas.clone(), typeface_data, 30.0),
        );

        let texture = atlas.lock().unwrap().clone().texture().clone();

        Fonts { fonts, texture }
    }

    pub fn texture(&self) -> &Texture {
        &self.texture
    }
}

impl std::ops::Index<TextStyle> for Fonts {
    type Output = Font;

    fn index(&self, text_style: TextStyle) -> &Font {
        &self.fonts[&text_style]
    }
}