shotgun-sh 0.2.29.dev2__py3-none-any.whl → 0.6.1.dev1__py3-none-any.whl

shotgun/tui/screens/chat_screen/messages.py

@@ -0,0 +1,219 @@
+"""Message types for ChatScreen communication.
+
+This module defines Textual message types used for communication
+between widgets and the ChatScreen, particularly for step checkpoints
+and cascade confirmation in the Router's Planning mode.
+"""
+
+from textual.message import Message
+
+from shotgun.agents.models import AgentType
+from shotgun.agents.router.models import CascadeScope, ExecutionPlan, ExecutionStep
+
+__all__ = [
+    # Step checkpoint messages (Stage 4)
+    "StepCompleted",
+    "CheckpointContinue",
+    "CheckpointModify",
+    "CheckpointStop",
+    # Cascade confirmation messages (Stage 5)
+    "CascadeConfirmationRequired",
+    "CascadeConfirmed",
+    "CascadeDeclined",
+    # Plan approval messages (Stage 7)
+    "PlanApprovalRequired",
+    "PlanApproved",
+    "PlanRejected",
+    # Sub-agent lifecycle messages (Stage 8)
+    "SubAgentStarted",
+    "SubAgentCompleted",
+    # Plan panel messages (Stage 11)
+    "PlanUpdated",
+    "PlanPanelClosed",
+]
+
+
+class StepCompleted(Message):
+    """Posted when a plan step completes in Planning mode.
+
+    This message triggers the checkpoint UI to appear, allowing the user
+    to choose whether to continue, modify the plan, or stop execution.
+
+    Attributes:
+        step: The step that was just completed.
+        next_step: The next step to execute, or None if this was the last step.
+    """
+
+    def __init__(self, step: ExecutionStep, next_step: ExecutionStep | None) -> None:
+        super().__init__()
+        self.step = step
+        self.next_step = next_step
+
+
+class CheckpointContinue(Message):
+    """Posted when user chooses to continue to next step.
+
+    This message indicates the user wants to proceed with the next
+    step in the execution plan.
+    """
+
+
+class CheckpointModify(Message):
+    """Posted when user wants to modify the plan.
+
+    This message indicates the user wants to return to the prompt input
+    to make adjustments to the plan before continuing.
+    """
+
+
+class CheckpointStop(Message):
+    """Posted when user wants to stop execution.
+
+    This message indicates the user wants to halt execution while
+    keeping the remaining steps in the plan as pending.
+    """
+
+
+# =============================================================================
+# Cascade Confirmation Messages (Stage 5)
+# =============================================================================
+
+
+class CascadeConfirmationRequired(Message):
+    """Posted when a file with dependents was modified and needs cascade confirmation.
+
+    In Planning mode, after modifying a file like specification.md that has
+    dependent files (plan.md, tasks.md), this message triggers the cascade
+    confirmation UI to appear.
+
+    Attributes:
+        updated_file: The file that was just updated (e.g., "specification.md").
+        dependent_files: List of files that depend on the updated file.
+    """
+
+    def __init__(self, updated_file: str, dependent_files: list[str]) -> None:
+        super().__init__()
+        self.updated_file = updated_file
+        self.dependent_files = dependent_files
+
+
+class CascadeConfirmed(Message):
+    """Posted when user confirms cascade update.
+
+    This message indicates the user wants to proceed with updating
+    dependent files based on the selected scope.
+
+    Attributes:
+        scope: The scope of files to update (ALL, PLAN_ONLY, TASKS_ONLY, NONE).
+    """
+
+    def __init__(self, scope: CascadeScope) -> None:
+        super().__init__()
+        self.scope = scope
+
+
+class CascadeDeclined(Message):
+    """Posted when user declines cascade update.
+
+    This message indicates the user does not want to update dependent
+    files and will handle them manually.
+    """
+
+
+# =============================================================================
+# Plan Approval Messages (Stage 7)
+# =============================================================================
+
+
+class PlanApprovalRequired(Message):
+    """Posted when a multi-step plan is created and needs user approval.
+
+    In Planning mode, after the router creates a plan with multiple steps,
+    this message triggers the approval UI to appear.
+
+    Attributes:
+        plan: The execution plan that needs user approval.
+    """
+
+    def __init__(self, plan: ExecutionPlan) -> None:
+        super().__init__()
+        self.plan = plan
+
+
+class PlanApproved(Message):
+    """Posted when user approves the plan.
+
+    This message indicates the user wants to proceed with executing
+    the plan ("Go Ahead").
+    """
+
+
+class PlanRejected(Message):
+    """Posted when user rejects the plan to clarify/modify.
+
+    This message indicates the user wants to return to the prompt input
+    to modify or clarify the request ("No, Let Me Clarify").
+    """
+
+
+# =============================================================================
+# Sub-Agent Lifecycle Messages (Stage 8)
+# =============================================================================
+
+
+class SubAgentStarted(Message):
+    """Posted when router starts delegating to a sub-agent.
+
+    This message triggers the mode indicator to show the active sub-agent
+    in the format "📋 Planning → Research".
+
+    Attributes:
+        agent_type: The type of sub-agent that started executing.
+    """
+
+    def __init__(self, agent_type: AgentType) -> None:
+        super().__init__()
+        self.agent_type = agent_type
+
+
+class SubAgentCompleted(Message):
+    """Posted when sub-agent delegation completes.
+
+    This message triggers the mode indicator to clear the sub-agent
+    display and return to showing just the mode.
+
+    Attributes:
+        agent_type: The type of sub-agent that completed.
+    """
+
+    def __init__(self, agent_type: AgentType) -> None:
+        super().__init__()
+        self.agent_type = agent_type
+
+
+# =============================================================================
+# Plan Panel Messages (Stage 11)
+# =============================================================================
+
+
+class PlanUpdated(Message):
+    """Posted when the current plan changes.
+
+    This message triggers the plan panel to auto-show/hide based on
+    whether a plan exists.
+
+    Attributes:
+        plan: The updated execution plan, or None if plan was cleared.
+    """
+
+    def __init__(self, plan: ExecutionPlan | None) -> None:
+        super().__init__()
+        self.plan = plan
+
+
+class PlanPanelClosed(Message):
+    """Posted when user closes the plan panel with × button.
+
+    This message indicates the user wants to dismiss the plan panel
+    temporarily. The panel will reopen when the plan changes.
+    """

