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

shotgun/tui/screens/chat/codebase_index_prompt_screen.py

@@ -1,16 +1,37 @@
 """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
+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."""
 
@@ -58,14 +79,79 @@ class CodebaseIndexPromptScreen(ModalScreen[bool]):
             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()
 
-        # Build the markdown content with privacy-first messaging
-        content = f"""
+        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:
@@ -85,24 +171,53 @@ If you're curious, you can review how Shotgun indexes/queries code by taking a l
 
 We take your privacy seriously. You can read our full [privacy policy](https://app.shotgun.sh/privacy) for more details.
 """
-
-        with Container(id="index-prompt-dialog"):
-            yield Label(
-                "Want to index your codebase so Shotgun can understand it?",
-                id="index-prompt-title",
-            )
-            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 Label(
+                    "Want to index your codebase?",
+                    id="index-prompt-title",
                 )
-                yield Button(
-                    "Index now",
-                    id="index-prompt-confirm",
-                    variant="primary",
+                # 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:
@@ -113,3 +228,16 @@ We take your privacy seriously. You can read our full [privacy policy](https://a
     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

@@ -360,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,
@@ -412,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/history/chat_history.py

@@ -92,42 +92,6 @@ class ChatHistory(Widget):
         self.items = messages
         filtered = list(self.filtered_items())
 
-        # Handle case where streaming inflated _rendered_count but final messages differ
-        # This happens when error replaces ModelResponse with HintMessage
-        if len(filtered) <= self._rendered_count and filtered:
-            # Check if the last rendered item type differs from what should be there
-            # Children: [UserQuestion, AgentResponse, ..., PartialResponse]
-            # We need to check the item before PartialResponseWidget
-            num_children = len(self.vertical_tail.children)
-            if num_children > 1:  # Has items besides PartialResponseWidget
-                last_widget = self.vertical_tail.children[-2]  # Item before Partial
-                last_filtered = filtered[-1]
-
-                # Check type mismatch
-                type_mismatch = (
-                    (
-                        isinstance(last_widget, AgentResponseWidget)
-                        and isinstance(last_filtered, HintMessage)
-                    )
-                    or (
-                        isinstance(last_widget, HintMessageWidget)
-                        and isinstance(last_filtered, ModelResponse)
-                    )
-                    or (
-                        isinstance(last_widget, AgentResponseWidget)
-                        and isinstance(last_filtered, ModelRequest)
-                    )
-                    or (
-                        isinstance(last_widget, UserQuestionWidget)
-                        and not isinstance(last_filtered, ModelRequest)
-                    )
-                )
-
-                if type_mismatch:
-                    # Remove the mismatched widget and adjust count
-                    last_widget.remove()
-                    self._rendered_count = len(filtered) - 1
-
         # Only mount new messages that haven't been rendered yet
         if len(filtered) > self._rendered_count:
             new_messages = filtered[self._rendered_count :]

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/model_picker.py

@@ -89,7 +89,7 @@ class ModelPickerScreen(Screen[ModelConfigUpdated | None]):
         ("escape", "done", "Back"),
     ]
 
-    selected_model: reactive[ModelName] = reactive(ModelName.GPT_5)
+    selected_model: reactive[ModelName] = reactive(ModelName.GPT_5_1)
 
     def compose(self) -> ComposeResult:
         with Vertical(id="titlebox"):
@@ -320,9 +320,15 @@ class ModelPickerScreen(Screen[ModelConfigUpdated | None]):
         """Get human-readable model name."""
         names = {
             ModelName.GPT_5: "GPT-5 (OpenAI)",
+            ModelName.GPT_5_MINI: "GPT-5 Mini (OpenAI)",
+            ModelName.GPT_5_1: "GPT-5.1 (OpenAI)",
+            ModelName.GPT_5_1_CODEX: "GPT-5.1 Codex (OpenAI)",
+            ModelName.GPT_5_1_CODEX_MINI: "GPT-5.1 Codex Mini (OpenAI)",
             ModelName.CLAUDE_OPUS_4_1: "Claude Opus 4.1 (Anthropic)",
             ModelName.CLAUDE_SONNET_4_5: "Claude Sonnet 4.5 (Anthropic)",
+            ModelName.CLAUDE_HAIKU_4_5: "Claude Haiku 4.5 (Anthropic)",
             ModelName.GEMINI_2_5_PRO: "Gemini 2.5 Pro (Google)",
+            ModelName.GEMINI_2_5_FLASH: "Gemini 2.5 Flash (Google)",
         }
         return names.get(model_name, model_name.value)
 

shotgun/tui/screens/onboarding.py

@@ -5,9 +5,12 @@ import webbrowser
 from textual import on
 from textual.app import ComposeResult
 from textual.containers import Container, Horizontal, VerticalScroll
+from textual.events import Resize
 from textual.screen import ModalScreen
 from textual.widgets import Button, Markdown, Static
 
+from shotgun.tui.layout import COMPACT_HEIGHT_THRESHOLD, TINY_HEIGHT_THRESHOLD
+
 
 class OnboardingModal(ModalScreen[None]):
     """Multi-page onboarding modal for new users.
@@ -120,6 +123,80 @@ class OnboardingModal(ModalScreen[None]):
             padding: 0;
             margin: 2 0 1 0;
         }
+
+        /* Tiny screen fallback */
+        #tiny-screen-container {
+            display: none;
+            width: auto;
+            height: auto;
+            padding: 1 2;
+            background: $surface;
+            text-align: center;
+        }
+
+        #tiny-screen-message {
+            padding: 1 0;
+        }
+
+        #tiny-screen-link {
+            padding: 1 0;
+            color: $accent;
+        }
+
+        /* Compact styles for short terminals */
+        #onboarding-container.compact {
+            padding: 1;
+            max-height: 98%;
+        }
+
+        #progress-sidebar.compact {
+            padding: 0;
+        }
+
+        .progress-item.compact {
+            padding: 0;
+        }
+
+        #onboarding-header.compact {
+            padding-bottom: 0;
+        }
+
+        #onboarding-content.compact {
+            padding: 0;
+        }
+
+        #page-indicator.compact {
+            padding: 0;
+        }
+
+        #buttons-container.compact {
+            padding: 0;
+        }
+
+        #resource-sections.compact {
+            padding: 0;
+        }
+
+        #resource-sections.compact Button {
+            margin: 0 0 1 0;
+        }
+
+        #video-section.compact {
+            margin: 0;
+        }
+
+        #docs-section.compact {
+            margin: 1 0 0 0;
+        }
+
+        /* Tiny mode - hide full onboarding, show minimal message */
+        OnboardingModal.tiny #onboarding-container {
+            display: none;
+        }
+
+        OnboardingModal.tiny #tiny-screen-container {
+            display: block;
+        }
     """
 
     BINDINGS = [
@@ -143,6 +220,20 @@ class OnboardingModal(ModalScreen[None]):
 
     def compose(self) -> ComposeResult:
         """Compose the onboarding modal."""
+        # Tiny screen fallback - shown when terminal is too small
+        with Container(id="tiny-screen-container"):
+            yield Static(
+                "Your screen is too small for the onboarding wizard.",
+                id="tiny-screen-message",
+            )
+            yield Static(
+                "[@click=screen.open_usage_guide]View usage instructions[/]",
+                id="tiny-screen-link",
+                markup=True,
+            )
+            yield Button("Start Shotgunning", id="tiny-close-button")
+
+        # Full onboarding container
         with Container(id="onboarding-container"):
             # Left sidebar for progress tracking
             with Container(id="progress-sidebar"):
@@ -192,6 +283,63 @@ class OnboardingModal(ModalScreen[None]):
     def on_mount(self) -> None:
         """Set up the modal after mounting."""
         self.update_page()
+        # Apply layout based on terminal height
+        self._apply_layout_for_height(self.app.size.height)
+
+    @on(Resize)
+    def handle_resize(self, event: Resize) -> None:
+        """Adjust layout based on terminal height."""
+        self._apply_layout_for_height(event.size.height)
+
+    def _apply_layout_for_height(self, height: int) -> None:
+        """Apply appropriate layout based on terminal height."""
+        if height < TINY_HEIGHT_THRESHOLD:
+            self.add_class("tiny")
+            self.remove_class("compact")
+        elif height < COMPACT_HEIGHT_THRESHOLD:
+            self.remove_class("tiny")
+            self._apply_compact_classes(True)
+        else:
+            self.remove_class("tiny")
+            self._apply_compact_classes(False)
+
+    def _apply_compact_classes(self, compact: bool) -> None:
+        """Apply or remove compact layout classes."""
+        container = self.query_one("#onboarding-container")
+        sidebar = self.query_one("#progress-sidebar")
+        header = self.query_one("#onboarding-header")
+        content = self.query_one("#onboarding-content")
+        page_indicator = self.query_one("#page-indicator")
+        buttons_container = self.query_one("#buttons-container")
+        resource_sections = self.query_one("#resource-sections")
+        progress_items = self.query(".progress-item")
+
+        if compact:
+            container.add_class("compact")
+            sidebar.add_class("compact")
+            header.add_class("compact")
+            content.add_class("compact")
+            page_indicator.add_class("compact")
+            buttons_container.add_class("compact")
+            resource_sections.add_class("compact")
+            for item in progress_items:
+                item.add_class("compact")
+        else:
+            container.remove_class("compact")
+            sidebar.remove_class("compact")
+            header.remove_class("compact")
+            content.remove_class("compact")
+            page_indicator.remove_class("compact")
+            buttons_container.remove_class("compact")
+            resource_sections.remove_class("compact")
+            for item in progress_items:
+                item.remove_class("compact")
+
+    def action_open_usage_guide(self) -> None:
+        """Open the usage guide in browser."""
+        webbrowser.open(
+            "https://github.com/shotgun-sh/shotgun?tab=readme-ov-file#-usage"
+        )
 
     def update_page(self) -> None:
         """Update the displayed page content and navigation buttons."""
@@ -412,6 +560,7 @@ Intelligently compress the conversation history while preserving important conte
             self.dismiss()
 
     @on(Button.Pressed, "#close-button")
+    @on(Button.Pressed, "#tiny-close-button")
     def handle_close(self) -> None:
         """Handle close button press."""
         self.dismiss()

shotgun/tui/screens/pipx_migration.py

@@ -7,9 +7,12 @@ from typing import TYPE_CHECKING
 from textual import on
 from textual.app import ComposeResult
 from textual.containers import Container, Horizontal, VerticalScroll
+from textual.events import Resize
 from textual.screen import ModalScreen
 from textual.widgets import Button, Label, Markdown
 
+from shotgun.tui.layout import COMPACT_HEIGHT_THRESHOLD
+
 if TYPE_CHECKING:
     pass
 
@@ -58,6 +61,24 @@ class PipxMigrationScreen(ModalScreen[None]):
             min-height: 1;
             text-align: center;
         }
+
+        /* Compact styles for short terminals */
+        #migration-container.compact {
+            padding: 1;
+            max-height: 98%;
+        }
+
+        #migration-content.compact {
+            padding: 0;
+        }
+
+        #buttons-container.compact {
+            padding: 1 0 0 0;
+        }
+
+        #migration-status.compact {
+            padding: 0;
+        }
     """
 
     BINDINGS = [
@@ -132,6 +153,31 @@ Or install permanently: `uv tool install shotgun-sh`
         """Focus the continue button and ensure scroll starts at top."""
         self.query_one("#continue", Button).focus()
         self.query_one("#migration-content", VerticalScroll).scroll_home(animate=False)
+        # 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("#migration-container")
+        content = self.query_one("#migration-content")
+        buttons_container = self.query_one("#buttons-container")
+        status = self.query_one("#migration-status")
+
+        if compact:
+            container.add_class("compact")
+            content.add_class("compact")
+            buttons_container.add_class("compact")
+            status.add_class("compact")
+        else:
+            container.remove_class("compact")
+            content.remove_class("compact")
+            buttons_container.remove_class("compact")
+            status.remove_class("compact")
 
     @on(Button.Pressed, "#copy-instructions")
     def _copy_instructions(self) -> None:

shotgun/tui/screens/provider_config.py

@@ -7,11 +7,13 @@ from typing import TYPE_CHECKING, cast
 from textual import on
 from textual.app import ComposeResult
 from textual.containers import Horizontal, Vertical
+from textual.events import Resize
 from textual.reactive import reactive
 from textual.screen import Screen
 from textual.widgets import Button, Input, Label, ListItem, ListView, Markdown, Static
 
 from shotgun.agents.config import ConfigManager, ProviderType
+from shotgun.tui.layout import COMPACT_HEIGHT_THRESHOLD
 
 if TYPE_CHECKING:
     from ..app import ShotgunApp
@@ -85,6 +87,30 @@ class ProviderConfigScreen(Screen[None]):
         #provider-status.error {
             color: $error;
         }
+
+        /* Compact styles for short terminals */
+        ProviderConfigScreen.compact #titlebox {
+            margin: 0;
+            padding: 0;
+            border: none;
+        }
+
+        ProviderConfigScreen.compact #provider-config-summary {
+            display: none;
+        }
+
+        ProviderConfigScreen.compact #provider-links {
+            display: none;
+        }
+
+        ProviderConfigScreen.compact #provider-list {
+            margin: 0;
+            padding: 0;
+        }
+
+        ProviderConfigScreen.compact #provider-actions {
+            padding: 0;
+        }
     """
 
     BINDINGS = [
@@ -131,6 +157,21 @@ class ProviderConfigScreen(Screen[None]):
         # Refresh UI asynchronously
         self.run_worker(self._refresh_ui(), exclusive=False)
 
+        # Apply layout based on terminal height
+        self._apply_layout_for_height(self.app.size.height)
+
+    @on(Resize)
+    def handle_resize(self, event: Resize) -> None:
+        """Adjust layout based on terminal height."""
+        self._apply_layout_for_height(event.size.height)
+
+    def _apply_layout_for_height(self, height: int) -> None:
+        """Apply appropriate layout based on terminal height."""
+        if height < COMPACT_HEIGHT_THRESHOLD:
+            self.add_class("compact")
+        else:
+            self.remove_class("compact")
+
     def on_screenresume(self) -> None:
         """Refresh provider status when screen is resumed.