roboticus-core 0.11.1

use std::io::IsTerminal;

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ColorMode {
    Auto,
    Always,
    Never,
}

impl ColorMode {
    pub fn from_flag(s: &str) -> Self {
        match s {
            "always" => Self::Always,
            "never" => Self::Never,
            _ => Self::Auto,
        }
    }
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ThemeVariant {
    AiPurple,
    CrtGreen,
    CrtOrange,
    Terminal,
}

impl ThemeVariant {
    pub fn from_flag(s: &str) -> Self {
        match s {
            "ai-purple" => Self::AiPurple,
            "crt-orange" => Self::CrtOrange,
            "terminal" => Self::Terminal,
            _ => Self::CrtGreen,
        }
    }
}

// ── Braille Spinner ─────────────────────────────────────────────
// Shared across TUI and CLI for a consistent "thinking" indicator.

/// Braille spinner frames for activity indicators.
/// Cycle through these at ~80-120ms per frame.
pub const SPINNER_FRAMES: &[char] = &['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];

/// Return the spinner frame for a given tick count.
pub fn spinner_frame(tick: usize) -> char {
    SPINNER_FRAMES[tick % SPINNER_FRAMES.len()]
}

/// CLI theme with selectable color palettes and optional typewriter effects.
///
/// Precedence: `--color` flag > `NO_COLOR` env var > TTY auto-detection.
/// Draw (typewriter) is enabled by default on interactive TTY, disabled with `--no-draw`.
#[derive(Debug, Clone)]
pub struct Theme {
    enabled: bool,
    draw: bool,
    variant: ThemeVariant,
    nerdmode: bool,
}

impl Theme {
    pub fn detect() -> Self {
        Self::resolve(ColorMode::Auto, ThemeVariant::CrtGreen)
    }

    pub fn from_flags(color_flag: &str, theme_flag: &str) -> Self {
        Self::resolve(
            ColorMode::from_flag(color_flag),
            ThemeVariant::from_flag(theme_flag),
        )
    }

    pub fn resolve(mode: ColorMode, variant: ThemeVariant) -> Self {
        let enabled = match mode {
            ColorMode::Always => true,
            ColorMode::Never => false,
            ColorMode::Auto => {
                let no_color = std::env::var("NO_COLOR")
                    .map(|v| !v.is_empty())
                    .unwrap_or(false);
                if no_color {
                    false
                } else {
                    std::io::stderr().is_terminal()
                }
            }
        };
        Self {
            enabled,
            draw: enabled,
            variant,
            nerdmode: false,
        }
    }

    pub fn plain() -> Self {
        Self {
            enabled: false,
            draw: false,
            variant: ThemeVariant::CrtGreen,
            nerdmode: false,
        }
    }

    pub fn with_draw(mut self, draw: bool) -> Self {
        self.draw = draw;
        self
    }

    pub fn with_nerdmode(mut self, nerd: bool) -> Self {
        if nerd {
            self.nerdmode = true;
            self.draw = true;
            if self.variant == ThemeVariant::Terminal {
                self.variant = ThemeVariant::CrtGreen;
            }
        }
        self
    }

    pub fn colors_enabled(&self) -> bool {
        self.enabled
    }

    pub fn draw_enabled(&self) -> bool {
        self.draw
    }

    pub fn variant(&self) -> ThemeVariant {
        self.variant
    }

    pub fn nerdmode(&self) -> bool {
        self.nerdmode
    }

    // ── Icon Accessors ───────────────────────────────────────────
    // Return emoji by default; ASCII when nerdmode is active.

    pub fn icon_ok(&self) -> &'static str {
        if self.nerdmode { "[OK]" } else { "\u{2705}" }
    }

    pub fn icon_action(&self) -> &'static str {
        if self.nerdmode {
            "[>>]"
        } else {
            "\u{26a1}\u{fe0f}"
        }
    }

