pi/

tui.rs

//! Terminal UI components using rich_rust.
//!
//! This module provides the interactive terminal interface for Pi,
//! built on rich_rust for beautiful markup-based output.

use std::io::{self, IsTerminal, Write};

use rich_rust::prelude::*;
use rich_rust::renderables::{Markdown, Syntax};
use rich_rust::segment::Segment;

/// Pi's console wrapper providing styled terminal output.
pub struct PiConsole {
    console: Console,
    is_tty: bool,
}

impl PiConsole {
    /// Create a new Pi console with auto-detected terminal capabilities.
    pub fn new() -> Self {
        Self::new_with_theme(None)
    }

    /// Create a new Pi console with an optional theme.
    pub fn new_with_theme(_theme: Option<crate::theme::Theme>) -> Self {
        let is_tty = io::stdout().is_terminal();
        let console = Console::builder().markup(is_tty).emoji(is_tty).build();

        Self { console, is_tty }
    }

    /// Create a console with forced color output (for testing).
    pub fn with_color() -> Self {
        Self {
            console: Console::builder()
                .markup(true)
                .emoji(true)
                .file(Box::new(io::sink()))
                .build(),
            is_tty: true,
        }
    }

    /// Check if we're running in a terminal.
    pub const fn is_terminal(&self) -> bool {
        self.is_tty
    }

    /// Get the terminal width.
    pub fn width(&self) -> usize {
        self.console.width()
    }

    // -------------------------------------------------------------------------
    // Text Output
    // -------------------------------------------------------------------------

    /// Print plain text without any styling.
    pub fn print_plain(&self, text: &str) {
        print!("{text}");
        let _ = io::stdout().flush();
    }

    /// Print text with rich markup (if TTY).
    pub fn print_markup(&self, markup: &str) {
        if self.is_tty {
            self.console.print(markup);
        } else {
            // Strip markup for non-TTY
            print!("{}", strip_markup(markup));
            let _ = io::stdout().flush();
        }
    }

    /// Print a newline.
    pub fn newline(&self) {
        println!();
    }

    /// Render Markdown (TTY → styled output; non-TTY → raw Markdown).
    pub fn render_markdown(&self, markdown: &str) {
        if self.is_tty {
            let mut segments = render_markdown_with_syntax(markdown, self.width());
            let mut ends_with_newline = false;
            for segment in segments.iter().rev() {
                let text = segment.text.as_ref();
                if text.is_empty() {
                    continue;
                }
                ends_with_newline = text.ends_with('\n');
                break;
            }
            if !ends_with_newline {
                segments.push(Segment::plain("\n"));
            }
            self.console.print_segments(&segments);
        } else {
            print!("{markdown}");
            if !markdown.ends_with('\n') {
                println!();
            }
            let _ = io::stdout().flush();
        }
    }

    // -------------------------------------------------------------------------
    // Agent Event Rendering
    // -------------------------------------------------------------------------

    /// Render streaming text from the assistant.
    pub fn render_text_delta(&self, text: &str) {
        print!("{text}");
        let _ = io::stdout().flush();
    }

    /// Render streaming thinking text (dimmed).
    pub fn render_thinking_delta(&self, text: &str) {
        if self.is_tty {
            // Dim style for thinking
            print!("\x1b[2m{text}\x1b[0m");
        } else {
            print!("{text}");
        }
        let _ = io::stdout().flush();
    }

    /// Render the start of a thinking block.
    pub fn render_thinking_start(&self) {
        if self.is_tty {
            self.print_markup("\n[dim italic]Thinking...[/]\n");
        }
    }

    /// Render the end of a thinking block.
    pub fn render_thinking_end(&self) {
        if self.is_tty {
            self.print_markup("[/dim]\n");
        }
    }

    /// Render tool execution start.
    pub fn render_tool_start(&self, name: &str, _input: &str) {
        if self.is_tty {
            self.print_markup(&format!("\n[bold yellow][[Running {name}...]][/]\n"));
        }
    }

    /// Render tool execution end.
    pub fn render_tool_end(&self, name: &str, is_error: bool) {
        if self.is_tty {
            if is_error {
                self.print_markup(&format!("[bold red][[{name} failed]][/]\n\n"));
            } else {
                self.print_markup(&format!("[bold green][[{name} done]][/]\n\n"));
            }
        }
    }

