shotgun-sh 0.2.17__py3-none-any.whl → 0.3.3.dev1__py3-none-any.whl

shotgun/tui/screens/chat/codebase_index_prompt_screen.py

@@ -1,12 +1,35 @@
 """Modal dialog for codebase indexing prompts."""
 
+import os
+import webbrowser
 from pathlib import Path
 
 from textual import on
 from textual.app import ComposeResult
-from textual.containers import Container
+from textual.containers import Container, VerticalScroll
+from textual.events import Resize
 from textual.screen import ModalScreen
-from textual.widgets import Button, Label, Static
+from textual.widgets import Button, Label, Markdown, Static
+
+from shotgun.tui.layout import COMPACT_HEIGHT_THRESHOLD
+from shotgun.utils.file_system_utils import get_shotgun_home
+
+
+def _is_home_directory() -> bool:
+    """Check if cwd is user's home directory.
+
+    Can be simulated with HOME_DIRECTORY_SIMULATE=true env var for testing.
+    """
+    if os.environ.get("HOME_DIRECTORY_SIMULATE", "").lower() == "true":
+        return True
+    return Path.cwd() == Path.home()
+
+
+def _track_event(event_name: str) -> None:
+    """Track an event to PostHog."""
+    from shotgun.posthog_telemetry import track_event
+
+    track_event(event_name)
 
 
 class CodebaseIndexPromptScreen(ModalScreen[bool]):
@@ -19,39 +42,182 @@ class CodebaseIndexPromptScreen(ModalScreen[bool]):
         }
 
         CodebaseIndexPromptScreen > #index-prompt-dialog {
-            width: 60%;
-            max-width: 60;
+            width: 80%;
+            max-width: 90;
             height: auto;
+            max-height: 85%;
             border: wide $primary;
             padding: 1 2;
             layout: vertical;
             background: $surface;
+        }
+
+        #index-prompt-title {
+            text-style: bold;
+            color: $text-accent;
+            text-align: center;
+            padding-bottom: 1;
+        }
+
+        #index-prompt-content {
             height: auto;
+            max-height: 1fr;
+        }
+
+        #index-prompt-info {
+            padding: 0 1;
         }
 
         #index-prompt-buttons {
             layout: horizontal;
             align-horizontal: right;
             height: auto;
