zjcode 0.0.1__py3-none-any.whl

deepagents_code/tui/widgets/skill_trust.py

@@ -0,0 +1,131 @@
+"""Trust prompt for skills that resolve outside trusted skill directories.
+
+When a `/skill:<name>` invocation reads a `SKILL.md` whose resolved path (via
+symlink) falls outside every trusted skill root, `load_skill_content` refuses
+the read. Rather than forcing the user to quit, edit env/config, and relaunch,
+this non-blocking modal asks for an in-the-moment decision. Allowing persists
+the resolved target directory to the skill trust store.
+"""
+
+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.screen import ModalScreen
+from textual.widgets import Static
+
+if TYPE_CHECKING:
+    from textual.app import ComposeResult
+
+
+class SkillTrustScreen(ModalScreen[bool | None]):
+    """Approval overlay for a skill resolving outside trusted directories.
+
+    Dismisses with `True` when the user allows the resolved target and `False`
+    when the user declines. Esc is treated as deny so the user is never forced
+    into reading from an untrusted location they did not explicitly choose.
+
+    Typed `bool | None` rather than `bool`: a programmatic pop, or the app's
+    priority Esc binding falling through to `dismiss(None)` (see
+    `action_cancel`), can yield `None`. The caller collapses `None` and `False`
+    to deny (`if not allowed`), so both dismiss values fail closed.
+    """
+
+    BINDINGS: ClassVar[list[BindingType]] = [
+        Binding("enter", "confirm", "Allow", show=False, priority=True),
+        Binding("escape", "cancel", "Deny", show=False, priority=True),
+    ]
+
+    CSS = """
+    SkillTrustScreen {
+        align: center middle;
+    }
+
+    SkillTrustScreen > Vertical {
+        width: 72;
+        max-width: 90%;
+        height: auto;
+        background: $surface;
+        border: solid $warning;
+        padding: 1 2;
+    }
+
+    SkillTrustScreen .skill-trust-title {
+        text-style: bold;
+        color: $warning;
+        text-align: center;
+        margin-bottom: 1;
+    }
+
+    SkillTrustScreen .skill-trust-body {
+        height: auto;
+        color: $text;
+        margin-bottom: 1;
+    }
+
+    SkillTrustScreen .skill-trust-help {
+        height: 1;
+        color: $text-muted;
+        text-style: italic;
+        text-align: center;
+    }
+    """
+
+    def __init__(self, skill_name: str, target_dir: str) -> None:
+        """Initialize the prompt.
+
+        Args:
+            skill_name: Name of the skill being invoked.
+            target_dir: Resolved directory the skill's `SKILL.md` lives in.
+        """
+        super().__init__()
+        self._skill_name = skill_name
+        self._target_dir = target_dir
+
+    def compose(self) -> ComposeResult:
+        """Compose the skill trust dialog.
+
+        Yields:
+            Title, body, and help-row widgets parented inside a `Vertical`.
+        """
+        with Vertical():
+            yield Static(
+                "Allow skill from outside trusted directories?",
+                classes="skill-trust-title",
+                markup=False,
+            )
+            yield Static(
+                Content.from_markup(
+                    "Skill [bold]$name[/bold] resolves to [bold]$dir[/bold], "
+                    "outside your trusted skill directories. Allowing reads "
+                    "instructions from there and remembers this location for "
+                    "future sessions.",
+                    name=self._skill_name,
+                    dir=self._target_dir,
+                ),
+                classes="skill-trust-body",
+                markup=False,
+            )
+            yield Static(
+                "Enter to allow, Esc to deny",
+                classes="skill-trust-help",
+                markup=False,
+            )
+
+    def action_confirm(self) -> None:
+        """Dismiss with `True`."""
+        self.dismiss(True)
+
+    def action_cancel(self) -> None:
+        """Dismiss with `False`.
+
+        The method name must stay `cancel`: the app owns a priority `escape`
+        binding that, for an active `ModalScreen`, dispatches to
+        `action_cancel` if present and otherwise falls through to
+        `dismiss(None)`. Renaming this would silently regress Esc to a
+        `None` dismiss instead of an explicit deny.
+        """
+        self.dismiss(False)

deepagents_code/tui/widgets/startup_tip.py

@@ -0,0 +1,91 @@
+"""Startup tip widget shown above the chat input."""
+
+from __future__ import annotations
+
+import random
+from typing import Any
+
+from textual.content import Content
+from textual.widgets import Static
+
+from deepagents_code._env_vars import HIDE_SPLASH_TIPS, is_env_truthy
+
+_TIPS: dict[str, int] = {
+    "Use @ to reference files and / for commands": 3,
+    "Try /threads to resume a previous conversation": 2,
+    "Use /offload when your conversation gets long": 2,
+    "Use /copy to copy the latest message": 3,
+    "Use /tools to list the tools available to the agent": 1,
+    "Use /mcp login <server> to authenticate MCP servers": 1,
+    "Use /remember to save learnings from this conversation": 1,
+    "Use /model to switch models mid-conversation": 2,
+    "Use /effort to change the current model's reasoning effort": 1,
+    "Press ctrl+x to compose prompts in your external editor": 1,
+    "Use /skill:<name> to invoke a skill directly": 1,
+    "Use /theme to customize the TUI's colors": 1,
+    "Use /skill-creator to build reusable agent skills": 1,
+    "Ask for a workflow to fan work out to subagents in parallel": 3,
+    "Use /timestamps to show or hide message timestamp footers": 1,
+    "Use /agents to browse and switch between your available agents": 2,
+    "Press Shift+Tab to toggle auto-approve mode": 2,
+    "Use !! for incognito shell commands that stay out of model context": 1,
+    "Deep Agents can explain its own features and look up its docs. Ask it how to use.": 3,  # noqa: E501
+}
+"""Tips shown above the chat input. One is chosen at random per launch,
+weighted by these relative selection weights."""
+
+# Fail fast at import if the registry is ever emptied or given a non-positive
+# weight: `random.choices` would otherwise raise a cryptic error at widget
+# construction. `_TIPS` is a hardcoded constant, so this never fires in
+# practice — it just guards future edits.
+if not _TIPS:
+    msg = "_TIPS must not be empty"
+    raise ValueError(msg)
+if any(weight <= 0 for weight in _TIPS.values()):
+    msg = "_TIPS weights must be positive"
+    raise ValueError(msg)
+
+
+def _pick_tip() -> str:
+    """Pick one startup tip using the configured relative weights.
+
+    Returns:
+        Tip text selected from `_TIPS`.
+    """
+    tips = list(_TIPS.keys())
+    weights = list(_TIPS.values())
+    return random.choices(tips, weights=weights, k=1)[0]  # noqa: S311
+
+
+def show_startup_tip() -> bool:
+    """Return whether startup tips should be shown.
+
+    Returns:
+        `True` when startup tips are enabled for the current process.
+    """
+    return not is_env_truthy(HIDE_SPLASH_TIPS)
+
+
+class StartupTip(Static):
+    """One startup tip displayed above the chat input."""
+
+    DEFAULT_CSS = """
+    StartupTip {
+        height: auto;
+        color: $text-muted;
+        text-style: dim italic;
+        padding: 0 1;
+    }
+    """
+
+    def __init__(self, tip: str | None = None, **kwargs: Any) -> None:
+        """Initialize the startup tip widget.
+
+        Args:
+            tip: Tip text to display. When omitted, one weighted tip is selected.
+            **kwargs: Additional arguments passed to `Static`.
+        """
+        self.tip: str = tip if tip is not None else _pick_tip()
+        # Styling (dim italic, muted color) is owned by DEFAULT_CSS, which
+        # applies to the whole widget — no need to restyle the spans here.
+        super().__init__(Content.assemble("Tip: ", self.tip), **kwargs)