shotgun/tui/screens/database_locked_dialog.py

@@ -0,0 +1,219 @@
+"""Dialog shown when the database is locked by another process."""
+
+import webbrowser
+
+import pyperclip  # type: ignore[import-untyped]
+from textual import on
+from textual.app import ComposeResult
+from textual.containers import Container, Horizontal
+from textual.events import Resize
+from textual.screen import ModalScreen
+from textual.widgets import Button, Label, Static
+
+from shotgun.exceptions import SHOTGUN_CONTACT_EMAIL
+from shotgun.posthog_telemetry import track_event
+from shotgun.tui.layout import COMPACT_HEIGHT_THRESHOLD
+from shotgun.tui.screens.confirmation_dialog import ConfirmationDialog
+from shotgun.tui.screens.models import LockedDialogAction
+
+# Discord invite link for support
+DISCORD_LINK = "https://discord.gg/5RmY6J2N7s"
+
+
+class DatabaseLockedDialog(ModalScreen[LockedDialogAction]):
+    """Dialog shown when the database is locked by another process.
+
+    This modal informs the user that the database is locked, which could mean
+    another instance is running OR a previous instance shut down unsafely
+    without releasing the lock.
+
+    Returns:
+        LockedDialogAction.RETRY if user wants to retry after closing other instances
+        LockedDialogAction.DELETE if user wants to delete the locked database
+        LockedDialogAction.QUIT if user wants to quit the application
+    """
+
+    DEFAULT_CSS = """
+        DatabaseLockedDialog {
+            align: center middle;
+            background: rgba(0, 0, 0, 0.0);
+        }
+
+        DatabaseLockedDialog > #dialog-container {
+            width: 70%;
+            max-width: 80;
+            height: auto;
+            border: wide $warning;
+            padding: 1 2;
+            layout: vertical;
+            background: $surface;
+        }
+
+        #dialog-title {
+            text-style: bold;
+            color: $warning;
+            padding-bottom: 1;
+        }
+
+        #dialog-message {
+            padding-bottom: 1;
+            color: $text-muted;
+        }
+
+        #support-buttons {
+            layout: horizontal;
+            height: auto;
+            padding-bottom: 1;
+        }
+
+        #support-buttons Button {
+            margin-right: 1;
+        }
+
+        #dialog-buttons {
+            layout: horizontal;
+            align-horizontal: right;
+            height: auto;
+        }
+
+        #dialog-buttons Button {
+            margin-left: 1;
+        }
+
+        #delete-section {
+            layout: horizontal;
+            height: auto;
+            padding-top: 1;
+            border-top: solid $warning-darken-2;
+        }
+
+        #delete-section Static {
+            width: 1fr;
+            color: $text-muted;
+        }
+
+        #delete-section Button {
+            margin-left: 1;
+        }
+
+        /* Compact styles for short terminals */
+        #dialog-container.compact {
+            padding: 0 2;
+            max-height: 98%;
+        }
+
+        #dialog-title.compact {
+            padding-bottom: 0;
+        }
+
+        #dialog-message.compact {
+            padding-bottom: 0;
+        }
+    """
+
+    def compose(self) -> ComposeResult:
+        """Compose the dialog widgets."""
+        with Container(id="dialog-container"):
+            yield Label("Codebase Index Unavailable", id="dialog-title")
+            message = (
+                "Unable to access the codebase index because it is locked.\n\n"
+                "We can't determine if another shotgun instance is currently running "
+                "or if a previous instance shut down unsafely without releasing the lock.\n\n"
+                "To resolve this:\n"
+                "1. Close any other shotgun instances and click Retry\n"
+                "2. If no other instance is running, you can delete the index\n\n"
+                "Need help? Contact support:"
+            )
+            yield Static(message, id="dialog-message")
+            with Horizontal(id="support-buttons"):
+                yield Button(
+                    f"Copy Support Email [{SHOTGUN_CONTACT_EMAIL}]", id="copy-email"
+                )
+                yield Button("Open Support Discord", id="open-discord")
+            with Container(id="dialog-buttons"):
+                yield Button("Retry", id="retry", variant="primary")
+                yield Button("Quit", id="cancel")
+            with Horizontal(id="delete-section"):
+                yield Static("Caution: Only delete if no other instance is running.")
+                yield Button("Delete Index", id="delete", variant="error")
+
+    def on_mount(self) -> None:
+        """Set up the dialog after mounting."""
+        # Track this event in PostHog
+        track_event("database_locked_dialog_shown", {})
+
+        # Focus retry button - user likely wants to retry after closing other instance
+        self.query_one("#retry", Button).focus()
+
+        # Apply compact layout if starting in a short terminal
+        self._apply_compact_layout(self.app.size.height < COMPACT_HEIGHT_THRESHOLD)
+
+    @on(Resize)
+    def handle_resize(self, event: Resize) -> None:
+        """Adjust layout based on terminal height."""
+        self._apply_compact_layout(event.size.height < COMPACT_HEIGHT_THRESHOLD)
+
+    def _apply_compact_layout(self, compact: bool) -> None:
+        """Apply or remove compact layout classes for short terminals."""
+        container = self.query_one("#dialog-container")
+        title = self.query_one("#dialog-title")
+        message = self.query_one("#dialog-message")
+
+        if compact:
+            container.add_class("compact")
+            title.add_class("compact")
+            message.add_class("compact")
+        else:
+            container.remove_class("compact")
+            title.remove_class("compact")
+            message.remove_class("compact")
+
+    @on(Button.Pressed, "#cancel")
+    def handle_cancel(self, event: Button.Pressed) -> None:
+        """Handle cancel button press."""
+        event.stop()
+        self.dismiss(LockedDialogAction.QUIT)
+
+    @on(Button.Pressed, "#retry")
+    def handle_retry(self, event: Button.Pressed) -> None:
+        """Handle retry button press."""
+        event.stop()
+        self.dismiss(LockedDialogAction.RETRY)
+
+    @on(Button.Pressed, "#delete")
+    async def handle_delete(self, event: Button.Pressed) -> None:
+        """Handle delete button press with confirmation."""
+        event.stop()
+        # Show confirmation dialog before proceeding
+        confirmed = await self.app.push_screen_wait(
+            ConfirmationDialog(
+                title="Delete Codebase Index?",
+                message=(
+                    "Have you checked that no other shotgun instance is running?\n\n"
+                    "Deleting while another instance is open could cause data loss. "
+                    "You will need to re-index the codebase after deletion."
+                ),
+                confirm_label="Delete Index",
+                cancel_label="Cancel",
+                confirm_variant="error",
+                danger=True,
+            )
+        )
+        if confirmed:
+            track_event("database_locked_dialog_delete", {})
+            self.dismiss(LockedDialogAction.DELETE)
+
+    @on(Button.Pressed, "#copy-email")
+    def handle_copy_email(self, event: Button.Pressed) -> None:
+        """Copy support email to clipboard."""
+        event.stop()
+        pyperclip.copy(SHOTGUN_CONTACT_EMAIL)
+        track_event("database_locked_dialog_copy_email", {})
+        self.notify("Email copied to clipboard", severity="information")
+
+    @on(Button.Pressed, "#open-discord")
+    def handle_open_discord(self, event: Button.Pressed) -> None:
+        """Open Discord link in browser."""
+        event.stop()
+        webbrowser.open(DISCORD_LINK)
+        track_event("database_locked_dialog_open_discord", {})

