shotgun-sh 0.4.0.dev1__py3-none-any.whl → 0.6.2__py3-none-any.whl

shotgun/tui/screens/kuzu_error_dialog.py

@@ -0,0 +1,135 @@
+"""Error dialog for Windows kuzu/graph database import failures."""
+
+from __future__ import annotations
+
+from textual.app import ComposeResult
+from textual.containers import Container, Horizontal
+from textual.screen import ModalScreen
+from textual.widgets import Button, Label, Static
+
+from shotgun.codebase.core.kuzu_compat import (
+    _VC_INSTALL_SCRIPT,
+    _VC_REDIST_URL,
+    copy_vcpp_script_to_clipboard,
+    open_vcpp_download_page,
+)
+from shotgun.logging_config import get_logger
+
+logger = get_logger(__name__)
+
+
+class KuzuErrorDialog(ModalScreen[bool]):
+    """Error dialog for Windows kuzu import failures with copy/open buttons."""
+
+    DEFAULT_CSS = """
+        KuzuErrorDialog {
+            align: center middle;
+            background: rgba(0, 0, 0, 0.5);
+        }
+
+        KuzuErrorDialog > #dialog-container {
+            width: 90%;
+            max-width: 100;
+            height: auto;
+            max-height: 90%;
+            border: wide $error;
+            padding: 1 2;
+            layout: vertical;
+            background: $surface;
+        }
+
+        #error-title {
+            text-style: bold;
+            color: $error;
+            padding-bottom: 1;
+        }
+
+        #error-message {
+            padding: 1 0;
+        }
+
+        #script-display {
+            padding: 1;
+            margin: 1 0;
+            background: $surface-darken-1;
+            border: round $primary;
+            overflow-x: auto;
+        }
+
+        #status-label {
+            color: $success;
+            padding: 1 0;
+            min-height: 1;
+        }
+
+        #dialog-buttons {
+            layout: horizontal;
+            align-horizontal: center;
+            height: auto;
+            padding-top: 1;
+        }
+
+        #dialog-buttons Button {
+            margin: 0 1;
+        }
+    """
+
+    BINDINGS = [
+        ("escape", "close", "Close"),
+    ]
+
+    def compose(self) -> ComposeResult:
+        with Container(id="dialog-container"):
+            yield Label("Code Indexing Requires Visual C++", id="error-title")
+            yield Static(
+                "The graph database library requires the Visual C++ Redistributable "
+                "to be installed on Windows.\n\n"
+                "Run this PowerShell script as Administrator:",
+                id="error-message",
+            )
+            yield Static(_VC_INSTALL_SCRIPT, id="script-display")
+            yield Static("", id="status-label")
+            with Horizontal(id="dialog-buttons"):
+                yield Button(
+                    "Copy Script to Clipboard", id="copy-btn", variant="primary"
+                )
+                yield Button("Open Download Page", id="open-btn", variant="default")
+                yield Button("Close", id="close-btn", variant="default")
+
+    def on_button_pressed(self, event: Button.Pressed) -> None:
+        """Handle button presses."""
+        event.stop()
+
+        if event.button.id == "copy-btn":
+            self._copy_script()
+        elif event.button.id == "open-btn":
+            self._open_download()
+        elif event.button.id == "close-btn":
+            self.dismiss(True)
+
+    def _copy_script(self) -> None:
+        """Copy the PowerShell installation script to clipboard."""
+        status_label = self.query_one("#status-label", Static)
+
+        if copy_vcpp_script_to_clipboard():
+            status_label.update("Copied script to clipboard!")
+            self.query_one("#copy-btn", Button).label = "Copied!"
+            logger.debug("Copied VC++ installation script to clipboard")
+        else:
+            status_label.update(f"Could not copy. Download from: {_VC_REDIST_URL}")
+            logger.warning("Failed to copy VC++ script to clipboard")
+
+    def _open_download(self) -> None:
+        """Open the download page in the default browser."""
+        status_label = self.query_one("#status-label", Static)
+
+        if open_vcpp_download_page():
+            status_label.update("Opened download page in browser")
+            logger.debug(f"Opened VC++ download page: {_VC_REDIST_URL}")
+        else:
+            status_label.update(f"Could not open browser. URL: {_VC_REDIST_URL}")
+            logger.warning("Failed to open VC++ download page")
+
+    def action_close(self) -> None:
+        """Close the dialog."""
+        self.dismiss(True)

