rustyclaw_core/

theme.rs

//! Terminal theme & spinner helpers.
//!
//! Mirrors openclaw's "lobster palette" (`src/terminal/palette.ts`) and
//! `src/terminal/theme.ts`.  Respects the `NO_COLOR` env-var and the
//! `--no-color` CLI flag.
//!
//! # Palette (from openclaw docs/cli/index.md)
//!
//! | Token          | Hex       | Usage                          |
//! |----------------|-----------|--------------------------------|
//! | accent         | `#FF5A2D` | headings, labels, primary      |
//! | accent_bright  | `#FF7A3D` | command names, emphasis         |
//! | accent_dim     | `#D14A22` | secondary highlight             |
//! | info           | `#FF8A5B` | informational values            |
//! | success        | `#2FBF71` | success states                  |
//! | warn           | `#FFB020` | warnings, fallbacks             |
//! | error          | `#E23D2D` | errors, failures                |
//! | muted          | `#8B7F77` | de-emphasis, metadata           |

use colored::Colorize;
use indicatif::{ProgressBar, ProgressStyle};
use std::sync::atomic::{AtomicBool, Ordering};
use std::time::Duration;

// ── Global color toggle ─────────────────────────────────────────────────────

static COLOR_DISABLED: AtomicBool = AtomicBool::new(false);

/// Call once at startup (after CLI parsing) to disable colour globally.
pub fn disable_color() {
    COLOR_DISABLED.store(true, Ordering::Relaxed);
    colored::control::set_override(false);
}

/// Initialise the colour system.  Checks `NO_COLOR` env-var and optional
/// `--no-color` flag.
pub fn init_color(no_color_flag: bool) {
    if no_color_flag
        || std::env::var("NO_COLOR")
            .map(|v| !v.is_empty())
            .unwrap_or(false)
    {
        disable_color();
    }
}

fn is_color() -> bool {
    !COLOR_DISABLED.load(Ordering::Relaxed)
}

// ── Lobster palette ─────────────────────────────────────────────────────────

/// Lobster palette hex values — source of truth.
pub mod palette {
    pub const ACCENT: (u8, u8, u8) = (0xFF, 0x5A, 0x2D);
    pub const ACCENT_BRIGHT: (u8, u8, u8) = (0xFF, 0x7A, 0x3D);
    pub const ACCENT_DIM: (u8, u8, u8) = (0xD1, 0x4A, 0x22);
    pub const INFO: (u8, u8, u8) = (0xFF, 0x8A, 0x5B);
    pub const SUCCESS: (u8, u8, u8) = (0x2F, 0xBF, 0x71);
    pub const WARN: (u8, u8, u8) = (0xFF, 0xB0, 0x20);
    pub const ERROR: (u8, u8, u8) = (0xE2, 0x3D, 0x2D);
    pub const MUTED: (u8, u8, u8) = (0x8B, 0x7F, 0x77);
}

// ── Themed formatting helpers ───────────────────────────────────────────────
//
// Each function returns a `String` so callers can `println!("{}", accent("…"))`.

fn apply(text: &str, rgb: (u8, u8, u8)) -> String {
    if is_color() {
        text.truecolor(rgb.0, rgb.1, rgb.2).to_string()
    } else {
        text.to_string()
    }
}

fn apply_bold(text: &str, rgb: (u8, u8, u8)) -> String {
    if is_color() {
        text.truecolor(rgb.0, rgb.1, rgb.2).bold().to_string()
    } else {
        text.to_string()
    }
}

/// Primary accent (headings, labels).
pub fn accent(text: &str) -> String {
    apply(text, palette::ACCENT)
}

/// Bright accent (command names, emphasis).
pub fn accent_bright(text: &str) -> String {
    apply(text, palette::ACCENT_BRIGHT)
}

/// Dim accent (secondary highlight).
pub fn accent_dim(text: &str) -> String {
    apply(text, palette::ACCENT_DIM)
}

/// Informational values.
pub fn info(text: &str) -> String {
    apply(text, palette::INFO)
}

/// Success state.
pub fn success(text: &str) -> String {
    apply(text, palette::SUCCESS)
}

/// Warning / attention.
pub fn warn(text: &str) -> String {
    apply(text, palette::WARN)
}

/// Error / failure.
pub fn error(text: &str) -> String {
    apply(text, palette::ERROR)
}

/// De-emphasis / metadata.
pub fn muted(text: &str) -> String {
    apply(text, palette::MUTED)
}

/// Bold heading in accent colour.
pub fn heading(text: &str) -> String {
    apply_bold(text, palette::ACCENT)
}

/// Bold text (no colour).
pub fn bold(text: &str) -> String {
    if is_color() {
        text.bold().to_string()
    } else {
        text.to_string()
    }
}

