iCode-TUI 0.29.1__py3-none-any.whl

chrys/app/tui/screens/dialogs/connection_test.py

@@ -0,0 +1,183 @@
+# Copyright (c) Huawei Technologies Co., Ltd. 2026. All rights reserved.
+
+"""ConnectionTestDialog — modal shown while testing an external connection.
+
+Shared by the agent-config Test buttons: MCP servers and ACP agents both
+run their one-shot connection tests behind this dialog (``subject_label``
+names the subject in the border captions).
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, ClassVar
+
+from rich.text import Text
+from textual import on
+from textual.binding import Binding
+from textual.containers import HorizontalGroup, VerticalGroup
+from textual.widgets import Button, LoadingIndicator, Static
+
+from chrys.app.tui.i18n import render_str, widget_localizer
+from chrys.app.tui.screens.dialogs.base import BaseDialog
+from chrys.app.tui.widgets import DialogButtonRow, DialogButtonSpec, StableAutoHeightScroll
+from chrys.app.tui.widgets.markdown import VirtualizedMarkdown
+from chrys.foundation.i18n import MessageRef, msg
+
+if TYPE_CHECKING:
+    from textual.app import ComposeResult
+    from textual.geometry import Size
+
+
+_TESTING_CONNECTION = msg(
+    "tui.connection_test.testing",
+    fallback="Testing {subject} Connection",
+)
+_OK = msg("tui.connection_test.button.ok", fallback="Ok")
+_SUCCESS = msg("tui.connection_test.success", fallback="Connection Successful")
+_FAILED = msg("tui.connection_test.failed", fallback="Connection Failed")
+_MCP_SERVER = msg("tui.connection_test.default_subject", fallback="MCP Server")
+_DEFAULT_MCP_SERVER = _MCP_SERVER.bind()
+
+
+class _ResultMarkdown(VirtualizedMarkdown, can_focus=False):
+    """Markdown result body that lets the wrapping dialog scroll own scrolling.
+
+    Mirroring the ask-user dialog pattern: the widget grows to its virtual
+    height so the ``StableAutoHeightScroll`` wrapper scrolls it — that keeps
+    the scrollbar flush with the container edge while the markdown's own
+    horizontal padding insets the content.
+    """
+
+    def watch_virtual_size(self, _old: Size, new: Size) -> None:
+        if new.height > 0:
+            self.styles.height = new.height
+
+
+class ConnectionTestDialog(BaseDialog[None]):
+    """Modal dialog shown during an external connection test (MCP or ACP).
+
+    Lifecycle:
+    - Opens with a ``LoadingIndicator`` and no buttons (the user cannot
+      dismiss while the test runs — Esc is a no-op).
+    - Once the caller invokes :meth:`set_result`, the spinner is replaced
+      by the result message and an ``Ok`` button appears.  Esc only
+      dismisses when ``allow_esc=True`` is passed (timeout case);
+      otherwise the user must explicitly click ``Ok``.
+    """
+
+    BINDINGS: ClassVar[list] = [
+        Binding("escape", "dismiss_if_allowed", show=False, priority=True),
+    ]
+
+    CSS_PATH = "connection_test.tcss"
+
+    def __init__(self, server_name: str = "", *, subject_label: MessageRef | str = _DEFAULT_MCP_SERVER) -> None:
+        self._server_name = server_name.strip()
+        self._subject_label = subject_label
+        self._resolved = False
+        self._esc_allowed = False
+        self._mounted = False
+        self._pending_result: tuple[bool, str, bool, bool] | None = None
+        super().__init__()
+
+    def compose(self) -> ComposeResult:
+        localizer = widget_localizer(self)
+        subject = (
+            self._subject_label if isinstance(self._subject_label, str) else render_str(localizer, self._subject_label)
+        )
+        with VerticalGroup(id="connection-test-container") as container:
+            # Identity rides the top border; the lifecycle state (testing →
+            # success/failure) rides the bottom one.
+            container.border_title = Text(self._server_name or subject)
+            container.border_subtitle = Text(render_str(localizer, _TESTING_CONNECTION.bind(subject=subject)))
+            with VerticalGroup(id="connection-test-inner"):
+                yield LoadingIndicator(id="connection-test-loading")
+                message = Static("", id="connection-test-message", markup=False)
+                message.display = False
+                yield message
+            buttons = DialogButtonRow(
+                DialogButtonSpec(Text(render_str(localizer, _OK.bind())), id="connection-test-ok", variant="primary"),
+                id="connection-test-buttons",
+            )
+            buttons.display = False
+            yield buttons
+
+    def on_mount(self) -> None:
+        self._mounted = True
+        if self._pending_result is not None:
+            success, message, allow_esc, markdown = self._pending_result
+            self._pending_result = None
+            self._apply_result(success, message, allow_esc, markdown)
+
+    def set_result(self, success: bool, message: str, allow_esc: bool = False, *, markdown: bool = False) -> None:
+        """Swap the loading state for a final result + Ok button.
+
+        ``allow_esc`` enables Esc-to-dismiss for the result state — pass
+        ``True`` only for timeout outcomes; otherwise the user must
+        explicitly click ``Ok``.
+
+        ``markdown`` renders ``message`` as Markdown in a scrollable body and
+        widens the dialog — used for structured reports such as the ACP
+        handshake summary. Plain results keep the compact text layout.
+
+        Safe to call before the dialog has finished mounting — the result is
+        buffered and applied in :meth:`on_mount`.
+        """
+        if self._resolved:
+            return
+        self._resolved = True
+
+        # Owners may still report after removal starts, which prunes the children before this
+        # dialog unmounts. Buffer as before mount; nothing will render it.
+        if not self._mounted or self._pruning or self._closing:
+            self._pending_result = (success, message, allow_esc, markdown)
+            return
+
+        self._apply_result(success, message, allow_esc, markdown)
+
+    def _apply_result(self, success: bool, message: str, allow_esc: bool, markdown: bool) -> None:
+        self._esc_allowed = allow_esc
+
+        if markdown:
+            # The Markdown body replaces the whole inner group (spinner and
+            # plain-text message included) and owns its own scrolling; the
+            # ``-markdown`` class widens the container to report size.
+            self.add_class("-markdown")
+            self.query_one("#connection-test-inner", VerticalGroup).remove()
+            container = self.query_one("#connection-test-container", VerticalGroup)
+            container.mount(
+                StableAutoHeightScroll(
+                    _ResultMarkdown(message, id="connection-test-markdown", open_links=False),
+                    id="connection-test-markdown-scroll",
+                ),
+                before="#connection-test-buttons",
+            )
+        else:
+            loading = self.query_one("#connection-test-loading", LoadingIndicator)
+            loading.display = False
+            loading.remove()
+
+            message_widget = self.query_one("#connection-test-message", Static)
+            message_widget.update(message)
+            message_widget.display = True
+
+        buttons = self.query_one("#connection-test-buttons", HorizontalGroup)
+        buttons.display = True
+
+        container = self.query_one("#connection-test-container", VerticalGroup)
+        localizer = widget_localizer(self)
+        container.border_subtitle = Text(render_str(localizer, (_SUCCESS if success else _FAILED).bind()))
+
+        self.add_class("-success" if success else "-error")
+        self.query_one("#connection-test-ok", Button).focus()
+
+    @on(Button.Pressed, "#connection-test-ok")
+    def _on_ok(self, event: Button.Pressed) -> None:
+        self.dismiss(None)
+
+    def action_dismiss_if_allowed(self) -> None:
+        if self._resolved and self._esc_allowed:
+            self.dismiss(None)
+
+    def _allow_click_outside_dismiss(self) -> bool:
+        return self._resolved and self._esc_allowed

chrys/app/tui/screens/dialogs/connection_test.tcss

@@ -0,0 +1,92 @@
+ConnectionTestDialog {
+    align: center middle;
+}
+
+#connection-test-container {
+    width: 60;
+    max-width: 90%;
+    height: auto;
+    min-height: 8;
+    max-height: 85%;
+    background: $surface;
+    border: round $tui-border-warning $border-opacity;
+    border-title-color: $warning;
+    border-title-style: bold;
+    border-subtitle-color: $warning;
+    border-subtitle-style: bold;
+}
+
+#connection-test-inner {
+    width: 100%;
+    height: auto;
+    min-height: 4;
+    background: $surface;
+    padding: 1 1;
+}
+
+#connection-test-loading {
+    width: 100%;
+    height: 1;
+    color: $warning;
+    margin: 1 0;
+}
+
+#connection-test-message {
+    width: 100%;
+    height: auto;
+    max-height: 14;
+    overflow-y: auto;
+    text-align: center;
+    margin: 0 1;
+}
+
+/* Markdown result mode (structured reports, e.g. the ACP handshake): the
+   container grows into a wide report pane and the Markdown body scrolls. */
+ConnectionTestDialog.-markdown #connection-test-container {
+    width: 90%;
+    max-width: 110;
+    max-height: 90%;
+}
+
+/* The scroll wrapper owns the scrollbar at the container's inner edge; the
+   markdown's own `padding: 0 2` (widget default) insets the content. */
+#connection-test-markdown-scroll {
+    margin: 1 0 0 0;
+    padding: 0;
+}
+
+#connection-test-markdown {
+    width: 100%;
+    height: auto;
+}
+
+#connection-test-buttons {
+    dock: bottom;
+    align: center top;
+    height: 3;
+    padding: 0 1;
+}
+
+#connection-test-buttons > Button {
+    margin: 0 1;
+}
+
+ConnectionTestDialog.-success #connection-test-container {
+    border: round $tui-border-success $border-opacity;
+    border-title-color: $success;
+    border-subtitle-color: $success;
+}
+
+ConnectionTestDialog.-success #connection-test-loading {
+    color: $success;
+}
+
+ConnectionTestDialog.-error #connection-test-container {
+    border: round $tui-border-error $border-opacity;
+    border-title-color: $error;
+    border-subtitle-color: $error;
+}
+
+ConnectionTestDialog.-error #connection-test-loading {
+    color: $error;
+}

chrys/app/tui/screens/dialogs/diagram.py

@@ -0,0 +1,102 @@
+# Copyright (c) Huawei Technologies Co., Ltd. 2026. All rights reserved.
+
+"""Modal viewer for a compiled terminal Mermaid diagram."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, ClassVar
+
+from rich.cells import cell_len
+from rich.text import Text
+from textual.containers import VerticalGroup
+
+from chrys.app.tui.binding_display import CLOSE_BINDING, COPY_BINDING, TOGGLE_VIEW_BINDING, localized_binding
+from chrys.app.tui.clipboard import copy_text_to_clipboards
+from chrys.app.tui.copy_messages import COPIED_TITLE
+from chrys.app.tui.i18n import render_str, widget_localizer
+from chrys.app.tui.screens.dialogs.base import BaseDialog
+from chrys.app.tui.widgets.markdown.diagram.model import CompiledDiagram
+from chrys.app.tui.widgets.markdown.diagram.viewer import DiagramViewport
+from chrys.foundation.i18n import msg
+
+if TYPE_CHECKING:
+    from textual.app import ComposeResult
+
+
+_TITLE = msg("tui.diagram.title", fallback="Mermaid diagram")
+_VIEW_HINT = msg(
+    "tui.diagram.view_hint",
+    fallback="Press Space to switch source/rendered view · Press c to copy",
+)
+_COPIED_SOURCE = msg("tui.diagram.copied_source", fallback="Copied Mermaid source")
+
+
+def _source_canvas(diagram: CompiledDiagram) -> CompiledDiagram:
+    """Build a control-safe, cell-measured canvas for the Mermaid source."""
+    source_lines = diagram.source.expandtabs(4).splitlines() or [""]
+    rows = tuple(
+        " " + "".join("�" if ord(char) < 32 or 0x7F <= ord(char) < 0xA0 else char for char in line)
+        for line in source_lines
+    )
+    return CompiledDiagram(
+        source=diagram.source,
+        kind=diagram.kind,
+        width=max((cell_len(row) for row in rows), default=1) + 1,
+        height=len(rows),
+        rows=rows,
+        diagnostics=diagram.diagnostics,
+    )
+
+
+class DiagramDialog(BaseDialog[None]):
+    """Large diagram view with native draggable scrollbars on both axes."""
+
+    CSS_PATH = "diagram.tcss"
+
+    BINDINGS: ClassVar[list] = [
+        localized_binding("escape", "close", CLOSE_BINDING, show=False, priority=True),
+        localized_binding("q", "close", CLOSE_BINDING, show=False),
+        localized_binding("space", "toggle_view", TOGGLE_VIEW_BINDING, show=False, priority=True),
+        localized_binding("c", "copy_source", COPY_BINDING, show=False),
+    ]
+
+    def __init__(self, diagram: CompiledDiagram) -> None:
+        self.diagram = diagram
+        self._source_canvas = _source_canvas(diagram)
+        self._showing_source = False
+        super().__init__()
+
+    def compose(self) -> ComposeResult:
+        localizer = widget_localizer(self)
+        with VerticalGroup(id="diagram-container") as container:
+            container.border_title = Text(render_str(localizer, _TITLE.bind()))
+            container.border_subtitle = Text(render_str(localizer, _VIEW_HINT.bind()))
+            yield DiagramViewport(self.diagram, id="diagram-viewport")
+
+    def on_mount(self) -> None:
+        """Put keyboard scrolling directly on the diagram surface."""
+        self.query_one(DiagramViewport).focus()
+
+    def action_copy_source(self) -> None:
+        """Copy the original Mermaid source without terminal decoration."""
+        copy_text_to_clipboards(self.app, self.diagram.source)
+        localizer = widget_localizer(self)
+        self.notify(
+            render_str(localizer, _COPIED_SOURCE.bind()),
+            title=render_str(localizer, COPIED_TITLE.bind()),
+            timeout=2,
+            markup=False,
+        )
+
+    def action_toggle_view(self) -> None:
+        """Switch the viewport between rendered cells and literal Mermaid source."""
+        self._showing_source = not self._showing_source
+        diagram = self._source_canvas if self._showing_source else self.diagram
+        self.query_one(DiagramViewport).set_diagram(diagram)
+
+    def action_close(self) -> None:
+        """Close the dialog."""
+        self.dismiss(None)
+
+
+__all__ = ["DiagramDialog"]

chrys/app/tui/screens/dialogs/diagram.tcss

@@ -0,0 +1,24 @@
+DiagramDialog {
+    align: center middle;
+    background: $background 40%;
+}
+
+#diagram-container {
+    width: 94%;
+    max-width: 140;
+    height: 90%;
+    max-height: 50;
+    border: round $tui-border-primary $border-opacity;
+    border-title-color: $primary;
+    border-title-style: bold;
+    border-subtitle-align: right;
+    border-subtitle-color: $primary;
+    background: $surface;
+    padding: 0;
+}
+
+#diagram-viewport {
+    width: 100%;
+    height: 1fr;
+    background: $surface;
+}