    /// Render an error message.
    pub fn render_error(&self, error: &str) {
        if self.is_tty {
            self.print_markup(&format!("\n[bold red]Error:[/] {error}\n"));
        } else {
            eprintln!("\nError: {error}");
        }
    }

    /// Render a warning message.
    pub fn render_warning(&self, warning: &str) {
        if self.is_tty {
            self.print_markup(&format!("[bold yellow]Warning:[/] {warning}\n"));
        } else {
            eprintln!("Warning: {warning}");
        }
    }

    /// Render a success message.
    pub fn render_success(&self, message: &str) {
        if self.is_tty {
            self.print_markup(&format!("[bold green]{message}[/]\n"));
        } else {
            println!("{message}");
        }
    }

    /// Render an info message.
    pub fn render_info(&self, message: &str) {
        if self.is_tty {
            self.print_markup(&format!("[bold blue]{message}[/]\n"));
        } else {
            println!("{message}");
        }
    }

    // -------------------------------------------------------------------------
    // Structured Output
    // -------------------------------------------------------------------------

    /// Render a panel with a title.
    pub fn render_panel(&self, content: &str, title: &str) {
        if self.is_tty {
            let panel = Panel::from_text(content)
                .title(title)
                .border_style(Style::parse("cyan").unwrap_or_default());
            self.console.print_renderable(&panel);
        } else {
            println!("--- {title} ---");
            println!("{content}");
            println!("---");
        }
    }

    /// Render a table.
    pub fn render_table(&self, headers: &[&str], rows: &[Vec<&str>]) {
        if self.is_tty {
            let mut table = Table::new().header_style(Style::parse("bold").unwrap_or_default());
            for header in headers {
                table = table.with_column(Column::new(*header));
            }
            for row in rows {
                table.add_row_cells(row.iter().copied());
            }
            self.console.print_renderable(&table);
        } else {
            // Simple text table for non-TTY
            println!("{}", headers.join("\t"));
            for row in rows {
                println!("{}", row.join("\t"));
            }
        }
    }

    /// Render a horizontal rule.
    pub fn render_rule(&self, title: Option<&str>) {
        if self.is_tty {
            let rule = title.map_or_else(Rule::new, Rule::with_title);
            self.console.print_renderable(&rule);
        } else if let Some(t) = title {
            println!("--- {t} ---");
        } else {
            println!("---");
        }
    }

    // -------------------------------------------------------------------------
    // Usage/Status Display
    // -------------------------------------------------------------------------

    /// Render token usage statistics.
    pub fn render_usage(&self, input_tokens: u32, output_tokens: u32, cost_usd: Option<f64>) {
        if self.is_tty {
            let cost_str = cost_usd
                .map(|c| format!(" [dim](${c:.4})[/]"))
                .unwrap_or_default();
            self.print_markup(&format!(
                "[dim]Tokens: {input_tokens} in / {output_tokens} out{cost_str}[/]\n"
            ));
        }
    }

    /// Render session info.
    pub fn render_session_info(&self, session_path: &str, message_count: usize) {
        if self.is_tty {
            self.print_markup(&format!(
                "[dim]Session: {session_path} ({message_count} messages)[/]\n"
            ));
        }
    }

    /// Render model info.
    pub fn render_model_info(&self, model: &str, thinking_level: Option<&str>) {
        if self.is_tty {
            let thinking_str = thinking_level
                .map(|t| format!(" [dim](thinking: {t})[/]"))
                .unwrap_or_default();
            self.print_markup(&format!("[dim]Model: {model}{thinking_str}[/]\n"));
        }
    }

    // -------------------------------------------------------------------------
    // Interactive Mode Helpers
    // -------------------------------------------------------------------------

    /// Render the input prompt.
    pub fn render_prompt(&self) {
        if self.is_tty {
            self.print_markup("[bold cyan]>[/] ");
        } else {
            print!("> ");
        }
        let _ = io::stdout().flush();
    }

    /// Render a user message echo.
    pub fn render_user_message(&self, message: &str) {
        if self.is_tty {
            self.print_markup(&format!("[bold]You:[/] {message}\n\n"));
        } else {
            println!("You: {message}\n");
        }
    }

    /// Render assistant message start.
    pub fn render_assistant_start(&self) {
        if self.is_tty {
            self.print_markup("[bold]Assistant:[/] ");
        } else {
            print!("Assistant: ");
        }
        let _ = io::stdout().flush();
    }

