claux 20260416.0.1

//! Theme system for claux.
//!
//! Provides RGB color definitions for the TUI with support for multiple themes
//! (dark, light, ansi) and runtime switching.

use ratatui::style::Color;

/// Available theme names.
#[derive(Debug, Clone, Copy, PartialEq, Default)]
pub enum ThemeName {
    /// Dark theme (default) - gruvbox-inspired
    #[default]
    Dark,
    /// Light theme
    Light,
    /// ANSI 16-color fallback
    Ansi,
    /// Dracula theme
    Dracula,
    /// Nord theme
    Nord,
    /// Catppuccin Mocha theme
    Catppuccin,
}

/// Color palette for a theme.
#[derive(Debug, Clone, Copy, PartialEq)]
pub struct Theme {
    // UI elements
    pub fg: Color,
    pub bg: Color,
    pub dim: Color,
    pub bold: Color,

    // Semantic colors
    pub success: Color,
    pub error: Color,
    pub warning: Color,
    pub info: Color,

    // Assistant/user message colors
    pub assistant: Color,
    pub assistant_bold: Color,
    pub user: Color,
    pub user_bold: Color,
    pub user_message_bg: Color, // Background for user messages (inverted)
    pub user_message_fg: Color, // Foreground for user messages (inverted)

    // Tool colors
    pub tool_name: Color,
    pub tool_summary: Color,
    pub tool_success: Color,
    pub tool_error: Color,

    // Diff colors
    pub diff_added: Color,
    pub diff_removed: Color,
    pub diff_added_dim: Color,
    pub diff_removed_dim: Color,

    // Status bar
    pub status_bg: Color,
    pub status_fg: Color,
    pub thinking: Color,

    // Borders and dividers
    pub border: Color,
    pub border_bold: Color,
    pub divider: Color,
}

impl Theme {
    /// Get the default dark theme.
    pub fn dark() -> Self {
        Self {
            fg: Color::Rgb(213, 196, 161),   // gruvbox fg2
            bg: Color::Rgb(40, 40, 40),      // gruvbox bg
            dim: Color::Rgb(146, 131, 116),  // gruvbox gray
            bold: Color::Rgb(250, 233, 213), // gruvbox fg1

            success: Color::Rgb(184, 187, 38), // gruvbox green
            error: Color::Rgb(251, 73, 52),    // gruvbox red
            warning: Color::Rgb(250, 189, 47), // gruvbox yellow
            info: Color::Rgb(131, 165, 152),   // gruvbox blue

            assistant: Color::Rgb(131, 165, 152), // gruvbox blue (purple alternative)
            assistant_bold: Color::Rgb(211, 134, 155), // gruvbox purple
            user: Color::Rgb(184, 187, 38),       // gruvbox green
            user_bold: Color::Rgb(250, 233, 213), // gruvbox fg1
            user_message_bg: Color::Rgb(60, 60, 60), // Darker than bg for bubble effect
            user_message_fg: Color::Rgb(213, 196, 161), // Same as fg for readability

            tool_name: Color::Rgb(250, 189, 47), // gruvbox yellow
            tool_summary: Color::Rgb(146, 131, 116), // gruvbox gray
            tool_success: Color::Rgb(184, 187, 38), // gruvbox green
            tool_error: Color::Rgb(251, 73, 52), // gruvbox red

            diff_added: Color::Rgb(152, 195, 121), // lighter green
            diff_removed: Color::Rgb(224, 108, 117), // lighter red
            diff_added_dim: Color::Rgb(99, 166, 71), // darker green
            diff_removed_dim: Color::Rgb(184, 68, 77), // darker red

            status_bg: Color::Rgb(60, 60, 60),
            status_fg: Color::Rgb(213, 196, 161),
            thinking: Color::Rgb(184, 187, 38), // gruvbox green

            border: Color::Rgb(100, 100, 100),
            border_bold: Color::Rgb(180, 180, 180),
            divider: Color::Rgb(80, 80, 80),
        }
    }