    pub fn icon_warn(&self) -> &'static str {
        if self.nerdmode {
            "[!!]"
        } else {
            "\u{26a0}\u{fe0f}"
        }
    }

    pub fn icon_detail(&self) -> &'static str {
        if self.nerdmode { ">" } else { "\u{25b8}" }
    }

    pub fn icon_error(&self) -> &'static str {
        if self.nerdmode { "[XX]" } else { "\u{26d3}" }
    }

    // ── Color Palette ────────────────────────────────────────────

    /// Emphasis/highlight color. Purple, green, orange, or bold depending on variant.
    pub fn accent(&self) -> &'static str {
        if !self.enabled {
            return "";
        }
        match self.variant {
            ThemeVariant::AiPurple => "\x1b[38;5;177m", // light purple (#c180ff)
            ThemeVariant::CrtGreen => "\x1b[38;5;46m",
            ThemeVariant::CrtOrange => "\x1b[38;5;208m",
            ThemeVariant::Terminal => "\x1b[1m",
        }
    }

    /// Secondary accent. Cyan for AI Purple, matches accent for others.
    pub fn secondary(&self) -> &'static str {
        if !self.enabled {
            return "";
        }
        match self.variant {
            ThemeVariant::AiPurple => "\x1b[96m", // bright cyan (#7de9ff)
            ThemeVariant::CrtGreen => "\x1b[38;5;46m",
            ThemeVariant::CrtOrange => "\x1b[38;5;208m",
            ThemeVariant::Terminal => "\x1b[96m",
        }
    }

    /// Body-text color. Matches the variant's base tone; empty for Terminal.
    pub fn dim(&self) -> &'static str {
        if !self.enabled {
            return "";
        }
        match self.variant {
            ThemeVariant::AiPurple => "\x1b[38;5;146m", // muted lavender (#9baad6)
            ThemeVariant::CrtGreen => "\x1b[38;5;40m",
            ThemeVariant::CrtOrange => "\x1b[38;5;172m",
            ThemeVariant::Terminal => "",
        }
    }

    /// Monospace-value color. Same as accent for CRT variants; bold for Terminal.
    pub fn mono(&self) -> &'static str {
        if !self.enabled {
            return "";
        }
        match self.variant {
            ThemeVariant::AiPurple => "\x1b[38;5;177m",
            ThemeVariant::CrtGreen => "\x1b[38;5;46m",
            ThemeVariant::CrtOrange => "\x1b[38;5;208m",
            ThemeVariant::Terminal => "\x1b[1m",
        }
    }

    /// Bright green. Explicit "all passed" summaries, enabled/active states.
    pub fn success(&self) -> &'static str {
        if self.enabled { "\x1b[92m" } else { "" }
    }

    /// Bright yellow. Warnings, fallback states, skipped items.
    pub fn warn(&self) -> &'static str {
        if self.enabled { "\x1b[93m" } else { "" }
    }

    /// Bright red. Errors, failures, disabled states.
    pub fn error(&self) -> &'static str {
        if self.enabled { "\x1b[91m" } else { "" }
    }

    /// Bright cyan. Auto-fix actions, debug info, discovery output.
    pub fn info(&self) -> &'static str {
        if self.enabled { "\x1b[96m" } else { "" }
    }

    // ── Typography modifiers ─────────────────────────────────────

    pub fn bold(&self) -> &'static str {
        if self.enabled { "\x1b[1m" } else { "" }
    }

    /// Soft reset: clears styles and re-tints to the variant's body color.
    /// For Terminal variant, this is a plain reset (no tint).
    pub fn reset(&self) -> &'static str {
        if !self.enabled {
            return "";
        }
        match self.variant {
            ThemeVariant::AiPurple => "\x1b[0m\x1b[38;5;146m",
            ThemeVariant::CrtGreen => "\x1b[0m\x1b[38;5;40m",
            ThemeVariant::CrtOrange => "\x1b[0m\x1b[38;5;172m",
            ThemeVariant::Terminal => "\x1b[0m",
        }
    }

    /// Hard reset: returns terminal to default colors. Use at program exit.
    pub fn hard_reset(&self) -> &'static str {
        if self.enabled { "\x1b[0m" } else { "" }
    }

    // ── Typewriter Effects ───────────────────────────────────────

    /// Typewrite to stderr, character-by-character. ANSI sequences emitted instantly.
    /// Skips delay when draw is disabled (instant print).
    pub fn typewrite(&self, text: &str, delay_ms: u64) {
        use std::io::Write;
        if !self.draw {
            eprint!("{text}");
            return;
        }
        let delay = std::time::Duration::from_millis(delay_ms);
        let mut chars = text.chars();
        while let Some(ch) = chars.next() {
            if ch == '\x1b' {
                let mut seq = String::from(ch);
                for c in chars.by_ref() {
                    seq.push(c);
                    if c == 'm' {
                        break;
                    }
                }
                eprint!("{seq}");
            } else if ch == '\n' {
                eprintln!();
            } else {
                eprint!("{ch}");
                std::io::stderr().flush().ok();
                std::thread::sleep(delay);
            }
        }
    }

    /// Typewrite to stderr + newline.
    pub fn typewrite_line(&self, text: &str, delay_ms: u64) {
        self.typewrite(text, delay_ms);
        eprintln!();
    }

    /// Typewrite to stdout, character-by-character.
    pub fn typewrite_stdout(&self, text: &str, delay_ms: u64) {
        use std::io::Write;
        if !self.draw {
            print!("{text}");
            return;
        }
        let delay = std::time::Duration::from_millis(delay_ms);
        let mut chars = text.chars();
        while let Some(ch) = chars.next() {
            if ch == '\x1b' {
                let mut seq = String::from(ch);
                for c in chars.by_ref() {
                    seq.push(c);
                    if c == 'm' {
                        break;
                    }
                }
                print!("{seq}");
            } else if ch == '\n' {
                println!();
            } else {
                print!("{ch}");
                std::io::stdout().flush().ok();
                std::thread::sleep(delay);
            }
        }
    }

    /// Typewrite to stdout + newline.
    pub fn typewrite_line_stdout(&self, text: &str, delay_ms: u64) {
        use std::io::Write;
        self.typewrite_stdout(text, delay_ms);
        println!();
        std::io::stdout().flush().ok();
    }
}