+            padding-top: 1;
+        }
+
+        #index-prompt-buttons Button {
+            margin: 0 1;
+            min-width: 12;
+        }
+
+        #index-prompt-warning {
+            background: $surface-lighten-1;
+            color: $text;
+            padding: 1 2;
+            margin-bottom: 1;
+            text-align: center;
+        }
+
+        #compact-link {
+            text-align: center;
+            padding: 1 0;
+            display: none;
+        }
+
+        /* Compact styles for short terminals */
+        #index-prompt-dialog.compact {
+            padding: 0 1;
+            border: none;
+            max-height: 100%;
+        }
+
+        #index-prompt-dialog.compact #index-prompt-content {
+            display: none;
+        }
+
+        #index-prompt-dialog.compact #compact-link {
+            display: block;
+        }
+
+        #index-prompt-dialog.compact #index-prompt-warning {
+            padding: 0;
+            margin-bottom: 0;
+            background: transparent;
+        }
+
+        #index-prompt-dialog.compact #index-prompt-title {
+            padding-bottom: 0;
+        }
+
+        #index-prompt-dialog.compact #index-prompt-buttons {
+            padding-top: 0;
         }
     """
 
     def compose(self) -> ComposeResult:
+        storage_path = get_shotgun_home() / "codebases"
+        cwd = Path.cwd()
+        is_home = _is_home_directory()
+
         with Container(id="index-prompt-dialog"):
-            yield Label("Index this codebase?", id="index-prompt-title")
-            yield Static(
-                f"Would you like to index the codebase at:\n{Path.cwd()}\n\n"
-                "This is required for the agent to understand your code and answer "
-                "questions about it. Without indexing, the agent cannot analyze "
-                "your codebase."
-            )
-            with Container(id="index-prompt-buttons"):
-                yield Button(
-                    "Index now",
-                    id="index-prompt-confirm",
-                    variant="primary",
+            if is_home:
+                # Show warning for home directory
+                yield Label(
+                    "Home directory detected",
+                    id="index-prompt-title",
+                )
+                yield Static(
+                    "Running from home directory isn't recommended.",
+                    id="index-prompt-warning",
+                )
+                with Container(id="index-prompt-buttons"):
+                    yield Button(
+                        "Quit",
+                        id="index-prompt-quit",
+                    )
+                    yield Button(
+                        "Continue without indexing",
+                        id="index-prompt-continue",
+                    )
+            else:
+                # Normal indexing prompt
+                content = f"""
+## 🔒 Your code never leaves your computer
+
+Shotgun will index the codebase at:
+**`{cwd}`**
+_(This is the current working directory where you started Shotgun)_
+
+### What happens during indexing:
+
+- **Stays on your computer**: Index is stored locally at `{storage_path}` - it will not be stored on a server
+- **Zero cost**: Indexing runs entirely on your machine
+- **Runs in the background**: Usually takes 1-3 minutes, and you can continue using Shotgun while it indexes
+- **Enable code understanding**: Allows Shotgun to answer questions about your codebase
+
+---
+
+If you're curious, you can review how Shotgun indexes/queries code by taking a look at the [source code](https://github.com/shotgun-sh/shotgun).
+
+We take your privacy seriously. You can read our full [privacy policy](https://app.shotgun.sh/privacy) for more details.
+"""
+                yield Label(
+                    "Want to index your codebase?",
+                    id="index-prompt-title",
+                )
+                # Compact mode: show only a link
+                yield Static(
+                    "[@click=screen.open_faq]Learn more about indexing[/]",
+                    id="compact-link",
+                    markup=True,
                 )
-                yield Button("Not now", id="index-prompt-cancel")
+                # Full mode: show detailed content
+                with VerticalScroll(id="index-prompt-content"):
+                    yield Markdown(content, id="index-prompt-info")
+                with Container(id="index-prompt-buttons"):
+                    yield Button(
+                        "Not now",
+                        id="index-prompt-cancel",
+                    )
+                    yield Button(
+                        "Index now",
+                        id="index-prompt-confirm",
+                        variant="primary",
+                    )
+
+    def on_mount(self) -> None:
+        """Track when the home directory warning screen is shown and apply compact layout."""
+        if _is_home_directory():
+            _track_event("home_directory_warning_shown")
+        # 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."""
+        dialog = self.query_one("#index-prompt-dialog")
+        if compact:
+            dialog.add_class("compact")
+        else:
+            dialog.remove_class("compact")
+
+    def action_open_faq(self) -> None:
+        """Open the FAQ page in a browser."""
+        webbrowser.open("https://github.com/shotgun-sh/shotgun?tab=readme-ov-file#faq")
 
     @on(Button.Pressed, "#index-prompt-cancel")
     def handle_cancel(self, event: Button.Pressed) -> None:
@@ -62,3 +228,16 @@ class CodebaseIndexPromptScreen(ModalScreen[bool]):
     def handle_confirm(self, event: Button.Pressed) -> None:
         event.stop()
         self.dismiss(True)
+
+    @on(Button.Pressed, "#index-prompt-continue")
+    def handle_continue(self, event: Button.Pressed) -> None:
+        """Continue without indexing when in home directory."""
+        event.stop()
+        _track_event("home_directory_warning_continue")
+        self.dismiss(False)
+
+    @on(Button.Pressed, "#index-prompt-quit")
+    def handle_quit(self, event: Button.Pressed) -> None:
+        event.stop()
+        _track_event("home_directory_warning_quit")
+        self.app.exit()

shotgun/tui/screens/chat_screen/command_providers.py

@@ -5,6 +5,7 @@ from textual.command import DiscoveryHit, Hit, Provider
 
 from shotgun.agents.models import AgentType
 from shotgun.codebase.models import CodebaseGraph
+from shotgun.tui.screens.chat_screen.hint_message import HintMessage
 from shotgun.tui.screens.model_picker import ModelPickerScreen
 from shotgun.tui.screens.provider_config import ProviderConfigScreen
 
@@ -271,8 +272,8 @@ class DeleteCodebasePaletteProvider(Provider):
         try:
             result = await self.chat_screen.codebase_sdk.list_codebases()
         except Exception as exc:  # pragma: no cover - defensive UI path
-            self.chat_screen.notify(
-                f"Unable to load codebases: {exc}", severity="error"
+            self.chat_screen.agent_manager.add_hint_message(
+                HintMessage(message=f"❌ Unable to load codebases: {exc}")
             )
             return []
         return result.graphs
@@ -359,6 +360,11 @@ class UnifiedCommandProvider(Provider):
             self.open_model_picker,
             help="🤖 Choose which AI model to use",
         )
+        yield DiscoveryHit(
+            "Share specs to workspace",
+            self.chat_screen.share_specs_command,
+            help="📤 Upload .shotgun/ files to share with your team",
+        )
         yield DiscoveryHit(
             "Show context",
             self.chat_screen.action_show_context,
@@ -411,6 +417,11 @@ class UnifiedCommandProvider(Provider):
                 self.open_model_picker,
                 "🤖 Choose which AI model to use",
             ),
+            (
+                "Share specs to workspace",
+                self.chat_screen.share_specs_command,
+                "📤 Upload .shotgun/ files to share with your team",
+            ),
             (
                 "Show context",
                 self.chat_screen.action_show_context,

shotgun/tui/screens/chat_screen/hint_message.py

@@ -1,14 +1,23 @@
 from typing import Literal
 
 from pydantic import BaseModel
+from textual import on
 from textual.app import ComposeResult
+from textual.containers import Horizontal
 from textual.widget import Widget
-from textual.widgets import Markdown
+from textual.widgets import Button, Label, Markdown, Static
+
+from shotgun.logging_config import get_logger
+
+logger = get_logger(__name__)
 
 
 class HintMessage(BaseModel):
     message: str
     kind: Literal["hint"] = "hint"
+    # Optional email copy functionality
+    email: str | None = None
+    markdown_after: str | None = None
 
 
 class HintMessageWidget(Widget):
@@ -30,6 +39,30 @@ class HintMessageWidget(Widget):
           }
         }
 
