shotgun-sh 0.2.11.dev2__py3-none-any.whl → 0.2.11.dev7__py3-none-any.whl

shotgun/tui/screens/welcome.py

@@ -136,14 +136,17 @@ class WelcomeScreen(Screen[None]):
 
     def on_mount(self) -> None:
         """Focus the first button on mount."""
-        # Update BYOK button text based on whether user has existing providers
+        self.query_one("#shotgun-button", Button).focus()
+        # Update BYOK button text asynchronously
+        self.run_worker(self._update_byok_button_text(), exclusive=False)
+
+    async def _update_byok_button_text(self) -> None:
+        """Update BYOK button text based on whether user has existing providers."""
         byok_button = self.query_one("#byok-button", Button)
         app = cast("ShotgunApp", self.app)
-        if app.config_manager.has_any_provider_key():
+        if await app.config_manager.has_any_provider_key():
             byok_button.label = "I'll stick with my BYOK setup"
 
-        self.query_one("#shotgun-button", Button).focus()
-
     @on(Button.Pressed, "#shotgun-button")
     def _on_shotgun_pressed(self) -> None:
         """Handle Shotgun Account button press."""
@@ -156,12 +159,12 @@ class WelcomeScreen(Screen[None]):
 
     async def _start_byok_config(self) -> None:
         """Launch BYOK provider configuration flow."""
-        self._mark_welcome_shown()
+        await self._mark_welcome_shown()
 
         app = cast("ShotgunApp", self.app)
 
         # If user already has providers, just dismiss and continue to chat
-        if app.config_manager.has_any_provider_key():
+        if await app.config_manager.has_any_provider_key():
             self.dismiss()
             return
 
@@ -171,7 +174,7 @@ class WelcomeScreen(Screen[None]):
         await self.app.push_screen_wait(ProviderConfigScreen())
 
         # Dismiss welcome screen after config if providers are now configured
-        if app.config_manager.has_any_provider_key():
+        if await app.config_manager.has_any_provider_key():
             self.dismiss()
 
     async def _start_shotgun_auth(self) -> None:
@@ -179,7 +182,7 @@ class WelcomeScreen(Screen[None]):
         from .shotgun_auth import ShotgunAuthScreen
 
         # Mark welcome screen as shown before auth
-        self._mark_welcome_shown()
+        await self._mark_welcome_shown()
 
         # Push the auth screen and wait for result
         await self.app.push_screen_wait(ShotgunAuthScreen())
@@ -187,9 +190,9 @@ class WelcomeScreen(Screen[None]):
         # Dismiss welcome screen after auth
         self.dismiss()
 
-    def _mark_welcome_shown(self) -> None:
+    async def _mark_welcome_shown(self) -> None:
         """Mark the welcome screen as shown in config."""
         app = cast("ShotgunApp", self.app)
-        config = app.config_manager.load()
+        config = await app.config_manager.load()
         config.shown_welcome_screen = True
-        app.config_manager.save(config)
+        await app.config_manager.save(config)

shotgun/tui/services/conversation_service.py

@@ -8,6 +8,8 @@ import logging
 from pathlib import Path
 from typing import TYPE_CHECKING
 
+import aiofiles.os
+
 from shotgun.agents.conversation_history import ConversationHistory, ConversationState
 from shotgun.agents.conversation_manager import ConversationManager
 from shotgun.agents.models import AgentType
@@ -48,7 +50,7 @@ class ConversationService:
         else:
             self.conversation_manager = ConversationManager()
 
-    def save_conversation(self, agent_manager: "AgentManager") -> bool:
+    async def save_conversation(self, agent_manager: "AgentManager") -> bool:
         """Save the current conversation to persistent storage.
 
         Args:
@@ -68,15 +70,15 @@ class ConversationService:
             conversation.set_agent_messages(state.agent_messages)
             conversation.set_ui_messages(state.ui_messages)
 
-            # Save to file
-            self.conversation_manager.save(conversation)
+            # Save to file (now async)
+            await self.conversation_manager.save(conversation)
             logger.debug("Conversation saved successfully")
             return True
         except Exception as e:
             logger.exception(f"Failed to save conversation: {e}")
             return False
 
-    def load_conversation(self) -> ConversationHistory | None:
+    async def load_conversation(self) -> ConversationHistory | None:
         """Load conversation from persistent storage.
 
         Returns:
@@ -84,7 +86,7 @@ class ConversationService:
             or if loading failed.
         """
         try:
-            conversation = self.conversation_manager.load()
+            conversation = await self.conversation_manager.load()
             if conversation is None:
                 logger.debug("No conversation file found")
                 return None
@@ -95,7 +97,7 @@ class ConversationService:
             logger.exception(f"Failed to load conversation: {e}")
             return None
 
-    def check_for_corrupted_conversation(self) -> bool:
+    async def check_for_corrupted_conversation(self) -> bool:
         """Check if a conversation backup exists (indicating corruption).
 
         Returns:
