zjcode 0.0.1__py3-none-any.whl

deepagents_code/tui/widgets/notification_detail.py

@@ -0,0 +1,257 @@
+"""Generic detail modal for a single pending notification.
+
+Used by `NotificationCenterScreen` when the user drills into an entry
+whose payload does not have a dedicated modal (e.g. missing-dependency
+notices). Update-available notifications continue to use
+`UpdateAvailableScreen`, which adds a changelog row on top of the
+action list.
+
+Dismisses with the selected `ActionId`, or `None` on Esc.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, ClassVar
+
+from textual.binding import Binding, BindingType
+from textual.containers import Vertical
+from textual.content import Content
+from textual.message import Message
+from textual.screen import ModalScreen
+from textual.widgets import Static
+
+if TYPE_CHECKING:
+    from textual.app import ComposeResult
+    from textual.events import Click
+
+    from deepagents_code.notifications import (
+        ActionId,
+        NotificationAction,
+        PendingNotification,
+    )
+
+from deepagents_code import theme
+from deepagents_code.config import get_glyphs, is_ascii_mode
+
+
+class DetailActionActivated(Message):
+    """Posted when an `_ActionOption` is clicked with the mouse."""
+
+    def __init__(self, action_id: ActionId) -> None:
+        """Initialize the message.
+
+        Args:
+            action_id: The action the clicked row represents.
+        """
+        super().__init__()
+        self.action_id = action_id
+
+
+class _ActionOption(Static):
+    """Clickable single-line action row."""
+
+    def __init__(self, action: NotificationAction, widget_id: str) -> None:
+        """Initialize the action row widget.
+
+        Args:
+            action: The action this row represents.
+            widget_id: DOM id assigned to the widget.
+        """
+        super().__init__(id=widget_id, classes="nd-action")
+        self._action = action
+        self._is_selected = False
+        self.update(self._render())
+
+    @property
+    def action(self) -> NotificationAction:
+        """Underlying action."""
+        return self._action
+
+    def set_selected(self, selected: bool) -> None:
+        """Toggle selection styling.
+
+        Args:
+            selected: Whether this row is currently under the cursor.
+        """
+        if self._is_selected == selected:
+            return
+        self._is_selected = selected
+        self.set_class(selected, "-selected")
+        self.update(self._render())
+
+    def _render(self) -> Content:
+        glyphs = get_glyphs()
+        cursor = glyphs.cursor if self._is_selected else " "
+        text = f"{cursor} {self._action.label}"
+        if self._action.primary:
+            return Content.styled(text, "bold")
+        return Content(text)
+
+    def on_click(self, event: Click) -> None:
+        """Dispatch a click as a `DetailActionActivated` message."""
+        event.stop()
+        self.post_message(DetailActionActivated(self._action.action_id))
+
+
+class NotificationDetailScreen(ModalScreen["ActionId | None"]):
+    """Modal displaying a single notification's title, body, and actions.
+
+    Activation returns the chosen `ActionId` via `dismiss()`; Esc
+    returns `None` so the caller can keep the underlying notification
+    center open.
+    """
+
+    BINDINGS: ClassVar[list[BindingType]] = [
+        Binding("escape", "cancel", "Back", show=False),
+        Binding("up", "move_up", "Up", show=False, priority=True),
+        Binding("k", "move_up", "Up", show=False, priority=True),
+        Binding("down", "move_down", "Down", show=False, priority=True),
+        Binding("j", "move_down", "Down", show=False, priority=True),
+        Binding("tab", "move_down", "Next", show=False, priority=True),
+        Binding("shift+tab", "move_up", "Previous", show=False, priority=True),
+        Binding("enter", "activate", "Select", show=False, priority=True),
+    ]
+
+    CSS = """
+    NotificationDetailScreen {
+        align: center middle;
+    }
+
+    NotificationDetailScreen > Vertical {
+        width: 68;
+        max-width: 90%;
+        height: auto;
+        max-height: 80%;
+        background: $surface;
+        border: solid $primary;
+        padding: 1 2;
+    }
+
+    NotificationDetailScreen .nd-title {
+        text-style: bold;
+        color: $primary;
+        text-align: center;
+        margin-bottom: 1;
+    }
+
+    NotificationDetailScreen .nd-body {
+        color: $text-muted;
+        margin-bottom: 1;
+    }
+
+    NotificationDetailScreen .nd-action {
+        height: auto;
+        padding: 0 1;
+        color: $text;
+    }
+
+    NotificationDetailScreen .nd-action:hover {
+        background: $surface-lighten-1;
+    }
+
+    NotificationDetailScreen .nd-action.-selected {
+        background: $surface-lighten-1;
+    }
+
+    NotificationDetailScreen .nd-help {
+        height: 1;
+        color: $text-muted;
+        text-style: italic;
+        margin-top: 1;
+        text-align: center;
+    }
+    """
+
+    def __init__(self, entry: PendingNotification) -> None:
+        """Initialize the screen.
+
+        Args:
+            entry: Notification to render. Its actions drive the row
+                list; dismissing via an action returns the chosen
+                `ActionId` to the caller.
+        """
+        super().__init__()
+        self._entry = entry
+        self._options: list[_ActionOption] = []
+        self._selected = 0
+
+    def compose(self) -> ComposeResult:
+        """Compose the modal layout.
+
+        Yields:
+            Title, optional body, one `_ActionOption` per action, and
+            a help footer.
+        """
+        glyphs = get_glyphs()
+        with Vertical():
+            yield Static(self._entry.title, classes="nd-title", markup=False)
+            if self._entry.body:
+                yield Static(self._entry.body, classes="nd-body", markup=False)
+            for idx, action in enumerate(self._entry.actions):
+                option = _ActionOption(action, f"nd-row-{idx}")
+                self._options.append(option)
+                yield option
+            help_text = (
+                f"{glyphs.arrow_up}/{glyphs.arrow_down} or Tab navigate "
+                f"{glyphs.bullet} Enter select "
+                f"{glyphs.bullet} Esc back"
+            )
+            yield Static(help_text, classes="nd-help")
+
+    def on_mount(self) -> None:
+        """Apply ASCII borders and highlight the primary action by default."""
+        if is_ascii_mode():
+            container = self.query_one(Vertical)
+            colors = theme.get_theme_colors(self)
+            container.styles.border = ("ascii", colors.primary)
+        if not self._options:
+            return
+        primary_idx = next(
+            (i for i, a in enumerate(self._entry.actions) if a.primary),
+            0,
+        )
+        self._set_selected(primary_idx)
+
+    def _set_selected(self, new_index: int) -> None:
+        """Move the selection cursor to *new_index*.
+
+        Raises:
+            IndexError: When *new_index* is outside `0..len(self._options)`.
+        """
+        if not self._options:
+            return
+        if not 0 <= new_index < len(self._options):
+            msg = f"selection {new_index} out of range 0..{len(self._options)}"
+            raise IndexError(msg)
+        if new_index != self._selected:
+            self._options[self._selected].set_selected(selected=False)
+        self._selected = new_index
+        self._options[new_index].set_selected(selected=True)
+
+    def action_move_up(self) -> None:
+        """Move the cursor up one row (wraps at the top)."""
+        if not self._options:
+            return
+        self._set_selected((self._selected - 1) % len(self._options))
+
+    def action_move_down(self) -> None:
+        """Move the cursor down one row (wraps at the bottom)."""
+        if not self._options:
+            return
+        self._set_selected((self._selected + 1) % len(self._options))
+
+    def action_activate(self) -> None:
+        """Dismiss with the highlighted action's id."""
+        if not self._options:
+            self.dismiss(None)
+            return
+        self.dismiss(self._options[self._selected].action.action_id)
+
+    def action_cancel(self) -> None:
+        """Close without firing any action."""
+        self.dismiss(None)
+
+    def on_detail_action_activated(self, message: DetailActionActivated) -> None:
+        """Handle a mouse click on an action row."""
+        message.stop()
+        self.dismiss(message.action_id)

