//! 8x8 font collection
use super::{
    color::{BackgroundColor, StrokeColor},
    FrameLine, PixelColor, PixelFrame,
};

use super::error::ScreenError;
pub use font8x8::{
    FontUnicode, UnicodeFonts, BASIC_FONTS, BLOCK_FONTS, BOX_FONTS, GREEK_FONTS, HIRAGANA_FONTS,
    LATIN_FONTS,
};
use std::collections::HashMap;
use std::fmt;

lazy_static! {
    /// A `static HashMap<char, FontUnicode>` that holds the entire set of fonts supported
    /// for the `Screen`.
    pub static ref FONT_HASHMAP: HashMap<char, FontUnicode> = default_hashmap();
    /// A `static FontCollection` that offers a higher-level API for working with
    /// pixel frames, clips, scrolls, etc.
    ///
    /// `FONT_COLLECTION.sanitize_str(&str)` returns a sanitized `FontString`,
    /// and use that to render pixel frames..
    ///
    /// `FONT_COLLECTION.get(font: char)` returns the low-level `FontUnicode` if the font
    /// is found in the collection.
    pub static ref FONT_COLLECTION: FontCollection = FontCollection(default_hashmap());
}

fn default_hashmap() -> HashMap<char, FontUnicode> {
    BASIC_FONTS
        .iter()
        .chain(LATIN_FONTS.iter())
        .chain(BLOCK_FONTS.iter())
        .chain(BOX_FONTS.iter())
        .chain(GREEK_FONTS.iter())
        .chain(HIRAGANA_FONTS.iter())
        .map(|x| (x.0, *x))
        .collect()
}

// A set of font symbols that can be printed on a `Screen`.
#[derive(Clone, Debug, PartialEq)]
//#[cfg_attr(feature = "serde-support", derive(Serialize, Deserialize))]
pub struct FontCollection(HashMap<char, FontUnicode>);

impl FontCollection {
    /// Create a default `FontCollection`, containing the Unicode constants
    /// from the [font8x8](https://github.com/saibatizoku/font8x8-rs) crate, except for
    /// `MISC_FONTS`, and `SGA_FONTS` (which are non-standard).
    pub fn new() -> Self {
        FontCollection(default_hashmap())
    }

    /// Create a `FontCollection` with a custom HashMap of font symbols.
    pub fn from_hashmap(hashmap: HashMap<char, FontUnicode>) -> Self {
        FontCollection(hashmap)
    }

    /// Get an `Option` with the symbol's byte rendering.
    pub fn get(&self, symbol: char) -> Option<&FontUnicode> {
        self.0.get(&symbol)
    }

    /// Search if collection has a symbol by its unicode key.
    pub fn contains_key(&self, symbol: char) -> bool {
        self.0.contains_key(&symbol)
    }

    /// Sanitize a `&str` and create a new `FontString`.
    pub fn sanitize_str(&self, s: &str) -> Result<FontString, ScreenError> {
        let valid = s
            .chars()
            .filter(|c| self.0.contains_key(c))
            .map(|sym| *self.get(sym).unwrap())
            .collect::<Vec<FontUnicode>>();
        Ok(FontString(valid))
    }
}

impl Default for FontCollection {
    fn default() -> Self {
        FontCollection::new()
    }
}

/// A `FontString` is a collection of `FontUnicode` which can be rendered to frames for the LED
/// Matrix.
#[derive(Clone, Debug, Default, PartialEq)]
pub struct FontString(Vec<FontUnicode>);

impl FontString {
    /// Create an empty `FontString`.
    pub fn new() -> Self {
        FontString(Default::default())
    }

    /// Render the font string as a collection of unicode value points, `Vec<char>`.
    pub fn chars(&self) -> Vec<char> {
        self.0.iter().map(|font| font.char()).collect::<Vec<char>>()
    }

    /// Returns a `Vec<FontFrame>` for each inner font.
    pub fn font_frames(&self, stroke: PixelColor, bg: PixelColor) -> Vec<FontFrame> {
        self.0
            .iter()
            .map(|font| FontFrame::new(*font, stroke, bg))
            .collect::<Vec<FontFrame>>()
    }

    /// Returns a `Vec<PixelFrame>` for each inner font.
    pub fn pixel_frames(&self, stroke: PixelColor, bg: PixelColor) -> Vec<PixelFrame> {
        self.font_frames(stroke, bg)
            .into_iter()
            .map(|f| f.into())
            .collect::<Vec<PixelFrame>>()
    }
}

impl fmt::Display for FontString {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(
            f,
            "{}",
            self.0.iter().map(|font| font.char()).collect::<String>()
        )
    }
}

