code-puppy 0.0.287__py3-none-any.whl → 0.0.323__py3-none-any.whl

code_puppy/mcp_/registry.py

@@ -12,6 +12,7 @@ import uuid
 from pathlib import Path
 from typing import Dict, List, Optional
 
+from code_puppy import config
 from .managed_server import ServerConfig
 
 # Configure logging
@@ -23,7 +24,7 @@ class ServerRegistry:
     Registry for managing MCP server configurations.
 
     Provides CRUD operations for server configurations with thread-safe access,
-    validation, and persistent storage to ~/.code_puppy/mcp_registry.json.
+    validation, and persistent storage to XDG_DATA_HOME/code_puppy/mcp_registry.json.
 
     All operations are thread-safe and use JSON serialization for ServerConfig objects.
     Handles file not existing gracefully and validates configurations according to
@@ -36,13 +37,12 @@ class ServerRegistry:
 
         Args:
             storage_path: Optional custom path for registry storage.
-                         Defaults to ~/.code_puppy/mcp_registry.json
+                         Defaults to XDG_DATA_HOME/code_puppy/mcp_registry.json
         """
         if storage_path is None:
-            home_dir = Path.home()
-            code_puppy_dir = home_dir / ".code_puppy"
-            code_puppy_dir.mkdir(exist_ok=True)
-            self._storage_path = code_puppy_dir / "mcp_registry.json"
+            data_dir = Path(config.DATA_DIR)
+            data_dir.mkdir(parents=True, exist_ok=True, mode=0o700)
+            self._storage_path = data_dir / "mcp_registry.json"
         else:
             self._storage_path = Path(storage_path)
 

code_puppy/messaging/__init__.py

@@ -1,3 +1,71 @@
+"""Code Puppy Messaging System.
+
+This package provides both the legacy messaging API and the new structured
+messaging system.
+
+Legacy API (backward compatible):
+    - emit_info(), emit_warning(), emit_error(), etc.
+    - MessageQueue, UIMessage, MessageType
+    - Used by existing code throughout the codebase
+
+New Structured Messaging API:
+    - MessageBus for bidirectional Agent <-> UI communication
+    - Pydantic message models (TextMessage, DiffMessage, etc.)
+    - Command models for UI -> Agent communication
+    - RichConsoleRenderer for presentation
+
+Example (legacy):
+    >>> from code_puppy.messaging import emit_info, emit_error
+    >>> emit_info("Operation complete")
+    >>> emit_error("Something went wrong")
+
+Example (new):
+    >>> from code_puppy.messaging import (
+    ...     MessageBus, get_message_bus,
+    ...     TextMessage, MessageLevel,
+    ...     RichConsoleRenderer,
+    ... )
+    >>> bus = get_message_bus()
+    >>> bus.emit(TextMessage(level=MessageLevel.INFO, text="Hello"))
+"""
+
+# =============================================================================
+# Apply Rich Markdown patches (left-justified headers)
+# =============================================================================
+from .markdown_patches import patch_markdown_headings
+
+patch_markdown_headings()
+
+# =============================================================================
+# Legacy API (backward compatible)
+# =============================================================================
+
+# Message bus
+from .bus import (
+    MessageBus,
+    emit_shell_line,
+    get_message_bus,
+    get_session_context,
+    reset_message_bus,
+    set_session_context,
+)
+from .bus import emit as bus_emit  # Convenience functions (new API versions)
+from .bus import emit_debug as bus_emit_debug
+from .bus import emit_error as bus_emit_error
+from .bus import emit_info as bus_emit_info
+from .bus import emit_success as bus_emit_success
+from .bus import emit_warning as bus_emit_warning
+
+# Command types (UI -> Agent)
+from .commands import (  # Base; Agent control; User interaction responses; Union type
+    AnyCommand,
+    BaseCommand,
+    CancelAgentCommand,
+    ConfirmationResponse,
+    InterruptShellCommand,
+    SelectionResponse,
+    UserInputResponse,
+)
 from .message_queue import (
     MessageQueue,
     MessageType,
@@ -19,14 +87,66 @@ from .message_queue import (
     get_global_queue,
     provide_prompt_response,
 )
+
+# Message types and enums
+from .messages import (  # Enums, Base, Text, File ops, Diff, Shell, Agent, etc.
+    AgentReasoningMessage,
+    AgentResponseMessage,
+    AnyMessage,
+    BaseMessage,
+    ConfirmationRequest,
+    DiffLine,
+    DiffMessage,
+    DividerMessage,
+    FileContentMessage,
+    FileEntry,
+    FileListingMessage,
+    GrepMatch,
+    GrepResultMessage,
+    MessageCategory,
+    MessageLevel,
+    SelectionRequest,
+    ShellLineMessage,
+    ShellOutputMessage,
+    ShellStartMessage,
+    SpinnerControl,
+    StatusPanelMessage,
+    SubAgentInvocationMessage,
+    SubAgentResponseMessage,
+    TextMessage,
+    UserInputRequest,
+    VersionCheckMessage,
+)
 from .queue_console import QueueConsole, get_queue_console
-from .renderers import InteractiveRenderer, SynchronousInteractiveRenderer, TUIRenderer
+from .renderers import InteractiveRenderer, SynchronousInteractiveRenderer
+
+# Renderer
+from .rich_renderer import (
+    DEFAULT_STYLES,
+    DIFF_STYLES,
+    RendererProtocol,
+    RichConsoleRenderer,
+)
+
+# =============================================================================
+# New Structured Messaging API
+# =============================================================================
+
+
+# =============================================================================
+# Export all public symbols
+# =============================================================================
 
 __all__ = [
+    # -------------------------------------------------------------------------
+    # Legacy API (backward compatible)
+    # -------------------------------------------------------------------------
+    # Message queue
     "MessageQueue",
     "MessageType",
     "UIMessage",
     "get_global_queue",
+    # Legacy emit functions
     "emit_message",
     "emit_info",
     "emit_success",
@@ -42,9 +162,71 @@ __all__ = [
     "emit_prompt",
     "provide_prompt_response",
     "get_buffered_startup_messages",
+    # Legacy renderers
     "InteractiveRenderer",
-    "TUIRenderer",
     "SynchronousInteractiveRenderer",
     "QueueConsole",
     "get_queue_console",
+    # -------------------------------------------------------------------------
+    # New Structured Messaging API
+    # -------------------------------------------------------------------------
+    # Enums
+    "MessageLevel",
+    "MessageCategory",
+    # Base classes
+    "BaseMessage",
+    "BaseCommand",
+    # Message types
+    "TextMessage",
+    "FileEntry",
+    "FileListingMessage",
+    "FileContentMessage",
+    "GrepMatch",
+    "GrepResultMessage",
+    "DiffLine",
+    "DiffMessage",
+    "ShellStartMessage",
+    "ShellLineMessage",
+    "ShellOutputMessage",
+    "emit_shell_line",
+    "AgentReasoningMessage",
+    "AgentResponseMessage",
+    "SubAgentInvocationMessage",
+    "SubAgentResponseMessage",
+    "UserInputRequest",
+    "ConfirmationRequest",
+    "SelectionRequest",
+    "SpinnerControl",
+    "DividerMessage",
+    "StatusPanelMessage",
+    "VersionCheckMessage",
+    "AnyMessage",
+    # Command types
+    "CancelAgentCommand",
+    "InterruptShellCommand",
+    "UserInputResponse",
+    "ConfirmationResponse",
+    "SelectionResponse",
+    "AnyCommand",
+    # Message bus
+    "MessageBus",
+    "get_message_bus",
+    "reset_message_bus",
+    # Session context
+    "set_session_context",
+    "get_session_context",
+    # New API convenience functions (prefixed to avoid collision)
+    "bus_emit",
+    "bus_emit_info",
+    "bus_emit_warning",
+    "bus_emit_error",
+    "bus_emit_success",
+    "bus_emit_debug",
+    # Renderer
+    "RendererProtocol",
+    "RichConsoleRenderer",
+    "DEFAULT_STYLES",
+    "DIFF_STYLES",
+    # Markdown patches
+    "patch_markdown_headings",
 ]