shotgun-sh 0.1.14.dev1__py3-none-any.whl → 0.1.15.dev1__py3-none-any.whl

shotgun/cli/feedback.py

@@ -0,0 +1,46 @@
+"""Configuration management CLI commands."""
+
+from typing import Annotated
+
+import typer
+from rich.console import Console
+
+from shotgun.agents.config import get_config_manager
+from shotgun.logging_config import get_logger
+from shotgun.posthog_telemetry import Feedback, FeedbackKind, submit_feedback_survey
+
+logger = get_logger(__name__)
+console = Console()
+
+app = typer.Typer(
+    name="feedback",
+    help="Send us feedback",
+    no_args_is_help=True,
+)
+
+
+@app.callback(invoke_without_command=True)
+def send_feedback(
+    description: Annotated[str, typer.Argument(help="Description of the feedback")],
+    kind: Annotated[
+        FeedbackKind,
+        typer.Option("--type", "-t", help="Feedback type"),
+    ],
+) -> None:
+    """Initialize Shotgun configuration."""
+    config_manager = get_config_manager()
+    config_manager.load()
+    user_id = config_manager.get_user_id()
+
+    if not description:
+        console.print(
+            '❌ Please add your feedback (shotgun feedback "<your feedback>").',
+            style="red",
+        )
+        raise typer.Exit(1)
+
+    feedback = Feedback(kind=kind, description=description, user_id=user_id)
+
+    submit_feedback_survey(feedback)
+
+    console.print("✅ Feedback sent. Thank you!")

shotgun/codebase/core/manager.py

@@ -1221,26 +1221,11 @@ class CodebaseGraphManager:
 
         removed_graphs = []
 
-        # Find all .kuzu files (can be files or directories)
+        # Find all .kuzu databases (files in v0.11.2, directories in newer versions)
         for path in self.storage_dir.glob("*.kuzu"):
             graph_id = path.stem
 
-            # If it's a plain file (not a directory), it's corrupted
-            # Valid Kuzu databases are always directories
-            if path.is_file():
-                logger.warning(
-                    f"Detected corrupted database file (should be directory) at {path}, removing it"
-                )
-                try:
-                    await anyio.to_thread.run_sync(path.unlink)
-                    removed_graphs.append(graph_id)
-                    logger.info(f"Removed corrupted database file: {graph_id}")
-                except Exception as e:
-                    logger.error(
-                        f"Failed to remove corrupted database file {graph_id}: {e}"
-                    )
-                continue
-
+            # Try to open and validate the database
             try:
                 # Try to open the database with a timeout to prevent hanging
                 async def try_open_database(
@@ -1346,7 +1331,7 @@ class CodebaseGraphManager:
         """
         graphs = []
 
-        # Find all .kuzu files
+        # Find all .kuzu database files (Kuzu v0.11.2 creates files, not directories)
         for path in self.storage_dir.glob("*.kuzu"):
             if path.is_file():
                 graph_id = path.stem
@@ -1439,6 +1424,8 @@ class CodebaseGraphManager:
         Args:
             graph_id: Graph to delete
         """
+        import shutil
+
         # Stop watcher if running
         if graph_id in self._watchers:
             await self.stop_watcher(graph_id)
@@ -1453,11 +1440,14 @@ class CodebaseGraphManager:
                 self._databases[graph_id].close()
                 del self._databases[graph_id]
 
-        # Delete files
+        # Delete database (files in v0.11.2, directories in newer versions)
         graph_path = self.storage_dir / f"{graph_id}.kuzu"
         if graph_path.exists():
-            # Delete the database file
-            await anyio.to_thread.run_sync(graph_path.unlink)
+            if graph_path.is_dir():
+                await anyio.to_thread.run_sync(shutil.rmtree, graph_path)
+            else:
+                # File-based database (Kuzu v0.11.2)
+                await anyio.to_thread.run_sync(graph_path.unlink)
 
         # Also delete the WAL file if it exists
         wal_path = self.storage_dir / f"{graph_id}.kuzu.wal"

shotgun/main.py

@@ -16,7 +16,17 @@ from dotenv import load_dotenv
 
 from shotgun import __version__
 from shotgun.agents.config import get_config_manager
-from shotgun.cli import codebase, config, export, plan, research, specify, tasks, update
+from shotgun.cli import (
+    codebase,
+    config,
+    export,
+    feedback,
+    plan,
+    research,
+    specify,
+    tasks,
+    update,
+)
 from shotgun.logging_config import configure_root_logger, get_logger
 from shotgun.posthog_telemetry import setup_posthog_observability
 from shotgun.sentry_telemetry import setup_sentry_observability
@@ -68,6 +78,7 @@ app.add_typer(specify.app, name="specify", help="Generate comprehensive specific
 app.add_typer(tasks.app, name="tasks", help="Generate task lists with agentic approach")
 app.add_typer(export.app, name="export", help="Export artifacts to various formats")
 app.add_typer(update.app, name="update", help="Check for and install updates")
+app.add_typer(feedback.app, name="feedback", help="Send us feedback")
 
 
 def version_callback(value: bool) -> None:

shotgun/posthog_telemetry.py

@@ -1,11 +1,14 @@
 """PostHog analytics setup for Shotgun."""
 
+from enum import Enum
 from typing import Any
 
 import posthog
+from pydantic import BaseModel
 
 from shotgun import __version__
 from shotgun.agents.config import get_config_manager
+from shotgun.agents.conversation_manager import ConversationManager
 from shotgun.logging_config import get_early_logger
 
 # Use early logger to prevent automatic StreamHandler creation
@@ -132,3 +135,51 @@ def shutdown() -> None:
             logger.warning("Error shutting down PostHog: %s", e)
         finally:
             _posthog_client = None
+
+
+class FeedbackKind(str, Enum):
+    BUG = "bug"
+    FEATURE = "feature"
+    OTHER = "other"
+
+
+class Feedback(BaseModel):
+    kind: FeedbackKind
+    description: str
+    user_id: str
+
+
+SURVEY_ID = "01999f81-9486-0000-4fa6-9632959f92f3"
+Q_KIND_ID = "aaa5fcc3-88ba-4c24-bcf5-1481fd5efc2b"
+Q_DESCRIPTION_ID = "a0ed6283-5d4b-452c-9160-6768d879db8a"
+
+
+def submit_feedback_survey(feedback: Feedback) -> None:
+    global _posthog_client
+    if _posthog_client is None:
+        logger.debug("PostHog not initialized, skipping feedback survey")
+        return
+
+    config_manager = get_config_manager()
+    config = config_manager.load()
+    conversation_manager = ConversationManager()
+    conversation = conversation_manager.load()
+    last_10_messages = []
+    if conversation is not None:
+        last_10_messages = conversation.get_agent_messages()[:10]
+
+    track_event(
+        "survey sent",
+        properties={
+            "$survey_id": SURVEY_ID,
+            "$survey_questions": [
+                {"id": Q_KIND_ID, "question": "Feedback type"},
+                {"id": Q_DESCRIPTION_ID, "question": "Feedback description"},
+            ],
+            f"$survey_response_{Q_KIND_ID}": feedback.kind,
+            f"$survey_response_{Q_DESCRIPTION_ID}": feedback.description,
+            "provider": config.default_provider.value,
+            "config_version": config.config_version,
+            "last_10_messages": last_10_messages,  # last 10 messages
+        },
+    )

shotgun/tui/app.py

@@ -13,6 +13,7 @@ from shotgun.utils.update_checker import perform_auto_update_async
 
 from .screens.chat import ChatScreen
 from .screens.directory_setup import DirectorySetupScreen
+from .screens.feedback import FeedbackScreen
 from .screens.provider_config import ProviderConfigScreen
 
 logger = get_logger(__name__)
@@ -23,10 +24,12 @@ class ShotgunApp(App[None]):
         "chat": ChatScreen,
         "provider_config": ProviderConfigScreen,
         "directory_setup": DirectorySetupScreen,
+        "feedback": FeedbackScreen,
     }
     BINDINGS = [
         Binding("ctrl+c", "quit", "Quit the app"),
     ]
+
     CSS_PATH = "styles.tcss"
 
     def __init__(
@@ -90,7 +93,22 @@ class ShotgunApp(App[None]):
         self.exit()
 
     def get_system_commands(self, screen: Screen[Any]) -> Iterable[SystemCommand]:
-        return []  # we don't want any system commands
+        return [
+            SystemCommand(
+                "Feedback", "Send us feedback or report a bug", self.action_feedback
+            )
+        ]  # we don't want any system commands
+
+    def action_feedback(self) -> None:
+        """Open feedback screen and submit feedback."""
+        from shotgun.posthog_telemetry import Feedback, submit_feedback_survey
+
+        def handle_feedback(feedback: Feedback | None) -> None:
+            if feedback is not None:
+                submit_feedback_survey(feedback)
+                self.notify("✅ Feedback sent. Thank you!")
+
+        self.push_screen("feedback", callback=handle_feedback)
 
 
 def run(no_update_check: bool = False, continue_session: bool = False) -> None:

shotgun/tui/screens/feedback.py

@@ -0,0 +1,193 @@
+"""Screen for submitting user feedback."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, cast
+
+from textual import on
+from textual.app import ComposeResult
+from textual.containers import Horizontal, Vertical
+from textual.reactive import reactive
+from textual.screen import Screen
+from textual.widgets import Button, Label, ListItem, ListView, Static, TextArea
+
+from shotgun.posthog_telemetry import Feedback, FeedbackKind
+
+if TYPE_CHECKING:
+    from ..app import ShotgunApp
+
+
+class FeedbackScreen(Screen[Feedback | None]):
+    """Collect feedback from users."""
+
+    CSS = """
+        FeedbackScreen {
+            layout: vertical;
+        }
+
+        FeedbackScreen > * {
+            height: auto;
+        }
+
+        Label {
+            padding: 0 1;
+        }
+
+        #titlebox {
+            height: auto;
+            margin: 2 0;
+            padding: 1;
+            border: hkey $border;
+            content-align: center middle;
+
+            & > * {
+                text-align: center;
+            }
+        }
+
+        #feedback-title {
+            padding: 1 0;
+            margin-bottom: 2;
+            text-style: bold;
+            color: $text-accent;
+        }
+
+        #feedback-type-list {
+            height: auto;
+            & > * {
+                padding: 1 0;
+            }
+        }
+
+        #feedback-description {
+            margin: 1 0;
+            height: 10;
+            border: solid $border;
+        }
+
+        #feedback-actions {
+            padding: 1;
+        }
+
+        #feedback-actions > * {
+            margin-right: 2;
+        }
+
+        #feedback-type-list {
+            padding: 1;
+        }
+    """
+
+    BINDINGS = [
+        ("escape", "cancel", "Cancel"),
+    ]
+
+    selected_kind: reactive[FeedbackKind] = reactive(FeedbackKind.BUG)
+
+    def compose(self) -> ComposeResult:
+        with Vertical(id="titlebox"):
+            yield Static("Send us feedback", id="feedback-title")
+            yield Static(
+                "Select the type of feedback and provide details below.",
+                id="feedback-summary",
+            )
+        yield ListView(*self._build_feedback_type_items(), id="feedback-type-list")
+        yield TextArea(
+            "",
+            id="feedback-description",
+        )
+        with Horizontal(id="feedback-actions"):
+            yield Button("Submit", variant="primary", id="submit")
+            yield Button("Cancel \\[ESC]", id="cancel")
+
+    def on_mount(self) -> None:
+        list_view = self.query_one(ListView)
+        if list_view.children:
+            list_view.index = 0
+        self.selected_kind = FeedbackKind.BUG
+        text_area = self.query_one("#feedback-description", TextArea)
+        text_area.focus()
+
+    def action_cancel(self) -> None:
+        self.dismiss(None)
+
+    @on(ListView.Highlighted)
+    def _on_kind_highlighted(self, event: ListView.Highlighted) -> None:
+        kind = self._kind_from_item(event.item)
+        if kind:
+            self.selected_kind = kind
+
+    @on(ListView.Selected)
+    def _on_kind_selected(self, event: ListView.Selected) -> None:
+        kind = self._kind_from_item(event.item)
+        if kind:
+            self.selected_kind = kind
+            self.set_focus(self.query_one("#feedback-description", TextArea))
+
+    @on(Button.Pressed, "#submit")
+    def _on_submit_pressed(self) -> None:
+        self._submit_feedback()
+
+    @on(Button.Pressed, "#cancel")
+    def _on_cancel_pressed(self) -> None:
+        self.action_cancel()
+
+    def watch_selected_kind(self, kind: FeedbackKind) -> None:
+        if not self.is_mounted:
+            return
+        # Update the placeholder in text area based on selected kind
+        text_area = self.query_one("#feedback-description", TextArea)
+        text_area.placeholder = self._placeholder_for_kind(kind)
+
+    def _build_feedback_type_items(self) -> list[ListItem]:
+        items: list[ListItem] = []
+        for kind in FeedbackKind:
+            label = Label(self._kind_label(kind), id=f"label-{kind.value}")
+            items.append(ListItem(label, id=f"kind-{kind.value}"))
+        return items
+
+    def _kind_from_item(self, item: ListItem | None) -> FeedbackKind | None:
+        if item is None or item.id is None:
+            return None
+        kind_id = item.id.removeprefix("kind-")
+        try:
+            return FeedbackKind(kind_id)
+        except ValueError:
+            return None
+
+    def _kind_label(self, kind: FeedbackKind) -> str:
+        display_names = {
+            FeedbackKind.BUG: "Bug Report",
+            FeedbackKind.FEATURE: "Feature Request",
+            FeedbackKind.OTHER: "Other",
+        }
+        return display_names.get(kind, kind.value.title())
+
+    def _placeholder_for_kind(self, kind: FeedbackKind) -> str:
+        placeholders = {
+            FeedbackKind.BUG: "Describe the bug you encountered...",
+            FeedbackKind.FEATURE: "Describe the feature you'd like to see...",
+            FeedbackKind.OTHER: "Tell us what's on your mind...",
+        }
+        return placeholders.get(kind, "Enter your feedback...")
+
+    def _submit_feedback(self) -> None:
+        text_area = self.query_one("#feedback-description", TextArea)
+        description = text_area.text.strip()
+
+        if not description:
+            self.notify(
+                "Please enter a description before submitting.", severity="error"
+            )
+            return
+
+        app = cast("ShotgunApp", self.app)
+        user_id = app.config_manager.get_user_id()
+
+        feedback = Feedback(
+            kind=self.selected_kind,
+            description=description,
+            user_id=user_id,
+        )
+
+        self.dismiss(feedback)

{shotgun_sh-0.1.14.dev1.dist-info → shotgun_sh-0.1.15.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shotgun-sh
-Version: 0.1.14.dev1
+Version: 0.1.15.dev1
 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

{shotgun_sh-0.1.14.dev1.dist-info → shotgun_sh-0.1.15.dev1.dist-info}/RECORD

@@ -1,8 +1,8 @@
 shotgun/__init__.py,sha256=P40K0fnIsb7SKcQrFnXZ4aREjpWchVDhvM1HxI4cyIQ,104
 shotgun/build_constants.py,sha256=RXNxMz46HaB5jucgMVpw8a2yCJqjbhTOh0PddyEVMN8,713
 shotgun/logging_config.py,sha256=UKenihvgH8OA3W0b8ZFcItYaFJVe9MlsMYlcevyW1HY,7440
-shotgun/main.py,sha256=Fbg4MNEOR-FPibcbIQ7LwdQIOdYMNYMbMCPPjasbJyA,4834
-shotgun/posthog_telemetry.py,sha256=tu5FLao8uXd76RcKMyWQ9RWPCCJos749XJ_9uT_NNwI,4294
+shotgun/main.py,sha256=670RwzIwEIz9QRil37IbVoxWuX66YATqXFLSYSqKw-w,4955
+shotgun/posthog_telemetry.py,sha256=ZD_BjRej1v4Mxh7VN3AlXGKV4jIU9SC0uBrH94VQa6c,5885
 shotgun/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 shotgun/sentry_telemetry.py,sha256=L7jFMNAnDIENWVeQYSLpyul2nmIm2w3wnOp2kDP_cic,2902
 shotgun/telemetry.py,sha256=Ves6Ih3hshpKVNVAUUmwRdtW8NkTjFPg8hEqvFKZ0t0,3208
@@ -50,6 +50,7 @@ shotgun/agents/tools/web_search/utils.py,sha256=GLJ5QV9bT2ubFMuFN7caMN7tK9OTJ0R3
 shotgun/cli/__init__.py,sha256=_F1uW2g87y4bGFxz8Gp8u7mq2voHp8vQIUtCmm8Tojo,40
 shotgun/cli/config.py,sha256=LbjxDNPdetYJiwlcyOYLnqwzALfgU-m54cfstUshbrs,8715
 shotgun/cli/export.py,sha256=3hIwK2_OM1MFYSTfzBxsGuuBGm5fo0XdxASfQ5Uqb3Y,2471
+shotgun/cli/feedback.py,sha256=Sh0aK93wE7rvvCI31QCbm0sU-AUEldg7s9QqoAm1oQM,1242
 shotgun/cli/models.py,sha256=LoajeEK7MEDUSnZXb1Li-dbhXqne812YZglx-LcVpiQ,181
 shotgun/cli/plan.py,sha256=T-eu-I9z-dSoKqJ-KI8X5i5Mm0VL1BfornxRiUjTgnk,2324
 shotgun/cli/research.py,sha256=qvBBtX3Wyn6pDZlJpcEvbeK-0iTOXegi71tm8HKVYaE,2490
@@ -69,7 +70,7 @@ shotgun/codebase/core/code_retrieval.py,sha256=_JVyyQKHDFm3dxOOua1mw9eIIOHIVz3-I
 shotgun/codebase/core/cypher_models.py,sha256=Yfysfa9lLguILftkmtuJCN3kLBFIo7WW7NigM-Zr-W4,1735
 shotgun/codebase/core/ingestor.py,sha256=yh6BEIuUUfXU3dVpP0Llk19SrxA-uo3pdGnfcQsDsSo,63368
 shotgun/codebase/core/language_config.py,sha256=vsqHyuFnumRPRBV1lMOxWKNOIiClO6FyfKQR0fGrtl4,8934
-shotgun/codebase/core/manager.py,sha256=nwyDv6bnyZRIerra8LSC41fJasmfSs7mGpVYbbKGeYs,66378
+shotgun/codebase/core/manager.py,sha256=USGLBdDUoFtq6fMFWRtUu2HBC_FI8d6lWcAV4l6fcvk,66000
 shotgun/codebase/core/nl_query.py,sha256=kPoSJXBlm5rLhzOofZhqPVMJ_Lj3rV2H6sld6BwtMdg,16115
 shotgun/codebase/core/parser_loader.py,sha256=LZRrDS8Sp518jIu3tQW-BxdwJ86lnsTteI478ER9Td8,4278
 shotgun/prompts/__init__.py,sha256=RswUm0HMdfm2m2YKUwUsEdRIwoczdbI7zlucoEvHYRo,132
@@ -102,7 +103,7 @@ shotgun/sdk/exceptions.py,sha256=qBcQv0v7ZTwP7CMcxZST4GqCsfOWtOUjSzGBo0-heqo,412
 shotgun/sdk/models.py,sha256=X9nOTUHH0cdkQW1NfnMEDu-QgK9oUsEISh1Jtwr5Am4,5496
 shotgun/sdk/services.py,sha256=J4PJFSxCQ6--u7rb3Ta-9eYtlYcxcbnzrMP6ThyCnw4,705
 shotgun/tui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-shotgun/tui/app.py,sha256=xosEcdzWzRv0Aa0ZLJ04DYlYPpJ6dk2BITEPNLyFxms,4330
+shotgun/tui/app.py,sha256=HAfPbTSS8oyWlqoGL5hYw356_nHEGxHWrERdIA0uh2Q,4998
 shotgun/tui/filtered_codebase_service.py,sha256=lJ8gTMhIveTatmvmGLP299msWWTkVYKwvY_2FhuL2s4,1687
 shotgun/tui/styles.tcss,sha256=ETyyw1bpMBOqTi5RLcAJUScdPWTvAWEqE9YcT0kVs_E,121
 shotgun/tui/commands/__init__.py,sha256=8D5lvtpqMW5-fF7Bg3oJtUzU75cKOv6aUaHYYszydU8,2518
@@ -113,6 +114,7 @@ shotgun/tui/components/vertical_tail.py,sha256=kROwTaRjUwVB7H35dtmNcUVPQqNYvvfq7
 shotgun/tui/screens/chat.py,sha256=HIDnqt8og2KMmtdOok1lL44130M8V7IP6Un_8HcRwsA,30178
 shotgun/tui/screens/chat.tcss,sha256=2Yq3E23jxsySYsgZf4G1AYrYVcpX0UDW6kNNI0tDmtM,437
 shotgun/tui/screens/directory_setup.py,sha256=lIZ1J4A6g5Q2ZBX8epW7BhR96Dmdcg22CyiM5S-I5WU,3237
+shotgun/tui/screens/feedback.py,sha256=cYtmuM3qqKwevstu8gJ9mmk7lkIKZvfAyDEBUOLh-yI,5660
 shotgun/tui/screens/provider_config.py,sha256=A_tvDHF5KLP5PV60LjMJ_aoOdT3TjI6_g04UIUqGPqM,7126
 shotgun/tui/screens/splash.py,sha256=E2MsJihi3c9NY1L28o_MstDxGwrCnnV7zdq00MrGAsw,706
 shotgun/tui/screens/chat_screen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -126,8 +128,8 @@ shotgun/utils/env_utils.py,sha256=8QK5aw_f_V2AVTleQQlcL0RnD4sPJWXlDG46fsHu0d8,10
 shotgun/utils/file_system_utils.py,sha256=l-0p1bEHF34OU19MahnRFdClHufThfGAjQ431teAIp0,1004
 shotgun/utils/source_detection.py,sha256=Co6Q03R3fT771TF3RzB-70stfjNP2S4F_ArZKibwzm8,454
 shotgun/utils/update_checker.py,sha256=IgzPHRhS1ETH7PnJR_dIx6lxgr1qHpCkMTgzUxvGjhI,7586
-shotgun_sh-0.1.14.dev1.dist-info/METADATA,sha256=3x81Mpn8f-zmc9tH8P_FVniZ7GqGV_SeH4nbDPqRIcU,11197
-shotgun_sh-0.1.14.dev1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-shotgun_sh-0.1.14.dev1.dist-info/entry_points.txt,sha256=asZxLU4QILneq0MWW10saVCZc4VWhZfb0wFZvERnzfA,45
-shotgun_sh-0.1.14.dev1.dist-info/licenses/LICENSE,sha256=YebsZl590zCHrF_acCU5pmNt0pnAfD2DmAnevJPB1tY,1065
-shotgun_sh-0.1.14.dev1.dist-info/RECORD,,
+shotgun_sh-0.1.15.dev1.dist-info/METADATA,sha256=EeBReM7IB4Iw_Nf9bkzFAkQL0LPGZzILv3N_A77FRqk,11197
+shotgun_sh-0.1.15.dev1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+shotgun_sh-0.1.15.dev1.dist-info/entry_points.txt,sha256=asZxLU4QILneq0MWW10saVCZc4VWhZfb0wFZvERnzfA,45
+shotgun_sh-0.1.15.dev1.dist-info/licenses/LICENSE,sha256=YebsZl590zCHrF_acCU5pmNt0pnAfD2DmAnevJPB1tY,1065
+shotgun_sh-0.1.15.dev1.dist-info/RECORD,,