Kigi-CLI/crates/codegen/kigi-tui/src/views/agent_status.rs

//! Agent status bar — composable right-aligned status items with separators.
//!
//! Provides [`AgentStatusBar`] which collects items as `Line<'static>` spans,
//! lays them out right-aligned with dim `│` separators, and renders into a
//! buffer row.  Returns hit-test areas keyed by item ID.
//!
//! # Example
//!
//! ```ignore
//! let mut status = AgentStatusBar::new(&theme);
//! status.push("context", context_line);
//! status.push("badge", badge_line);
//! let areas = status.render(buf, status_bar_rect);
//! let context_area = areas.get("context");
//! ```

use std::collections::HashMap;

use ratatui::buffer::Buffer;
use ratatui::layout::Rect;
use ratatui::style::Style;
use ratatui::text::{Line, Span};

use super::context_bar::SEPARATOR;
use super::turn_status::SPINNER_DIVISOR;
use crate::app::agent::{GoalDisplayPhase, GoalDisplayState, GoalDisplayStatus};
use crate::app::agent_view::McpInitProgress;
use crate::theme::Theme;

/// A named status bar item.
struct StatusEntry {
    /// Identifier for hit-test lookup (e.g., "context", "badge").
    id: &'static str,
    line: Line<'static>,
    /// Display width in columns.
    width: u16,
}

/// Builder for the agent status bar.
///
/// Collect items with [`push`], then call [`render`] to lay them out
/// right-aligned with separators and get back hit-test areas.
pub struct AgentStatusBar<'a> {
    items: Vec<StatusEntry>,
    theme: &'a Theme,
    /// Padding from the right edge of the status bar area.
    right_pad: u16,
}

impl<'a> AgentStatusBar<'a> {
    pub fn new(theme: &'a Theme) -> Self {
        Self {
            items: Vec::new(),
            theme,
            right_pad: 0,
        }
    }

    /// Add an item to the status bar.
    ///
    /// Items are rendered left-to-right in push order, but the entire
    /// group is right-aligned within the status bar area.
    pub fn push(&mut self, id: &'static str, line: Line<'static>) {
        let width = line.width() as u16;
        self.items.push(StatusEntry { id, line, width });
    }

    /// Build a separator span: ` │ ` in dim color.
    fn separator(&self) -> Span<'static> {
        Span::styled(
            format!(" {SEPARATOR} "),
            Style::default()
                .fg(self.theme.gray_dim)
                .bg(self.theme.bg_base),
        )
    }

    /// Render all items right-aligned into the given area.
    ///
    /// Layout: `··· item0 │ item1 │ item2` — separators appear only *between*
    /// items, never before the first or after the last.
    ///
    /// Returns a map of item ID → screen `Rect` for hit-testing.
    pub fn render(self, buf: &mut Buffer, area: Rect) -> HashMap<&'static str, Rect> {
        if area.height == 0 || area.width == 0 || self.items.is_empty() {
            return HashMap::new();
        }

        buf.set_style(area, Style::default().bg(self.theme.bg_base));

        let sep = self.separator();
        let sep_w = sep.width() as u16;

        // Total width: items plus the separators *between* them only — no
        // leading separator before the first item or trailing one after the
        // last.
        let items_width: u16 = self.items.iter().map(|e| e.width).sum();
        let num_seps = (self.items.len() as u16).saturating_sub(1);
        let total_width = items_width + num_seps * sep_w;

        // Right-align: compute starting x
        let start_x = area
            .x
            .saturating_add(area.width.saturating_sub(self.right_pad + total_width));

        let mut x = start_x;
        let mut areas = HashMap::new();

        for (i, entry) in self.items.iter().enumerate() {
            // Separator before every item except the first.
            if i > 0 {
                buf.set_span(x, area.y, &sep, sep_w);
                x += sep_w;
            }

            buf.set_line(x, area.y, &entry.line, entry.width);
            areas.insert(
                entry.id,
                Rect {
                    x,
                    y: area.y,
                    width: entry.width,
                    height: 1,
                },
            );
            x += entry.width;
        }

        areas
    }
}

/// Format a token count compactly: `500`, `1.5k`, `50k`, `1.5M`.
pub(crate) fn format_tokens_compact(tokens: i64) -> String {
    let sign = if tokens < 0 { "-" } else { "" };
    let abs = tokens.unsigned_abs();
    if abs >= 1_000_000 {
        let m = abs as f64 / 1_000_000.0;
        format!("{sign}{}", format!("{m:.1}M").replace(".0M", "M"))
    } else if abs >= 1_000 {
        let k = abs as f64 / 1_000.0;
        format!("{sign}{}", format!("{k:.1}k").replace(".0k", "k"))
    } else {
        tokens.to_string()
    }
}

/// Format elapsed milliseconds compactly: `5s`, `3m`, `2h`.
fn format_elapsed_compact(ms: u64) -> String {
    let secs = ms / 1000;
    if secs >= 3600 {
        format!("{}h", secs / 3600)
    } else if secs >= 60 {
        format!("{}m", secs / 60)
    } else {
        format!("{}s", secs)
    }
}

/// Build the status-chip label. Paused variants render their
/// `pause_label()`, Budget → "Budget", Done → "Done"; an Active goal
/// uses the shared [`active_phase_label`] suffix.
fn goal_phase_label(goal: &GoalDisplayState) -> String {
    match goal.status {
        GoalDisplayStatus::UserPaused
        | GoalDisplayStatus::BackOffPaused
        | GoalDisplayStatus::NoProgressPaused
        | GoalDisplayStatus::InfraPaused
        | GoalDisplayStatus::Blocked => goal.status.pause_label().into(),
        GoalDisplayStatus::BudgetLimited => "Budget".into(),
        GoalDisplayStatus::Complete => "Done".into(),
        GoalDisplayStatus::Active => active_phase_label(goal),
    }
}

/// Live phase suffix for an Active goal — the single source of truth
/// shared by the status chip and the goal-detail modal so they cannot
/// disagree. The transient `verifying_completion` overlay wins, then
/// `planning`, then the steady-state phase.
pub fn active_phase_label(goal: &GoalDisplayState) -> String {
    if goal.verifying_completion {
        let attempts = classifier_attempts_label(goal);
        // Omit the "(n/m)" suffix until the first counter arrives so the
        // chip reads "Verifying" instead of a confusing "Verifying (0/0)".
        return if attempts.is_empty() {
            "Verifying".into()
        } else {
            format!("Verifying ({attempts})")
        };
    }
    if goal.planning {
        return "Planning".into();
    }
    match goal.phase {
        GoalDisplayPhase::Idle => "Idle".into(),
        GoalDisplayPhase::Planning => "Planning".into(),
        GoalDisplayPhase::Executing => "Executing".into(),
    }
}

/// Format the classifier "attempts: n/m" counter for both the
/// status chip and the modal so the two displays cannot drift.
/// Returns the empty string when both fields are absent / zero — no
/// classifier run has been reserved yet, so there is no meaningful
/// counter. Callers render it only when non-empty: the chip drops the
/// `(n/m)` suffix, the modal falls back to an em-dash.
pub fn classifier_attempts_label(goal: &GoalDisplayState) -> String {
    let attempt = goal.classifier_runs_attempted.unwrap_or(0);
    let max = goal.classifier_max_runs.unwrap_or(0);
    if attempt == 0 && max == 0 {
        return String::new();
    }
    format!("{attempt}/{max}")
}