deepagents_code/tui/widgets/notification_settings.py

@@ -0,0 +1,170 @@
+"""Notification settings screen for `/notifications` command."""
+
+from __future__ import annotations
+
+import asyncio
+import logging
+from typing import TYPE_CHECKING, ClassVar
+
+from textual.binding import Binding, BindingType
+from textual.containers import VerticalGroup
+from textual.screen import ModalScreen
+from textual.widgets import Checkbox, Static
+
+if TYPE_CHECKING:
+    from textual.app import ComposeResult
+
+from deepagents_code import theme
+from deepagents_code.config import get_glyphs, is_ascii_mode
+
+logger = logging.getLogger(__name__)
+
+# Warning keys and their user-facing labels.
+# Checked = warning is shown at startup (not suppressed). Unchecked = suppressed.
+WARNING_TOGGLES: list[tuple[str, str]] = [
+    ("ripgrep", "Warn when ripgrep is not installed"),
+    ("tavily", "Warn when TAVILY_API_KEY is not set (web search)"),
+]
+
+
+class NotificationSettingsScreen(ModalScreen[None]):
+    """Modal dialog for managing startup warning preferences.
+
+    Each checkbox maps to a key in `[warnings].suppress` in
+    `~/.deepagents/config.toml`. Toggling a checkbox immediately
+    persists the change.
+    """
+
+    BINDINGS: ClassVar[list[BindingType]] = [
+        Binding("escape", "cancel", "Close", show=False),
+        Binding("up", "app.focus_previous", "Previous", show=False, priority=True),
+        Binding("down", "app.focus_next", "Next", show=False, priority=True),
+        Binding("tab", "app.focus_next", "Next", show=False, priority=True),
+        Binding(
+            "shift+tab",
+            "app.focus_previous",
+            "Previous",
+            show=False,
+            priority=True,
+        ),
+    ]
+
+    CSS = """
+    NotificationSettingsScreen {
+        align: center middle;
+        background: transparent;
+    }
+
+    NotificationSettingsScreen > VerticalGroup {
+        width: 65;
+        max-width: 90%;
+        height: auto;
+        max-height: 80%;
+        background: $surface;
+        border: solid $primary;
+        padding: 1 2;
+    }
+
+    NotificationSettingsScreen .ns-title {
+        text-style: bold;
+        color: $primary;
+        text-align: center;
+        margin-bottom: 1;
+    }
+
+    NotificationSettingsScreen .ns-help {
+        height: 1;
+        color: $text-muted;
+        text-style: italic;
+        margin-top: 1;
+        text-align: center;
+    }
+
+    NotificationSettingsScreen Checkbox {
+        margin: 0;
+        border: none;
+        &:focus {
+            border: none;
+        }
+    }
+    """
+
+    def __init__(self, suppressed: set[str]) -> None:
+        """Initialize the notification settings screen.
+
+        Args:
+            suppressed: Set of currently suppressed warning keys.
+        """
+        super().__init__()
+        self._suppressed = suppressed
+
+    def compose(self) -> ComposeResult:
+        """Compose the screen layout.
+
+        Yields:
+            Widgets for the notification settings UI.
+        """
+        glyphs = get_glyphs()
+        with VerticalGroup():
+            yield Static("Notification Settings", classes="ns-title")
+            for key, label in WARNING_TOGGLES:
+                yield Checkbox(
+                    label,
+                    value=key not in self._suppressed,
+                    id=f"ns-{key}",
+                )
+            help_text = (
+                f"{glyphs.arrow_up}/{glyphs.arrow_down} or Tab navigate"
+                f" {glyphs.bullet} Space toggle"
+                f" {glyphs.bullet} Esc close"
+            )
+            yield Static(help_text, classes="ns-help")
+
+    def on_mount(self) -> None:
+        """Apply ASCII border if needed."""
+        if is_ascii_mode():
+            container = self.query_one(VerticalGroup)
+            colors = theme.get_theme_colors(self)
+            container.styles.border = ("ascii", colors.success)
+
+    def on_checkbox_changed(self, event: Checkbox.Changed) -> None:
+        """Persist warning suppression toggle to config.toml on change."""
+        event.stop()
+        checkbox_id = event.checkbox.id
+        if not checkbox_id or not checkbox_id.startswith("ns-"):
+            return
+        key = checkbox_id.removeprefix("ns-")
+        enabled = event.value
+
+        async def _persist() -> None:
+            from deepagents_code.model_config import (
+                suppress_warning,
+                unsuppress_warning,
+            )
+
+            try:
+                if enabled:
+                    ok = await asyncio.to_thread(unsuppress_warning, key)
+                else:
+                    ok = await asyncio.to_thread(suppress_warning, key)
+            except Exception:
+                logger.warning(
+                    "Failed to persist notification setting for %r",
+                    key,
+                    exc_info=True,
+                )
+                ok = False
+            if not ok:
+                self.app.notify(
+                    "Could not save notification preference. "
+                    "Check file permissions for ~/.deepagents/config.toml.",
+                    severity="warning",
+                    timeout=6,
+                    markup=False,
+                )
+
+        self.call_later(_persist)
+
+    def action_cancel(self) -> None:
+        """Close the screen."""
+        self.dismiss(None)