+        HintMessageWidget .email-copy-row {
+            width: auto;
+            height: auto;
+            margin: 1 0;
+        }
+
+        HintMessageWidget .email-text {
+            width: auto;
+            margin-right: 1;
+            content-align: left middle;
+        }
+
+        HintMessageWidget .copy-btn {
+            width: auto;
+            min-width: 12;
+        }
+
+        HintMessageWidget #copy-status {
+            height: 1;
+            width: 100%;
+            margin-top: 1;
+            content-align: left middle;
+        }
+
     """
 
     def __init__(self, message: HintMessage) -> None:
@@ -37,4 +70,46 @@ class HintMessageWidget(Widget):
         self.message = message
 
     def compose(self) -> ComposeResult:
+        # Main message markdown
         yield Markdown(markdown=f"{self.message.message}")
+
+        # Optional email copy section
+        if self.message.email:
+            # Email + copy button on same line
+            with Horizontal(classes="email-copy-row"):
+                yield Static(f"Contact: {self.message.email}", classes="email-text")
+                yield Button("Copy email", id="copy-email-btn", classes="copy-btn")
+
+            # Status feedback label
+            yield Label("", id="copy-status")
+
+            # Optional markdown after email
+            if self.message.markdown_after:
+                yield Markdown(self.message.markdown_after)
+
+    @on(Button.Pressed, "#copy-email-btn")
+    def _copy_email(self) -> None:
+        """Copy email address to clipboard when button is pressed."""
+        if not self.message.email:
+            return
+
+        status_label = self.query_one("#copy-status", Label)
+
+        try:
+            import pyperclip  # type: ignore[import-untyped]  # noqa: PGH003
+
+            pyperclip.copy(self.message.email)
+            status_label.update("✓ Copied to clipboard!")
+            logger.debug(
+                f"Successfully copied email to clipboard: {self.message.email}"
+            )
+
+        except ImportError:
+            status_label.update(
+                f"⚠️ Clipboard unavailable. Please manually copy: {self.message.email}"
+            )
+            logger.warning("pyperclip not available for clipboard operations")
+
+        except Exception as e:
+            status_label.update(f"⚠️ Copy failed: {e}")
+            logger.error(f"Failed to copy email to clipboard: {e}", exc_info=True)

shotgun/tui/screens/confirmation_dialog.py

@@ -5,9 +5,12 @@ 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
+
 ButtonVariant = Literal["default", "primary", "success", "warning", "error"]
 
 
@@ -87,6 +90,20 @@ class ConfirmationDialog(ModalScreen[bool]):
         #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__(
@@ -138,6 +155,29 @@ class ConfirmationDialog(ModalScreen[bool]):
         # Focus cancel button by default for safety
         self.query_one("#cancel", 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."""

