"""
Assistant message component - displays streaming assistant text and thinking blocks.
"""
from __future__ import annotations
from typing import Any
from ..theme import DARK_THEME, RESET, Theme, hex_to_bg
from ..tui import Component, Container
from .markdown import DefaultMarkdownTheme, Markdown
from .spacer import Spacer
class AssistantMessage(Component):
    """Component that renders assistant message with text and thinking blocks."""
    def __init__(self, theme: Theme | None = None, debug_layout: bool = False) -> None:
        """Initialize assistant message component."""
        self._theme = theme or DARK_THEME
        self._debug_layout = debug_layout
        self._content_container = Container()
        self._text_content = ""
        self._thinking_content = ""
        self._thinking_intensity = "medium"  # minimal, low, medium, high
        # Keep references to components for efficient updates
        self._thinking_md: Markdown | None = None
        self._text_md: Markdown | None = None
        self._thinking_spacer: Spacer | None = None
    def append_text(self, delta: str) -> None:
        """Append text delta to current text content."""
        self._text_content += delta
        if self._text_md:

            self._text_md.set_text(self._text_content.strip())
        else:
            # First text delta - add spacer if we have thinking content
            if self._thinking_content and self._thinking_content.strip():
                self._thinking_spacer = Spacer(
                    1, debug_label="after-thinking", debug_layout=self._debug_layout
                )
                self._content_container.add_child(self._thinking_spacer)
            self._text_md = Markdown(
                self._text_content.strip(),
                padding_x=2,
                padding_y=self._theme.message_padding_y,
                theme=DefaultMarkdownTheme(self._theme),
                gutter_prefix=self._theme.assistant_gutter,
            )
            self._content_container.add_child(self._text_md)
    def append_thinking(self, delta: str) -> None:
        """Append thinking delta to current thinking content."""
        self._thinking_content += delta
        if self._thinking_md:

            thinking_text = f"thinking()\n\n{self._thinking_content.strip()}"
            self._thinking_md.set_text(thinking_text)
        else:
            self._rebuild_content()
    def set_text(self, text: str) -> None:
        """Set complete text content."""
        self._text_content = text
        # Always rebuild to ensure proper spacing between thinking and text
        self._rebuild_content()
    def set_thinking(self, thinking: str) -> None:
        """Set complete thinking content."""
        self._thinking_content = thinking
        # Always rebuild to ensure proper spacing between thinking and text
        self._rebuild_content()
    def set_thinking_intensity(self, intensity: str) -> None:
        """Set thinking intensity level (minimal, low, medium, high)."""
        self._thinking_intensity = intensity
        self._rebuild_content()
    def clear(self) -> None:
        """Clear all content."""
        self._text_content = ""
        self._thinking_content = ""
        self._thinking_md = None
        self._text_md = None
        self._thinking_spacer = None
        self._content_container.clear()
    def invalidate(self) -> None:
        """Invalidate content container."""
        self._content_container.invalidate()
    def _rebuild_content(self) -> None:
        """Rebuild content container from current text and thinking."""
        self._content_container.clear()
        self._thinking_md = None
        self._text_md = None
        self._thinking_spacer = None
        # Note: We don't add a spacer here - the previous component (UserMessage)
        # already has padding_y=1 which provides spacing. Adding a spacer here
        # would overwrite that colored padding during differential re-rendering.
        # Render thinking blocks first (if any)
        if self._thinking_content and self._thinking_content.strip():
            # Format thinking like a tool call with background and gutter prefix
            thinking_text = f"thinking()\n\n{self._thinking_content.strip()}"
            if hasattr(self._theme, "thinking_bg_fn"):
                bg_fn = self._theme.thinking_bg_fn
            else:
                def bg_fn(x: str) -> str:
                    return f"{hex_to_bg(self._theme.tool_pending_bg)}{x}{RESET}"
            self._thinking_md = Markdown(
                thinking_text,
                padding_x=2,
                padding_y=self._theme.thinking_padding_y,
                theme=DefaultMarkdownTheme(self._theme),
                bg_fn=bg_fn,
                fg_fn=self._theme.thinking_text_fg,
                gutter_prefix=self._theme.assistant_gutter,
            )
            self._content_container.add_child(self._thinking_md)
        if (
            self._thinking_content
            and self._thinking_content.strip()
            and self._text_content
            and self._text_content.strip()
        ):
            self._content_container.add_child(
                Spacer(1, debug_label="after-thinking", debug_layout=self._debug_layout)
            )
        # Render text content (if any)
        if self._text_content and self._text_content.strip():
            self._text_md = Markdown(
                self._text_content.strip(),
                padding_x=2,
                padding_y=self._theme.message_padding_y,
                theme=DefaultMarkdownTheme(self._theme),
                gutter_prefix=self._theme.assistant_gutter,
            )
            self._content_container.add_child(self._text_md)
    def render(self, width: int) -> list[str]:
        """Render assistant message."""
        return self._content_container.render(width)
    def debug_state(self) -> dict[str, Any]:
        """Return debug state as JSON-serializable dict."""
        return {
            "type": "AssistantMessage",
            "thinking_content": self._thinking_content[:100] + "..."
            if len(self._thinking_content) > 100
            else self._thinking_content,
            "thinking_content_length": len(self._thinking_content),
            "text_content": self._text_content[:100] + "..."
            if len(self._text_content) > 100
            else self._text_content,
            "text_content_length": len(self._text_content),
            "has_thinking_md": self._thinking_md is not None,
            "has_text_md": self._text_md is not None,
            "has_thinking_spacer": self._thinking_spacer is not None,
            "container_children_count": len(self._content_container.children),
            "container_children_types": [
                type(child).__name__ for child in self._content_container.children
            ],
        }