shotgun/tui/screens/database_timeout_dialog.py

@@ -0,0 +1,158 @@
+"""Dialog shown when database operation times out."""
+
+from typing import Literal
+
+from textual import on
+from textual.app import ComposeResult
+from textual.containers import Container
+from textual.events import Resize
+from textual.screen import ModalScreen
+from textual.widgets import Button, Label, Static
+
+from shotgun.tui.layout import COMPACT_HEIGHT_THRESHOLD
+
+TimeoutAction = Literal["retry", "skip", "cancel"]
+
+
+class DatabaseTimeoutDialog(ModalScreen[TimeoutAction]):
+    """Dialog shown when database operation takes longer than expected.
+
+    This modal informs the user that the database operation is taking longer
+    than expected (can happen with large codebases) and offers options to
+    wait longer, skip, or cancel.
+
+    Args:
+        codebase_name: Name of the codebase that timed out
+        timeout_seconds: The timeout that was exceeded
+
+    Returns:
+        "retry" - Wait longer (90s timeout)
+        "skip" - Skip this database and continue
+        "cancel" - Cancel the operation
+    """
+
+    DEFAULT_CSS = """
+        DatabaseTimeoutDialog {
+            align: center middle;
+            background: rgba(0, 0, 0, 0.0);
+        }
+
+        DatabaseTimeoutDialog > #dialog-container {
+            width: 60%;
+            max-width: 70;
+            height: auto;
+            border: wide $warning;
+            padding: 1 2;
+            layout: vertical;
+            background: $surface;
+        }
+
+        #dialog-title {
+            text-style: bold;
+            color: $warning;
+            padding-bottom: 1;
+        }
+
+        #dialog-message {
+            padding-bottom: 1;
+            color: $text-muted;
+        }
+
+        #dialog-buttons {
+            layout: horizontal;
+            align-horizontal: right;
+            height: auto;
+        }
+
+        #dialog-buttons Button {
+            margin-left: 1;
+        }
+
+        /* Compact styles for short terminals */
+        #dialog-container.compact {
+            padding: 0 2;
+            max-height: 98%;
+        }
+
+        #dialog-title.compact {
+            padding-bottom: 0;
+        }
+
+        #dialog-message.compact {
+            padding-bottom: 0;
+        }
+    """
+
+    def __init__(self, codebase_name: str = "", timeout_seconds: float = 10.0) -> None:
+        """Initialize the dialog.
+
+        Args:
+            codebase_name: Name of the codebase that timed out
+            timeout_seconds: The timeout that was exceeded
+        """
+        super().__init__()
+        self.codebase_name = codebase_name
+        self.timeout_seconds = timeout_seconds
+
+    def compose(self) -> ComposeResult:
+        """Compose the dialog widgets."""
+        with Container(id="dialog-container"):
+            yield Label("Database Taking Longer Than Expected", id="dialog-title")
+            message = (
+                f"The database operation exceeded {self.timeout_seconds:.0f} seconds.\n\n"
+                "This can happen with large codebases. "
+                "Would you like to wait longer (90 seconds)?"
+            )
+            if self.codebase_name:
+                message = f"Codebase: {self.codebase_name}\n\n" + message
+            yield Static(message, id="dialog-message")
+            with Container(id="dialog-buttons"):
+                yield Button("Wait Longer", id="retry", variant="primary")
+                yield Button("Skip", id="skip")
+                yield Button("Cancel", id="cancel")
+
+    def on_mount(self) -> None:
+        """Set up the dialog after mounting."""
+        # Focus "Wait Longer" button - most likely what user wants
+        self.query_one("#retry", Button).focus()
+
+        # Apply compact layout if starting in a short terminal
+        self._apply_compact_layout(self.app.size.height < COMPACT_HEIGHT_THRESHOLD)
+
+    @on(Resize)
+    def handle_resize(self, event: Resize) -> None:
+        """Adjust layout based on terminal height."""
+        self._apply_compact_layout(event.size.height < COMPACT_HEIGHT_THRESHOLD)
+
+    def _apply_compact_layout(self, compact: bool) -> None:
+        """Apply or remove compact layout classes for short terminals."""
+        container = self.query_one("#dialog-container")
+        title = self.query_one("#dialog-title")
+        message = self.query_one("#dialog-message")
+
+        if compact:
+            container.add_class("compact")
+            title.add_class("compact")
+            message.add_class("compact")
+        else:
+            container.remove_class("compact")
+            title.remove_class("compact")
+            message.remove_class("compact")
+
+    @on(Button.Pressed, "#cancel")
+    def handle_cancel(self, event: Button.Pressed) -> None:
+        """Handle cancel button press."""
+        event.stop()
+        self.dismiss("cancel")
+
+    @on(Button.Pressed, "#skip")
+    def handle_skip(self, event: Button.Pressed) -> None:
+        """Handle skip button press."""
+        event.stop()
+        self.dismiss("skip")
+
+    @on(Button.Pressed, "#retry")
+    def handle_retry(self, event: Button.Pressed) -> None:
+        """Handle retry button press."""
+        event.stop()
+        self.dismiss("retry")