shotgun-sh 0.2.8.dev2__py3-none-any.whl → 0.3.3.dev1__py3-none-any.whl

shotgun/tui/screens/chat/codebase_index_prompt_screen.py

@@ -0,0 +1,243 @@
+"""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, VerticalScroll
+from textual.events import Resize
+from textual.screen import ModalScreen
+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]):
+    """Modal dialog asking whether to index the detected codebase."""
+
+    DEFAULT_CSS = """
+        CodebaseIndexPromptScreen {
+            align: center middle;
+            background: rgba(0, 0, 0, 0.0);
+        }
+
+        CodebaseIndexPromptScreen > #index-prompt-dialog {
+            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"):
+            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,
+                )
+                # 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:
+        event.stop()
+        self.dismiss(False)
+
+    @on(Button.Pressed, "#index-prompt-confirm")
+    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/codebase_index_selection.py

@@ -0,0 +1,12 @@
+"""Codebase indexing selection models."""
+
+from pathlib import Path
+
+from pydantic import BaseModel
+
+
+class CodebaseIndexSelection(BaseModel):
+    """User-selected repository path and name for indexing."""
+
+    repo_path: Path
+    name: str

shotgun/tui/screens/chat/help_text.py

@@ -0,0 +1,40 @@
+"""Helper functions for chat screen help text."""
+
+
+def help_text_with_codebase(already_indexed: bool = False) -> str:
+    """Generate help text for when a codebase is available.
+
+    Args:
+        already_indexed: Whether the codebase is already indexed.
+
+    Returns:
+        Formatted help text string.
+    """
+    return (
+        "Howdy! Welcome to Shotgun - Spec Driven Development for Developers and AI Agents.\n\n"
+        "Shotgun writes codebase-aware specs for your AI coding agents so they don't derail.\n\n"
+        f"{'It' if already_indexed else 'Once your codebase is indexed, it'} can help you:\n"
+        "- Research your codebase and spec out new features\n"
+        "- Create implementation plans that fit your architecture\n"
+        "- Generate AGENTS.md files for AI coding agents\n"
+        "- Onboard to existing projects or plan refactors\n\n"
+        "Ready to build something? Let's go.\n"
+    )
+
+
+def help_text_empty_dir() -> str:
+    """Generate help text for empty directory.
+
+    Returns:
+        Formatted help text string.
+    """
+    return (
+        "Howdy! Welcome to Shotgun - Spec Driven Development for Developers and AI Agents.\n\n"
+        "Shotgun writes codebase-aware specs for your AI coding agents so they don't derail.\n\n"
+        "It can help you:\n"
+        "- Research your codebase and spec out new features\n"
+        "- Create implementation plans that fit your architecture\n"
+        "- Generate AGENTS.md files for AI coding agents\n"
+        "- Onboard to existing projects or plan refactors\n\n"
+        "Ready to build something? Let's go.\n"
+    )

shotgun/tui/screens/chat/prompt_history.py

@@ -0,0 +1,48 @@
+"""Prompt history management for chat screen."""
+
+from pydantic import BaseModel, Field
+
+
+class PromptHistory(BaseModel):
+    """Manages prompt history for navigation in chat input."""
+
+    prompts: list[str] = Field(default_factory=lambda: ["Hello there!"])
+    curr: int | None = None
+
+    def next(self) -> str:
+        """Navigate to next prompt in history.
+
+        Returns:
+            The next prompt in history.
+        """
+        if self.curr is None:
+            self.curr = -1
+        else:
+            self.curr = -1
+        return self.prompts[self.curr]
+
+    def prev(self) -> str:
+        """Navigate to previous prompt in history.
+
+        Returns:
+            The previous prompt in history.
+
+        Raises:
+            Exception: If current entry is None.
+        """
+        if self.curr is None:
+            raise Exception("current entry is none")
+        if self.curr == -1:
+            self.curr = None
+            return ""
+        self.curr += 1
+        return ""
+
+    def append(self, text: str) -> None:
+        """Add a new prompt to history.
+
+        Args:
+            text: The prompt text to add.
+        """
+        self.prompts.append(text)
+        self.curr = None

shotgun/tui/screens/chat.tcss

@@ -38,6 +38,17 @@ ModeIndicator {
 }
 
 
+#right-footer-indicators {
+  width: auto;
+  height: auto;
+  layout: vertical;
+}
+
+#context-indicator {
+  text-align: end;
+  height: 1;
+}
+
 #indexing-job-display {
   text-align: end;
 }

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
 
@@ -130,6 +131,38 @@ class UsageProvider(Provider):
                 )
 
 
+class ContextProvider(Provider):
+    """Command provider for showing conversation context analysis."""
+
+    @property
+    def chat_screen(self) -> "ChatScreen":
+        from shotgun.tui.screens.chat import ChatScreen
+
+        return cast(ChatScreen, self.screen)
+
+    async def discover(self) -> AsyncGenerator[DiscoveryHit, None]:
+        """Provide context command when palette opens."""
+        yield DiscoveryHit(
+            "Show context",
+            self.chat_screen.action_show_context,
+            help="Display conversation context composition and statistics",
+        )
+
+    async def search(self, query: str) -> AsyncGenerator[Hit, None]:
+        """Search for context command."""
+        matcher = self.matcher(query)
+
+        async for discovery_hit in self.discover():
+            score = matcher.match(discovery_hit.text or "")
+            if score > 0:
+                yield Hit(
+                    score,
+                    matcher.highlight(discovery_hit.text or ""),
+                    discovery_hit.command,
+                    help=discovery_hit.help,
+                )
+
+
 class ProviderSetupProvider(Provider):
     """Command palette entries for provider configuration."""
 
@@ -145,7 +178,9 @@ class ProviderSetupProvider(Provider):
 
     def open_model_picker(self) -> None:
         """Show the model picker screen."""
-        self.chat_screen.app.push_screen(ModelPickerScreen())
+        self.chat_screen.app.push_screen(
+            ModelPickerScreen(), callback=self.chat_screen.handle_model_selected
+        )
 
     async def discover(self) -> AsyncGenerator[DiscoveryHit, None]:
         yield DiscoveryHit(
@@ -237,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
@@ -288,11 +323,18 @@ class UnifiedCommandProvider(Provider):
 
     def open_model_picker(self) -> None:
         """Show the model picker screen."""
-        self.chat_screen.app.push_screen(ModelPickerScreen())
+        self.chat_screen.app.push_screen(
+            ModelPickerScreen(), callback=self.chat_screen.handle_model_selected
+        )
 
     async def discover(self) -> AsyncGenerator[DiscoveryHit, None]:
         """Provide commands in alphabetical order when palette opens."""
         # Alphabetically ordered commands
+        yield DiscoveryHit(
+            "Clear Conversation",
+            self.chat_screen.action_clear_conversation,
+            help="Clear the entire conversation history",
+        )
         yield DiscoveryHit(
             "Codebase: Delete Codebase Index",
             self.chat_screen.delete_codebase_command,
@@ -303,6 +345,11 @@ class UnifiedCommandProvider(Provider):
             self.chat_screen.index_codebase_command,
             help="Index a repository into the codebase graph",
         )
+        yield DiscoveryHit(
+            "Compact Conversation",
+            self.chat_screen.action_compact_conversation,
+            help="Reduce conversation size by compacting message history",
+        )
         yield DiscoveryHit(
             "Open Provider Setup",
             self.open_provider_config,
@@ -313,11 +360,26 @@ 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,
+            help="Display conversation context composition and statistics",
+        )
         yield DiscoveryHit(
             "Show usage",
             self.chat_screen.action_show_usage,
             help="Display usage information for the current session",
         )
+        yield DiscoveryHit(
+            "View Onboarding",
+            self.chat_screen.action_view_onboarding,
+            help="View the onboarding tutorial and helpful resources",
+        )
 
     async def search(self, query: str) -> AsyncGenerator[Hit, None]:
         """Search for commands in alphabetical order."""
@@ -325,6 +387,11 @@ class UnifiedCommandProvider(Provider):
 
         # Define all commands in alphabetical order
         commands = [
+            (
+                "Clear Conversation",
+                self.chat_screen.action_clear_conversation,
+                "Clear the entire conversation history",
+            ),
             (
                 "Codebase: Delete Codebase Index",
                 self.chat_screen.delete_codebase_command,
@@ -335,6 +402,11 @@ class UnifiedCommandProvider(Provider):
                 self.chat_screen.index_codebase_command,
                 "Index a repository into the codebase graph",
             ),
+            (
+                "Compact Conversation",
+                self.chat_screen.action_compact_conversation,
+                "Reduce conversation size by compacting message history",
+            ),
             (
                 "Open Provider Setup",
                 self.open_provider_config,
@@ -345,11 +417,26 @@ 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,
+                "Display conversation context composition and statistics",
+            ),
             (
                 "Show usage",
                 self.chat_screen.action_show_usage,
                 "Display usage information for the current session",
             ),
+            (
+                "View Onboarding",
+                self.chat_screen.action_view_onboarding,
+                "View the onboarding tutorial and helpful resources",
+            ),
         ]
 
         for title, callback, help_text in commands:

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/chat_screen/history/__init__.py

@@ -0,0 +1,22 @@
+"""Chat history package - displays conversation messages in the TUI.
+
+This package provides widgets for displaying chat history including:
+- User questions
+- Agent responses
+- Tool calls
+- Streaming/partial responses
+"""
+
+from .agent_response import AgentResponseWidget
+from .chat_history import ChatHistory
+from .formatters import ToolFormatter
+from .partial_response import PartialResponseWidget
+from .user_question import UserQuestionWidget
+
+__all__ = [
+    "ChatHistory",
+    "PartialResponseWidget",
+    "AgentResponseWidget",
+    "UserQuestionWidget",
+    "ToolFormatter",
+]