shotgun/tui/screens/model_picker.py

@@ -326,10 +326,8 @@ class ModelPickerScreen(Screen[ModelConfigUpdated | None]):
         """Get human-readable model name."""
         names = {
             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.GPT_5_2: "GPT-5.2 (OpenAI)",
             ModelName.CLAUDE_OPUS_4_5: "Claude Opus 4.5 (Anthropic)",
-            ModelName.CLAUDE_SONNET_4: "Claude Sonnet 4 (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)",

shotgun/tui/screens/models.py

@@ -0,0 +1,11 @@
+"""Models and enums for TUI screens."""
+
+from enum import StrEnum
+
+
+class LockedDialogAction(StrEnum):
+    """Actions available in the database locked dialog."""
+
+    RETRY = "retry"
+    DELETE = "delete"
+    QUIT = "quit"

shotgun/tui/state/processing_state.py

@@ -7,6 +7,7 @@ This module provides centralized management of processing state including:
 - Providing clean cancellation API
 """
 
+import asyncio
 from typing import TYPE_CHECKING, Any
 
 from shotgun.logging_config import get_logger
@@ -61,6 +62,7 @@ class ProcessingStateManager:
         self._spinner_widget: Spinner | None = None
         self._default_spinner_text = "Processing..."
         self._telemetry_context = telemetry_context or {}
+        self._cancellation_event: asyncio.Event | None = None
 
     @property
     def is_working(self) -> bool:
@@ -71,6 +73,15 @@ class ProcessingStateManager:
         """
         return self._working
 
+    @property
+    def cancellation_event(self) -> asyncio.Event | None:
+        """Get the current cancellation event for agent deps.
+
+        Returns:
+            The cancellation event for the current operation, or None if not processing
+        """
+        return self._cancellation_event
+
     def bind_spinner(self, spinner: "Spinner") -> None:
         """Bind a spinner widget for state coordination.
 
@@ -95,6 +106,9 @@ class ProcessingStateManager:
         self._working = True
         text = spinner_text or self._default_spinner_text
 
+        # Create a new cancellation event for this operation
+        self._cancellation_event = asyncio.Event()
+
         # Update screen's reactive working state
         if hasattr(self.screen, "working"):
             self.screen.working = True
@@ -113,6 +127,7 @@ class ProcessingStateManager:
 
         self._working = False
         self._current_worker = None
+        self._cancellation_event = None
 
         # Update screen's reactive working state
         if hasattr(self.screen, "working"):
@@ -151,6 +166,10 @@ class ProcessingStateManager:
             return False
 
         try:
+            # Set the cancellation event first for immediate effect on streaming
+            if self._cancellation_event:
+                self._cancellation_event.set()
+
             self._current_worker.cancel()
             logger.info("Operation cancelled successfully")
 

shotgun/tui/widgets/widget_coordinator.py

@@ -24,6 +24,7 @@ from shotgun.tui.screens.chat_screen.history.chat_history import ChatHistory
 
 if TYPE_CHECKING:
     from shotgun.agents.context_analyzer.models import ContextAnalysis
+    from shotgun.attachments import FileAttachment
     from shotgun.tui.screens.chat import ChatScreen
     from shotgun.tui.screens.chat_screen.hint_message import HintMessage
 
@@ -261,3 +262,20 @@ class WidgetCoordinator:
             context_indicator.set_streaming(streaming)
         except Exception as e:
             logger.exception(f"Failed to set context streaming: {e}")
+
+    def update_attachment_bar(self, attachment: "FileAttachment | None") -> None:
+        """Update the attachment bar with pending attachment.
+
+        Args:
+            attachment: FileAttachment to display, or None to hide bar.
+        """
+        if not self.screen.is_mounted:
+            return
+
+        try:
+            from shotgun.tui.components.attachment_bar import AttachmentBar
+
+            attachment_bar = self.screen.query_one(AttachmentBar)
+            attachment_bar.update_attachment(attachment)
+        except Exception as e:
+            logger.exception(f"Failed to update attachment bar: {e}")

shotgun/utils/file_system_utils.py

@@ -1,5 +1,6 @@
 """File system utility functions."""
 
+import os
 from pathlib import Path
 
 import aiofiles
@@ -24,7 +25,9 @@ def get_shotgun_home() -> Path:
     if custom_home := settings.dev.home:
         return Path(custom_home)
 
-    return Path.home() / ".shotgun-sh"
+    # Use os.path.join for explicit path separator handling on Windows
+    # This avoids potential edge cases with pathlib's / operator
+    return Path(os.path.join(os.path.expanduser("~"), ".shotgun-sh"))
 
 
 def ensure_shotgun_directory_exists() -> Path:

{shotgun_sh-0.4.0.dev1.dist-info → shotgun_sh-0.6.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shotgun-sh
-Version: 0.4.0.dev1
+Version: 0.6.2
 Summary: AI-powered research, planning, and task management CLI tool
 Project-URL: Homepage, https://shotgun.sh/
 Project-URL: Repository, https://github.com/shotgun-sh/shotgun
@@ -30,14 +30,16 @@ Requires-Dist: jinja2>=3.1.0
 Requires-Dist: logfire>=2.0.0
 Requires-Dist: openai>=1.0.0
 Requires-Dist: packaging>=23.0
+Requires-Dist: pathspec>=0.12.0
 Requires-Dist: posthog>=3.0.0
+Requires-Dist: psutil>=5.9.0
 Requires-Dist: pydantic-ai>=1.26.0
 Requires-Dist: pydantic-settings>=2.0.0
+Requires-Dist: pymupdf>=1.26.7
 Requires-Dist: pyperclip>=1.10.0
-Requires-Dist: real-ladybug>=0.12.0
+Requires-Dist: real-ladybug==0.14.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: sentencepiece>=0.2.0
-Requires-Dist: sentry-sdk[pure-eval]>=2.0.0
 Requires-Dist: tenacity>=8.0.0
 Requires-Dist: textual-dev>=1.7.0
 Requires-Dist: textual-serve>=0.1.0
@@ -51,11 +53,6 @@ Requires-Dist: tree-sitter-typescript>=0.23.0
 Requires-Dist: tree-sitter>=0.21.0
 Requires-Dist: typer>=0.12.0
 Requires-Dist: watchdog>=4.0.0
-Provides-Extra: dev
-Requires-Dist: commitizen>=3.13.0; extra == 'dev'
-Requires-Dist: lefthook>=1.12.0; extra == 'dev'
-Requires-Dist: mypy>=1.11.0; extra == 'dev'
-Requires-Dist: ruff>=0.6.0; extra == 'dev'
 Description-Content-Type: text/markdown
 
 <div align="center">
@@ -97,10 +94,6 @@ Description-Content-Type: text/markdown
 
 It includes research on existing patterns, implementation plans that respect your architecture, and task breakdowns ready to export as **AGENTS.md** files. Each spec is complete enough that your AI agent can work longer and further without losing context or creating conflicts.
 
-<p align="center">
-  <img src="https://github.com/user-attachments/assets/9c7ca014-1ed3-4935-b310-9147b275fdc7" alt="Shotgun Demo" />
-</p>
-
 </td>
 </tr>
 </table>
@@ -141,12 +134,52 @@ curl -LsSf https://astral.sh/uv/install.sh | sh
 <td>
 
 ```powershell
-powershell -c "irm https://astral.sh/uv/install.ps1 | iex"       
+powershell -c "irm https://astral.sh/uv/install.ps1 | iex"
 ```
 </td>
 </tr>
 </table>
 
+<details>
+<summary><strong>Windows Installation</strong></summary>
+
+```powershell
+# Set execution policy (one-time)
+Set-ExecutionPolicy RemoteSigned -Scope CurrentUser -Force
+
+# Install uv
+powershell -c "irm https://astral.sh/uv/install.ps1 | iex"
+
+# Add to PATH (or restart terminal)
+$env:Path = "C:\Users\$env:USERNAME\.local\bin;$env:Path"
+
+# Run Shotgun (ephemeral)
+uvx --python 3.12 shotgun-sh@latest
+
+# Or install permanently
+uv tool install --python 3.12 shotgun-sh
+```
+
+| Supported | Not Supported |
+|-----------|---------------|
+| Windows x64 (regular PCs) | 32-bit Python |
+| Python 3.11-3.13 | Python 3.14+ (no wheels yet) |
+
+**Important:** Run in **PowerShell**, not Command Prompt or VS Developer shells.
+
+#### Optional: Enable Code Indexing
+
+Code indexing requires the Visual C++ Redistributable. Run this in PowerShell (as Administrator):
+
+```powershell
+# Download and install Visual C++ Redistributable
+Import-Module BitsTransfer
+Start-BitsTransfer -Source "https://aka.ms/vs/17/release/vc_redist.x64.exe" -Destination "$env:TEMP\vc_redist.x64.exe"
+Start-Process -FilePath "$env:TEMP\vc_redist.x64.exe" -ArgumentList "/install", "/quiet", "/norestart" -Wait
+```
+
+</details>
+
 _💡 Restart your terminal after installation_
 
 ### 2. Run Shotgun
@@ -218,17 +251,25 @@ _Click the image above to watch the full demo on YouTube_
 
 # 🎯 Usage
 
-Shotgun's Terminal UI guides you through **5 specialized modes** — from research to export. Each mode has a dedicated AI agent optimized for that phase.
-
 ### Launch Shotgun in your project directory:
 
 | Already Installed | First Time / Try It Out |
 |-------------------|------------------------|
 | `shotgun` | `uvx shotgun-sh@latest` |
 
-_The TUI opens automatically. **Press `Shift+Tab` to switch modes** or `Ctrl+P` for the command palette._
+Shotgun's Terminal UI runs a Router that orchestrates your work using two execution modes: **Planning** and **Drafting**.
+
+### Planning vs Drafting
+
+| Mode | How It Works | When to Use It |
+|------|--------------|---------------|
+| **Planning** (default) | Shotgun proposes an execution plan, shows each step, and asks for confirmation before running agents that change files. You get checkpoints, can refine the plan, and can confirm or skip cascaded updates when one change affects other docs. | When you want control, visibility, and the ability to refine the plan before execution. |
+| **Drafting** | Shotgun runs the full plan in one go, without intermediate confirmations. Progress is still tracked internally, but you won’t be prompted at each step. | When you’re confident in the plan and want fast, end-to-end execution. |
+
+_The TUI opens automatically. **Press `Shift+Tab` to switch between Planning & Drafting** or `Ctrl+P` for the command palette._
 
-### The 5-Phase Workflow
+### How the Router Works Internally
+Under the hood, the Router relies on specialized sub-agents. You don’t select or manage them manually.
 
 <table>
 <tr>
@@ -244,22 +285,10 @@ _The TUI opens automatically. **Press `Shift+Tab` to switch modes** or `Ctrl+P`
 </tr>
 </table>
 
-_Each phase builds on the previous one, creating a complete specification ready for AI coding agents._
-
-### Mode Reference
-
-| Mode | What It Does | Example Prompt | Output |
-|:-----|:-------------|:---------------|:-------|
-| **🔬&nbsp;Research** | Searches codebase + web, identifies patterns | `How do we handle authentication in this codebase?` | `research.md` |
-| **📝&nbsp;Specify** | Creates technical specs aware of architecture | `Add OAuth2 authentication with refresh token support` | `specification.md` |
-| **📋&nbsp;Plan** | Generates implementation roadmap | `Create an implementation plan for the payment system` | `plan.md` |
-| **✅&nbsp;Tasks** | Breaks plans into actionable items | `Break down the user dashboard plan into tasks` | `tasks.md` |
-| **📤&nbsp;Export** | Formats for AI coding agents | `Export everything to AGENTS.md` | `AGENTS.md` |
+> Planning and Drafting are the only execution modes you control; everything else is handled by the Router.
 
 _**Mode switching:** `Shift+Tab` cycles through modes_
 
-_**Visual status:** See current mode and progress at bottom_
-
 ### ⌨️ Keyboard Shortcuts
 
 | Shortcut | Action |
@@ -277,7 +306,7 @@ _**Visual status:** See current mode and progress at bottom_
 | ✅ `Research how we handle auth` | ❌ Jump straight to building |
 | ✅ `Shotgun please ask me questions first` | ❌ Assume Shotgun knows your needs |
 | ✅ `I'm working on payments, need refunds` | ❌ `Add refunds` (no context) |
-| ✅ Follow Research → Specify → Plan → Tasks | ❌ Skip phases |
+| ✅ Start in Planning mode, let Shotgun propose and refine a plan with you, then run it | ❌ Blast everything in one go without reviewing the plan first (unless you intentionally switch to Drafting mode) |
 
 **Result:** Your AI coding agent gets complete context—what exists, why, and what to build.
 
@@ -285,6 +314,30 @@ _**Visual status:** See current mode and progress at bottom_
 
 ---
 
+# 🤝 Share Specs with Your Team
+
+Sharing specs to a workspace is available on **paid Shotgun plans**.
+
+Shotgun lets you share specs externally by publishing them to a **workspace**. This creates a versioned, shareable snapshot your team can access outside the repo.
+
+### How to Share a Spec
+
+1. Hit `Ctrl+P` → select _Share specs to workspace_
+2. Choose one option:
+- **Create new spec** — publish a fresh spec from your current `.shotgun/` files
+- **Add new version** — publish an updated version of an existing spec
+3. Wait for upload to complete. When finished, you can:
+- **Open in Browser** — view the shared spec in the workspace
+- **Copy URL** — share the link with your team
+- **Done** — return to Shotgun
+<img width="516" height="181" alt="image" src="https://github.com/user-attachments/assets/6dd9412c-345e-4dab-a40a-ad5f1c994d34" />
+
+
+Your local `.shotgun/*.md` files remain unchanged.  
+The workspace contains a **shareable, versioned snapshot** of the spec.
+
+---
+
 # ✨ Features
 
 ### What Makes Shotgun Different
@@ -329,7 +382,7 @@ Single-agent or one-size-fits-all prompts.
 <tr>
 <td><strong>Structured Workflow</strong></td>
 <td>
-5-phase journey with checkpoints: Research → Spec → Plan → Tasks → Export
+Router-driven flow with Planning and Drafting modes; internally it runs Research → Spec → Plan → Tasks → Export with checkpoints in Planning mode.
 </td>
 <td>
 No structure. Just "prompt and hope."
@@ -376,7 +429,7 @@ We had to implement payments. Cursor, Claude Code, and Copilot all suggested bui
 
 **Q: Does Shotgun collect any stats or data?**
 
-A: We only gather minimal, anonymous events (e.g., install, server start, tool call). We don't collect the content itself—only that an event occurred. We use Sentry for error reporting to improve stability.
+A: We only gather minimal, anonymous events (e.g., install, server start, tool call). We don't collect the content itself—only that an event occurred. We use PostHog for analytics and error reporting to improve stability.
 
 **Q: Does my code leave my computer when indexing?**
 
@@ -433,7 +486,7 @@ Shotgun is open-source and we welcome contributions. Whether you're fixing bugs,
 
 ## 🚀 Ready to Stop AI Agents from Derailing?
 
-**Research → Specify → Plan → Tasks → Export** — Five phases that give AI agents the full picture.
+**Planning → Drafting** — Two execution modes that give AI agents the full picture, backed by internal phases for Research → Specify → Plan → Tasks → Export.
 
 ```bash
 uvx shotgun-sh@latest