pythinker-code 0.8.0__py3-none-any.whl

pythinker_code/ui/shell/visualize/_worklog.py

@@ -0,0 +1,245 @@
+from __future__ import annotations
+
+import time
+from dataclasses import dataclass
+from enum import Enum
+
+from pythinker_core.tooling import BriefDisplayBlock, DisplayBlock
+from rich.console import Group, RenderableType
+from rich.panel import Panel
+from rich.style import Style
+from rich.text import Text
+
+from pythinker_code.tools.display import (
+    BackgroundTaskDisplayBlock,
+    DiffDisplayBlock,
+    TodoDisplayBlock,
+)
+from pythinker_code.utils.rich.columns import BulletColumns
+from pythinker_code.utils.rich.diff_render import (
+    collect_diff_hunks,
+    render_diff_preview,
+    render_diff_summary_panel,
+)
+from pythinker_code.utils.rich.markdown import Markdown
+
+
+class WorkLogState(Enum):
+    RUNNING = "running"
+    COMPLETED = "completed"
+    FAILED = "failed"
+    DENIED = "denied"
+    INTERRUPTED = "interrupted"
+
+
+@dataclass(frozen=True, slots=True)
+class ToolStyle:
+    label: str
+    icon: str
+    style: str
+
+
+_TOOL_STYLES: dict[str, ToolStyle] = {
+    "Read": ToolStyle("Read", "->", "cyan"),
+    "ReadFile": ToolStyle("Read", "->", "cyan"),
+    "Grep": ToolStyle("Search", "*", "blue"),
+    "Glob": ToolStyle("Find", "*", "blue"),
+    "Edit": ToolStyle("Edit", "<-", "magenta"),
+    "Replace": ToolStyle("Edit", "<-", "magenta"),
+    "Write": ToolStyle("Write", "<-", "magenta"),
+    "WriteFile": ToolStyle("Write", "<-", "magenta"),
+    "ApplyPatch": ToolStyle("Patch", "◆", "magenta"),
+    "Bash": ToolStyle("Shell", "$", "green"),
+    "Shell": ToolStyle("Shell", "$", "green"),
+    "TodoWrite": ToolStyle("Todo", "☑", "yellow"),
+    "Agent": ToolStyle("Subagent", "●", "grey50"),
+    "Task": ToolStyle("Subagent", "●", "grey50"),
+    "AskUser": ToolStyle("Ask", "?", "yellow"),
+    "FetchURL": ToolStyle("Fetch", "%", "blue"),
+    "WebFetch": ToolStyle("Fetch", "%", "blue"),
+    "WebSearch": ToolStyle("Search", "◈", "blue"),
+    "Skill": ToolStyle("Skill", "◇", "cyan"),
+}
+
+
+_STATE_STYLE = {
+    WorkLogState.RUNNING: "bright_white",
+    WorkLogState.COMPLETED: "grey50",
+    WorkLogState.FAILED: "red",
+    WorkLogState.DENIED: "grey50 strike",
+    WorkLogState.INTERRUPTED: "yellow",
+}
+
+_STATE_ICON_STYLE = {
+    WorkLogState.COMPLETED: Style(color="green"),
+    WorkLogState.FAILED: Style(color="red"),
+    WorkLogState.DENIED: Style(color="grey50", strike=True),
+    WorkLogState.INTERRUPTED: Style(color="yellow"),
+}
+
+
+def _state_icon(state: WorkLogState) -> tuple[str, Style]:
+    if state == WorkLogState.RUNNING:
+        glyph = "●" if int(time.monotonic() / 0.8) % 2 == 0 else " "
+        return glyph, Style(color="grey50")
+    return "●", _STATE_ICON_STYLE[state]
+
+
+def tool_style(name: str) -> ToolStyle:
+    return _TOOL_STYLES.get(name, ToolStyle(name, "⚙", "blue"))
+
+
+def denied_error(message: str) -> bool:
+    lowered = message.lower().strip()
+    if lowered == "denied" or lowered.startswith("denied:"):
+        return True
+    return any(
+        needle in lowered
+        for needle in (
+            "questionrejectederror",
+            "rejected permission",
+            "specified a rule",
+            "user dismissed",
+            "tool calls are disabled",
+        )
+    )
+
+
+def render_worklog_entry(
+    *,
+    label: str,
+    target: str | None = None,
+    state: WorkLogState,
+    detail: str | None = None,
+    icon: str = "•",
+    icon_style: str = "blue",
+    icon_renderable: RenderableType | None = None,
+    children: list[RenderableType] | None = None,
+) -> RenderableType:
+    line = Text()
+    if icon_renderable is None:
+        glyph, glyph_style = _state_icon(state)
+        line.append(glyph, style=glyph_style)
+        line.append(" ")
+    line.append(label, style="bold")
+    if target:
+        line.append(" ")
+        line.append(target, style="grey70")
+    line.append(" ")
+    line.append(state.value, style=_STATE_STYLE[state])
+    if detail:
+        line.append(" · ", style="grey50")
+        line.append(detail, style=_STATE_STYLE[state])
+    if icon_renderable is not None:
+        return BulletColumns(
+            line if not children else Group(line, *children),
+            bullet=icon_renderable,
+        )
+    if not children:
+        return line
+    return Group(line, *children)
+
+
+def render_worklog_card(
+    title: str,
+    body: RenderableType,
+    *,
+    subtitle: str | None = None,
+    border_style: str = "grey39",
+) -> Panel:
+    return Panel(
+        body,
+        title=title,
+        title_align="left",
+        subtitle=subtitle,
+        subtitle_align="left",
+        border_style=border_style,
+        padding=(0, 1),
+        expand=False,
+    )
+
+
+def render_display_blocks(
+    display: list[DisplayBlock], *, is_error: bool = False
+) -> list[RenderableType]:
+    rendered: list[RenderableType] = []
+    idx = 0
+    while idx < len(display):
+        block = display[idx]
+        if isinstance(block, DiffDisplayBlock):
+            path = block.path
+            diff_blocks: list[DiffDisplayBlock] = []
+            while idx < len(display):
+                candidate = display[idx]
+                if not isinstance(candidate, DiffDisplayBlock) or candidate.path != path:
+                    break
+                diff_blocks.append(candidate)
+                idx += 1
+            if any(item.is_summary for item in diff_blocks):
+                rendered.append(
+                    render_worklog_card(
+                        "Diff", render_diff_summary_panel(path, diff_blocks), subtitle=path
+                    )
+                )
+                continue
+            hunks, added_total, removed_total = collect_diff_hunks(diff_blocks)
+            if hunks:
+                preview_lines, _ = render_diff_preview(
+                    path,
+                    hunks,
+                    added_total,
+                    removed_total,
+                    max_lines=8,
+                )
+                rendered.append(
+                    render_worklog_card(
+                        "Diff",
+                        Group(*preview_lines),
+                    )
+                )
+            continue
+        if isinstance(block, BriefDisplayBlock):
+            text = block.text.strip()
+            if text:
+                title = "Error" if is_error else "Report"
+                style = "red" if is_error else "grey70"
+                if "\n" in text or len(text) > 100:
+                    rendered.append(
+                        render_worklog_card(
+                            title,
+                            Markdown(text, style=style),
+                            border_style="red" if is_error else "grey39",
+                        )
+                    )
+                else:
+                    rendered.append(Markdown(text, style=style))
+            idx += 1
+            continue
+        if isinstance(block, TodoDisplayBlock):
+            lines: list[str] = []
+            for todo in block.items:
+                match todo.status:
+                    case "done":
+                        marker = "✓"
+                    case "in_progress":
+                        marker = "→"
+                    case _:
+                        marker = "·"
+                lines.append(f"{marker} {todo.title}")
+            rendered.append(render_worklog_card("Todos", Text("\n".join(lines), style="grey70")))
+            idx += 1
+            continue
+        if isinstance(block, BackgroundTaskDisplayBlock):
+            rendered.append(
+                render_worklog_card(
+                    "Background task",
+                    Text(
+                        f"{block.task_id} [{block.status}] {block.kind}: {block.description}",
+                        style="grey70",
+                    ),
+                )
+            )
+            idx += 1
+            continue
+        idx += 1
+    return rendered