shotgun/tui/screens/directory_setup.py

@@ -1,4 +1,4 @@
-"""Screen for setting up the local .shotgun directory."""
+"""Screen for displaying .shotgun directory creation errors."""
 
 from __future__ import annotations
 
@@ -8,13 +8,20 @@ from textual import on
 from textual.app import ComposeResult
 from textual.containers import Horizontal, Vertical
 from textual.screen import Screen
-from textual.widgets import Button, Static
-
-from shotgun.utils.file_system_utils import ensure_shotgun_directory_exists
+from textual.widgets import Button, Label, Static
 
 
 class DirectorySetupScreen(Screen[None]):
-    """Prompt the user to initialize the .shotgun directory."""
+    """Display an error when .shotgun directory creation fails."""
+
+    def __init__(self, error_message: str) -> None:
+        """Initialize the error screen.
+
+        Args:
+            error_message: The error message to display to the user.
+        """
+        super().__init__()
+        self.error_message = error_message
 
     CSS = """
         DirectorySetupScreen {
@@ -56,58 +63,55 @@ class DirectorySetupScreen(Screen[None]):
         #directory-actions > * {
             margin-right: 2;
         }
+
+        #directory-status {
+            height: auto;
+            padding: 0 1;
+            min-height: 1;
+            color: $error;
+            text-align: center;
+        }
     """
 
     BINDINGS = [
-        ("enter", "confirm", "Initialize"),
+        ("enter", "retry", "Retry"),
         ("escape", "cancel", "Exit"),
     ]
 
     def compose(self) -> ComposeResult:
         with Vertical(id="titlebox"):