/// Build a compact goal status `Line` for the agent status bar.
///
/// Format: `[Goal: {label}]  {tokens}  {elapsed}`
///
/// When `hovered` is true the label is bolded/underlined to signal
/// clickability.  When the goal is `Active`, a braille spinner driven
/// by `tick` is prepended.
pub fn goal_status_line(
    goal: &GoalDisplayState,
    theme: &Theme,
    hovered: bool,
    tick: usize,
    context_used: Option<u64>,
    active_subagent_tokens: u64,
) -> Line<'static> {
    let label = goal_phase_label(goal);

    let tokens_str =
        format_tokens_compact(goal.live_tokens_used(context_used, active_subagent_tokens));
    let tokens_display = match goal.token_budget {
        Some(budget) if budget > 0 => {
            format!("{}/{} tokens", tokens_str, format_tokens_compact(budget))
        }
        _ => format!("{} tokens", tokens_str),
    };

    let elapsed_str = format_elapsed_compact(goal.live_elapsed_ms());

    let dim_style = Style::default().fg(theme.gray_dim).bg(theme.bg_base);
    // Paused goals use an inverted warning-colour chip so the chip background
    // visually matches the modal's `theme.warning` status row.
    let mut label_style = if goal.status.is_paused() {
        Style::default().fg(theme.bg_base).bg(theme.warning)
    } else {
        Style::default().fg(theme.accent_plan).bg(theme.bg_base)
    };

    if hovered {
        label_style = label_style
            .add_modifier(ratatui::style::Modifier::BOLD)
            .add_modifier(ratatui::style::Modifier::UNDERLINED);
    }

    let is_active = matches!(goal.status, GoalDisplayStatus::Active);

    let goal_text = if is_active {
        let frames = crate::glyphs::dot_spinner_frames();
        let frame = frames[(tick / 4) % frames.len()];
        format!("{frame} Goal: {label}")
    } else {
        format!("Goal: {label}")
    };

    Line::from(vec![
        Span::styled("[", dim_style),
        Span::styled(goal_text, label_style),
        Span::styled("]", dim_style),
        Span::styled(format!("  {tokens_display}  {elapsed_str}"), dim_style),
    ])
}

/// Build the compact `/graph` status chip: node progress, the current
/// node, and spend. Same chip idiom as [`goal_status_line`] — dim
/// brackets, paused chips invert onto `theme.warning`, active chips
/// animate.
pub fn graph_status_line(
    graph: &crate::app::agent::GraphDisplayState,
    theme: &Theme,
    tick: usize,
) -> Line<'static> {
    let dim_style = Style::default().fg(theme.gray_dim).bg(theme.bg_base);
    let label_style = if graph.status.is_paused() {
        Style::default().fg(theme.bg_base).bg(theme.warning)
    } else {
        Style::default().fg(theme.accent_plan).bg(theme.bg_base)
    };
    let is_active = matches!(graph.status, GoalDisplayStatus::Active);

    let mut progress = format!("{}/{}", graph.achieved_nodes, graph.total_nodes);
    if graph.failed_nodes > 0 {
        progress.push_str(&format!(" ({} failed)", graph.failed_nodes));
    }
    // Planner titles are uncapped; clamp so one long title can't push
    // the whole status bar off-screen.
    let clamped_title = graph.current_node_title.as_deref().map(|t| {
        if t.chars().count() > 40 {
            let head: String = t.chars().take(39).collect();
            format!("{head}…")
        } else {
            t.to_owned()
        }
    });
    let label = match (&graph.status, clamped_title.as_deref()) {
        (GoalDisplayStatus::Active, Some(title)) => format!("{progress} · {title}"),
        (GoalDisplayStatus::Active, None) if graph.running_nodes > 1 => {
            format!("{progress} · {} nodes in flight", graph.running_nodes)
        }
        (GoalDisplayStatus::Complete, _) => format!("{progress} · complete"),
        (GoalDisplayStatus::BudgetLimited, _) => format!("{progress} · budget limit"),
        (status, _) if status.is_paused() => format!("{progress} · paused"),
        _ => progress,
    };
    let graph_text = if is_active {
        let frames = crate::glyphs::dot_spinner_frames();
        let frame = frames[(tick / 4) % frames.len()];
        format!("{frame} Graph: {label}")
    } else {
        format!("Graph: {label}")
    };

    let tokens_str = format_tokens_compact(graph.tokens_spent.max(0));
    let tokens_display = match graph.token_budget {
        Some(budget) if budget > 0 => {
            format!("{}/{} tokens", tokens_str, format_tokens_compact(budget))
        }
        _ => format!("{tokens_str} tokens"),
    };

    Line::from(vec![
        Span::styled("[", dim_style),
        Span::styled(graph_text, label_style),
        Span::styled("]", dim_style),
        Span::styled(format!("  {tokens_display}"), dim_style),
    ])
}