pythinker_code/ui/theme.py

@@ -0,0 +1,395 @@
+"""Centralized terminal color theme definitions.
+
+All UI-facing colors live here so that switching between dark and light
+terminal themes only requires changing the active ``ThemeName``.
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Literal
+
+from prompt_toolkit.styles import Style as PTKStyle
+from rich.style import Style as RichStyle
+
+type ThemeName = Literal["dark", "light"]
+
+
+# ---------------------------------------------------------------------------
+# Diff colors (used by utils/rich/diff_render.py)
+# ---------------------------------------------------------------------------
+
+
+@dataclass(frozen=True, slots=True)
+class DiffColors:
+    add_bg: RichStyle
+    del_bg: RichStyle
+    add_hl: RichStyle
+    del_hl: RichStyle
+
+
+_DIFF_DARK = DiffColors(
+    add_bg=RichStyle(bgcolor="#12261e"),
+    del_bg=RichStyle(bgcolor="#2d1214"),
+    add_hl=RichStyle(bgcolor="#1a4a2e"),
+    del_hl=RichStyle(bgcolor="#5c1a1d"),
+)
+
+_DIFF_LIGHT = DiffColors(
+    add_bg=RichStyle(bgcolor="#dafbe1"),
+    del_bg=RichStyle(bgcolor="#ffebe9"),
+    add_hl=RichStyle(bgcolor="#aff5b4"),
+    del_hl=RichStyle(bgcolor="#ffc1c0"),
+)
+
+
+# ---------------------------------------------------------------------------
+# Task browser colors (used by ui/shell/task_browser.py)
+# ---------------------------------------------------------------------------
+
+
+def _task_browser_style_dark() -> PTKStyle:
+    return PTKStyle.from_dict(
+        {
+            "header": "bg:#1f2937 #e5e7eb",
+            "header.title": "bg:#1f2937 #67e8f9 bold",
+            "header.meta": "bg:#1f2937 #9ca3af",
+            "status.running": "bg:#1f2937 #86efac bold",
+            "status.success": "bg:#1f2937 #86efac",
+            "status.warning": "bg:#1f2937 #fbbf24",
+            "status.error": "bg:#1f2937 #fca5a5",
+            "status.info": "bg:#1f2937 #93c5fd",
+            "task-list": "bg:#111827 #d1d5db",
+            "task-list.checked": "bg:#164e63 #ecfeff bold",
+            "frame.border": "#155e75",
+            "frame.label": "bg:#0f172a #67e8f9 bold",
+            "footer": "bg:#0f172a #cbd5e1",
+            "footer.key": "bg:#0f172a #67e8f9 bold",
+            "footer.text": "bg:#0f172a #cbd5e1",
+            "footer.warning": "bg:#7f1d1d #fecaca bold",
+            "footer.meta": "bg:#0f172a #94a3b8",
+        }
+    )
+
+
+def _task_browser_style_light() -> PTKStyle:
+    return PTKStyle.from_dict(
+        {
+            "header": "bg:#e5e7eb #1f2937",
+            "header.title": "bg:#e5e7eb #0e7490 bold",
+            "header.meta": "bg:#e5e7eb #6b7280",
+            "status.running": "bg:#e5e7eb #166534 bold",
+            "status.success": "bg:#e5e7eb #166534",
+            "status.warning": "bg:#e5e7eb #92400e",
+            "status.error": "bg:#e5e7eb #991b1b",
+            "status.info": "bg:#e5e7eb #1e40af",
+            "task-list": "bg:#f9fafb #374151",
+            "task-list.checked": "bg:#cffafe #164e63 bold",
+            "frame.border": "#0e7490",
+            "frame.label": "bg:#f1f5f9 #0e7490 bold",
+            "footer": "bg:#f1f5f9 #475569",
+            "footer.key": "bg:#f1f5f9 #0e7490 bold",
+            "footer.text": "bg:#f1f5f9 #475569",
+            "footer.warning": "bg:#fee2e2 #991b1b bold",
+            "footer.meta": "bg:#f1f5f9 #64748b",
+        }
+    )
+
+
+# ---------------------------------------------------------------------------
+# Prompt / completion menu colors (used by ui/shell/prompt.py)
+# ---------------------------------------------------------------------------
+
+
+_PROMPT_STYLE_DARK = {
+    "bottom-toolbar": "noreverse",
+    # Input area — minimal: no background bar, only the prompt glyph is
+    # colored. Lets the terminal background show through so the input row
+    # reads as a single line of text rather than a chrome panel.
+    "compact-input": "",
+    "compact-input.prompt": "fg:#8abeb7 bold",
+    "compact-input.frame": "fg:#5f87ff",
+    "running-prompt-placeholder": "fg:#7c8594 italic",
+    "running-prompt-separator": "fg:#4a5568",
+    # Slash completion menu — selected row gets the same selected-bg as cards.
+    "slash-completion-menu": "",
+    "slash-completion-menu.separator": "fg:#4a5568",
+    "slash-completion-menu.marker": "fg:#4a5568",
+    "slash-completion-menu.marker.current": "fg:#8abeb7 bold",
+    "slash-completion-menu.command": "fg:#a6adba",
+    "slash-completion-menu.command.match": "fg:#8abeb7 bold",
+    "slash-completion-menu.meta": "fg:#7c8594",
+    "slash-completion-menu.command.current": "bg:#3a3a4a fg:#8abeb7 bold",
+    "slash-completion-menu.command.match.current": "bg:#3a3a4a fg:#8abeb7 bold",
+    "slash-completion-menu.meta.current": "bg:#3a3a4a fg:#a6adba",
+    "slash-completion-menu.row.current": "bg:#3a3a4a",
+}
+
+_PROMPT_STYLE_LIGHT = {
+    "bottom-toolbar": "noreverse",
+    "compact-input": "",
+    "compact-input.prompt": "fg:#5a8080 bold",
+    "compact-input.frame": "fg:#547da7",
+    "running-prompt-placeholder": "fg:#6b7280 italic",
+    "running-prompt-separator": "fg:#d1d5db",
+    "slash-completion-menu": "",
+    "slash-completion-menu.separator": "fg:#d1d5db",
+    "slash-completion-menu.marker": "fg:#9ca3af",
+    "slash-completion-menu.marker.current": "fg:#5a8080 bold",
+    "slash-completion-menu.command": "fg:#4b5563",
+    "slash-completion-menu.command.match": "fg:#5a8080 bold",
+    "slash-completion-menu.meta": "fg:#6b7280",
+    "slash-completion-menu.command.current": "bg:#d0d0e0 fg:#5a8080 bold",
+    "slash-completion-menu.command.match.current": "bg:#d0d0e0 fg:#5a8080 bold",
+    "slash-completion-menu.meta.current": "bg:#d0d0e0 fg:#4b5563",
+    "slash-completion-menu.row.current": "bg:#d0d0e0",
+}
+
+
+# ---------------------------------------------------------------------------
+# Bottom toolbar fragment colors (used by ui/shell/prompt.py)
+# ---------------------------------------------------------------------------
+
+
+@dataclass(frozen=True, slots=True)
+class ToolbarColors:
+    separator: str
+    yolo_label: str
+    auto_label: str
+    plan_label: str
+    plan_prompt: str
+    cwd: str
+    bg_tasks: str
+    tip: str
+    tip_key: str
+
+
+_TOOLBAR_DARK = ToolbarColors(
+    separator="fg:#4d4d4d",
+    yolo_label="bold fg:#ffff00",
+    auto_label="bold fg:#ff8800",
+    plan_label="bold fg:#00aaff",
+    plan_prompt="fg:#00aaff",
+    cwd="fg:#666666",
+    bg_tasks="fg:#888888",
+    tip="fg:#555555",
+    tip_key="fg:#777777 bold",
+)
+
+_TOOLBAR_LIGHT = ToolbarColors(
+    separator="fg:#d1d5db",
+    yolo_label="bold fg:#b45309",
+    auto_label="bold fg:#c2410c",
+    plan_label="bold fg:#2563eb",
+    plan_prompt="fg:#2563eb",
+    cwd="fg:#6b7280",
+    bg_tasks="fg:#4b5563",
+    tip="fg:#9ca3af",
+    tip_key="fg:#6b7280 bold",
+)
+
+
+# ---------------------------------------------------------------------------
+# MCP status prompt colors (used by ui/shell/mcp_status.py)
+# ---------------------------------------------------------------------------
+
+
+@dataclass(frozen=True, slots=True)
+class MCPPromptColors:
+    text: str
+    detail: str
+    connected: str
+    connecting: str
+    pending: str
+    failed: str
+
+
+_MCP_PROMPT_DARK = MCPPromptColors(
+    text="fg:#d4d4d4",
+    detail="fg:#7c8594",
+    connected="fg:#56d364",
+    connecting="fg:#56a4ff",
+    pending="fg:#f2cc60",
+    failed="fg:#ff7b72",
+)
+
+_MCP_PROMPT_LIGHT = MCPPromptColors(
+    text="fg:#374151",
+    detail="fg:#6b7280",
+    connected="fg:#166534",
+    connecting="fg:#1d4ed8",
+    pending="fg:#92400e",
+    failed="fg:#dc2626",
+)
+
+
+# ---------------------------------------------------------------------------
+# Public API — resolve by theme name
+# ---------------------------------------------------------------------------
+
+_active_theme: ThemeName = "dark"
+
+
+def set_active_theme(theme: ThemeName) -> None:
+    global _active_theme
+    _active_theme = theme
+
+
+def get_active_theme() -> ThemeName:
+    return _active_theme
+
+
+def get_diff_colors() -> DiffColors:
+    return _DIFF_LIGHT if _active_theme == "light" else _DIFF_DARK
+
+
+def get_task_browser_style() -> PTKStyle:
+    return _task_browser_style_light() if _active_theme == "light" else _task_browser_style_dark()
+
+
+def get_prompt_style() -> PTKStyle:
+    d = _PROMPT_STYLE_LIGHT if _active_theme == "light" else _PROMPT_STYLE_DARK
+    return PTKStyle.from_dict(d)
+
+
+def get_toolbar_colors() -> ToolbarColors:
+    return _TOOLBAR_LIGHT if _active_theme == "light" else _TOOLBAR_DARK
+
+
+def get_mcp_prompt_colors() -> MCPPromptColors:
+    return _MCP_PROMPT_LIGHT if _active_theme == "light" else _MCP_PROMPT_DARK
+
+
+# ---------------------------------------------------------------------------
+# Pythinker semantic TUI tokens (used by ui/shell/components/* and the tool
+# renderer registry). Default semantic token palette
+# and light themes so the Pythinker code path renders with the reference
+# palette. Existing pythinker styles continue to work — these tokens add a
+# parallel naming layer keyed by *semantic role* rather than concrete color.
+# ---------------------------------------------------------------------------
+
+
+@dataclass(frozen=True, slots=True)
+class TuiTokens:
+    """Pythinker semantic theme tokens.
+
+    Values are hex strings (``"#rrggbb"``) or the empty string for "use
+    terminal default". Background tokens (``*_bg``) are intended for
+    Rich ``bgcolor=`` arguments; foreground tokens for ``color=``.
+    """
+
+    # Core
+    accent: str
+    border: str
+    border_accent: str
+    border_muted: str
+    success: str
+    error: str
+    warning: str
+    muted: str
+    dim: str
+    text: str
+    thinking_text: str
+    # Backgrounds
+    selected_bg: str
+    user_message_bg: str
+    user_message_text: str
+    custom_message_bg: str
+    custom_message_text: str
+    custom_message_label: str
+    tool_pending_bg: str
+    tool_success_bg: str
+    tool_error_bg: str
+    tool_title: str
+    tool_output: str
+    # Diffs
+    tool_diff_added: str
+    tool_diff_removed: str
+    tool_diff_context: str
+    # Bash mode accent
+    bash_mode: str
+
+
+_TUI_TOKENS_DARK = TuiTokens(
+    accent="#8abeb7",
+    border="#5f87ff",
+    border_accent="#00d7ff",
+    border_muted="#505050",
+    success="#b5bd68",
+    error="#cc6666",
+    warning="#ffff00",
+    muted="#808080",
+    dim="#666666",
+    text="",
+    thinking_text="#808080",
+    selected_bg="#3a3a4a",
+    user_message_bg="#343541",
+    user_message_text="",
+    custom_message_bg="#2d2838",
+    custom_message_text="",
+    custom_message_label="#9575cd",
+    tool_pending_bg="#282832",
+    tool_success_bg="#283228",
+    tool_error_bg="#3c2828",
+    tool_title="",
+    tool_output="#808080",
+    tool_diff_added="#b5bd68",
+    tool_diff_removed="#cc6666",
+    tool_diff_context="#808080",
+    bash_mode="#b5bd68",
+)
+
+
+_TUI_TOKENS_LIGHT = TuiTokens(
+    accent="#5a8080",
+    border="#547da7",
+    border_accent="#5a8080",
+    border_muted="#b0b0b0",
+    success="#588458",
+    error="#aa5555",
+    warning="#9a7326",
+    muted="#6c6c6c",
+    dim="#767676",
+    text="",
+    thinking_text="#6c6c6c",
+    selected_bg="#d0d0e0",
+    user_message_bg="#e8e8e8",
+    user_message_text="",
+    custom_message_bg="#ede7f6",
+    custom_message_text="",
+    custom_message_label="#7e57c2",
+    tool_pending_bg="#e8e8f0",
+    tool_success_bg="#e8f0e8",
+    tool_error_bg="#f0e8e8",
+    tool_title="",
+    tool_output="#6c6c6c",
+    tool_diff_added="#588458",
+    tool_diff_removed="#aa5555",
+    tool_diff_context="#6c6c6c",
+    bash_mode="#588458",
+)
+
+
+def get_tui_tokens(theme: ThemeName | None = None) -> TuiTokens:
+    """Return Pythinker semantic tokens for *theme* (defaults to active)."""
+    name = theme if theme is not None else _active_theme
+    return _TUI_TOKENS_LIGHT if name == "light" else _TUI_TOKENS_DARK
+
+
+def tui_rich_style(token: str, *, theme: ThemeName | None = None) -> RichStyle:
+    """Resolve a TuiTokens field name to a Rich Style.
+
+    Background tokens (suffix ``_bg``) produce a style with ``bgcolor``;
+    everything else produces a style with ``color``. Empty hex values
+    (``""``) yield an empty style — Rich falls back to terminal defaults.
+
+    Raises:
+        AttributeError: If *token* is not a known TuiTokens field.
+    """
+    tokens = get_tui_tokens(theme)
+    value = getattr(tokens, token)
+    if not value:
+        return RichStyle()
+    if token.endswith("_bg"):
+        return RichStyle(bgcolor=value)
+    return RichStyle(color=value)