    /// Get the light theme.
    pub fn light() -> Self {
        Self {
            fg: Color::Rgb(60, 60, 60),     // dark gray
            bg: Color::Rgb(250, 250, 250),  // off-white
            dim: Color::Rgb(140, 140, 140), // medium gray
            bold: Color::Rgb(30, 30, 30),   // near black

            success: Color::Rgb(84, 128, 0),  // dark green
            error: Color::Rgb(200, 60, 60),   // red
            warning: Color::Rgb(200, 150, 0), // amber
            info: Color::Rgb(40, 80, 180),    // blue

            assistant: Color::Rgb(80, 120, 180),        // blue
            assistant_bold: Color::Rgb(140, 80, 160),   // purple
            user: Color::Rgb(84, 128, 0),               // green
            user_bold: Color::Rgb(30, 30, 30),          // near black
            user_message_bg: Color::Rgb(220, 220, 220), // Lighter than bg for bubble effect
            user_message_fg: Color::Rgb(60, 60, 60),    // Dark fg for readability

            tool_name: Color::Rgb(200, 150, 0),      // amber
            tool_summary: Color::Rgb(140, 140, 140), // gray
            tool_success: Color::Rgb(84, 128, 0),    // green
            tool_error: Color::Rgb(200, 60, 60),     // red

            diff_added: Color::Rgb(120, 180, 60),
            diff_removed: Color::Rgb(200, 100, 100),
            diff_added_dim: Color::Rgb(180, 220, 150),
            diff_removed_dim: Color::Rgb(240, 200, 200),

            status_bg: Color::Rgb(220, 220, 220),
            status_fg: Color::Rgb(60, 60, 60),
            thinking: Color::Rgb(84, 128, 0), // green

            border: Color::Rgb(180, 180, 180),
            border_bold: Color::Rgb(100, 100, 100),
            divider: Color::Rgb(200, 200, 200),
        }
    }

    /// Get the ANSI 16-color theme (fallback for limited terminals).
    pub fn ansi() -> Self {
        Self {
            fg: Color::White,
            bg: Color::Black,
            dim: Color::DarkGray,
            bold: Color::White,

            success: Color::Green,
            error: Color::Red,
            warning: Color::Yellow,
            info: Color::Blue,

            assistant: Color::Blue,
            assistant_bold: Color::Magenta,
            user: Color::Green,
            user_bold: Color::White,
            user_message_bg: Color::DarkGray, // Inverted bg for ANSI
            user_message_fg: Color::White,    // Inverted fg for ANSI

            tool_name: Color::Yellow,
            tool_summary: Color::DarkGray,
            tool_success: Color::Green,
            tool_error: Color::Red,

            diff_added: Color::Green,
            diff_removed: Color::Red,
            diff_added_dim: Color::Rgb(99, 166, 71), // darker green fallback
            diff_removed_dim: Color::Rgb(184, 68, 77), // darker red fallback

            status_bg: Color::DarkGray,
            status_fg: Color::White,
            thinking: Color::Green,

            border: Color::White,
            border_bold: Color::White,
            divider: Color::DarkGray,
        }
    }

    /// Get the Dracula theme.
    /// Based on https://draculatheme.com/
    pub fn dracula() -> Self {
        Self {
            fg: Color::Rgb(248, 248, 242), // p:fg
            bg: Color::Rgb(40, 42, 54),    // p:bg
            dim: Color::Rgb(98, 114, 164), // p:comment
            bold: Color::Rgb(255, 255, 255),

            success: Color::Rgb(139, 233, 253), // p:cyan
            error: Color::Rgb(255, 85, 85),     // p:red
            warning: Color::Rgb(255, 184, 108), // p:yellow
            info: Color::Rgb(139, 233, 253),    // p:cyan

            assistant: Color::Rgb(189, 147, 249), // p:purple
            assistant_bold: Color::Rgb(255, 121, 198), // p:pink
            user: Color::Rgb(80, 250, 123),       // p:green
            user_bold: Color::Rgb(248, 248, 242),
            user_message_bg: Color::Rgb(68, 71, 90), // Dracula overlay
            user_message_fg: Color::Rgb(248, 248, 242),

            tool_name: Color::Rgb(255, 184, 108),   // p:yellow
            tool_summary: Color::Rgb(98, 114, 164), // p:comment
            tool_success: Color::Rgb(80, 250, 123), // p:green
            tool_error: Color::Rgb(255, 85, 85),    // p:red

            diff_added: Color::Rgb(139, 233, 253),
            diff_removed: Color::Rgb(255, 184, 108),
            diff_added_dim: Color::Rgb(60, 180, 200),
            diff_removed_dim: Color::Rgb(220, 150, 80),

            status_bg: Color::Rgb(68, 71, 90),
            status_fg: Color::Rgb(248, 248, 242),
            thinking: Color::Rgb(80, 250, 123),

            border: Color::Rgb(98, 114, 164),
            border_bold: Color::Rgb(189, 147, 249),
            divider: Color::Rgb(68, 71, 90),
        }
    }

    /// Get the Nord theme.
    /// Based on https://www.nordtheme.com/
    pub fn nord() -> Self {
        Self {
            fg: Color::Rgb(236, 239, 244), // n:fg
            bg: Color::Rgb(46, 52, 64),    // n:bg
            dim: Color::Rgb(94, 120, 156), // n:comment
            bold: Color::Rgb(248, 250, 252),

            success: Color::Rgb(163, 190, 140), // n:green
            error: Color::Rgb(191, 97, 106),    // n:red
            warning: Color::Rgb(235, 203, 139), // n:yellow
            info: Color::Rgb(88, 166, 224),     // n:blue

            assistant: Color::Rgb(143, 188, 187),      // n:cyan
            assistant_bold: Color::Rgb(180, 142, 173), // n:magenta
            user: Color::Rgb(163, 190, 140),           // n:green
            user_bold: Color::Rgb(236, 239, 244),
            user_message_bg: Color::Rgb(59, 66, 82), // Nord overlay
            user_message_fg: Color::Rgb(236, 239, 244),

            tool_name: Color::Rgb(235, 203, 139),    // n:yellow
            tool_summary: Color::Rgb(94, 120, 156),  // n:comment
            tool_success: Color::Rgb(163, 190, 140), // n:green
            tool_error: Color::Rgb(191, 97, 106),    // n:red

            diff_added: Color::Rgb(163, 190, 140),
            diff_removed: Color::Rgb(191, 97, 106),
            diff_added_dim: Color::Rgb(130, 160, 110),
            diff_removed_dim: Color::Rgb(160, 70, 80),

            status_bg: Color::Rgb(59, 66, 82),
            status_fg: Color::Rgb(236, 239, 244),
            thinking: Color::Rgb(163, 190, 140),

            border: Color::Rgb(94, 120, 156),
            border_bold: Color::Rgb(143, 188, 187),
            divider: Color::Rgb(59, 66, 82),
        }
    }

    /// Get the Catppuccin Mocha theme.
    /// Based on https://catppuccin.com/
    pub fn catppuccin() -> Self {
        Self {
            fg: Color::Rgb(205, 214, 244),  // base: text
            bg: Color::Rgb(30, 34, 44),     // base: base
            dim: Color::Rgb(148, 157, 180), // mantle: subtext0
            bold: Color::Rgb(249, 250, 255),

            success: Color::Rgb(166, 227, 161), // green
            error: Color::Rgb(243, 139, 168),   // red
            warning: Color::Rgb(250, 179, 135), // peach
            info: Color::Rgb(137, 180, 250),    // blue

            assistant: Color::Rgb(180, 194, 255), // lavender
            assistant_bold: Color::Rgb(231, 130, 198), // pink
            user: Color::Rgb(166, 227, 161),      // green
            user_bold: Color::Rgb(205, 214, 244),
            user_message_bg: Color::Rgb(41, 45, 60), // Catppuccin overlay
            user_message_fg: Color::Rgb(205, 214, 244),

            tool_name: Color::Rgb(250, 179, 135),    // peach
            tool_summary: Color::Rgb(148, 157, 180), // subtext0
            tool_success: Color::Rgb(166, 227, 161), // green
            tool_error: Color::Rgb(243, 139, 168),   // red

            diff_added: Color::Rgb(166, 227, 161),
            diff_removed: Color::Rgb(243, 139, 168),
            diff_added_dim: Color::Rgb(130, 190, 125),
            diff_removed_dim: Color::Rgb(210, 100, 135),

            status_bg: Color::Rgb(41, 45, 60),
            status_fg: Color::Rgb(205, 214, 244),
            thinking: Color::Rgb(166, 227, 161),

            border: Color::Rgb(148, 157, 180),
            border_bold: Color::Rgb(180, 194, 255),
            divider: Color::Rgb(41, 45, 60),
        }
    }

    /// Get the theme by name.
    pub fn from_name(name: ThemeName) -> Self {
        match name {
            ThemeName::Dark => Self::dark(),
            ThemeName::Light => Self::light(),
            ThemeName::Ansi => Self::ansi(),
            ThemeName::Dracula => Self::dracula(),
            ThemeName::Nord => Self::nord(),
            ThemeName::Catppuccin => Self::catppuccin(),
        }
    }
}

/// Get the default theme.
pub fn default_theme() -> Theme {
    Theme::dark()
}

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

    #[test]
    fn test_dark_theme_colors() {
        let theme = Theme::dark();
        assert_eq!(theme.fg, Color::Rgb(213, 196, 161));
        assert_eq!(theme.bg, Color::Rgb(40, 40, 40));
        assert_eq!(theme.success, Color::Rgb(184, 187, 38));
    }

    #[test]
    fn test_light_theme_colors() {
        let theme = Theme::light();
        assert_eq!(theme.fg, Color::Rgb(60, 60, 60));
        assert_eq!(theme.bg, Color::Rgb(250, 250, 250));
        assert_eq!(theme.success, Color::Rgb(84, 128, 0));
    }

    #[test]
    fn test_theme_from_name() {
        assert_eq!(Theme::from_name(ThemeName::Dark), Theme::dark());
        assert_eq!(Theme::from_name(ThemeName::Light), Theme::light());
        assert_eq!(Theme::from_name(ThemeName::Ansi), Theme::ansi());
    }
}