/// Convenience sleep in milliseconds.
pub fn sleep_ms(ms: u64) {
    std::thread::sleep(std::time::Duration::from_millis(ms));
}

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

    #[test]
    fn plain_theme_returns_empty_strings() {
        let t = Theme::plain();
        assert!(!t.colors_enabled());
        assert!(!t.draw_enabled());
        assert_eq!(t.accent(), "");
        assert_eq!(t.dim(), "");
        assert_eq!(t.mono(), "");
        assert_eq!(t.success(), "");
        assert_eq!(t.warn(), "");
        assert_eq!(t.error(), "");
        assert_eq!(t.info(), "");
        assert_eq!(t.bold(), "");
        assert_eq!(t.reset(), "");
        assert_eq!(t.hard_reset(), "");
    }

    #[test]
    fn always_mode_forces_color_and_draw() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::CrtGreen);
        assert!(t.colors_enabled());
        assert!(t.draw_enabled());
        assert!(!t.accent().is_empty());
        assert!(!t.reset().is_empty());
    }

    #[test]
    fn with_draw_false_disables_draw() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::CrtGreen).with_draw(false);
        assert!(t.colors_enabled());
        assert!(!t.draw_enabled());
    }

    #[test]
    fn crt_green_reset_includes_green_tint() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::CrtGreen);
        let r = t.reset();
        assert!(r.contains("\x1b[0m"), "reset should clear styles");
        assert!(
            r.contains("\x1b[38;5;40m"),
            "CrtGreen reset should tint green"
        );
    }

    #[test]
    fn crt_orange_palette() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::CrtOrange);
        assert!(t.accent().contains("208"), "CrtOrange accent should be 208");
        assert!(t.dim().contains("172"), "CrtOrange dim should be 172");
        assert!(
            t.reset().contains("172"),
            "CrtOrange reset should tint orange"
        );
    }

    #[test]
    fn terminal_variant_no_tint() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::Terminal);
        assert_eq!(t.reset(), "\x1b[0m", "Terminal reset should be plain");
        assert_eq!(t.dim(), "", "Terminal dim should be empty");
        assert_eq!(t.accent(), "\x1b[1m", "Terminal accent should be bold");
    }

    #[test]
    fn hard_reset_is_plain() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::CrtGreen);
        assert_eq!(t.hard_reset(), "\x1b[0m");
    }

    #[test]
    fn never_mode_disables_everything() {
        let t = Theme::resolve(ColorMode::Never, ThemeVariant::CrtGreen);
        assert!(!t.colors_enabled());
        assert!(!t.draw_enabled());
        assert_eq!(t.accent(), "");
    }

    #[test]
    fn from_flag_parses_correctly() {
        assert_eq!(ColorMode::from_flag("always"), ColorMode::Always);
        assert_eq!(ColorMode::from_flag("never"), ColorMode::Never);
        assert_eq!(ColorMode::from_flag("auto"), ColorMode::Auto);
        assert_eq!(ColorMode::from_flag("garbage"), ColorMode::Auto);
    }

    #[test]
    fn theme_variant_from_flag() {
        assert_eq!(ThemeVariant::from_flag("ai-purple"), ThemeVariant::AiPurple);
        assert_eq!(ThemeVariant::from_flag("crt-green"), ThemeVariant::CrtGreen);
        assert_eq!(
            ThemeVariant::from_flag("crt-orange"),
            ThemeVariant::CrtOrange
        );
        assert_eq!(ThemeVariant::from_flag("terminal"), ThemeVariant::Terminal);
        assert_eq!(ThemeVariant::from_flag("garbage"), ThemeVariant::CrtGreen);
    }

    #[test]
    fn semantic_colors_same_across_variants() {
        let purple = Theme::resolve(ColorMode::Always, ThemeVariant::AiPurple);
        let green = Theme::resolve(ColorMode::Always, ThemeVariant::CrtGreen);
        let orange = Theme::resolve(ColorMode::Always, ThemeVariant::CrtOrange);
        let term = Theme::resolve(ColorMode::Always, ThemeVariant::Terminal);

        assert_eq!(purple.success(), green.success());
        assert_eq!(green.success(), orange.success());
        assert_eq!(green.success(), term.success());
        assert_eq!(purple.warn(), green.warn());
        assert_eq!(green.warn(), orange.warn());
        assert_eq!(purple.error(), green.error());
        assert_eq!(green.error(), orange.error());
        assert_eq!(purple.info(), green.info());
        assert_eq!(green.info(), orange.info());
    }

    #[test]
    fn nerdmode_forces_ascii_icons() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::CrtGreen).with_nerdmode(true);
        assert_eq!(t.icon_ok(), "[OK]");
        assert_eq!(t.icon_action(), "[>>]");
        assert_eq!(t.icon_warn(), "[!!]");
        assert_eq!(t.icon_detail(), ">");
        assert_eq!(t.icon_error(), "[XX]");
    }

    #[test]
    fn nerdmode_overrides_terminal_to_green() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::Terminal).with_nerdmode(true);
        assert_eq!(t.variant(), ThemeVariant::CrtGreen);
        assert!(t.reset().contains("\x1b[38;5;40m"));
    }

    #[test]
    fn nerdmode_respects_orange() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::CrtOrange).with_nerdmode(true);
        assert_eq!(t.variant(), ThemeVariant::CrtOrange);
        assert!(t.accent().contains("208"));
    }

    #[test]
    fn nerdmode_forces_draw() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::CrtGreen)
            .with_draw(false)
            .with_nerdmode(true);
        assert!(t.draw_enabled());
    }

    #[test]
    fn default_icons_are_emoji() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::CrtGreen);
        assert!(!t.nerdmode());
        assert_eq!(t.icon_ok(), "\u{2705}");
        assert_eq!(t.icon_action(), "\u{26a1}\u{fe0f}");
        assert_eq!(t.icon_warn(), "\u{26a0}\u{fe0f}");
        assert_eq!(t.icon_detail(), "\u{25b8}");
        assert_eq!(t.icon_error(), "\u{26d3}");
    }

    #[test]
    fn from_flags_produces_correct_theme() {
        let t = Theme::from_flags("always", "crt-orange");
        assert!(t.colors_enabled());
        assert_eq!(t.variant(), ThemeVariant::CrtOrange);

        let t2 = Theme::from_flags("never", "terminal");
        assert!(!t2.colors_enabled());
        assert_eq!(t2.variant(), ThemeVariant::Terminal);
    }

    #[test]
    fn from_flags_unknown_defaults() {
        let t = Theme::from_flags("auto", "garbage");
        assert_eq!(t.variant(), ThemeVariant::CrtGreen);
    }

    #[test]
    fn detect_returns_a_theme() {
        let t = Theme::detect();
        assert_eq!(t.variant(), ThemeVariant::CrtGreen);
    }

    #[test]
    fn mono_colors_per_variant() {
        let purple = Theme::resolve(ColorMode::Always, ThemeVariant::AiPurple);
        assert_eq!(purple.mono(), "\x1b[38;5;177m");

        let green = Theme::resolve(ColorMode::Always, ThemeVariant::CrtGreen);
        assert_eq!(green.mono(), "\x1b[38;5;46m");

        let orange = Theme::resolve(ColorMode::Always, ThemeVariant::CrtOrange);
        assert_eq!(orange.mono(), "\x1b[38;5;208m");

        let term = Theme::resolve(ColorMode::Always, ThemeVariant::Terminal);
        assert_eq!(term.mono(), "\x1b[1m");
    }

    #[test]
    fn typewrite_instant_when_draw_disabled() {
        let t = Theme::plain();
        assert!(!t.draw_enabled());
        t.typewrite("hello", 100);
        t.typewrite("with \x1b[1m ansi \x1b[0m codes", 100);
    }

    #[test]
    fn typewrite_line_instant_when_draw_disabled() {
        let t = Theme::plain();
        t.typewrite_line("hello line", 100);
    }

    #[test]
    fn typewrite_stdout_instant_when_draw_disabled() {
        let t = Theme::plain();
        t.typewrite_stdout("stdout text", 100);
    }

    #[test]
    fn typewrite_line_stdout_instant_when_draw_disabled() {
        let t = Theme::plain();
        t.typewrite_line_stdout("stdout line", 100);
    }

    #[test]
    fn typewrite_with_draw_enabled_processes_ansi() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::CrtGreen);
        assert!(t.draw_enabled());
        t.typewrite("ab\x1b[1mc\x1b[0m\nend", 0);
    }

    #[test]
    fn typewrite_stdout_with_draw_enabled() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::CrtGreen);
        t.typewrite_stdout("ab\x1b[1mc\x1b[0m\nend", 0);
    }

    #[test]
    fn sleep_ms_does_not_panic() {
        sleep_ms(0);
        sleep_ms(1);
    }

    #[test]
    fn with_draw_true_enables_draw() {
        let t = Theme::plain().with_draw(true);
        assert!(t.draw_enabled());
    }

    #[test]
    fn nerdmode_false_is_noop() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::Terminal).with_nerdmode(false);
        assert!(!t.nerdmode());
    }

    #[test]
    fn semantic_colors_plain_theme() {
        let t = Theme::plain();
        assert_eq!(t.success(), "");
        assert_eq!(t.warn(), "");
        assert_eq!(t.error(), "");
        assert_eq!(t.info(), "");
        assert_eq!(t.bold(), "");
    }

    #[test]
    fn dim_per_variant() {
        let purple = Theme::resolve(ColorMode::Always, ThemeVariant::AiPurple);
        assert_eq!(purple.dim(), "\x1b[38;5;146m");

        let green = Theme::resolve(ColorMode::Always, ThemeVariant::CrtGreen);
        assert_eq!(green.dim(), "\x1b[38;5;40m");

        let orange = Theme::resolve(ColorMode::Always, ThemeVariant::CrtOrange);
        assert_eq!(orange.dim(), "\x1b[38;5;172m");

        let term = Theme::resolve(ColorMode::Always, ThemeVariant::Terminal);
        assert_eq!(term.dim(), "");
    }

    #[test]
    fn reset_per_variant() {
        let purple = Theme::resolve(ColorMode::Always, ThemeVariant::AiPurple);
        assert!(purple.reset().contains("\x1b[38;5;146m"));

        let green = Theme::resolve(ColorMode::Always, ThemeVariant::CrtGreen);
        assert!(green.reset().contains("\x1b[38;5;40m"));

        let orange = Theme::resolve(ColorMode::Always, ThemeVariant::CrtOrange);
        assert!(orange.reset().contains("\x1b[38;5;172m"));

        let term = Theme::resolve(ColorMode::Always, ThemeVariant::Terminal);
        assert_eq!(term.reset(), "\x1b[0m");
    }

    #[test]
    fn ai_purple_palette() {
        let t = Theme::resolve(ColorMode::Always, ThemeVariant::AiPurple);
        assert!(t.accent().contains("177"), "AiPurple accent should be 177");
        assert_eq!(
            t.secondary(),
            "\x1b[96m",
            "AiPurple secondary should be bright cyan"
        );
        assert!(t.dim().contains("146"), "AiPurple dim should be 146");
        assert!(
            t.reset().contains("146"),
            "AiPurple reset should tint lavender"
        );
    }

    #[test]
    fn spinner_frame_cycles() {
        assert_eq!(spinner_frame(0), '⠋');
        assert_eq!(spinner_frame(1), '⠙');
        assert_eq!(spinner_frame(9), '⠏');
        // Wraps around
        assert_eq!(spinner_frame(10), '⠋');
        assert_eq!(spinner_frame(15), '⠴');
    }

    #[test]
    fn spinner_frames_has_10_entries() {
        assert_eq!(SPINNER_FRAMES.len(), 10);
    }
}