/// Dimmed text (terminal dim attribute).
pub fn dim(text: &str) -> String {
    if is_color() {
        text.dimmed().to_string()
    } else {
        text.to_string()
    }
}

// ── Composite icons ─────────────────────────────────────────────────────────
//
// openclaw uses ✓ / ✗ / ⚠ with colour.

/// Green ✓
pub fn icon_ok(label: &str) -> String {
    format!("{} {}", success("✓"), label)
}

/// Red ✗
pub fn icon_fail(label: &str) -> String {
    format!("{} {}", error("✗"), label)
}

/// Yellow ⚠
pub fn icon_warn(label: &str) -> String {
    format!("{} {}", warn("⚠"), label)
}

/// Muted dash —
pub fn icon_muted(label: &str) -> String {
    format!("{} {}", muted("·"), muted(label))
}

// ── Labelled key : value ────────────────────────────────────────────────────

/// Format "  Label  : value" with the label dimmed and the value in accent.
pub fn label_value(label: &str, value: &str) -> String {
    format!("  {} : {}", muted(label), info(value))
}

// ── Spinner helpers ─────────────────────────────────────────────────────────

/// Spinner character set mimicking openclaw's clack spinners.
const SPINNER_CHARS: &[&str] = &["◒", "◐", "◓", "◑"];

/// Create an indeterminate spinner with a message.
///
/// Returns a `ProgressBar` that the caller should call `.finish_with_message()`
/// or `.finish_and_clear()` on when done.
pub fn spinner(message: &str) -> ProgressBar {
    let pb = ProgressBar::new_spinner();
    let style = if is_color() {
        ProgressStyle::with_template(&format!(
            "{{spinner:.{}}}  {{msg}}",
            "red" // indicatif colour name closest to lobster accent
        ))
        .unwrap()
        .tick_strings(SPINNER_CHARS)
    } else {
        ProgressStyle::with_template("{spinner}  {msg}")
            .unwrap()
            .tick_strings(SPINNER_CHARS)
    };
    pb.set_style(style);
    pb.set_message(message.to_string());
    pb.enable_steady_tick(Duration::from_millis(80));
    pb
}

/// Finish a spinner with a success icon + message.
pub fn spinner_ok(pb: &ProgressBar, message: &str) {
    pb.finish_with_message(icon_ok(message));
}

/// Finish a spinner with a failure icon + message.
pub fn spinner_fail(pb: &ProgressBar, message: &str) {
    pb.finish_with_message(icon_fail(message));
}

/// Finish a spinner with a warning icon + message.
pub fn spinner_warn(pb: &ProgressBar, message: &str) {
    pb.finish_with_message(icon_warn(message));
}

// ── Box drawing (for onboarding banner etc.) ────────────────────────────────

/// Print a styled header box (like openclaw's `intro()` from clack).
pub fn print_header(title: &str) {
    use unicode_width::UnicodeWidthStr;

    let display_w = UnicodeWidthStr::width(title);
    // Inner width = display width of title + at least 4 chars padding (2 each side)
    let inner = (display_w + 4).max(42);
    let pad = inner - display_w;
    let left = pad / 2;
    let right = pad - left;
    println!();
    println!("{}", accent(&format!("┌{}┐", "─".repeat(inner))));
    println!(
        "{}",
        accent(&format!(
            "│{}{}{}│",
            " ".repeat(left),
            title,
            " ".repeat(right)
        ))
    );
    println!("{}", accent(&format!("└{}┘", "─".repeat(inner))));
    println!();
}

// ── Tests ───────────────────────────────────────────────────────────────────

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

    #[test]
    fn test_no_color_output() {
        // Force no-color mode (both our flag AND the colored crate).
        COLOR_DISABLED.store(true, Ordering::Relaxed);
        colored::control::set_override(false);
        assert_eq!(accent("hello"), "hello");
        assert_eq!(success("ok"), "ok");
        assert_eq!(error("fail"), "fail");
        assert_eq!(icon_ok("done"), "✓ done");
        assert_eq!(icon_fail("bad"), "✗ bad");
        // Reset for other tests.
        colored::control::unset_override();
        COLOR_DISABLED.store(false, Ordering::Relaxed);
    }

    #[test]
    fn test_label_value() {
        COLOR_DISABLED.store(true, Ordering::Relaxed);
        let out = label_value("Key", "/some/path");
        assert!(out.contains("Key"));
        assert!(out.contains("/some/path"));
        COLOR_DISABLED.store(false, Ordering::Relaxed);
    }
}

// ── Ratatui palette ─────────────────────────────────────────────────────────
//
// The TUI palette has moved to the `rustyclaw-tui` crate.
// See `crates/rustyclaw-tui/src/tui_palette.rs`.