use gpui::{div, relative, IntoElement, Keystroke, ParentElement as _, RenderOnce, Styled as _};

use crate::ActiveTheme;

/// A key binding tag
#[derive(IntoElement)]
pub struct Kbd {
    stroke: gpui::Keystroke,
}

impl From<Keystroke> for Kbd {
    fn from(stroke: Keystroke) -> Self {
        Self { stroke }
    }
}

impl Kbd {
    pub fn new(stroke: Keystroke) -> Self {
        Self { stroke }
    }

    /// Return the Platform specific keybinding string by KeyStroke
    pub fn format(key: &Keystroke) -> String {
        if cfg!(target_os = "macos") {
            return format!("{}", key);
        }

        let mut parts = vec![];
        if key.modifiers.control {
            parts.push("Ctrl");
        }
        if key.modifiers.alt {
            parts.push("Alt");
        }
        if key.modifiers.platform {
            parts.push("Win");
        }
        if key.modifiers.shift {
            parts.push("Shift");
        }

        // Capitalize the first letter
        let key = if let Some(first_c) = key.key.chars().next() {
            format!("{}{}", first_c.to_uppercase(), &key.key[1..])
        } else {
            key.key.to_string()
        };

        parts.push(&key);
        parts.join("+")
    }
}

impl RenderOnce for Kbd {
    fn render(self, _: &mut gpui::Window, cx: &mut gpui::App) -> impl gpui::IntoElement {
        div()
            .border_1()
            .border_color(cx.theme().border)
            .text_color(cx.theme().muted_foreground)
            .bg(cx.theme().background)
            .py_0p5()
            .px_1()
            .rounded_md()
            .line_height(relative(1.))
            .text_xs()
            .child(Self::format(&self.stroke))
    }
}

#[cfg(test)]
mod tests {
    #[test]
    fn test_format() {
        use super::Kbd;
        use gpui::Keystroke;

        if cfg!(target_os = "windows") {
            assert_eq!(Kbd::format(&Keystroke::parse("a").unwrap()), "A");
            assert_eq!(Kbd::format(&Keystroke::parse("ctrl-a").unwrap()), "Ctrl+A");
            assert_eq!(
                Kbd::format(&Keystroke::parse("ctrl-alt-a").unwrap()),
                "Ctrl+Alt+A"
            );
            assert_eq!(
                Kbd::format(&Keystroke::parse("ctrl-alt-shift-a").unwrap()),
                "Ctrl+Alt+Shift+A"
            );
            assert_eq!(
                Kbd::format(&Keystroke::parse("ctrl-alt-shift-win-a").unwrap()),
                "Ctrl+Alt+Win+Shift+A"
            );
            assert_eq!(
                Kbd::format(&Keystroke::parse("ctrl-shift-backspace").unwrap()),
                "Ctrl+Shift+Backspace"
            );
        } else {
            assert_eq!(Kbd::format(&Keystroke::parse("cmd-a").unwrap()), "⌘A");
            assert_eq!(Kbd::format(&Keystroke::parse("cmd-ctrl-a").unwrap()), "^⌘A");
            assert_eq!(
                Kbd::format(&Keystroke::parse("cmd-ctrl-shift-a").unwrap()),
                "^⌘⇧A"
            );
            assert_eq!(
                Kbd::format(&Keystroke::parse("cmd-ctrl-shift-alt-a").unwrap()),
                "^⌥⌘⇧A"
            );
        }
    }
}