deepagents_code/tui/widgets/restart_prompt.py

@@ -0,0 +1,158 @@
+"""Confirmation modal offered when a change needs an owned-server respawn.
+
+Some changes take effect only when the app-owned LangGraph server subprocess
+spawns: provider/sandbox extras and `--package` installs are imported at spawn
+time, and a Tavily key saved via `/auth` binds the `web_search` tool only at
+spawn time. A `/restart` respawns the subprocess without exiting the TUI, so
+rather than make the user type `/restart` by hand, this modal offers to run
+that restart immediately while leaving deferral one keypress away. The title
+`verb` and `body` are caller-supplied so one modal serves each flow.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, ClassVar, Literal
+
+from textual.binding import Binding, BindingType
+from textual.containers import Vertical
+from textual.content import Content
+from textual.screen import ModalScreen
+from textual.widgets import Static
+
+from deepagents_code.config import get_glyphs
+
+if TYPE_CHECKING:
+    from textual.app import ComposeResult
+
+
+RestartChoice = Literal["restart", "later"]
+"""Outcome of the prompt: restart the server now or defer."""
+
+
+class RestartPromptScreen(ModalScreen[RestartChoice]):
+    """Modal asking whether to restart the server for a spawn-time change.
+
+    Serves both the post-install offer and the post-`/auth` web-search offer;
+    the caller supplies the title `verb` and `body` copy.
+
+    Dismisses with `"restart"` when the user accepts and `"later"` when the
+    user defers. Esc is treated as "later" so the user is never forced into a
+    restart they did not explicitly choose.
+    """
+
+    BINDINGS: ClassVar[list[BindingType]] = [
+        Binding("enter", "restart", "Restart", show=False, priority=True),
+        Binding("escape", "later", "Later", show=False, priority=True),
+    ]
+
+    CSS = """
+    RestartPromptScreen {
+        align: center middle;
+    }
+
+    RestartPromptScreen > Vertical {
+        width: 64;
+        max-width: 90%;
+        height: auto;
+        background: $surface;
+        border: solid $primary;
+        padding: 1 2;
+    }
+
+    RestartPromptScreen .restart-prompt-title {
+        text-style: bold;
+        color: $primary;
+        text-align: center;
+        margin-bottom: 1;
+    }
+
+    RestartPromptScreen .restart-prompt-body {
+        height: auto;
+        color: $text;
+        margin-bottom: 1;
+    }
+
+    RestartPromptScreen .restart-prompt-help {
+        height: 1;
+        color: $text-muted;
+        text-style: italic;
+        text-align: center;
+    }
+    """
+
+    _DEFAULT_BODY = "Restart the server to load it now, or defer with `/restart`."
+
+    def __init__(
+        self,
+        label: str,
+        *,
+        verb: str,
+        body: str | None = None,
+    ) -> None:
+        """Initialize the prompt.
+
+        Args:
+            label: The subject surfaced in the title (e.g. an installed extra
+                name, or a saved credential like ``"Tavily API key"``).
+            verb: Past-tense action shown before `label` in the title — e.g.
+                `"Installed"` for the post-install flow or `"Saved"` for a
+                saved credential. Required (no default) so each flow states its
+                own intent and a future caller can't inherit install-only copy.
+            body: Optional override for the explanatory line under the title.
+                Defaults to the generic restart copy.
+        """
+        super().__init__()
+        self._label = label
+        self._verb = verb
+        self._body = body or self._DEFAULT_BODY
+
+    def compose(self) -> ComposeResult:
+        """Compose the confirmation dialog.
+
+        Yields:
+            Title, body, and help-row widgets parented inside a `Vertical`.
+        """
+        glyphs = get_glyphs()
+        with Vertical():
+            yield Static(
+                Content.from_markup(
+                    "$check $verb [bold]$name[/bold]",
+                    check=glyphs.checkmark,
+                    verb=self._verb,
+                    name=self._label,
+                ),
+                classes="restart-prompt-title",
+                markup=False,
+            )
+            yield Static(
+                self._body,
+                classes="restart-prompt-body",
+                markup=False,
+            )
+            yield Static(
+                "Enter to restart, Esc to defer",
+                classes="restart-prompt-help",
+                markup=False,
+            )
+
+    def action_restart(self) -> None:
+        """Dismiss with `"restart"`."""
+        self.dismiss("restart")
+
+    def action_later(self) -> None:
+        """Dismiss with `"later"`."""
+        self.dismiss("later")
+
+    def action_cancel(self) -> None:
+        """Alias for `action_later` so Esc resolves to a deliberate defer.
+
+        The app's `action_interrupt` (`escape` binding, `priority=True`)
+        fires before this screen's own `escape` binding. When the active
+        screen is a `ModalScreen`, it dispatches to `action_cancel` if
+        present, else falls through to `dismiss(None)`. The current caller
+        no-ops on both `None` and `"later"`, but defining this alias pins Esc
+        to an explicit `"later"` — matching the sibling reconnect/cwd-switch
+        modals and keeping the outcome unambiguous for any future caller that
+        branches on a deliberate defer versus a programmatic dismiss.
+        """
+        self.action_later()