tetro-tui 3.5.2

use std::time::Duration;

use crate::core_game_engine::{Button, ExtNonNegF64, Input, Tetromino};
use crossterm::event::{KeyCode, KeyModifiers};

use crate::settings::{GameKeybinds, MiniTetrominoSymbols};

pub type KeybindsLegend = Vec<(/*(KeyCode, KeyModifiers)*/ String, &'static str)>;

pub fn generate_timestamp() -> String {
    chrono::Utc::now().format("%Y-%m-%d_%H:%M.%S").to_string()
}

pub trait BoolAsOnOff {
    fn on_off(self) -> &'static str;
}

impl BoolAsOnOff for bool {
    fn on_off(self) -> &'static str {
        if self { "on" } else { "off" }
    }
}

// FIXME: Consider showing full millisecond precision? This would agree with the chosen precision of inputs that we currently handle.
pub fn fmt_duration(dur: Duration) -> String {
    if dur.as_secs() / 60 == 0 {
        format!("{}.{:02}s", dur.as_secs() % 60, dur.as_millis() % 1000 / 10)
    } else {
        format!(
            "{}min {}.{:02}s",
            dur.as_secs() / 60,
            dur.as_secs() % 60,
            dur.as_millis() % 1000 / 10
        )
    }
}

pub fn fmt_hertz(f: ExtNonNegF64) -> String {
    const MIN: f64 = 1. / 100.0;
    const MAX: f64 = 10000.0; // "20 tiles per (60 Hz) frame."
    if f.is_zero() {
        "0 Hz".to_owned()
    } else if f.get() <= MIN {
        "~0 Hz".to_owned()
    } else if f.is_infinite() {
        "∞ Hz".to_owned()
    } else if MAX <= f.get() {
        "~∞ Hz".to_owned()
    } else {
        format!("{:.02} Hz", f.get())
    }
}

pub fn fmt_tetromino_counts(
    counts: &[u32; Tetromino::VARIANTS.len()],
    mini_tet_symbols: &MiniTetrominoSymbols,
) -> String {
    counts
        .iter()
        .zip(Tetromino::VARIANTS)
        .map(|(n, t)| {
            format!(
                "{n}{}",
                mini_tet_symbols.tets[t as usize].to_ascii_lowercase()
            )
        })
        .collect::<Vec<_>>()
        .join(" ")
}

pub fn fmt_player_input(input: Input, button_symbols: [char; Button::VARIANTS.len()]) -> String {
    match input {
        Input::Activate(b) => format!("++|{}|", button_symbols[b]),
        Input::Deactivate(b) => format!("--|{}|", button_symbols[b]),
    }
}

pub fn fmt_key(key: KeyCode) -> String {
    use KeyCode as K;
    use crossterm::event::ModifierKeyCode as M;
    match key {
        K::Backspace => "Back",
        //K::Enter => "Enter",
        K::Left => "←",
        K::Right => "→",
        K::Up => "↑",
        K::Down => "↓",
        //K::Home => "Home",
        //K::End => "End",
        //K::Insert => "Insert",
        K::Delete => "Del",
        //K::Menu => "Menu",
        K::PageUp => "PgUp",
        K::PageDown => "PgDn",
        //K::Tab => "Tab",
        //K::CapsLock => "CapsLock",
        K::F(k) => return format!("F{k}"),
        K::Char(' ') => "Space",
        K::Char(c) => {
            return c.to_string();
        }
        //K::Esc => "Esc",
        K::Modifier(M::LeftAlt) => "LAlt",
        K::Modifier(M::RightAlt) => "RAlt",
        K::Modifier(M::LeftShift) => "LShift",
        K::Modifier(M::RightShift) => "RShift",
        K::Modifier(M::LeftControl) => "LCtrl",
        K::Modifier(M::RightControl) => "RCtrl",
        K::Modifier(M::IsoLevel3Shift) => "AltGr",
        K::Modifier(M::IsoLevel5Shift) => "Iso5",
        K::Modifier(M::LeftSuper) => "LSuper",
        K::Modifier(M::RightSuper) => "RSuper",
        K::Modifier(M::LeftHyper) => "LHyper",
        K::Modifier(M::RightHyper) => "RHyper",
        K::Modifier(M::LeftMeta) => "LMeta",
        K::Modifier(M::RightMeta) => "RMeta",
        keycode => return format!("{keycode:?}"),
    }
    .to_string()
}

pub fn fmt_keymods(keymods: KeyModifiers) -> String {
    use KeyModifiers as KMs;
    [
        keymods.contains(KMs::CONTROL).then_some("Ctrl"),
        keymods.contains(KMs::SHIFT).then_some("Shift"),
        keymods.contains(KMs::ALT).then_some("Alt"),
        keymods.contains(KMs::SUPER).then_some("Super"),
        keymods.contains(KMs::HYPER).then_some("Hyper"),
        keymods.contains(KMs::META).then_some("Meta"),
    ]
    .into_iter()
    .flatten()
    .collect::<Vec<_>>()
    .join("+")
}

pub fn fmt_key_with_keymods((key, keymods): (KeyCode, KeyModifiers)) -> String {
    if keymods.is_empty() {
        format!("[{}]", fmt_key(key))
    } else {
        format!("[{}+{}]", fmt_keymods(keymods), fmt_key(key))
    }
}

pub fn fmt_button_keybinds(button: Button, keybinds: &GameKeybinds, sep: &str) -> String {
    keybinds
        .iter()
        .filter_map(|(key_keymods, b)| (*b == button).then_some(fmt_key_with_keymods(*key_keymods)))
        .collect::<Vec<_>>()
        .join(sep)
}

