kimun_notes/settings/themes/

mod.rs

use ratatui::style::{Color, Style};
use serde::{Deserialize, Deserializer, Serialize, Serializer};
use std::fmt::Display;

/// Built-in theme definitions (`Theme::gruvbox_dark()`, `Theme::nord()`, …).
mod builtin;

#[derive(Debug, Clone, PartialEq)]
pub enum ThemeColor {
    Rgb(u8, u8, u8),
    /// Terminal ANSI color index (0–15 for the standard palette, up to 255 for
    /// 256-color mode). The actual color is determined by the user's terminal.
    Ansi(u8),
    /// The terminal's default foreground or background color.
    Reset,
}

impl Serialize for ThemeColor {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: Serializer,
    {
        match self {
            ThemeColor::Rgb(r, g, b) => {
                serializer.serialize_str(&format!("#{:02x}{:02x}{:02x}", r, g, b))
            }
            ThemeColor::Ansi(n) => serializer.serialize_str(&format!("ansi:{}", n)),
            ThemeColor::Reset => serializer.serialize_str("reset"),
        }
    }
}

impl<'de> Deserialize<'de> for ThemeColor {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: Deserializer<'de>,
    {
        let s = String::deserialize(deserializer)?;
        ThemeColor::from_string(&s).map_err(serde::de::Error::custom)
    }
}

impl ThemeColor {
    pub fn new(r: u8, g: u8, b: u8) -> Self {
        ThemeColor::Rgb(r, g, b)
    }

    /// Convert to the corresponding ratatui `Color`.
    ///
    /// ANSI indices 0–15 map to ratatui's named color variants so they emit
    /// the standard SGR codes (30–37 / 90–97) the terminal's palette is keyed
    /// to, rather than the 256-color `38;5;n` form which some terminals
    /// remap inconsistently for the low 16 slots.
    pub fn to_ratatui(&self) -> Color {
        match self {
            ThemeColor::Rgb(r, g, b) => Color::Rgb(*r, *g, *b),
            ThemeColor::Ansi(n) => match n {
                0 => Color::Black,
                1 => Color::Red,
                2 => Color::Green,
                3 => Color::Yellow,
                4 => Color::Blue,
                5 => Color::Magenta,
                6 => Color::Cyan,
                7 => Color::Gray,
                8 => Color::DarkGray,
                9 => Color::LightRed,
                10 => Color::LightGreen,
                11 => Color::LightYellow,
                12 => Color::LightBlue,
                13 => Color::LightMagenta,
                14 => Color::LightCyan,
                15 => Color::White,
                _ => Color::Indexed(*n),
            },
            ThemeColor::Reset => Color::Reset,
        }
    }

    /// Parse a color from a string in various formats:
    /// - RGB: "rgb(255, 128, 0)"
    /// - 3-char hex: "#abc" (expanded to #aabbcc)
    /// - 6-char hex: "#aabbcc"
    /// - ANSI index: "ansi:4" (0–255)
    /// - Terminal default: "reset"
    pub fn from_string(s: &str) -> Result<Self, String> {
        let s = s.trim();

        if s.starts_with('#') {
            Self::from_hex(s)
        } else if s.starts_with("rgb(") && s.ends_with(')') {
            Self::from_rgb_string(s)
        } else if s == "reset" {
            Ok(ThemeColor::Reset)
        } else if let Some(rest) = s.strip_prefix("ansi:") {
            rest.parse::<u8>()
                .map(ThemeColor::Ansi)
                .map_err(|_| format!("Invalid ANSI color index: {}", rest))
        } else {
            Err(format!("Invalid color format: {}", s))
        }
    }

    /// Parse hex color string (#abc or #aabbcc)
    fn from_hex(s: &str) -> Result<Self, String> {
        if !s.starts_with('#') {
            return Err("Hex color must start with #".to_string());
        }

        let hex = &s[1..];

        match hex.len() {
            3 => Self::from_hex_3char(hex),
            6 => Self::from_hex_6char(hex),
            _ => Err(format!(
                "Invalid hex color length: expected 3 or 6 chars, got {}",
                hex.len()
            )),
        }
    }

    /// Parse 3-character hex color (e.g., "abc" -> r=0xaa, g=0xbb, b=0xcc)
    fn from_hex_3char(hex: &str) -> Result<Self, String> {
        if hex.len() != 3 {
            return Err("Expected 3 hex characters".to_string());
        }

        let r = u8::from_str_radix(&hex[0..1].repeat(2), 16)
            .map_err(|_| format!("Invalid hex character in red component: {}", &hex[0..1]))?;
        let g = u8::from_str_radix(&hex[1..2].repeat(2), 16)
            .map_err(|_| format!("Invalid hex character in green component: {}", &hex[1..2]))?;
        let b = u8::from_str_radix(&hex[2..3].repeat(2), 16)
            .map_err(|_| format!("Invalid hex character in blue component: {}", &hex[2..3]))?;

        Ok(ThemeColor::Rgb(r, g, b))
    }

    /// Parse 6-character hex color (e.g., "aabbcc")
    fn from_hex_6char(hex: &str) -> Result<Self, String> {
        if hex.len() != 6 {
            return Err("Expected 6 hex characters".to_string());
        }

        let r = u8::from_str_radix(&hex[0..2], 16)
            .map_err(|_| format!("Invalid hex characters in red component: {}", &hex[0..2]))?;
        let g = u8::from_str_radix(&hex[2..4], 16)
            .map_err(|_| format!("Invalid hex characters in green component: {}", &hex[2..4]))?;
        let b = u8::from_str_radix(&hex[4..6], 16)
            .map_err(|_| format!("Invalid hex characters in blue component: {}", &hex[4..6]))?;

        Ok(ThemeColor::Rgb(r, g, b))
    }

    /// Parse RGB string format (e.g., "rgb(255, 128, 0)")
    fn from_rgb_string(s: &str) -> Result<Self, String> {
        if !s.starts_with("rgb(") || !s.ends_with(')') {
            return Err("RGB format must be rgb(r, g, b)".to_string());
        }

        let inner = &s[4..s.len() - 1];
        let parts: Vec<&str> = inner.split(',').map(|p| p.trim()).collect();

        if parts.len() != 3 {
            return Err(format!("RGB format requires 3 values, got {}", parts.len()));
        }

        let r = parts[0]
            .parse::<u8>()
            .map_err(|_| format!("Invalid red value: {}", parts[0]))?;
        let g = parts[1]
            .parse::<u8>()
            .map_err(|_| format!("Invalid green value: {}", parts[1]))?;
        let b = parts[2]
            .parse::<u8>()
            .map_err(|_| format!("Invalid blue value: {}", parts[2]))?;

        Ok(ThemeColor::Rgb(r, g, b))
    }
}

impl Display for ThemeColor {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            ThemeColor::Rgb(r, g, b) => write!(f, "rgb({},{},{})", r, g, b),
            ThemeColor::Ansi(n) => write!(f, "ansi:{}", n),
            ThemeColor::Reset => write!(f, "reset"),
        }
    }
}

/// Theme for the TUI application.
///
/// Fields are named after the UI roles they fill, making it straightforward to
/// map any popular terminal color scheme (Gruvbox, Catppuccin, Tokyo Night, …)
/// to this struct.  Custom themes can be placed as `.toml` files in the themes
/// config directory and will be loaded automatically at startup.
///
/// # Example theme file (`~/.config/kimun/themes/mytheme.toml`)
/// ```toml
/// name = "My Theme"
/// bg               = "#1e1e2e"
/// bg_panel         = "#181825"
/// bg_selected      = "#313244"
/// fg               = "#cdd6f4"
/// fg_secondary     = "#a6adc8"
/// fg_muted         = "#6c7086"
/// fg_selected      = "#cdd6f4"
/// border           = "#45475a"
/// border_focused   = "#89b4fa"
/// accent           = "#89b4fa"
/// color_directory  = "#89dceb"
/// color_journal_date = "#94e2d5"
/// color_search_match = "#a6e3a1"
/// ```
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct Theme {
    pub name: String,

    // ── Backgrounds ─────────────────────────────────────────────────────────
    /// Main/editor background.
    pub bg: ThemeColor,
    /// Sidebar / panel background (usually slightly offset from `bg`).
    pub bg_panel: ThemeColor,
    /// Background of the currently selected row in lists.
    pub bg_selected: ThemeColor,

    // ── Foreground / text ────────────────────────────────────────────────────
    /// Primary text color.
    pub fg: ThemeColor,
    /// Secondary text: filenames, metadata, subdued hints.
    pub fg_secondary: ThemeColor,
    /// Very dim text: placeholders, separators, disabled items.
    pub fg_muted: ThemeColor,
    /// Text color of a selected/highlighted row (often brighter than `fg`).
    pub fg_selected: ThemeColor,

    // ── Borders ──────────────────────────────────────────────────────────────
    /// Default (unfocused) border color.
    pub border: ThemeColor,
    /// Border color when the pane has keyboard focus.
    pub border_focused: ThemeColor,

    // ── Accent ───────────────────────────────────────────────────────────────
    /// Primary accent: title bars, active markers, cursor highlights.
    pub accent: ThemeColor,

    // ── Semantic colors for file-list entries ────────────────────────────────
    /// Color used for directory entries in the file list.
    pub color_directory: ThemeColor,
    /// Color for the journal-date annotation line in journal entries.
    pub color_journal_date: ThemeColor,
    /// Color for highlighted search-match text.
    pub color_search_match: ThemeColor,
    /// Color for #hashtag label spans in the editor.
    #[serde(default = "default_color_tag")]
    pub color_tag: ThemeColor,
    /// Color of the `│` blockquote bar drawn in place of `>` markers.
    #[serde(default = "default_blockquote_bar")]
    pub blockquote_bar: ThemeColor,
    /// Background of fenced and indented code blocks (the "code box").
    /// Inline `code` uses `bg_selected`, not this.
    #[serde(default = "default_code_bg")]
    pub code_bg: ThemeColor,
}

/// Serde default for `color_tag` — used when deserializing older theme TOML
/// files that do not contain the field. Falls back to Gruvbox Dark's orange.
fn default_color_tag() -> ThemeColor {
    ThemeColor::from_string("#fe8019").unwrap()
}

/// Serde default for `blockquote_bar` — Gruvbox Dark's accent yellow.
fn default_blockquote_bar() -> ThemeColor {
    ThemeColor::from_string("#fabd2f").unwrap()
}

/// Serde default for `code_bg` — Gruvbox Dark's panel background.
fn default_code_bg() -> ThemeColor {
    ThemeColor::from_string("#32302f").unwrap()
}

impl Default for Theme {
    fn default() -> Self {
        Self::gruvbox_dark()
    }
}

impl Theme {
    /// Returns the appropriate border style depending on focus state.
    pub fn border_style(&self, focused: bool) -> Style {
        if focused {
            Style::default().fg(self.border_focused.to_ratatui())
        } else {
            Style::default().fg(self.border.to_ratatui())
        }
    }

    /// Base style for most surfaces: theme fg on theme bg.
    pub fn base_style(&self) -> Style {
        Style::default()
            .fg(self.fg.to_ratatui())
            .bg(self.bg.to_ratatui())
    }

    /// Panel style for sidebars and panels: theme fg on bg_panel.
    pub fn panel_style(&self) -> Style {
        Style::default()
            .fg(self.fg.to_ratatui())
            .bg(self.bg_panel.to_ratatui())
    }
}

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

    #[test]
    fn test_border_style_focused() {
        let theme = Theme::gruvbox_dark();
        let style = theme.border_style(true);
        assert_eq!(
            style,
            Style::default().fg(theme.border_focused.to_ratatui())
        );
    }

    #[test]
    fn test_border_style_unfocused() {
        let theme = Theme::gruvbox_dark();
        let style = theme.border_style(false);
        assert_eq!(style, Style::default().fg(theme.border.to_ratatui()));
    }

    #[test]
    fn test_from_hex_6char() {
        assert_eq!(
            ThemeColor::from_string("#ff8800").unwrap(),
            ThemeColor::Rgb(255, 136, 0)
        );
    }

    #[test]
    fn test_from_hex_6char_lowercase() {
        assert_eq!(
            ThemeColor::from_string("#abcdef").unwrap(),
            ThemeColor::Rgb(171, 205, 239)
        );
    }

    #[test]
    fn test_from_hex_6char_uppercase() {
        assert_eq!(
            ThemeColor::from_string("#ABCDEF").unwrap(),
            ThemeColor::Rgb(171, 205, 239)
        );
    }

    #[test]
    fn test_from_hex_3char() {
        assert_eq!(
            ThemeColor::from_string("#f80").unwrap(),
            ThemeColor::Rgb(255, 136, 0)
        );
    }

    #[test]
    fn test_from_hex_3char_expansion() {
        assert_eq!(
            ThemeColor::from_string("#abc").unwrap(),
            ThemeColor::Rgb(170, 187, 204)
        );
    }

    #[test]
    fn test_from_hex_3char_black() {
        assert_eq!(
            ThemeColor::from_string("#000").unwrap(),
            ThemeColor::Rgb(0, 0, 0)
        );
    }

    #[test]
    fn test_from_hex_3char_white() {
        assert_eq!(
            ThemeColor::from_string("#fff").unwrap(),
            ThemeColor::Rgb(255, 255, 255)
        );
    }

    #[test]
    fn test_from_rgb_string() {
        assert_eq!(
            ThemeColor::from_string("rgb(255, 128, 0)").unwrap(),
            ThemeColor::Rgb(255, 128, 0)
        );
    }

    #[test]
    fn test_from_rgb_string_no_spaces() {
        assert_eq!(
            ThemeColor::from_string("rgb(255,128,0)").unwrap(),
            ThemeColor::Rgb(255, 128, 0)
        );
    }

    #[test]
    fn test_from_rgb_string_extra_spaces() {
        assert_eq!(
            ThemeColor::from_string("rgb( 255 , 128 , 0 )").unwrap(),
            ThemeColor::Rgb(255, 128, 0)
        );
    }

    #[test]
    fn test_from_rgb_string_min_max() {
        assert_eq!(
            ThemeColor::from_string("rgb(0, 255, 0)").unwrap(),
            ThemeColor::Rgb(0, 255, 0)
        );
    }

    #[test]
    fn test_from_string_with_whitespace() {
        assert_eq!(
            ThemeColor::from_string("  #ff8800  ").unwrap(),
            ThemeColor::Rgb(255, 136, 0)
        );
    }

    #[test]
    fn test_ansi_to_ratatui() {
        // Low 16 ANSI indices map to named ratatui variants
        assert_eq!(ThemeColor::Ansi(0).to_ratatui(), Color::Black);
        assert_eq!(ThemeColor::Ansi(4).to_ratatui(), Color::Blue);
        assert_eq!(ThemeColor::Ansi(7).to_ratatui(), Color::Gray);
        assert_eq!(ThemeColor::Ansi(8).to_ratatui(), Color::DarkGray);
        assert_eq!(ThemeColor::Ansi(15).to_ratatui(), Color::White);
        // Indices >= 16 still use 256-color
        assert_eq!(ThemeColor::Ansi(42).to_ratatui(), Color::Indexed(42));
        assert_eq!(ThemeColor::Reset.to_ratatui(), Color::Reset);
    }

    #[test]
    fn test_invalid_hex_length() {
        let result = ThemeColor::from_string("#ff880");
        assert!(result.is_err());
        assert!(result.unwrap_err().contains("Invalid hex color length"));
    }

    #[test]
    fn test_invalid_hex_chars() {
        let result = ThemeColor::from_string("#gghhii");
        assert!(result.is_err());
    }

    #[test]
    fn test_missing_hash() {
        let result = ThemeColor::from_string("ff8800");
        assert!(result.is_err());
        assert!(result.unwrap_err().contains("Invalid color format"));
    }

    #[test]
    fn test_invalid_rgb_format() {
        let result = ThemeColor::from_string("rgb(255, 128)");
        assert!(result.is_err());
        assert!(result.unwrap_err().contains("requires 3 values"));
    }

    #[test]
    fn test_rgb_value_out_of_range() {
        let result = ThemeColor::from_string("rgb(256, 128, 0)");
        assert!(result.is_err());
    }

    #[test]
    fn test_rgb_negative_value() {
        let result = ThemeColor::from_string("rgb(-1, 128, 0)");
        assert!(result.is_err());
    }

    #[test]
    fn test_rgb_non_numeric() {
        let result = ThemeColor::from_string("rgb(abc, 128, 0)");
        assert!(result.is_err());
        assert!(result.unwrap_err().contains("Invalid red value"));
    }

    #[test]
    fn test_invalid_format() {
        let result = ThemeColor::from_string("not a color");
        assert!(result.is_err());
        assert!(result.unwrap_err().contains("Invalid color format"));
    }

    #[test]
    fn test_empty_string() {
        let result = ThemeColor::from_string("");
        assert!(result.is_err());
    }

    #[test]
    fn test_new_constructor() {
        assert_eq!(ThemeColor::new(255, 128, 0), ThemeColor::Rgb(255, 128, 0));
    }

    #[test]
    fn test_to_ratatui() {
        let color = ThemeColor::new(131, 165, 152);
        assert_eq!(color.to_ratatui(), Color::Rgb(131, 165, 152));
    }

    #[test]
    fn test_theme_color_serialize() {
        #[derive(Serialize)]
        struct Wrapper {
            color: ThemeColor,
        }
        let wrapper = Wrapper {
            color: ThemeColor::new(59, 130, 246),
        };
        let serialized = toml::to_string(&wrapper).unwrap();
        assert!(serialized.contains("color = \"#3b82f6\""));
    }

    #[test]
    fn test_theme_color_deserialize() {
        #[derive(Deserialize)]
        struct Wrapper {
            color: ThemeColor,
        }
        let toml_str = r###"color = "#3b82f6""###;
        let wrapper: Wrapper = toml::from_str(toml_str).unwrap();
        assert_eq!(wrapper.color, ThemeColor::Rgb(59, 130, 246));
    }

    #[test]
    fn test_theme_color_roundtrip() {
        #[derive(Serialize, Deserialize)]
        struct Wrapper {
            color: ThemeColor,
        }
        let original = Wrapper {
            color: ThemeColor::new(239, 68, 68),
        };
        let serialized = toml::to_string(&original).unwrap();
        let deserialized: Wrapper = toml::from_str(&serialized).unwrap();
        assert_eq!(original.color, deserialized.color);
    }

    #[test]
    fn test_theme_serialize_to_toml() {
        let theme = Theme::gruvbox_dark();
        let toml_string = toml::to_string_pretty(&theme).unwrap();

        assert!(toml_string.contains("name = \"Gruvbox Dark\""));
        assert!(toml_string.contains("bg = \"#282828\""));
        assert!(toml_string.contains("bg_panel = \"#32302f\""));
        assert!(toml_string.contains("border_focused = \"#fabd2f\""));
        assert!(toml_string.contains("color_journal_date = \"#8ec07c\""));
    }

    #[test]
    fn test_theme_deserialize_from_toml() {
        let toml_str = r###"
            name = "Test Theme"
            bg                 = "#282828"
            bg_panel           = "#32302f"
            bg_selected        = "#504945"
            fg                 = "#ebdbb2"
            fg_secondary       = "#a89984"
            fg_muted           = "#7c6f64"
            fg_selected        = "#fbf1c7"
            border             = "#504945"
            border_focused     = "#fabd2f"
            accent             = "#fabd2f"
            color_directory    = "#83a598"
            color_journal_date = "#8ec07c"
            color_search_match = "#b8bb26"
            color_tag          = "#fe8019"
        "###;

        let theme: Theme = toml::from_str(toml_str).unwrap();
        assert_eq!(theme.name, "Test Theme");
        assert_eq!(theme.bg, ThemeColor::new(0x28, 0x28, 0x28));
        assert_eq!(theme.border_focused, ThemeColor::new(0xfa, 0xbd, 0x2f));
        assert_eq!(theme.color_journal_date, ThemeColor::new(0x8e, 0xc0, 0x7c));
    }

    #[test]
    fn test_theme_roundtrip() {
        let original = Theme::tokyo_night();
        let toml_string = toml::to_string_pretty(&original).unwrap();
        let deserialized: Theme = toml::from_str(&toml_string).unwrap();

        assert_eq!(original.name, deserialized.name);
        assert_eq!(original.bg, deserialized.bg);
        assert_eq!(original.fg, deserialized.fg);
        assert_eq!(original.border_focused, deserialized.border_focused);
        assert_eq!(original.color_journal_date, deserialized.color_journal_date);
    }

    #[test]
    fn test_theme_color_serialize_lowercase_hex() {
        #[derive(Serialize)]
        struct Wrapper {
            color: ThemeColor,
        }
        let wrapper = Wrapper {
            color: ThemeColor::new(171, 205, 239),
        };
        let serialized = toml::to_string(&wrapper).unwrap();
        assert!(serialized.contains("color = \"#abcdef\""));
    }

    #[test]
    fn test_theme_deserialize_uppercase_hex() {
        #[derive(Deserialize)]
        struct Wrapper {
            color: ThemeColor,
        }
        let toml_str = r###"color = "#ABCDEF""###;
        let wrapper: Wrapper = toml::from_str(toml_str).unwrap();
        assert_eq!(wrapper.color, ThemeColor::Rgb(171, 205, 239));
    }

    #[test]
    fn test_theme_deserialize_3char_hex() {
        #[derive(Deserialize)]
        struct Wrapper {
            color: ThemeColor,
        }
        let toml_str = r###"color = "#abc""###;
        let wrapper: Wrapper = toml::from_str(toml_str).unwrap();
        assert_eq!(wrapper.color, ThemeColor::Rgb(170, 187, 204));
    }

    #[test]
    fn test_from_ansi_index() {
        assert_eq!(
            ThemeColor::from_string("ansi:4").unwrap(),
            ThemeColor::Ansi(4)
        );
        assert_eq!(
            ThemeColor::from_string("ansi:255").unwrap(),
            ThemeColor::Ansi(255)
        );
    }

    #[test]
    fn test_from_reset() {
        assert_eq!(ThemeColor::from_string("reset").unwrap(), ThemeColor::Reset);
    }

    #[test]
    fn test_all_builtin_themes_serialize() {
        let themes = vec![
            Theme::ansi(),
            Theme::gruvbox_dark(),
            Theme::gruvbox_light(),
            Theme::catppuccin_mocha(),
            Theme::catppuccin_latte(),
            Theme::tokyo_night(),
            Theme::tokyo_night_storm(),
            Theme::solarized_dark(),
            Theme::solarized_light(),
            Theme::nord(),
        ];
        for theme in themes {
            let toml_string = toml::to_string_pretty(&theme).unwrap();
            let roundtrip: Theme = toml::from_str(&toml_string).unwrap();
            assert_eq!(theme.name, roundtrip.name);
            assert_eq!(theme.bg, roundtrip.bg);
        }
    }

    #[test]
    fn test_ansi_theme() {
        let theme = Theme::ansi();
        assert_eq!(theme.name, "ANSI");
        assert_eq!(theme.bg, ThemeColor::Reset);
        assert_eq!(theme.fg, ThemeColor::Reset);
        assert_eq!(theme.bg_selected, ThemeColor::Ansi(4));
        assert_eq!(theme.border_focused, ThemeColor::Ansi(6));
        assert_eq!(theme.color_directory, ThemeColor::Ansi(12));
    }

    #[test]
    fn new_decoration_fields_present_and_deserialize_default() {
        // Built-in theme exposes the fields.
        let t = Theme::gruvbox_dark();
        assert_eq!(
            t.blockquote_bar,
            ThemeColor::from_string("#fabd2f").unwrap()
        );
        assert_eq!(t.code_bg, ThemeColor::from_string("#32302f").unwrap());

        // Old TOML without the fields still deserializes (serde defaults kick in).
        let toml = r##"
            name = "Old"
            bg = "#000000"
            bg_panel = "#111111"
            bg_selected = "#222222"
            fg = "#ffffff"
            fg_secondary = "#cccccc"
            fg_muted = "#888888"
            fg_selected = "#ffffff"
            border = "#333333"
            border_focused = "#444444"
            accent = "#55aaff"
            color_directory = "#66ccee"
            color_journal_date = "#77ddcc"
            color_search_match = "#88eeaa"
        "##;
        let parsed: Theme = toml::from_str(toml).expect("old theme TOML must still parse");
        assert_eq!(parsed.blockquote_bar, default_blockquote_bar());
        assert_eq!(parsed.code_bg, default_code_bg());
    }
}