-            yield Static("Directory setup", id="directory-setup-title")
-            yield Static("Shotgun keeps workspace data in a .shotgun directory.\n")
-            yield Static("Initialize it in the current directory?\n")
-            yield Static(f"[$foreground-muted]({Path.cwd().resolve()})[/]")
-        with Horizontal(id="directory-actions"):
-            yield Button(
-                "Initialize and proceed \\[ENTER]", variant="primary", id="initialize"
+            yield Static(
+                "Failed to create .shotgun directory", id="directory-setup-title"
+            )
+            yield Static("Shotgun was unable to create the .shotgun directory in:\n")
+            yield Static(f"[$foreground-muted]({Path.cwd().resolve()})[/]\n")
+            yield Static(f"[bold red]Error:[/] {self.error_message}\n")
+            yield Static(
+                "This directory is required for storing workspace data. "
+                "Please check permissions and try again."
             )
-            yield Button("Exit without setup \\[ESC]", variant="default", id="exit")
+        yield Label("", id="directory-status")
+        with Horizontal(id="directory-actions"):
+            yield Button("Retry \\[ENTER]", variant="primary", id="retry")
+            yield Button("Exit \\[ESC]", variant="default", id="exit")
 
     def on_mount(self) -> None:
-        self.set_focus(self.query_one("#initialize", Button))
+        self.set_focus(self.query_one("#retry", Button))
 
-    def action_confirm(self) -> None:
-        self._initialize_directory()
+    def action_retry(self) -> None:
+        """Retry by dismissing the screen, which will trigger refresh_startup_screen."""
+        self.dismiss()
 
     def action_cancel(self) -> None:
-        self._exit_application()
+        """Exit the application."""
+        self.app.exit()
 
-    @on(Button.Pressed, "#initialize")
-    def _on_initialize_pressed(self) -> None:
-        self._initialize_directory()
+    @on(Button.Pressed, "#retry")
+    def _on_retry_pressed(self) -> None:
+        """Retry by dismissing the screen."""
+        self.dismiss()
 
     @on(Button.Pressed, "#exit")
     def _on_exit_pressed(self) -> None:
-        self._exit_application()
-
-    def _initialize_directory(self) -> None:
-        try:
-            path = ensure_shotgun_directory_exists()
-        except Exception as exc:  # pragma: no cover - defensive; textual path
-            self.notify(f"Failed to initialize directory: {exc}", severity="error")
-            return
-
-        # Double-check a directory now exists; guard against unexpected filesystem state.
-        if not path.is_dir():
-            self.notify(
-                "Unable to initialize .shotgun directory due to filesystem conflict.",
-                severity="error",
-            )
-            return
-
-        self.dismiss()
-
-    def _exit_application(self) -> None:
+        """Exit the application."""
         self.app.exit()

shotgun/tui/screens/feedback.py

@@ -76,6 +76,13 @@ class FeedbackScreen(Screen[Feedback | None]):
         #feedback-type-list {
             padding: 1;
         }
+
+        #feedback-status {
+            height: auto;
+            padding: 0 1;
+            min-height: 1;
+            color: $error;
+        }
     """
 
     BINDINGS = [
@@ -96,6 +103,7 @@ class FeedbackScreen(Screen[Feedback | None]):
             "",
             id="feedback-description",
         )
+        yield Label("", id="feedback-status")
         with Horizontal(id="feedback-actions"):
             yield Button("Submit", variant="primary", id="submit")
             yield Button("Cancel \\[ESC]", id="cancel")
@@ -176,9 +184,8 @@ class FeedbackScreen(Screen[Feedback | None]):
         description = text_area.text.strip()
 
         if not description:
-            self.notify(
-                "Please enter a description before submitting.", severity="error"
-            )
+            status_label = self.query_one("#feedback-status", Label)
+            status_label.update("❌ Please enter a description before submitting.")
             return
 
         app = cast("ShotgunApp", self.app)

shotgun/tui/screens/github_issue.py

@@ -6,7 +6,7 @@ from textual import on
 from textual.app import ComposeResult
 from textual.containers import Container, Vertical
 from textual.screen import ModalScreen
-from textual.widgets import Button, Markdown, Static
+from textual.widgets import Button, Label, Markdown, Static
 
 
 class GitHubIssueScreen(ModalScreen[None]):
@@ -47,6 +47,13 @@ class GitHubIssueScreen(ModalScreen[None]):
             margin: 1 1;
             min-width: 20;
         }
+
+        #issue-status {
+            height: auto;
+            padding: 1;
+            min-height: 1;
+            text-align: center;
+        }
     """
 
     BINDINGS = [
@@ -85,6 +92,7 @@ We review all issues and will respond as soon as possible!
                     id="issue-markdown",
                 )
             with Vertical(id="issue-buttons"):
+                yield Label("", id="issue-status")
                 yield Button(
                     "🐙 Open GitHub Issues", id="github-button", variant="primary"
                 )
@@ -94,7 +102,8 @@ We review all issues and will respond as soon as possible!
     def handle_github(self) -> None:
         """Open GitHub issues page in browser."""
         webbrowser.open("https://github.com/shotgun-sh/shotgun/issues")
-        self.notify("Opening GitHub Issues in your browser...")
+        status_label = self.query_one("#issue-status", Label)
+        status_label.update("✓ Opening GitHub Issues in your browser...")
 
     @on(Button.Pressed, "#close-button")
     def handle_close(self) -> None: