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

shotgun/tui/dependencies.py

@@ -1,13 +1,44 @@
 """Dependency creation utilities for TUI components."""
 
+from typing import Any
+
 from pydantic_ai import RunContext
 
 from shotgun.agents.config import get_provider_model
+from shotgun.agents.config.models import ModelConfig
 from shotgun.agents.models import AgentDeps
+from shotgun.agents.router.models import RouterDeps, RouterMode
+from shotgun.codebase.service import CodebaseService
 from shotgun.tui.filtered_codebase_service import FilteredCodebaseService
 from shotgun.utils import get_shotgun_home
 
 
+async def _get_tui_config() -> tuple[ModelConfig, CodebaseService]:
+    """Get common TUI configuration components.
+
+    Returns:
+        Tuple of (model_config, codebase_service) for TUI deps.
+    """
+    model_config = await get_provider_model()
+    storage_dir = get_shotgun_home() / "codebases"
+    codebase_service = FilteredCodebaseService(storage_dir)
+    return model_config, codebase_service
+
+
+def _placeholder_system_prompt_fn(ctx: RunContext[Any]) -> str:
+    """Placeholder system prompt that should never be called.
+
+    Agents provide their own system_prompt_fn via their create functions.
+    This placeholder exists only to satisfy the AgentDeps requirement.
+
+    Raises:
+        RuntimeError: Always, as this should never be invoked.
+    """
+    raise RuntimeError(
+        "This should not be called - agents provide their own system_prompt_fn"
+    )
+
+
 async def create_default_tui_deps() -> AgentDeps:
     """Create default AgentDeps for TUI components.
 
@@ -21,14 +52,7 @@ async def create_default_tui_deps() -> AgentDeps:
     Returns:
         Configured AgentDeps instance ready for TUI use.
     """
-    model_config = await get_provider_model()
-    storage_dir = get_shotgun_home() / "codebases"
-    codebase_service = FilteredCodebaseService(storage_dir)
-
-    def _placeholder_system_prompt_fn(ctx: RunContext[AgentDeps]) -> str:
-        raise RuntimeError(
-            "This should not be called - agents provide their own system_prompt_fn"
-        )
+    model_config, codebase_service = await _get_tui_config()
 
     return AgentDeps(
         interactive_mode=True,
@@ -37,3 +61,29 @@ async def create_default_tui_deps() -> AgentDeps:
         codebase_service=codebase_service,
         system_prompt_fn=_placeholder_system_prompt_fn,
     )
+
+
+async def create_default_router_deps() -> RouterDeps:
+    """Create default RouterDeps for TUI components with router mode.
+
+    This creates a RouterDeps configuration suitable for interactive
+    TUI usage with:
+    - Router mode always starts in PLANNING (not persisted)
+    - Interactive mode enabled
+    - TUI context flag set
+    - Filtered codebase service (restricted to CWD)
+    - Placeholder system prompt (router provides its own)
+
+    Returns:
+        Configured RouterDeps instance ready for TUI use.
+    """
+    model_config, codebase_service = await _get_tui_config()
+
+    return RouterDeps(
+        interactive_mode=True,
+        is_tui_context=True,
+        llm_model=model_config,
+        codebase_service=codebase_service,
+        system_prompt_fn=_placeholder_system_prompt_fn,
+        router_mode=RouterMode.PLANNING,
+    )

shotgun/tui/protocols.py

@@ -43,3 +43,58 @@ class ProcessingStateProvider(Protocol):
             True if an agent is processing, False otherwise.
         """
         ...
+
+
+@runtime_checkable
+class RouterModeProvider(Protocol):
+    """Protocol for screens that provide router mode state.
+
+    This protocol allows components to check the current router mode
+    (Planning or Drafting) without importing the concrete ChatScreen class.
+    """
+
+    @property
+    def router_mode(self) -> str | None:
+        """The current router mode.
+
+        Returns:
+            'planning' or 'drafting' if in router mode, None otherwise.
+        """
+        ...
+
+
+@runtime_checkable
+class ActiveSubAgentProvider(Protocol):
+    """Protocol for screens that provide active sub-agent state.
+
+    This protocol allows components to check which sub-agent is currently
+    executing during router delegation without importing ChatScreen.
+    """
+
+    @property
+    def active_sub_agent(self) -> str | None:
+        """The currently executing sub-agent type.
+
+        Returns:
+            The sub-agent type string (e.g., 'research', 'specify') if
+            a sub-agent is executing, None if idle.
+        """
+        ...
+
+
+@runtime_checkable
+class QuitConfirmationProvider(Protocol):
+    """Protocol for apps that provide quit confirmation state.
+
+    This protocol allows components (like StatusBar) to check if a quit
+    confirmation is pending without importing the concrete App class.
+    """
+
+    @property
+    def quit_pending(self) -> bool:
+        """Whether a quit confirmation is pending.
+
+        Returns:
+            True if user pressed Ctrl+C and needs to confirm quit.
+        """
+        ...

shotgun/tui/screens/chat/chat.tcss

@@ -16,11 +16,19 @@ ModeIndicator {
   height: auto;
 }
 
+ModeIndicator.mode-planning {
+  /* Blue/cyan accent for planning mode */
+}
+
+ModeIndicator.mode-drafting {
+  /* Green accent for drafting mode */
+}
+
 #footer {
   dock: bottom;
   height: auto;
   padding: 1 1 1 2;
-  max-height: 14;
+  max-height: 24;
 }
 
 #window {
@@ -51,4 +59,19 @@ ModeIndicator {
 
 #indexing-job-display {
   text-align: end;
+}
+
+AttachmentBar {
+  height: auto;
+  padding: 0 1;
+  background: $secondary-background;
+  margin-bottom: 1;
+}
+
+AttachmentBar.hidden {
+  display: none;
+}
+
+AttachmentBar Static {
+  color: $text-accent;
 }