/// A font that can be rendered as a `PixelFrame` with a `stroke` color, and a `background` color.
#[derive(Debug, PartialEq)]
pub struct FontFrame {
    /// `UTF16` font
    font: FontUnicode,
    /// Color for the font stroke
    stroke: PixelColor,
    /// Color for the font background
    background: PixelColor,
}

impl FontFrame {
    /// Create a new font frame with a `stroke` color, and a `background` color.
    pub fn new(font: FontUnicode, stroke: PixelColor, background: PixelColor) -> Self {
        FontFrame {
            font,
            stroke,
            background,
        }
    }

    /// The `PixelFrame` for this font.
    pub fn pixel_frame(&self) -> PixelFrame {
        let pixels =
            font_to_pixel_color_array_with_bg(self.font.byte_array(), self.stroke, self.background);
        pixels.into()
    }
}

impl From<FontFrame> for PixelFrame {
    fn from(font: FontFrame) -> Self {
        font.pixel_frame()
    }
}

impl BackgroundColor for FontFrame {
    fn set_background_color(&mut self, color: PixelColor) {
        self.background = color;
    }
    fn get_background_color(&self) -> PixelColor {
        self.background
    }
}

impl StrokeColor for FontFrame {
    fn set_stroke_color(&mut self, color: PixelColor) {
        self.stroke = color;
    }
    fn get_stroke_color(&self) -> PixelColor {
        self.stroke
    }
}

// Render a font symbol with a stroke color and a background color.
fn font_to_pixel_color_array_with_bg(
    symbol: [u8; 8],
    color: PixelColor,
    background: PixelColor,
) -> [PixelColor; 64] {
    let mut pixels = [background; 64];
    for (row_idx, encoded_row) in symbol.iter().enumerate() {
        for col_idx in 0..8 {
            if (*encoded_row & 1 << col_idx) > 0 {
                pixels[row_idx * 8 + col_idx] = color;
            }
        }
    }
    pixels
}

// Render a font symbol with a `PixelColor` into a `[PixelColor; 64]`.
fn font_to_pixel_color_array(symbol: [u8; 8], color: PixelColor) -> [PixelColor; 64] {
    font_to_pixel_color_array_with_bg(symbol, color, Default::default())
}

/// Render a font symbol with a `PixelColor` into a `FrameLine`.
pub fn font_to_pixel_frame(symbol: [u8; 8], color: PixelColor) -> PixelFrame {
    let pixels = font_to_pixel_color_array(symbol, color);
    PixelFrame::new(&pixels)
}

/// Render a font symbol with a `PixelColor` into a `FrameLine`.
pub fn font_to_frame(symbol: [u8; 8], color: PixelColor) -> FrameLine {
    let pixels = font_to_pixel_color_array(symbol, color);
    FrameLine::from_pixels(&pixels)
}

#[cfg(test)]
mod tests {
    use super::*;

    const BLK: PixelColor = PixelColor::BLACK;
    const RED: PixelColor = PixelColor::RED;
    const GRN: PixelColor = PixelColor::GREEN;
    const BLU: PixelColor = PixelColor::BLUE;
    const YLW: PixelColor = PixelColor::YELLOW;
    const BASIC_FONT: [PixelColor; 64] = [
        BLU, BLU, BLK, BLK, BLK, BLU, BLU, BLK, //
        BLU, BLU, BLU, BLK, BLU, BLU, BLU, BLK, //
        BLU, BLU, BLU, BLU, BLU, BLU, BLU, BLK, //
        BLU, BLU, BLU, BLU, BLU, BLU, BLU, BLK, //
        BLU, BLU, BLK, BLU, BLK, BLU, BLU, BLK, //
        BLU, BLU, BLK, BLK, BLK, BLU, BLU, BLK, //
        BLU, BLU, BLK, BLK, BLK, BLU, BLU, BLK, //
        BLK, BLK, BLK, BLK, BLK, BLK, BLK, BLK, //
    ];
    const BOX_FONT: [PixelColor; 64] = [
        BLK, BLK, BLK, GRN, BLK, BLK, BLK, BLK, //
        BLK, BLK, BLK, GRN, BLK, BLK, BLK, BLK, //
        BLK, BLK, BLK, GRN, BLK, BLK, BLK, BLK, //
        BLK, BLK, BLK, GRN, GRN, GRN, GRN, GRN, //
        GRN, GRN, GRN, GRN, GRN, GRN, GRN, GRN, //
        BLK, BLK, BLK, BLK, BLK, BLK, BLK, BLK, //
        BLK, BLK, BLK, BLK, BLK, BLK, BLK, BLK, //
        BLK, BLK, BLK, BLK, BLK, BLK, BLK, BLK, //
    ];
    const BOX_FONT_BG: [PixelColor; 64] = [
        YLW, YLW, YLW, BLU, YLW, YLW, YLW, YLW, //
        YLW, YLW, YLW, BLU, YLW, YLW, YLW, YLW, //
        YLW, YLW, YLW, BLU, YLW, YLW, YLW, YLW, //
        YLW, YLW, YLW, BLU, BLU, BLU, BLU, BLU, //
        BLU, BLU, BLU, BLU, BLU, BLU, BLU, BLU, //
        YLW, YLW, YLW, YLW, YLW, YLW, YLW, YLW, //
        YLW, YLW, YLW, YLW, YLW, YLW, YLW, YLW, //
        YLW, YLW, YLW, YLW, YLW, YLW, YLW, YLW, //
    ];
    const HIRAGANA_FONT: [PixelColor; 64] = [
        BLK, BLK, BLK, RED, BLK, BLK, BLK, BLK, //
        BLK, RED, RED, RED, RED, RED, RED, BLK, //
        BLK, BLK, BLK, RED, BLK, BLK, BLK, BLK, //
        BLK, BLK, RED, RED, RED, RED, BLK, BLK, //
        BLK, BLK, BLK, BLK, BLK, BLK, RED, BLK, //
        BLK, BLK, BLK, BLK, BLK, BLK, RED, BLK, //
        BLK, BLK, BLK, RED, RED, RED, BLK, BLK, //
        BLK, BLK, BLK, BLK, BLK, BLK, BLK, BLK, //
    ];