@@ -104,9 +106,9 @@ class ConversationService:
         backup_path = self.conversation_manager.conversation_path.with_suffix(
             ".json.backup"
         )
-        return backup_path.exists()
+        return await aiofiles.os.path.exists(str(backup_path))
 
-    def restore_conversation(
+    async def restore_conversation(
         self,
         agent_manager: "AgentManager",
         usage_manager: "SessionUsageManager | None" = None,
@@ -123,11 +125,11 @@ class ConversationService:
             - error_message: Error message if restoration failed, None otherwise
             - restored_agent_type: The agent type from restored conversation
         """
-        conversation = self.load_conversation()
+        conversation = await self.load_conversation()
 
         if conversation is None:
             # Check for corruption
-            if self.check_for_corrupted_conversation():
+            if await self.check_for_corrupted_conversation():
                 return (
                     False,
                     "⚠️ Previous session was corrupted and has been backed up. Starting fresh conversation.",
@@ -151,7 +153,7 @@ class ConversationService:
 
             # Restore usage state if manager provided
             if usage_manager:
-                usage_manager.restore_usage_state()
+                await usage_manager.restore_usage_state()
 
             restored_type = AgentType(conversation.last_agent_model)
             logger.info(f"Conversation restored successfully (mode: {restored_type})")
@@ -165,7 +167,7 @@ class ConversationService:
                 None,
             )
 
-    def clear_conversation(self) -> bool:
+    async def clear_conversation(self) -> bool:
         """Clear the saved conversation file.
 
         Returns:
@@ -173,8 +175,8 @@ class ConversationService:
         """
         try:
             conversation_path = self.conversation_manager.conversation_path
-            if conversation_path.exists():
-                conversation_path.unlink()
+            if await aiofiles.os.path.exists(str(conversation_path)):
+                await aiofiles.os.unlink(str(conversation_path))
                 logger.info("Conversation file cleared")
             return True
         except Exception as e:

shotgun/tui/utils/mode_progress.py

@@ -3,6 +3,8 @@
 import random
 from pathlib import Path
 
+import aiofiles
+
 from shotgun.agents.models import AgentType
 from shotgun.utils.file_system_utils import get_shotgun_base_path
 
@@ -30,7 +32,7 @@ class ModeProgressChecker:
         """
         self.base_path = base_path or get_shotgun_base_path()
 
-    def has_mode_content(self, mode: AgentType) -> bool:
+    async def has_mode_content(self, mode: AgentType) -> bool:
         """Check if a mode has meaningful content.
 
         Args:
@@ -52,7 +54,8 @@ class ModeProgressChecker:
                 for item in export_path.glob("*"):
                     if item.is_file() and not item.name.startswith("."):
                         try:
-                            content = item.read_text(encoding="utf-8")
+                            async with aiofiles.open(item, encoding="utf-8") as f:
+                                content = await f.read()
                             if len(content.strip()) > self.MIN_CONTENT_SIZE:
                                 return True
                         except (OSError, UnicodeDecodeError):
@@ -65,13 +68,16 @@ class ModeProgressChecker:
             return False
 
         try:
-            content = file_path.read_text(encoding="utf-8")
+            async with aiofiles.open(file_path, encoding="utf-8") as f:
+                content = await f.read()
             # Check if file has meaningful content
             return len(content.strip()) > self.MIN_CONTENT_SIZE
         except (OSError, UnicodeDecodeError):
             return False
 
-    def get_next_suggested_mode(self, current_mode: AgentType) -> AgentType | None:
+    async def get_next_suggested_mode(
+        self, current_mode: AgentType
+    ) -> AgentType | None:
         """Get the next suggested mode based on current progress.
 
         Args:
@@ -94,7 +100,7 @@ class ModeProgressChecker:
             return None
 
         # Check if current mode has content
-        if not self.has_mode_content(current_mode):
+        if not await self.has_mode_content(current_mode):
             # Current mode is empty, no suggestion for next mode
             return None
 
@@ -222,8 +228,9 @@ class PlaceholderHints:
         if current_mode not in self.HINTS:
             return f"Enter your {current_mode.value} mode prompt (SHIFT+TAB to switch modes)"
 
-        # Determine if mode has content
-        has_content = self.progress_checker.has_mode_content(current_mode)
+        # For placeholder text, we default to "no content" state (initial hints)
+        # This avoids async file system checks in the UI rendering path
+        has_content = False
 
         # Get hint variations for this mode and state
         hints_list = self.HINTS[current_mode][has_content]

shotgun/tui/widgets/widget_coordinator.py

@@ -245,3 +245,18 @@ class WidgetCoordinator:
                 spinner.text = text
         except Exception as e:
             logger.exception(f"Failed to update spinner text: {e}")
+
+    def set_context_streaming(self, streaming: bool) -> None:
+        """Enable or disable context indicator streaming animation.
+
+        Args:
+            streaming: Whether to show streaming animation.
+        """
+        if not self.screen.is_mounted:
+            return
+
+        try:
+            context_indicator = self.screen.query_one(ContextIndicator)
+            context_indicator.set_streaming(streaming)
+        except Exception as e:
+            logger.exception(f"Failed to set context streaming: {e}")

shotgun/utils/file_system_utils.py

@@ -2,6 +2,8 @@
 
 from pathlib import Path
 
+import aiofiles
+
 from shotgun.settings import settings
 
 
@@ -35,3 +37,20 @@ def ensure_shotgun_directory_exists() -> Path:
     shotgun_dir.mkdir(exist_ok=True)
     # Note: Removed logger to avoid circular dependency with logging_config
     return shotgun_dir
+
+
+async def async_copy_file(src: Path, dst: Path) -> None:
+    """Asynchronously copy a file from src to dst.
+
+    Args:
+        src: Source file path
+        dst: Destination file path
+
+    Raises:
+        FileNotFoundError: If source file doesn't exist
+        OSError: If copy operation fails
+    """
+    async with aiofiles.open(src, "rb") as src_file:
+        content = await src_file.read()
+    async with aiofiles.open(dst, "wb") as dst_file:
+        await dst_file.write(content)

shotgun/utils/marketing.py

@@ -0,0 +1,110 @@
+"""Marketing message management for Shotgun CLI."""
+
+from collections.abc import Callable
+from datetime import datetime, timezone
+from pathlib import Path
+from typing import TYPE_CHECKING
+
+from shotgun.agents.config.models import MarketingConfig, MarketingMessageRecord
+from shotgun.agents.models import FileOperation
+
+if TYPE_CHECKING:
+    from shotgun.agents.config.manager import ConfigManager
+
+# Marketing message IDs
+GITHUB_STAR_MESSAGE_ID = "github_star_v1"
+
+# Spec files that trigger the GitHub star message
+SPEC_FILES = {"research.md", "specification.md", "plan.md", "tasks.md"}
+
+
+class MarketingManager:
+    """Manages marketing messages shown to users."""
+
+    @staticmethod
+    def should_show_github_star_message(
+        marketing_config: MarketingConfig, file_operations: list[FileOperation]
+    ) -> bool:
+        """
+        Check if the GitHub star message should be shown.
+
+        Args:
+            marketing_config: Current marketing configuration
+            file_operations: List of file operations from the current agent run
+
+        Returns:
+            True if message should be shown, False otherwise
+        """
+        # Check if message has already been shown
+        if GITHUB_STAR_MESSAGE_ID in marketing_config.messages:
+            return False
+
+        # Check if any spec file was written
+        for operation in file_operations:
+            # operation.file_path is a string, so we convert to Path to get the filename
+            file_name = Path(operation.file_path).name
+            if file_name in SPEC_FILES:
+                return True
+
+        return False
+
+    @staticmethod
+    def mark_message_shown(
+        marketing_config: MarketingConfig, message_id: str
+    ) -> MarketingConfig:
+        """
+        Mark a marketing message as shown.
+
+        Args:
+            marketing_config: Current marketing configuration
+            message_id: ID of the message to mark as shown
+
+        Returns:
+            Updated marketing configuration
+        """
+        # Create a new record with current timestamp
+        record = MarketingMessageRecord(shown_at=datetime.now(timezone.utc))
+
+        # Update the messages dict
+        marketing_config.messages[message_id] = record
+
+        return marketing_config
+
+    @staticmethod
+    def get_github_star_message() -> str:
+        """Get the GitHub star marketing message text."""
+        return "⭐ Enjoying Shotgun? Star us on GitHub: https://github.com/shotgun-sh/shotgun"
+
+    @staticmethod
+    async def check_and_display_messages(
+        config_manager: "ConfigManager",
+        file_operations: list[FileOperation],
+        display_callback: Callable[[str], None],
+    ) -> None:
+        """
+        Check if any marketing messages should be shown and display them.
+
+        This is the main entry point for marketing message handling. It checks
+        all configured messages, displays them if appropriate, and updates the
+        config to mark them as shown.
+
+        Args:
+            config_manager: Config manager to load/save configuration
+            file_operations: List of file operations from the current agent run
+            display_callback: Callback function to display messages to the user
+        """
+        config = await config_manager.load()
+
+        # Check GitHub star message
+        if MarketingManager.should_show_github_star_message(
+            config.marketing, file_operations
+        ):
+            # Display the message
+            message = MarketingManager.get_github_star_message()
+            display_callback(message)
+
+            # Mark as shown and save
+            MarketingManager.mark_message_shown(
+                config.marketing, GITHUB_STAR_MESSAGE_ID
+            )
+            await config_manager.save(config)

{shotgun_sh-0.2.11.dev2.dist-info → shotgun_sh-0.2.11.dev7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shotgun-sh
-Version: 0.2.11.dev2
+Version: 0.2.11.dev7
 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
@@ -21,6 +21,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: Utilities
 Requires-Python: >=3.11
+Requires-Dist: aiofiles>=24.0.0
 Requires-Dist: anthropic>=0.39.0
 Requires-Dist: dependency-injector>=4.41.0
 Requires-Dist: genai-prices>=0.0.27