roba 0.2.1

//! Presentation / "eye candy" subsystem.
//!
//! Centralizes the decisions about when to render markdown, when to
//! colorize, and how to format the various surfaces (body, footer,
//! refusal warnings, errors, tool-call lines). Everything keys off a
//! single [`Style`] computed once at startup from CLI args + env +
//! TTY detection.

use std::io::IsTerminal;

use crate::cli::AskArgs;

/// Resolved presentation policy for one roba invocation. Built once
/// and threaded through wherever output happens.
#[derive(Debug, Clone, Copy)]
pub struct Style {
    /// Render markdown bodies through termimad.
    pub render_markdown: bool,
    /// Use ANSI color anywhere we emit decoration.
    pub color: bool,
    /// Draw a spinner during the claude call (non-streaming path).
    pub spinner: bool,
}

impl Style {
    /// Resolve from CLI args + environment + TTY detection.
    pub fn detect(args: &AskArgs) -> Self {
        let stdout_tty = std::io::stdout().is_terminal();
        let stderr_tty = std::io::stderr().is_terminal();
        let no_color = std::env::var_os("NO_COLOR").is_some();
        let plain = args.plain;

        // Markdown rendering is for human consumption. Skip when the
        // user explicitly opted out, when stdout isn't a TTY, when
        // they asked for a structured / extracted form, or in stream
        // mode (we'd have to buffer to render markdown, defeating the
        // purpose of streaming).
        let render_markdown =
            !plain && stdout_tty && !args.json && !args.stream && args.code.is_none();

        // Color governs the footer, refusal warning, error prefixes,
        // and tool call lines. Off when piping, when --plain, or when
        // NO_COLOR is set.
        let color = !plain && stdout_tty && !no_color;

        // Spinner draws on stderr. Skip in --stream (tokens are
        // already arriving), --plain, or when stderr isn't a TTY
        // (e.g. captured for logging).
        let spinner = !plain && stderr_tty && !args.stream;

        Self {
            render_markdown,
            color,
            spinner,
        }
    }

    /// Build a style with all visual features off. Useful when the
    /// caller has already decided rendering is inappropriate (e.g.
    /// inside `--out` writing).
    pub fn plain() -> Self {
        Self {
            render_markdown: false,
            color: false,
            spinner: false,
        }
    }

    /// Best-effort style for surfaces that fire before AskArgs is
    /// fully resolved (most notably the fatal-error path in main).
    /// Honors NO_COLOR + stderr-TTY only -- no --plain visibility.
    pub fn detect_for_error() -> Self {
        let stderr_tty = std::io::stderr().is_terminal();
        let no_color = std::env::var_os("NO_COLOR").is_some();
        Self {
            render_markdown: false,
            color: stderr_tty && !no_color,
            spinner: false,
        }
    }

    /// Style for subcommands (history, last, profile, cost) that
    /// emit display output but don't have AskArgs in scope. Renders
    /// markdown on TTY, honors NO_COLOR, no spinner (no live call).
    pub fn detect_for_subcommand() -> Self {
        let stdout_tty = std::io::stdout().is_terminal();
        let no_color = std::env::var_os("NO_COLOR").is_some();
        Self {
            render_markdown: stdout_tty && !no_color,
            color: stdout_tty && !no_color,
            spinner: false,
        }
    }
}

/// Print the answer body to stdout, optionally with markdown
/// rendering applied. The cargo-style 3-space body indent is
/// imposed by this function; plain mode prints with no indent so
/// pipe consumers get raw text. Termimad is asked to wrap to
/// `terminal_width - 3` so soft-wrapping doesn't push lines past
/// the right edge.
pub fn print_body(text: &str, style: &Style) {
    if !style.render_markdown {
        println!("{text}");
        return;
    }
    let skin = build_skin(style.color);
    let width = terminal_width().saturating_sub(3).max(20);
    let rendered = skin.text(text, Some(width));
    let rendered_string = format!("{rendered}");
    let trimmed = rendered_string.trim_end_matches('\n');
    for line in trimmed.split('\n') {
        if line.is_empty() {
            println!();
        } else {
            println!("   {line}");
        }
    }
}

fn terminal_width() -> usize {
    termimad::crossterm::terminal::size()
        .map(|(w, _)| w as usize)
        .unwrap_or(80)
}

fn build_skin(color: bool) -> termimad::MadSkin {
    if color {
        termimad::MadSkin::default()
    } else {
        termimad::MadSkin::no_style()
    }
}

/// Print a metadata line on stderr (cost footer, tool rollup, etc.).
/// Dim gray when color is on; otherwise just the raw text.
pub fn print_meta(line: &str, style: &Style) {
    if style.color {
        eprintln!("\x1b[2m{line}\x1b[0m");
    } else {
        eprintln!("{line}");
    }
}

/// Print a blank line on stderr -- used as a separator before the
/// footer block.
pub fn print_meta_blank() {
    eprintln!();
}

/// Print a warning line on stderr (refusal detection, soft warning).
/// Bold yellow with a `◆ warning:` prefix when color is on; plain
/// `warning:` prefix otherwise.
pub fn print_warning(message: &str, style: &Style) {
    if style.color {
        eprintln!("\x1b[1;33m◆ warning:\x1b[0m {message}");
    } else {
        eprintln!("warning: {message}");
    }
}

/// Print an error line on stderr (hard failures from main).
/// Bold red with a `◆ error:` prefix when color is on; plain
/// `error:` prefix otherwise.
pub fn print_error(message: &str, style: &Style) {
    if style.color {
        eprintln!("\x1b[1;31m◆ error:\x1b[0m {message}");
    } else {
        eprintln!("error: {message}");
    }
}

/// Stream a thinking-block delta chunk to stderr. Chunks arrive
/// piecewise without newlines so this uses `eprint!` (no trailing
/// newline) and wraps each chunk in dim/reset when color is on, so
/// ANSI state never bleeds past one chunk.
pub fn print_thinking_delta(text: &str, style: &Style) {
    use std::io::Write;
    let stderr = std::io::stderr();
    let mut lock = stderr.lock();
    if style.color {
        let _ = write!(lock, "\x1b[2m{text}\x1b[0m");
    } else {
        let _ = write!(lock, "{text}");
    }
    let _ = lock.flush();
}

/// Print a tool-call indicator on stderr during --stream. Dim with
/// a `▸` glyph when color is on; plain `>` prefix otherwise. Same
/// 3-space cargo-style indent as the rendered body, so tools sit
/// as visible sub-actions under the response.
pub fn print_tool_call(line: &str, style: &Style) {
    if style.color {
        eprintln!("   \x1b[2m▸ {line}\x1b[0m");
    } else {
        eprintln!("   > {line}");
    }
}

/// Build a minimal spinner that redraws on stderr while a future is
/// awaited. Returns `None` when the style says no spinner -- the
/// caller is responsible for handling the `Option`.
///
/// Display: `⠋ 3.2s`, redrawn every 80ms via indicatif's steady
/// tick. `finish_and_clear` removes the line before the next bit
/// of output is written.
pub fn spinner() -> indicatif::ProgressBar {
    let pb = indicatif::ProgressBar::new_spinner();
    pb.enable_steady_tick(std::time::Duration::from_millis(80));
    pb.set_style(
        indicatif::ProgressStyle::with_template("{spinner} {elapsed}").expect("static template"),
    );
    pb
}