pub fn fmt_lineclear_name(lineclears: u32) -> &'static str {
    match lineclears {
        1 => "Mono",
        2 => "Duo",
        3 => "Tri",
        4 => "Tetra",
        5 => "Penta",
        6 => "Hexa",
        7 => "Hepta",
        8 => "Octa",
        9 => "Ennea",
        10 => "Deca",
        11 => "Hendeca",
        12 => "Dodeca",
        13 => "Triadeca",
        14 => "Tessaradeca",
        15 => "Penteeca",
        16 => "Hexadeca",
        17 => "Heptadeca",
        18 => "Octadeca",
        19 => "Enneadeca",
        20 => "Eicosa",
        _ => "Paralogo",
    }
}

pub fn to_roman(mut num: u32) -> String {
    // Large roman numerals should be uncommon and have little convention
    // (<https://en.wikipedia.org/wiki/Roman_numerals#Large_numbers>),
    // return fallback decimal representation.
    if 4000 <= num {
        return num.to_string();
    }

    const ADDITIVE_NUMERAL_PARTS: [(&str, u32); 13] = [
        ("M", 1000),
        ("CM", 900),
        ("D", 500),
        ("CD", 400),
        ("C", 100),
        ("XC", 90),
        ("L", 50),
        ("XL", 40),
        ("X", 10),
        ("IX", 9),
        ("V", 5),
        ("IV", 4),
        ("I", 1),
    ];

    let mut string = String::new();
    for (str, value) in ADDITIVE_NUMERAL_PARTS {
        while num >= value {
            num -= value;
            string.push_str(str);
        }
    }

    string
}

// Sidenote:
// In an ideal world, a char is just a 1-char `str`.
// Unfortunately, converting between `u8`, `char` and `&str` is painful. :-(
//                     char --let mut bs=vec![0;len_utf8()];c.encode_utf8(&mut bs)--> &str
//    u8 --b.into()--> char --c.to_string()--> String ------------------s.as_str()--> &str
//    u8 --------------------------------------------str::from_utf8(&[b]).unwrap()--> &str

/// 'Derivative' game modes use some string that ends in "[UNSIGNED_INTEGER]".
/// Find if game mode name ends like that, and increment the integer.
/// Otherwise push " [1]" to the end to signal the first proper derivative.
pub fn increment_game_mode_derivative(game_mode_title: &mut String) {
    let mut rev_chars = game_mode_title.chars().rev();
    // Get the last char.
    if let Some(last) = rev_chars.next()
        && last == ']'
    {
        let mut n = 0;
        for (i, ch) in rev_chars.enumerate() {
            if let Some(digit) = ch.to_digit(10) {
                // Digits continuing.
                // Accumulate.
                n += digit * 10u32.pow(i as u32);
                continue;
            } else if ch == '[' {
                if i == 0 {
                    // Closing bracket without digits. Not a proper tag.
                    break;
                } else {
                    // Closing bracket with some digits. Happy path!
                    game_mode_title
                        .truncate(game_mode_title.len() - 1/*bracket*/ - i /*chars*/);
                    game_mode_title.push_str(&format!("{}]", n + 1));
                    return;
                }
            } else {
                // Neither digit or closing bracket. Not a proper tag.
                break;
            }
        }
    }
    // Not properly tagged, push new derivative tag.
    game_mode_title.push_str(" [1]");
}

// Sanity checks. Because ad-hoc parsing and string manipulation sucks.
#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn inc_gamemode_deriv_empty() {
        let mut s = "".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, " [1]");
    }

    #[test]
    fn inc_gamemode_deriv_a() {
        let mut s = "a".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "a [1]");
    }

    #[test]
    fn inc_gamemode_deriv_abc() {
        let mut s = "abc".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "abc [1]");
    }

    #[test]
    fn inc_gamemode_deriv_open() {
        let mut s = "[".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "[ [1]");
    }

    #[test]
    fn inc_gamemode_deriv_open_9() {
        let mut s = "[9".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "[9 [1]");
    }

    #[test]
    fn inc_gamemode_deriv_close() {
        let mut s = "]".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "] [1]");
    }

    #[test]
    fn inc_gamemode_deriv_9_close() {
        let mut s = "9]".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "9] [1]");
    }

    #[test]
    fn inc_gamemode_deriv_openclose() {
        let mut s = "[]".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "[] [1]");
    }

    #[test]
    fn inc_gamemode_deriv_open_m1_close() {
        let mut s = "[-1]".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "[-1] [1]");
    }

    #[test]
    fn inc_gamemode_deriv_open_0_close() {
        let mut s = "[0]".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "[1]");
    }

    #[test]
    fn inc_gamemode_deriv_a_open_0_close() {
        let mut s = "a [0]".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "a [1]");
    }

    #[test]
    fn inc_gamemode_deriv_open_1_close() {
        let mut s = "[1]".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "[2]");
    }

    #[test]
    fn inc_gamemode_deriv_a_open_1_close() {
        let mut s = "a [1]".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "a [2]");
    }

    #[test]
    fn inc_gamemode_deriv_open_41_close() {
        let mut s = "[41]".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "[42]");
    }

    #[test]
    fn inc_gamemode_deriv_a_open_41_close() {
        let mut s = "a [41]".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "a [42]");
    }

    #[test]
    fn inc_gamemode_deriv_a_41_close() {
        let mut s = "a 41]".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "a 41] [1]");
    }

    #[test]
    fn inc_gamemode_deriv_a_open_41() {
        let mut s = "a [41".to_owned();
        increment_game_mode_derivative(&mut s);
        assert_eq!(s, "a [41 [1]");
    }
}