    /// Clear the current line (for progress updates).
    pub fn clear_line(&self) {
        if self.is_tty {
            print!("\r\x1b[K");
            let _ = io::stdout().flush();
        }
    }

    /// Move cursor up N lines.
    pub fn cursor_up(&self, n: usize) {
        if self.is_tty && n > 0 {
            print!("\x1b[{n}A");
            let _ = io::stdout().flush();
        }
    }
}

impl Default for PiConsole {
    fn default() -> Self {
        Self::new()
    }
}

// Thread-safe console for use across async tasks
impl Clone for PiConsole {
    fn clone(&self) -> Self {
        Self {
            console: Console::builder()
                .markup(self.is_tty)
                .emoji(self.is_tty)
                .build(),
            is_tty: self.is_tty,
        }
    }
}

#[derive(Debug, Clone)]
enum MarkdownChunk {
    Text(String),
    CodeBlock {
        language: Option<String>,
        code: String,
    },
}

fn parse_fenced_code_language(info: &str) -> Option<String> {
    let language_tag = info
        .split_whitespace()
        .next()
        .unwrap_or_default()
        .split(',')
        .next()
        .unwrap_or_default()
        .trim();
    if language_tag.is_empty() {
        None
    } else {
        Some(language_tag.to_ascii_lowercase())
    }
}

fn split_markdown_fenced_code_blocks(markdown: &str) -> Vec<MarkdownChunk> {
    let mut chunks = Vec::new();

    let mut text_buf = String::new();
    let mut code_buf = String::new();
    let mut in_code_block = false;
    let mut fence_len = 0usize;
    let mut code_language: Option<String> = None;

    for line in markdown.split_inclusive('\n') {
        let trimmed_start = line.trim_start();
        let trimmed_line = trimmed_start.trim_end_matches(['\r', '\n']);

        let backtick_count = trimmed_line.chars().take_while(|ch| *ch == '`').count();
        let is_fence = backtick_count >= 3 && trimmed_line.starts_with("```");

        if !in_code_block {
            if is_fence {
                fence_len = backtick_count;
                let info = trimmed_line.get(fence_len..).unwrap_or_default();

                // CommonMark: The info string may not contain any backtick characters.
                // If it does, this is likely an inline code span at the start of a line, not a fence.
                if info.contains('`') {
                    text_buf.push_str(line);
                    continue;
                }

                if !text_buf.is_empty() {
                    chunks.push(MarkdownChunk::Text(std::mem::take(&mut text_buf)));
                }

                code_language = parse_fenced_code_language(info);
                in_code_block = true;
                code_buf.clear();
                continue;
            }

            text_buf.push_str(line);
            continue;
        }

        if is_fence
            && backtick_count >= fence_len
            && trimmed_line[backtick_count..].trim().is_empty()
        {
            chunks.push(MarkdownChunk::CodeBlock {
                language: code_language.take(),
                code: std::mem::take(&mut code_buf),
            });
            in_code_block = false;
            fence_len = 0;
            continue;
        }

        code_buf.push_str(line);
    }

    if in_code_block {
        // Unterminated fence; fall back to treating the entire input as plain markdown.
        return vec![MarkdownChunk::Text(markdown.to_string())];
    }

    if !text_buf.is_empty() {
        chunks.push(MarkdownChunk::Text(text_buf));
    }

    chunks
}

fn has_multiple_non_none_styles(segments: &[Segment<'_>]) -> bool {
    use std::collections::HashSet;

    let mut seen = HashSet::new();
    for segment in segments {
        let Some(style) = &segment.style else {
            continue;
        };
        if segment.text.as_ref().trim().is_empty() {
            continue;
        }

        seen.insert(style.clone());
        if seen.len() > 1 {
            return true;
        }
    }

    false
}

fn render_syntax_line_by_line(
    code: &str,
    language: &str,
    width: usize,
) -> Option<Vec<Segment<'static>>> {
    let mut rendered: Vec<Segment<'static>> = Vec::new();
    for line in code.split_inclusive('\n') {
        let syntax = Syntax::new(line, language);
        let items = syntax.render(Some(width)).ok()?;
        rendered.extend(items.into_iter().map(Segment::into_owned));
    }
    Some(rendered)
}

fn render_markdown_with_syntax(markdown: &str, width: usize) -> Vec<Segment<'static>> {
    if !markdown.contains("```") {
        return Markdown::new(markdown)
            .render(width)
            .into_iter()
            .map(Segment::into_owned)
            .collect();
    }

    let chunks = split_markdown_fenced_code_blocks(markdown);
    let mut segments: Vec<Segment<'static>> = Vec::new();

    for chunk in chunks {
        match chunk {
            MarkdownChunk::Text(text) => {
                if text.is_empty() {
                    continue;
                }
                segments.extend(
                    Markdown::new(text)
                        .render(width)
                        .into_iter()
                        .map(Segment::into_owned),
                );
            }
            MarkdownChunk::CodeBlock { language, mut code } => {
                if !code.ends_with('\n') {
                    code.push('\n');
                }

                let language = language.unwrap_or_else(|| "text".to_string());
                let require_variation = matches!(language.as_str(), "typescript" | "ts" | "tsx");
                let mut candidates: Vec<&str> = Vec::new();
                match language.as_str() {
                    // syntect's built-in set doesn't always include TypeScript; prefer `ts` if
                    // available, otherwise fall back to JavaScript highlighting.
                    "typescript" | "ts" | "tsx" => candidates.extend(["ts", "javascript"]),
                    _ => candidates.push(language.as_str()),
                }
                candidates.push("text");

                let mut rendered_items: Option<Vec<Segment<'static>>> = None;
                for candidate in candidates {
                    let syntax = Syntax::new(code.as_str(), candidate);
                    if let Ok(items) = syntax.render(Some(width)) {
                        if require_variation
                            && candidate != "text"
                            && !has_multiple_non_none_styles(&items)
                        {
                            if candidate == "javascript" {
                                if let Some(line_items) =
                                    render_syntax_line_by_line(code.as_str(), candidate, width)
                                {
                                    if has_multiple_non_none_styles(&line_items) {
                                        rendered_items = Some(line_items);
                                        break;
                                    }
                                }
                            }
                            continue;
                        }
                        rendered_items = Some(items.into_iter().map(Segment::into_owned).collect());
                        break;
                    }
                }

                if let Some(items) = rendered_items {
                    segments.extend(items);
                } else {
                    segments.extend(
                        Markdown::new(format!("```\n{code}```\n"))
                            .render(width)
                            .into_iter()
                            .map(Segment::into_owned),
                    );
                }
            }
        }
    }

    segments
}

/// Strip rich markup tags from text.
fn strip_markup(text: &str) -> String {
    let mut result = String::with_capacity(text.len());
    let mut buffer = String::new();
    let mut in_tag = false;

    for c in text.chars() {
        if in_tag {
            if c == ']' {
                // End of potential tag
                // Check heuristics:
                // 1. Not pure digits (e.g. [0])
                // 2. Contains only allowed characters
                let is_pure_digits =
                    !buffer.is_empty() && buffer.chars().all(|ch| ch.is_ascii_digit());
                let contains_invalid_chars = buffer.chars().any(|ch| {
                    !ch.is_ascii_alphanumeric()
                        && !matches!(
                            ch,
                            ' ' | '/'
                                | ','
                                | '#'
                                | '='
                                | '.'
                                | ':'
                                | '-'
                                | '_'
                                | '?'
                                | '&'
                                | '%'
                                | '+'
                                | '~'
                                | ';'
                                | '*'
                                | '\''
                                | '('
                                | ')'
                        )
                });

                if is_pure_digits || contains_invalid_chars || buffer.is_empty() {
                    // Not a tag, restore literal
                    result.push('[');
                    result.push_str(&buffer);
                    result.push(']');
                } else {
                    // Valid tag, discard (strip it)
                }
                buffer.clear();
                in_tag = false;
            } else if c == '[' {
                result.push('[');
                if buffer.is_empty() {
                    // Escaped bracket: `[[` becomes `[`
                    in_tag = false;
                } else {
                    // Nested '[' means the previous '[' was literal.
                    // Flush previous '[' and buffer, start new tag candidate.
                    result.push_str(&buffer);
                    buffer.clear();
                    // Stay in_tag for this new '['
                }
            } else {
                buffer.push(c);
            }
        } else if c == '[' {
            in_tag = true;
        } else {
            result.push(c);
        }
    }

    // Flush any open tag at end of string
    if in_tag {
        result.push('[');
        result.push_str(&buffer);
    }

    result
}

/// Spinner styles for different operations.
pub enum SpinnerStyle {
    /// Default dots spinner for general operations.
    Dots,
    /// Line spinner for file operations.
    Line,
    /// Simple ASCII spinner for compatibility.
    Simple,
}

impl SpinnerStyle {
    /// Get the spinner frames for this style.
    pub const fn frames(&self) -> &'static [&'static str] {
        match self {
            Self::Dots => &["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"],
            Self::Line => &["⎺", "⎻", "⎼", "⎽", "⎼", "⎻"],
            Self::Simple => &["|", "/", "-", "\\"],
        }
    }

    /// Get the frame interval in milliseconds.
    pub const fn interval_ms(&self) -> u64 {
        match self {
            Self::Dots => 80,
            Self::Line | Self::Simple => 100,
        }
    }
}

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

    use std::collections::HashSet;
    use std::sync::{Arc, Mutex};

    fn capture_markdown_segments(markdown: &str) -> Vec<Segment<'static>> {
        let console = PiConsole::with_color();
        console.console.begin_capture();
        console.render_markdown(markdown);
        console.console.end_capture()
    }

    fn segments_text(segments: &[Segment<'static>]) -> String {
        segments.iter().map(|s| s.text.as_ref()).collect()
    }

    fn unique_style_debug_for_tokens(
        segments: &[Segment<'static>],
        tokens: &[&str],
    ) -> HashSet<String> {
        segments
            .iter()
            .filter(|segment| {
                let text = segment.text.as_ref();
                tokens.iter().any(|token| text.contains(token))
            })
            .map(|segment| format!("{:?}", segment.style))
            .collect()
    }

    #[derive(Clone)]
    struct SharedBufferWriter {
        buffer: Arc<Mutex<Vec<u8>>>,
    }

    impl io::Write for SharedBufferWriter {
        fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
            self.buffer
                .lock()
                .expect("lock buffer")
                .extend_from_slice(buf);
            Ok(buf.len())
        }

        fn flush(&mut self) -> io::Result<()> {
            Ok(())
        }
    }

    #[test]
    fn test_strip_markup() {
        assert_eq!(strip_markup("[bold]Hello[/]"), "Hello");
        assert_eq!(strip_markup("[red]A[/] [blue]B[/]"), "A B");
        assert_eq!(strip_markup("No markup"), "No markup");
        assert_eq!(strip_markup("[bold red on blue]Text[/]"), "Text");
        assert_eq!(strip_markup("array[0]"), "array[0]");
        assert_eq!(strip_markup("[#ff0000]Hex[/]"), "Hex");
        assert_eq!(strip_markup("[link=https://example.com]Link[/]"), "Link");
    }

    #[test]
    fn render_markdown_emits_ansi_when_tty() {
        let buffer = Arc::new(Mutex::new(Vec::new()));
        let writer = SharedBufferWriter {
            buffer: Arc::clone(&buffer),
        };
        let console = Console::builder()
            .markup(true)
            .emoji(false)
            .force_terminal(true)
            .color_system(ColorSystem::TrueColor)
            .file(Box::new(writer))
            .build();

        let pi_console = PiConsole {
            console,
            is_tty: true,
        };

        pi_console.render_markdown("# Title\n\n- Item 1\n- Item 2\n\n**bold**");

        let output = String::from_utf8(buffer.lock().expect("lock buffer").clone()).expect("utf-8");

        assert!(
            output.contains("\u{1b}["),
            "expected ANSI escape codes, got: {output:?}"
        );
        assert!(!output.contains("**bold**"));
        assert!(output.contains("bold"));
    }

    #[test]
    fn test_spinner_frames() {
        let dots = SpinnerStyle::Dots;
        assert_eq!(dots.frames().len(), 10);
        assert_eq!(dots.interval_ms(), 80);

        let simple = SpinnerStyle::Simple;
        assert_eq!(simple.frames().len(), 4);
    }

    #[test]
    fn test_console_creation() {
        let console = PiConsole::with_color();
        assert!(console.width() > 0);
    }

    #[test]
    fn render_markdown_produces_styled_segments() {
        let console = PiConsole::with_color();

        console.console.begin_capture();
        console.render_markdown("# Title\n\nThis is **bold**.\n\n- Item 1\n- Item 2");
        let segments = console.console.end_capture();

        let captured: String = segments.iter().map(|s| s.text.as_ref()).collect();
        assert!(captured.contains("Title"));
        assert!(captured.contains("bold"));
        assert!(segments.iter().any(|s| s.style.is_some()));
    }

    #[test]
    fn render_markdown_code_fence_uses_syntax_highlighting_when_language_present() {
        let console = PiConsole::with_color();

        console.console.begin_capture();
        console.render_markdown("```rust\nfn main() {\n    println!(\"hi\");\n}\n```");
        let segments = console.console.end_capture();

        let code_styles = unique_style_debug_for_tokens(&segments, &["fn", "println"]);

        assert!(
            code_styles.len() > 1,
            "expected multiple token styles from syntax highlighting, got {code_styles:?}"
        );
    }

    #[test]
    fn parse_fenced_code_language_extracts_first_tag() {
        assert_eq!(parse_fenced_code_language("rust"), Some("rust".to_string()));
        assert_eq!(
            parse_fenced_code_language(" RuSt "),
            Some("rust".to_string())
        );
        assert_eq!(
            parse_fenced_code_language("rust,ignore"),
            Some("rust".to_string())
        );
        assert_eq!(parse_fenced_code_language(""), None);
        assert_eq!(parse_fenced_code_language("   "), None);
    }

    #[test]
    fn split_markdown_fenced_code_blocks_splits_text_and_code() {
        let input = "Intro\n\n```rust\nfn main() {}\n```\n\nTail\n";
        let chunks = split_markdown_fenced_code_blocks(input);

        assert_eq!(chunks.len(), 3);
        assert!(matches!(chunks[0], MarkdownChunk::Text(_)));
        assert!(
            matches!(
                &chunks[1],
                MarkdownChunk::CodeBlock { language, code }
                    if language.as_deref() == Some("rust") && code.contains("fn main")
            ),
            "expected rust code block, got {chunks:?}"
        );
        assert!(matches!(chunks[2], MarkdownChunk::Text(_)));
    }

    #[test]
    fn split_markdown_fenced_code_blocks_unterminated_fence_falls_back_to_text() {
        let input = "Intro\n\n```rust\nfn main() {}\n";
        let chunks = split_markdown_fenced_code_blocks(input);

        assert_eq!(
            chunks.len(),
            1,
            "expected a single Text chunk, got {chunks:?}"
        );
        let MarkdownChunk::Text(text) = &chunks[0] else {
            unreachable!("expected text fallback, got {chunks:?}");
        };
        assert!(text.contains("```rust"));
        assert!(text.contains("fn main"));
    }

    #[test]
    fn render_markdown_strips_inline_markers_and_renders_headings_lists_links() {
        let segments = capture_markdown_segments(
            r"
# H1
## H2
### H3
#### H4
##### H5
###### H6

This is **bold**, *italic*, ~~strike~~, `code`, and [link](https://example.com).

- Bullet 1
1. Numbered 1

Nested: **bold and *italic*** and ~~**strike bold**~~.
",
        );

        let captured = segments_text(&segments);
        for needle in [
            "H1",
            "H2",
            "H3",
            "H4",
            "H5",
            "H6",
            "bold",
            "italic",
            "strike",
            "code",
            "link",
            "Bullet 1",
            "Numbered 1",
            "Nested",
        ] {
            assert!(
                captured.contains(needle),
                "expected output to contain {needle:?}, got: {captured:?}"
            );
        }

        assert!(
            !captured.contains("**"),
            "expected bold markers to be stripped, got: {captured:?}"
        );
        assert!(
            !captured.contains("~~"),
            "expected strikethrough markers to be stripped, got: {captured:?}"
        );
        assert!(
            !captured.contains('`'),
            "expected inline code markers to be stripped, got: {captured:?}"
        );
        assert!(
            !captured.contains("]("),
            "expected link markers to be stripped, got: {captured:?}"
        );

        assert!(
            segments.iter().any(|s| s.style.is_some()),
            "expected styled segments, got: {segments:?}"
        );
    }

    // ── strip_markup edge cases ──────────────────────────────────────
    #[test]
    fn strip_markup_nested_tags() {
        assert_eq!(strip_markup("[bold][red]text[/][/]"), "text");
    }

    #[test]
    fn strip_markup_empty_tag() {
        // `[]` has empty buffer → not treated as tag, preserved
        assert_eq!(strip_markup("before[]after"), "before[]after");
    }

    #[test]
    fn strip_markup_adjacent_tags() {
        assert_eq!(strip_markup("[bold]A[/][red]B[/]"), "AB");
    }

    #[test]
    fn strip_markup_only_closing_tag() {
        assert_eq!(strip_markup("[/]"), "");
    }

    #[test]
    fn strip_markup_unclosed_bracket_at_end() {
        assert_eq!(strip_markup("text[unclosed"), "text[unclosed");
    }

    #[test]
    fn strip_markup_bracket_with_special_chars() {
        // Characters like ! or @ are not in the tag heuristic set → not a tag
        assert_eq!(strip_markup("[hello!]world"), "[hello!]world");
        assert_eq!(strip_markup("[hello@world]text"), "[hello@world]text");
    }

    #[test]
    fn strip_markup_pure_digits_preserved() {
        assert_eq!(strip_markup("array[0]"), "array[0]");
        assert_eq!(strip_markup("arr[123]"), "arr[123]");
        assert_eq!(strip_markup("x[0][1][2]"), "x[0][1][2]");
    }

    #[test]
    fn strip_markup_mixed_digit_alpha_is_tag() {
        // "dim" is not pure digits → treated as tag
        assert_eq!(strip_markup("[dim]faded[/]"), "faded");
    }

    #[test]
    fn strip_markup_empty_input() {
        assert_eq!(strip_markup(""), "");
    }

    #[test]
    fn strip_markup_no_brackets() {
        assert_eq!(
            strip_markup("plain text without brackets"),
            "plain text without brackets"
        );
    }

    #[test]
    fn strip_markup_hash_color_tag() {
        assert_eq!(strip_markup("[#aabbcc]colored[/]"), "colored");
    }

    #[test]
    fn strip_markup_tag_with_equals() {
        assert_eq!(strip_markup("[link=https://example.com]click[/]"), "click");
    }

    #[test]
    fn strip_markup_multiple_lines() {
        let input = "[bold]line1[/]\n[red]line2[/]\n";
        assert_eq!(strip_markup(input), "line1\nline2\n");
    }

    // ── split_markdown_fenced_code_blocks edge cases ───────────────────
    #[test]
    fn split_markdown_multiple_code_blocks() {
        let input = "text1\n```rust\ncode1\n```\ntext2\n```python\ncode2\n```\ntext3\n";
        let chunks = split_markdown_fenced_code_blocks(input);
        assert_eq!(chunks.len(), 5, "expected 5 chunks: {chunks:?}");
        assert!(matches!(&chunks[0], MarkdownChunk::Text(_)));
        assert!(
            matches!(&chunks[1], MarkdownChunk::CodeBlock { language, .. } if language.as_deref() == Some("rust"))
        );
        assert!(matches!(&chunks[2], MarkdownChunk::Text(_)));
        assert!(
            matches!(&chunks[3], MarkdownChunk::CodeBlock { language, .. } if language.as_deref() == Some("python"))
        );
        assert!(matches!(&chunks[4], MarkdownChunk::Text(_)));
    }

    #[test]
    fn split_markdown_code_block_no_language() {
        let input = "```\nplain code\n```\n";
        let chunks = split_markdown_fenced_code_blocks(input);
        assert_eq!(chunks.len(), 1);
        assert!(matches!(
            &chunks[0],
            MarkdownChunk::CodeBlock { language, code }
                if language.is_none() && code.contains("plain code")
        ));
    }

    #[test]
    fn split_markdown_empty_code_block() {
        let input = "```rust\n```\n";
        let chunks = split_markdown_fenced_code_blocks(input);
        assert_eq!(chunks.len(), 1);
        assert!(matches!(
            &chunks[0],
            MarkdownChunk::CodeBlock { language, code }
                if language.as_deref() == Some("rust") && code.is_empty()
        ));
    }

    #[test]
    fn split_markdown_four_backtick_fence() {
        // 4-backtick fence should work (>= 3 backticks)
        let input = "````rust\ncode\n````\n";
        let chunks = split_markdown_fenced_code_blocks(input);
        assert_eq!(chunks.len(), 1);
        assert!(matches!(&chunks[0], MarkdownChunk::CodeBlock { .. }));
    }

    #[test]
    fn split_markdown_nested_fence_shorter_doesnt_close() {
        // Inner 3-backtick fence shouldn't close a 4-backtick fence
        let input = "````\nsome ```inner``` text\n````\n";
        let chunks = split_markdown_fenced_code_blocks(input);
        assert_eq!(chunks.len(), 1);
        assert!(matches!(
            &chunks[0],
            MarkdownChunk::CodeBlock { code, .. }
                if code.contains("```inner```")
        ));
    }

    #[test]
    fn split_markdown_no_code_blocks() {
        let input = "Just plain markdown\n\n# Heading\n";
        let chunks = split_markdown_fenced_code_blocks(input);
        assert_eq!(chunks.len(), 1);
        assert!(matches!(&chunks[0], MarkdownChunk::Text(t) if t.contains("plain markdown")));
    }

    #[test]
    fn split_markdown_code_block_at_start() {
        let input = "```js\nconsole.log('hi')\n```\ntext after";
        let chunks = split_markdown_fenced_code_blocks(input);
        assert_eq!(chunks.len(), 2);
        assert!(matches!(&chunks[0], MarkdownChunk::CodeBlock { .. }));
        assert!(matches!(&chunks[1], MarkdownChunk::Text(t) if t.contains("text after")));
    }

    // ── has_multiple_non_none_styles ────────────────────────────────────
    #[test]
    fn has_multiple_styles_empty() {
        assert!(!has_multiple_non_none_styles(&[]));
    }

    #[test]
    fn has_multiple_styles_all_none() {
        let segments = vec![Segment::plain("text1"), Segment::plain("text2")];
        assert!(!has_multiple_non_none_styles(&segments));
    }

    #[test]
    fn has_multiple_styles_single_style() {
        let style = Style::parse("bold").unwrap();
        let segments = vec![
            Segment::styled("text1", style.clone()),
            Segment::styled("text2", style),
        ];
        assert!(!has_multiple_non_none_styles(&segments));
    }

    #[test]
    fn has_multiple_styles_two_different() {
        let bold = Style::parse("bold").unwrap();
        let red = Style::parse("red").unwrap();
        let segments = vec![
            Segment::styled("text1", bold),
            Segment::styled("text2", red),
        ];
        assert!(has_multiple_non_none_styles(&segments));
    }

    #[test]
    fn has_multiple_styles_ignores_whitespace_only() {
        let bold = Style::parse("bold").unwrap();
        let red = Style::parse("red").unwrap();
        let segments = vec![
            Segment::styled("text1", bold),
            Segment::styled("   ", red), // whitespace-only, should be ignored
        ];
        assert!(!has_multiple_non_none_styles(&segments));
    }

    // ── SpinnerStyle ───────────────────────────────────────────────────
    #[test]
    fn spinner_line_frames_and_interval() {
        let line = SpinnerStyle::Line;
        assert_eq!(line.frames().len(), 6);
        assert_eq!(line.interval_ms(), 100);
    }

    #[test]
    fn spinner_all_frames_non_empty() {
        for style in [SpinnerStyle::Dots, SpinnerStyle::Line, SpinnerStyle::Simple] {
            for frame in style.frames() {
                assert!(!frame.is_empty(), "empty frame in {:?}", style.frames());
            }
        }
    }

    // ── parse_fenced_code_language additional ───────────────────────────
    #[test]
    fn parse_fenced_code_language_with_info_string() {
        // Info string like "rust,no_run" → language is "rust"
        assert_eq!(
            parse_fenced_code_language("rust,no_run"),
            Some("rust".to_string())
        );
    }

    #[test]
    fn parse_fenced_code_language_with_space_and_attr() {
        // "python attrs" → language is "python"
        assert_eq!(
            parse_fenced_code_language("python {.highlight}"),
            Some("python".to_string())
        );
    }

    #[test]
    fn render_markdown_code_fences_highlight_multiple_languages_and_fallback_unknown() {
        let segments = capture_markdown_segments(
            r#"
```rust
fn main() { println!("hi"); }
```

```python
def foo():
    print("hi")
```

```javascript
function foo() { console.log("hi"); }
```

```typescript
interface Foo { x: number }
const foo: Foo = { x: 1 };
const greeting = "hi";
```

```notalanguage
some_code_here();
```
"#,
        );

        for (language, tokens) in [
            ("rust", vec!["fn", "println", "\"hi\""]),
            ("python", vec!["def", "print", "\"hi\""]),
            ("javascript", vec!["function", "console", "\"hi\""]),
            ("typescript", vec!["interface", "const", "\"hi\""]),
        ] {
            let styles = unique_style_debug_for_tokens(&segments, &tokens);
            assert!(
                styles.len() > 1,
                "expected multiple styles for {language} tokens {tokens:?}, got {styles:?}"
            );
        }

        let captured = segments_text(&segments);
        assert!(
            captured.contains("some_code_here"),
            "expected unknown language fence to still render code, got: {captured:?}"
        );
    }
}