    #[test]
    fn font_collection_sanitizes_text_by_filtering_known_unicode_points() {
        let font_set = FontCollection::new();
        let valid_text = font_set.sanitize_str("hola niño @¶øþ¥").unwrap();
        assert_eq!(format!("{}", valid_text), "hola niño @¶øþ¥");
    }

    #[test]
    fn font_collection_sanitizes_text_by_removing_symbols_not_in_set() {
        let font_set = FontCollection::new();
        let invalid_text = font_set.sanitize_str("ŧ←→ł").unwrap();
        assert_eq!(format!("{}", invalid_text), "");

        let font_set = FontCollection::from_hashmap(HashMap::new());
        let invalid_text = font_set.sanitize_str("hola niño @¶øþ¥").unwrap();
        assert_eq!(format!("{}", invalid_text), "");
    }

    #[test]
    fn font_collection_gets_optional_symbol_by_unicode_key() {
        let font_set = FontCollection::new();
        let symbol = font_set.get('ñ');
        assert!(symbol.is_some());
    }

    #[test]
    fn font_collection_searches_for_symbols_by_unicode_key() {
        let font_set = FontCollection::new();
        let has_symbol = font_set.contains_key('ñ');
        assert!(has_symbol);
    }

    #[test]
    fn font_string_new_method_starts_emtpy_instance() {
        let font_string = FontString::new();
        assert_eq!(font_string.0, Vec::new());
    }

    #[test]
    fn font_string_chars_method_returns_vec_of_chars() {
        let font_set = FontCollection::new();
        let font_string = font_set.sanitize_str("┷│││┯").unwrap();
        assert_eq!(font_string.chars(), vec!['┷', '│', '│', '│', '┯']);
    }

    #[test]
    fn font_string_implements_display_trait() {
        let font_set = FontCollection::new();
        let font_string = font_set.sanitize_str("┷│││┯").unwrap();
        assert_eq!(format!("{}", font_string), "┷│││┯".to_string());
    }

    #[test]
    fn font_string_font_frames_returns_a_vec_of_font_frame() {
        let font_set = FontCollection::new();
        let font_string = font_set.sanitize_str("Mち┶").unwrap();
        let bas_font = font_set.get('M').unwrap();
        let hir_font = font_set.get('ち').unwrap();
        let box_font = font_set.get('┶').unwrap();
        let ft_frames = font_string.font_frames(PixelColor::YELLOW, PixelColor::BLACK);
        assert_eq!(
            ft_frames,
            vec![
                FontFrame {
                    font: *bas_font,
                    stroke: PixelColor::YELLOW,
                    background: PixelColor::BLACK,
                },
                FontFrame {
                    font: *hir_font,
                    stroke: PixelColor::YELLOW,
                    background: PixelColor::BLACK,
                },
                FontFrame {
                    font: *box_font,
                    stroke: PixelColor::YELLOW,
                    background: PixelColor::BLACK,
                },
            ]
        );
    }

    #[test]
    fn font_string_font_frames_returns_a_vec_of_pixel_frame() {
        let font_set = FontCollection::new();
        let font_string = font_set.sanitize_str("MM").unwrap();
        let px_frames = font_string.pixel_frames(PixelColor::BLUE, PixelColor::BLACK);
        assert_eq!(
            px_frames,
            vec![PixelFrame::from(BASIC_FONT), PixelFrame::from(BASIC_FONT),]
        );
    }