/// Build the compact MCP-connecting indicator for the agent status bar.
///
/// Format: `⠋ MCP (1/4)` — a braille spinner (driven by `tick`, same cadence as
/// the turn-status spinner) followed by the connected/total server count.
/// Rendered in `theme.gray_dim` so it reads as dim, matching the directory path
/// shown on the same row.
///
/// Returns `None` while `progress.total == 0` (a startup seed). That state
/// renders `⠋ Starting session…` above the prompt (see
/// [`crate::views::turn_status`]) rather than as a chip here — the top-bar chip
/// only shows real server counts once the shell reports `total > 0`.
pub fn mcp_status_line(
    progress: &McpInitProgress,
    tick: u64,
    theme: &Theme,
) -> Option<Line<'static>> {
    if progress.total == 0 {
        return None;
    }
    let frames = crate::glyphs::braille_spinner_frames();
    let frame_idx = (tick / SPINNER_DIVISOR) as usize % frames.len();
    let style = Style::default().fg(theme.gray_dim).bg(theme.bg_base);
    Some(Line::from(vec![
        Span::styled(format!("{} ", frames[frame_idx]), style),
        Span::styled(
            format!("MCP ({}/{})", progress.connected, progress.total),
            style,
        ),
    ]))
}

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

    #[test]
    fn tokens_compact_sub_thousand() {
        assert_eq!(format_tokens_compact(0), "0");
        assert_eq!(format_tokens_compact(500), "500");
        assert_eq!(format_tokens_compact(999), "999");
    }

    #[test]
    fn tokens_compact_thousands() {
        assert_eq!(format_tokens_compact(1000), "1k");
        assert_eq!(format_tokens_compact(1500), "1.5k");
        assert_eq!(format_tokens_compact(12300), "12.3k");
        assert_eq!(format_tokens_compact(50000), "50k");
        assert_eq!(format_tokens_compact(100000), "100k");
    }

    #[test]
    fn tokens_compact_negative() {
        assert_eq!(format_tokens_compact(-500), "-500");
        assert_eq!(format_tokens_compact(-1500), "-1.5k");
        assert_eq!(format_tokens_compact(-1_000_000), "-1M");
    }

    #[test]
    fn tokens_compact_millions() {
        assert_eq!(format_tokens_compact(1_000_000), "1M");
        assert_eq!(format_tokens_compact(1_500_000), "1.5M");
        assert_eq!(format_tokens_compact(10_000_000), "10M");
    }

    #[test]
    fn elapsed_compact_seconds() {
        assert_eq!(format_elapsed_compact(0), "0s");
        assert_eq!(format_elapsed_compact(5_000), "5s");
        assert_eq!(format_elapsed_compact(59_999), "59s");
    }

    #[test]
    fn elapsed_compact_minutes() {
        assert_eq!(format_elapsed_compact(60_000), "1m");
        assert_eq!(format_elapsed_compact(180_000), "3m");
        assert_eq!(format_elapsed_compact(3_599_000), "59m");
    }

    #[test]
    fn elapsed_compact_hours() {
        assert_eq!(format_elapsed_compact(3_600_000), "1h");
        assert_eq!(format_elapsed_compact(7_200_000), "2h");
    }

    fn make_goal(
        status: GoalDisplayStatus,
        phase: GoalDisplayPhase,
        idx: Option<u32>,
        total: u32,
        completed: u32,
    ) -> GoalDisplayState {
        GoalDisplayState {
            goal_id: "g-1".into(),
            objective: "Build widget".into(),
            status,
            phase,
            token_budget: Some(50_000),
            tokens_used: 12_300,
            elapsed_ms: 180_000,
            total_deliverables: total,
            completed_deliverables: completed,
            current_deliverable_id: idx,
            current_deliverable_title: Some("Add CSS vars".into()),
            current_subagent_role: None,
            total_worker_rounds: 3,
            total_verify_rounds: 1,
            live_subagent_tokens: None,
            live_tokens_by_model: Vec::new(),
            live_context_pct: None,
            live_turn_count: None,
            live_tool_call_count: None,
            last_event: None,
            last_event_detail: None,
            last_event_timestamp: None,
            token_baseline: 0,
            finished_subagent_tokens: 0,
            deliverables: vec![],
            pause_message: None,
            classifier_runs_attempted: None,
            classifier_max_runs: None,
            last_classifier_verdict: None,
            last_classifier_details_path: None,
            last_classifier_details_exists: false,
            verifying_completion: false,
            planning: false,
            received_at: std::time::Instant::now(),
            elapsed_floor_ms: 0,
        }
    }

    #[test]
    fn phase_label_active_executing() {
        let g = make_goal(
            GoalDisplayStatus::Active,
            GoalDisplayPhase::Executing,
            None,
            0,
            0,
        );
        assert_eq!(goal_phase_label(&g), "Executing");
    }

    #[test]
    fn phase_label_active_planning() {
        let g = make_goal(
            GoalDisplayStatus::Active,
            GoalDisplayPhase::Planning,
            None,
            0,
            0,
        );
        assert_eq!(goal_phase_label(&g), "Planning");
    }

    #[test]
    fn phase_label_paused_variants() {
        for (status, expected) in [
            (GoalDisplayStatus::UserPaused, "Paused"),
            (GoalDisplayStatus::BackOffPaused, "Paused (back-off)"),
            (GoalDisplayStatus::NoProgressPaused, "Paused (no progress)"),
            (GoalDisplayStatus::InfraPaused, "Paused (error)"),
            (GoalDisplayStatus::Blocked, "Paused (verification blocked)"),
        ] {
            let g = make_goal(status, GoalDisplayPhase::Executing, Some(0), 2, 0);
            assert_eq!(goal_phase_label(&g), expected, "for {status:?}");
        }
    }

    #[test]
    fn goal_line_contains_pause_label_for_infra_paused() {
        let g = make_goal(
            GoalDisplayStatus::InfraPaused,
            GoalDisplayPhase::Executing,
            Some(1),
            2,
            0,
        );
        let t = Theme::current();
        let line = goal_status_line(&g, &t, false, 0, None, 0);
        let text: String = line.spans.iter().map(|s| s.content.as_ref()).collect();
        assert!(text.contains("Paused (error)"));
    }

    #[test]
    fn status_chip_shows_verifying_completion_when_flag_set() {
        // Status-chip behaviour: an Active goal with
        // `verifying_completion = true` renders the "Verifying (n/m)"
        // label instead of the regular phase label so the user can see
        // the classifier run.
        let mut g = make_goal(
            GoalDisplayStatus::Active,
            GoalDisplayPhase::Executing,
            None,
            0,
            0,
        );
        g.verifying_completion = true;
        g.classifier_runs_attempted = Some(2);
        g.classifier_max_runs = Some(3);
        assert_eq!(goal_phase_label(&g), "Verifying (2/3)");

        let t = Theme::current();
        let line = goal_status_line(&g, &t, false, 0, None, 0);
        let text: String = line.spans.iter().map(|s| s.content.as_ref()).collect();
        assert!(text.contains("Verifying (2/3)"));
    }

    #[test]
    fn status_chip_verifying_omits_counter_when_counts_absent() {
        // Before the first counter arrives (both fields None) the chip reads
        // "Verifying", not "Verifying (0/0)".
        let mut g = make_goal(
            GoalDisplayStatus::Active,
            GoalDisplayPhase::Executing,
            None,
            0,
            0,
        );
        g.verifying_completion = true;
        g.classifier_runs_attempted = None;
        g.classifier_max_runs = None;
        assert_eq!(goal_phase_label(&g), "Verifying");
    }

    #[test]
    fn classifier_attempts_label_empty_when_both_absent_or_zero() {
        let mut g = make_goal(
            GoalDisplayStatus::Active,
            GoalDisplayPhase::Executing,
            None,
            0,
            0,
        );
        // Both None → empty (no run reserved yet).
        assert_eq!(classifier_attempts_label(&g), "");
        // Explicit zeros also count as "no counter".
        g.classifier_runs_attempted = Some(0);
        g.classifier_max_runs = Some(0);
        assert_eq!(classifier_attempts_label(&g), "");
        // A configured cap (max > 0) makes the counter meaningful.
        g.classifier_max_runs = Some(3);
        assert_eq!(classifier_attempts_label(&g), "0/3");
        g.classifier_runs_attempted = Some(2);
        assert_eq!(classifier_attempts_label(&g), "2/3");
    }

    #[test]
    fn live_elapsed_ms_clamps_to_carried_floor() {
        // The displayed clock must never tick below the carried monotonic
        // floor, even when the latest authoritative base is lower (the
        // pager's extrapolation outran the shell's flush point).
        let mut g = make_goal(
            GoalDisplayStatus::UserPaused,
            GoalDisplayPhase::Idle,
            None,
            0,
            0,
        );
        g.elapsed_ms = 1_000;
        g.elapsed_floor_ms = 5_000;
        assert_eq!(g.live_elapsed_ms(), 5_000);
    }

    #[test]
    fn live_elapsed_ms_uses_live_value_when_above_floor() {
        let mut g = make_goal(
            GoalDisplayStatus::UserPaused,
            GoalDisplayPhase::Idle,
            None,
            0,
            0,
        );
        g.elapsed_ms = 9_000;
        g.elapsed_floor_ms = 5_000;
        assert_eq!(g.live_elapsed_ms(), 9_000);
    }

    #[test]
    fn status_chip_shows_planning_when_flag_set() {
        // An Active goal with `planning = true` renders the "Planning"
        // label instead of the regular phase label so the user can see
        // the planner subagent run while it executes.
        let mut g = make_goal(
            GoalDisplayStatus::Active,
            GoalDisplayPhase::Idle,
            None,
            0,
            0,
        );
        g.planning = true;
        assert_eq!(goal_phase_label(&g), "Planning");

        let t = Theme::current();
        let line = goal_status_line(&g, &t, false, 0, None, 0);
        let text: String = line.spans.iter().map(|s| s.content.as_ref()).collect();
        assert!(text.contains("Planning"));
    }

    #[test]
    fn status_chip_verifying_wins_over_planning() {
        // Deterministic precedence: the two flags never overlap in
        // practice, but if both were set `verifying_completion` wins.
        let mut g = make_goal(
            GoalDisplayStatus::Active,
            GoalDisplayPhase::Idle,
            None,
            0,
            0,
        );
        g.planning = true;
        g.verifying_completion = true;
        g.classifier_runs_attempted = Some(1);
        g.classifier_max_runs = Some(2);
        assert_eq!(goal_phase_label(&g), "Verifying (1/2)");
    }

    #[test]
    fn status_chip_planning_suppressed_on_non_active_status() {
        // The "Planning" label is gated on `Active`: a paused goal that
        // somehow still carries `planning = true` shows its terminal
        // label, not the in-flight one.
        let mut g = make_goal(
            GoalDisplayStatus::UserPaused,
            GoalDisplayPhase::Idle,
            None,
            0,
            0,
        );
        g.planning = true;
        assert_eq!(goal_phase_label(&g), "Paused");
    }

    #[test]
    fn status_chip_verifying_suppressed_on_non_active_status() {
        // The chip text is gated on `Active`: a paused / complete /
        // budget-limited goal that somehow still carries
        // `verifying_completion = true` must show its terminal label,
        // not the in-flight one.
        let mut g = make_goal(
            GoalDisplayStatus::UserPaused,
            GoalDisplayPhase::Executing,
            None,
            0,
            0,
        );
        g.verifying_completion = true;
        assert_eq!(goal_phase_label(&g), "Paused");
    }

    #[test]
    fn goal_line_paused_chip_uses_warning_background() {
        // Paused chips render with the
        // `theme.warning` background to visually warn the user. Pin the
        // background colour on the label span so a regression that drops
        // the chip-vs-modal colour alignment gets caught.
        //
        // Use the unquantized `kiginight()` theme directly so warning and
        // bg_base remain distinguishable in the test env — `Theme::current()`
        // collapses both to ANSI `Reset` on 16-colour terminals, which
        // would defeat the assertion.
        let g = make_goal(
            GoalDisplayStatus::UserPaused,
            GoalDisplayPhase::Executing,
            Some(0),
            2,
            0,
        );
        let t = Theme::kiginight();
        let line = goal_status_line(&g, &t, false, 0, None, 0);
        // The label span is the one whose content starts with "Goal:".
        let label_span = line
            .spans
            .iter()
            .find(|s| s.content.contains("Goal:"))
            .expect("label span with `Goal:` prefix");
        assert_eq!(label_span.style.bg, Some(t.warning));
        assert_ne!(label_span.style.bg, Some(t.bg_base));
    }

    #[test]
    fn goal_line_active_chip_does_not_use_warning_background() {
        // Negative companion: an Active goal must keep the standard
        // accent-plan-on-bg-base chip style.
        let g = make_goal(
            GoalDisplayStatus::Active,
            GoalDisplayPhase::Executing,
            Some(0),
            2,
            0,
        );
        let t = Theme::kiginight();
        let line = goal_status_line(&g, &t, false, 0, None, 0);
        let label_span = line
            .spans
            .iter()
            .find(|s| s.content.contains("Goal:"))
            .expect("label span with `Goal:` prefix");
        assert_eq!(label_span.style.bg, Some(t.bg_base));
        assert_ne!(label_span.style.bg, Some(t.warning));
    }

    #[test]
    fn phase_label_budget_limited() {
        let g = make_goal(
            GoalDisplayStatus::BudgetLimited,
            GoalDisplayPhase::Executing,
            Some(1),
            2,
            0,
        );
        assert_eq!(goal_phase_label(&g), "Budget");
    }

    #[test]
    fn phase_label_complete() {
        let g = make_goal(
            GoalDisplayStatus::Complete,
            GoalDisplayPhase::Idle,
            None,
            3,
            3,
        );
        assert_eq!(goal_phase_label(&g), "Done");
    }

    #[test]
    fn phase_label_active_idle() {
        let g = make_goal(
            GoalDisplayStatus::Active,
            GoalDisplayPhase::Idle,
            None,
            0,
            0,
        );
        assert_eq!(goal_phase_label(&g), "Idle");
    }

    #[test]
    fn phase_label_executing_ignores_deliverables() {
        let g = make_goal(
            GoalDisplayStatus::Active,
            GoalDisplayPhase::Executing,
            None,
            4,
            2,
        );
        assert_eq!(goal_phase_label(&g), "Executing");
    }

    #[test]
    fn goal_line_contains_expected_text() {
        let g = make_goal(
            GoalDisplayStatus::Active,
            GoalDisplayPhase::Executing,
            Some(1),
            4,
            1,
        );
        let t = Theme::current();
        let line = goal_status_line(&g, &t, false, 0, None, 0);
        let text: String = line.spans.iter().map(|s| s.content.as_ref()).collect();
        assert!(text.contains("Goal: Exec"));
        assert!(text.contains("12.3k/50k tokens"));
        assert!(text.contains("3m"));
    }

    #[test]
    fn goal_line_without_budget() {
        let mut g = make_goal(
            GoalDisplayStatus::Active,
            GoalDisplayPhase::Planning,
            None,
            0,
            0,
        );
        g.token_budget = None;
        g.tokens_used = 500;
        let t = Theme::current();
        let line = goal_status_line(&g, &t, false, 0, None, 0);
        let text: String = line.spans.iter().map(|s| s.content.as_ref()).collect();
        assert!(text.contains("500 tokens"));
        assert!(!text.contains('/'));
    }

    #[test]
    fn goal_line_no_title_in_status_bar() {
        let g = make_goal(
            GoalDisplayStatus::Active,
            GoalDisplayPhase::Executing,
            None,
            1,
            0,
        );
        let t = Theme::current();
        let line = goal_status_line(&g, &t, false, 0, None, 0);
        let text: String = line.spans.iter().map(|s| s.content.as_ref()).collect();
        // Title should NOT appear in status bar (only in the modal)
        assert!(!text.contains("Add CSS vars"));
        assert!(!text.contains("Build widget"));
    }

    #[test]
    fn mcp_status_line_renders_compact_count() {
        // total > 0 renders the compact `MCP (connected/total)` chip.
        let progress = McpInitProgress {
            total: 4,
            connected: 1,
            started_at: std::time::Instant::now(),
        };
        let t = Theme::current();
        let line = mcp_status_line(&progress, 0, &t).expect("total > 0 must render a line");
        let text: String = line.spans.iter().map(|s| s.content.as_ref()).collect();
        assert!(
            text.contains("MCP (1/4)"),
            "expected 'MCP (1/4)', got: {text:?}"
        );
    }

    #[test]
    fn mcp_status_line_uses_dim_directory_color() {
        // The chip must render in `theme.gray_dim` to match the directory path.
        let t = Theme::kiginight();
        let progress = McpInitProgress {
            total: 2,
            connected: 0,
            started_at: std::time::Instant::now(),
        };
        let line = mcp_status_line(&progress, 0, &t).expect("total > 0 must render a line");
        for span in &line.spans {
            assert_eq!(
                span.style.fg,
                Some(t.gray_dim),
                "MCP chip spans must use theme.gray_dim"
            );
        }
    }

    #[test]
    fn mcp_status_line_hidden_for_zero_total() {
        // total == 0 (startup seed) renders nothing in the top bar — that state
        // shows "Starting session…" above the prompt instead.
        let progress = McpInitProgress {
            total: 0,
            connected: 0,
            started_at: std::time::Instant::now(),
        };
        let t = Theme::current();
        assert!(mcp_status_line(&progress, 0, &t).is_none());
    }

    /// Separators appear only *between* items — never before the first item or
    /// after the last (no leading/trailing divider).
    #[test]
    fn status_bar_separators_only_between_items() {
        let theme = Theme::current();
        let mut bar = AgentStatusBar::new(&theme);
        bar.push("a", Line::from("AA"));
        bar.push("b", Line::from("BB"));
        bar.push("c", Line::from("CC"));

        let area = Rect::new(0, 0, 40, 1);
        let mut buf = Buffer::empty(area);
        bar.render(&mut buf, area);

        let row: String = (0..area.width).map(|x| buf[(x, 0)].symbol()).collect();
        let trimmed = row.trim();

        // Exactly two dividers (between the three items), none at the ends.
        assert_eq!(trimmed.matches(SEPARATOR).count(), 2, "row = {trimmed:?}");
        assert!(
            !trimmed.starts_with(SEPARATOR),
            "no leading divider, row = {trimmed:?}"
        );
        assert!(
            !trimmed.ends_with(SEPARATOR),
            "no trailing divider, row = {trimmed:?}"
        );
        assert_eq!(trimmed, format!("AA {SEPARATOR} BB {SEPARATOR} CC"));
    }

    /// A single item renders with no separators at all (it is both first and
    /// last).
    #[test]
    fn status_bar_single_item_has_no_separators() {
        let theme = Theme::current();
        let mut bar = AgentStatusBar::new(&theme);
        bar.push("only", Line::from("XX"));

        let area = Rect::new(0, 0, 20, 1);
        let mut buf = Buffer::empty(area);
        bar.render(&mut buf, area);

        let row: String = (0..area.width).map(|x| buf[(x, 0)].symbol()).collect();
        assert_eq!(row.trim(), "XX");
        assert!(!row.contains(SEPARATOR));
    }
}