    #[test]
    fn fn_font_to_pixel_color_array_with_bg_creates_new_array() {
        let font_set = FontCollection::new();
        let font = font_set.get('┶').unwrap();
        let px_array = font_to_pixel_color_array_with_bg(
            font.byte_array(),
            PixelColor::BLUE,
            PixelColor::YELLOW,
        );
        for (idx, px) in px_array.iter().enumerate() {
            assert_eq!(*px, BOX_FONT_BG[idx]);
        }
    }

    #[test]
    fn fn_font_to_pixel_color_array_creates_new_array() {
        let font_set = FontCollection::new();
        let font = font_set.get('M').unwrap();
        let px_array = font_to_pixel_color_array(font.byte_array(), PixelColor::BLUE);
        for (idx, px) in px_array.iter().enumerate() {
            assert_eq!(*px, BASIC_FONT[idx]);
        }
    }

    #[test]
    fn fn_font_to_pixel_frame_takes_a_byte_array_and_pixel_color() {
        let font_set = FontCollection::new();
        let chi_font = font_set.get('ち').unwrap();
        let px_frame = font_to_pixel_frame(chi_font.byte_array(), PixelColor::RED);
        assert_eq!(px_frame, PixelFrame::from(HIRAGANA_FONT));
    }

    #[test]
    fn fn_font_to_frame_takes_a_byte_array_and_pixel_color() {
        let font_set = FontCollection::new();
        let box_font = font_set.get('┶').unwrap();
        let px_frame_line = font_to_frame(box_font.byte_array(), PixelColor::GREEN);
        assert_eq!(px_frame_line, PixelFrame::from(BOX_FONT).frame_line());
    }

    #[test]
    fn font_frames_are_created_from_ut16_font_a_stroke_and_a_background_color() {
        let font_set = FontCollection::new();
        let letter_a = font_set.get('a').unwrap();
        let font_frame = FontFrame::new(letter_a.clone(), PixelColor::WHITE, PixelColor::BLACK);
        assert_eq!(
            font_frame,
            FontFrame {
                font: *letter_a,
                stroke: PixelColor::WHITE,
                background: PixelColor::BLACK
            }
        );
    }

    #[test]
    fn font_frames_is_represented_as_a_pixel_frame() {
        let font_set = FontCollection::new();
        let hiragana_font = font_set.get('ち').unwrap();
        let font_frame = FontFrame::new(hiragana_font.clone(), PixelColor::RED, PixelColor::BLACK);
        let px_frame = font_frame.pixel_frame();
        assert_eq!(px_frame, PixelFrame::from(HIRAGANA_FONT));
    }

    #[test]
    fn pixel_frame_implements_from_font_frame_trait() {
        let font_set = FontCollection::new();
        let hiragana_font = font_set.get('ち').unwrap();
        let font_frame = FontFrame::new(hiragana_font.clone(), PixelColor::RED, PixelColor::BLACK);
        let px_frame = PixelFrame::from(font_frame);
        assert_eq!(px_frame, PixelFrame::from(HIRAGANA_FONT));
    }

    #[test]
    fn font_frame_sets_background_color() {
        let font_set = FontCollection::new();
        let letter_a = font_set.get('a').unwrap();
        let mut font_frame = FontFrame::new(letter_a.clone(), PixelColor::WHITE, PixelColor::BLACK);
        font_frame.set_background_color(PixelColor::RED);
        assert_eq!(
            font_frame,
            FontFrame {
                font: *letter_a,
                stroke: PixelColor::WHITE,
                background: PixelColor::RED
            }
        );
    }

    #[test]
    fn font_frame_gets_background_color() {
        let font_set = FontCollection::new();
        let letter_a = font_set.get('a').unwrap();
        let font_frame = FontFrame::new(letter_a.clone(), PixelColor::WHITE, PixelColor::GREEN);
        assert_eq!(font_frame.get_background_color(), PixelColor::GREEN);
    }

    #[test]
    fn font_frame_sets_stroke_color() {
        let font_set = FontCollection::new();
        let letter_a = font_set.get('a').unwrap();
        let mut font_frame = FontFrame::new(letter_a.clone(), PixelColor::WHITE, PixelColor::BLACK);
        font_frame.set_stroke_color(PixelColor::YELLOW);
        assert_eq!(
            font_frame,
            FontFrame {
                font: *letter_a,
                stroke: PixelColor::YELLOW,
                background: PixelColor::BLACK
            }
        );
    }

    #[test]
    fn font_frame_gets_stroke_color() {
        let font_set = FontCollection::new();
        let letter_a = font_set.get('a').unwrap();
        let font_frame = FontFrame::new(letter_a.clone(), PixelColor::BLUE, PixelColor::WHITE);
        assert_eq!(font_frame.get_stroke_color(), PixelColor::BLUE);
    }
}