code-puppy 0.0.354__py3-none-any.whl → 0.0.356__py3-none-any.whl

code_puppy/api/pty_manager.py

@@ -0,0 +1,446 @@
+"""PTY Manager for terminal emulation with cross-platform support.
+
+Provides pseudo-terminal (PTY) functionality for interactive shell sessions
+via WebSocket connections. Supports Unix (pty module) and Windows (pywinpty).
+"""
+
+import asyncio
+import logging
+import os
+import signal
+import struct
+import sys
+from dataclasses import dataclass, field
+from typing import Any, Callable, Optional
+
+logger = logging.getLogger(__name__)
+
+# Platform detection
+IS_WINDOWS = sys.platform == "win32"
+
+# Conditional imports based on platform
+if IS_WINDOWS:
+    try:
+        import winpty  # type: ignore
+
+        HAS_WINPTY = True
+    except ImportError:
+        HAS_WINPTY = False
+        winpty = None
+else:
+    import fcntl
+    import pty
+    import termios
+
+    HAS_WINPTY = False
+
+
+@dataclass
+class PTYSession:
+    """Represents an active PTY session."""
+
+    session_id: str
+    master_fd: Optional[int] = None  # Unix only
+    slave_fd: Optional[int] = None  # Unix only
+    pid: Optional[int] = None  # Unix only
+    winpty_process: Any = None  # Windows only
+    cols: int = 80
+    rows: int = 24
+    on_output: Optional[Callable[[bytes], None]] = None
+    _reader_task: Optional[asyncio.Task] = None  # type: ignore
+    _running: bool = field(default=False, init=False)
+
+    def is_alive(self) -> bool:
+        """Check if the PTY session is still active."""
+        if IS_WINDOWS:
+            return self.winpty_process is not None and self.winpty_process.isalive()
+        else:
+            if self.pid is None:
+                return False
+            try:
+                os.waitpid(self.pid, os.WNOHANG)
+                return True
+            except ChildProcessError:
+                return False
+
+
+class PTYManager:
+    """Manages PTY sessions for terminal emulation.
+
+    Provides cross-platform terminal emulation with support for:
+    - Unix systems via the pty module
+    - Windows via pywinpty (optional dependency)
+
+    Example:
+        manager = PTYManager()
+        session = await manager.create_session(
+            session_id="my-terminal",
+            on_output=lambda data: print(data.decode())
+        )
+        await manager.write(session.session_id, b"ls -la\n")
+        await manager.close_session(session.session_id)
+    """
+
+    def __init__(self) -> None:
+        self._sessions: dict[str, PTYSession] = {}
+        self._lock = asyncio.Lock()
+
+    @property
+    def sessions(self) -> dict[str, PTYSession]:
+        """Get all active sessions."""
+        return self._sessions.copy()
+
+    async def create_session(
+        self,
+        session_id: str,
+        cols: int = 80,
+        rows: int = 24,
+        on_output: Optional[Callable[[bytes], None]] = None,
+        shell: Optional[str] = None,
+    ) -> PTYSession:
+        """Create a new PTY session.
+
+        Args:
+            session_id: Unique identifier for the session
+            cols: Terminal width in columns
+            rows: Terminal height in rows
+            on_output: Callback for terminal output
+            shell: Shell to spawn (defaults to user's shell or /bin/bash)
+
+        Returns:
+            PTYSession: The created session
+
+        Raises:
+            RuntimeError: If session creation fails
+        """
+        async with self._lock:
+            if session_id in self._sessions:
+                logger.warning(f"Session {session_id} already exists, closing old one")
+                await self._close_session_internal(session_id)
+
+            if IS_WINDOWS:
+                session = await self._create_windows_session(
+                    session_id, cols, rows, on_output, shell
+                )
+            else:
+                session = await self._create_unix_session(
+                    session_id, cols, rows, on_output, shell
+                )
+
+            self._sessions[session_id] = session
+            logger.info(f"Created PTY session: {session_id}")
+            return session
+
+    async def _create_unix_session(
+        self,
+        session_id: str,
+        cols: int,
+        rows: int,
+        on_output: Optional[Callable[[bytes], None]],
+        shell: Optional[str],
+    ) -> PTYSession:
+        """Create a PTY session on Unix systems."""
+        shell = shell or os.environ.get("SHELL", "/bin/bash")
+
+        # Fork a new process with a PTY
+        pid, master_fd = pty.fork()
+
+        if pid == 0:
+            # Child process - exec the shell
+            os.execlp(shell, shell, "-i")  # noqa: S606
+        else:
+            # Parent process
+            # Set terminal size
+            self._set_unix_winsize(master_fd, rows, cols)
+
+            # Make master_fd non-blocking
+            flags = fcntl.fcntl(master_fd, fcntl.F_GETFL)
+            fcntl.fcntl(master_fd, fcntl.F_SETFL, flags | os.O_NONBLOCK)
+
+            session = PTYSession(
+                session_id=session_id,
+                master_fd=master_fd,
+                pid=pid,
+                cols=cols,
+                rows=rows,
+                on_output=on_output,
+            )
+            session._running = True
+
+            # Start reader task
+            session._reader_task = asyncio.create_task(self._unix_reader_loop(session))
+
+            return session
+
+    async def _create_windows_session(
+        self,
+        session_id: str,
+        cols: int,
+        rows: int,
+        on_output: Optional[Callable[[bytes], None]],
+        shell: Optional[str],
+    ) -> PTYSession:
+        """Create a PTY session on Windows systems."""
+        if not HAS_WINPTY:
+            raise RuntimeError(
+                "pywinpty is required for Windows terminal support. "
+                "Install it with: pip install pywinpty"
+            )
+
+        shell = shell or os.environ.get("COMSPEC", "cmd.exe")
+
+        # Create winpty process
+        winpty_process = winpty.PtyProcess.spawn(
+            shell,
+            dimensions=(rows, cols),
+        )
+
+        session = PTYSession(
+            session_id=session_id,
+            winpty_process=winpty_process,
+            cols=cols,
+            rows=rows,
+            on_output=on_output,
+        )
+        session._running = True
+
+        # Start reader task
+        session._reader_task = asyncio.create_task(self._windows_reader_loop(session))
+
+        return session
+
+    def _set_unix_winsize(self, fd: int, rows: int, cols: int) -> None:
+        """Set the terminal window size on Unix."""
+        winsize = struct.pack("HHHH", rows, cols, 0, 0)
+        fcntl.ioctl(fd, termios.TIOCSWINSZ, winsize)
+
+    async def _unix_reader_loop(self, session: PTYSession) -> None:
+        """Read output from Unix PTY and forward to callback."""
+        loop = asyncio.get_event_loop()
+
+        try:
+            while session._running and session.master_fd is not None:
+                try:
+                    data = await loop.run_in_executor(
+                        None, self._read_unix_pty, session.master_fd
+                    )
+
+                    if data is None:
+                        # No data available, wait a bit
+                        await asyncio.sleep(0.01)
+                        continue
+                    elif data == b"":
+                        # EOF - process terminated
+                        break
+                    elif session.on_output:
+                        session.on_output(data)
+
+                except asyncio.CancelledError:
+                    break
+
+        except Exception as e:
+            logger.error(f"Unix reader loop error: {e}")
+        finally:
+            session._running = False
+
+    def _read_unix_pty(self, fd: int) -> bytes | None:
+        """Read from Unix PTY file descriptor.
+
+        Returns:
+            bytes: Data read from PTY
+            None: No data available (would block)
+            b'': EOF (process terminated)
+        """
+        try:
+            data = os.read(fd, 4096)
+            return data
+        except BlockingIOError:
+            return None
+        except OSError:
+            return b""
+
+    async def _windows_reader_loop(self, session: PTYSession) -> None:
+        """Read output from Windows PTY and forward to callback."""
+        loop = asyncio.get_event_loop()
+
+        try:
+            while (
+                session._running
+                and session.winpty_process is not None
+                and session.winpty_process.isalive()
+            ):
+                try:
+                    data = await loop.run_in_executor(
+                        None, session.winpty_process.read, 4096
+                    )
+                    if data and session.on_output:
+                        session.on_output(
+                            data.encode() if isinstance(data, str) else data
+                        )
+                except EOFError:
+                    break
+                except asyncio.CancelledError:
+                    break
+
+                await asyncio.sleep(0.01)
+
+        except Exception as e:
+            logger.error(f"Windows reader loop error: {e}")
+        finally:
+            session._running = False
+
+    async def write(self, session_id: str, data: bytes) -> bool:
+        """Write data to a PTY session.
+
+        Args:
+            session_id: The session to write to
+            data: Data to write
+
+        Returns:
+            bool: True if write succeeded
+        """
+        session = self._sessions.get(session_id)
+        if not session:
+            logger.warning(f"Session {session_id} not found")
+            return False
+
+        try:
+            if IS_WINDOWS:
+                if session.winpty_process:
+                    session.winpty_process.write(
+                        data.decode() if isinstance(data, bytes) else data
+                    )
+                    return True
+            else:
+                if session.master_fd is not None:
+                    os.write(session.master_fd, data)
+                    return True
+        except Exception as e:
+            logger.error(f"Write error for session {session_id}: {e}")
+
+        return False
+
+    async def resize(self, session_id: str, cols: int, rows: int) -> bool:
+        """Resize a PTY session.
+
+        Args:
+            session_id: The session to resize
+            cols: New width in columns
+            rows: New height in rows
+
+        Returns:
+            bool: True if resize succeeded
+        """
+        session = self._sessions.get(session_id)
+        if not session:
+            logger.warning(f"Session {session_id} not found")
+            return False
+
+        try:
+            if IS_WINDOWS:
+                if session.winpty_process:
+                    session.winpty_process.setwinsize(rows, cols)
+            else:
+                if session.master_fd is not None:
+                    self._set_unix_winsize(session.master_fd, rows, cols)
+
+            session.cols = cols
+            session.rows = rows
+            logger.debug(f"Resized session {session_id} to {cols}x{rows}")
+            return True
+
+        except Exception as e:
+            logger.error(f"Resize error for session {session_id}: {e}")
+            return False
+
+    async def close_session(self, session_id: str) -> bool:
+        """Close a PTY session.
+
+        Args:
+            session_id: The session to close
+
+        Returns:
+            bool: True if session was closed
+        """
+        async with self._lock:
+            return await self._close_session_internal(session_id)
+
+    async def _close_session_internal(self, session_id: str) -> bool:
+        """Internal session close without lock."""
+        session = self._sessions.pop(session_id, None)
+        if not session:
+            return False
+
+        session._running = False
+
+        # Cancel reader task
+        if session._reader_task:
+            session._reader_task.cancel()
+            try:
+                await session._reader_task
+            except asyncio.CancelledError:
+                pass
+
+        # Clean up platform-specific resources
+        if IS_WINDOWS:
+            if session.winpty_process:
+                try:
+                    session.winpty_process.terminate()
+                except Exception as e:
+                    logger.debug(f"Error terminating winpty: {e}")
+        else:
+            # Close file descriptors
+            if session.master_fd is not None:
+                try:
+                    os.close(session.master_fd)
+                except OSError:
+                    pass
+
+            # Terminate child process
+            if session.pid is not None:
+                try:
+                    os.kill(session.pid, signal.SIGTERM)
+                    os.waitpid(session.pid, 0)
+                except (OSError, ChildProcessError):
+                    pass
+
+        logger.info(f"Closed PTY session: {session_id}")
+        return True
+
+    async def close_all(self) -> None:
+        """Close all PTY sessions."""
+        session_ids = list(self._sessions.keys())
+        for session_id in session_ids:
+            await self.close_session(session_id)
+        logger.info("Closed all PTY sessions")
+
+    def get_session(self, session_id: str) -> Optional[PTYSession]:
+        """Get a session by ID.
+
+        Args:
+            session_id: The session ID
+
+        Returns:
+            PTYSession or None if not found
+        """
+        return self._sessions.get(session_id)
+
+    def list_sessions(self) -> list[str]:
+        """List all active session IDs.
+
+        Returns:
+            List of session IDs
+        """
+        return list(self._sessions.keys())
+
+
+# Global PTY manager instance
+_pty_manager: Optional[PTYManager] = None
+
+
+def get_pty_manager() -> PTYManager:
+    """Get or create the global PTY manager instance."""
+    global _pty_manager
+    if _pty_manager is None:
+        _pty_manager = PTYManager()
+    return _pty_manager

code_puppy/api/routers/__init__.py

@@ -0,0 +1,12 @@
+"""API routers for Code Puppy REST endpoints.
+
+This package contains the FastAPI router modules for different API domains:
+    - config: Configuration management endpoints
+    - commands: Command execution endpoints
+    - sessions: Session management endpoints
+    - agents: Agent-related endpoints
+"""
+
+from code_puppy.api.routers import agents, commands, config, sessions
+
+__all__ = ["config", "commands", "sessions", "agents"]

code_puppy/api/routers/agents.py

@@ -0,0 +1,36 @@
+"""Agents API endpoints for agent management.
+
+This router provides REST endpoints for:
+- Listing all available agents with their metadata
+"""
+
+from typing import Any, Dict, List
+
+from fastapi import APIRouter
+
+router = APIRouter()
+
+
+@router.get("/")
+async def list_agents() -> List[Dict[str, Any]]:
+    """List all available agents.
+
+    Returns a list of all agents registered in the system,
+    including their name, display name, and description.
+
+    Returns:
+        List[Dict[str, Any]]: List of agent information dictionaries.
+    """
+    from code_puppy.agents import get_agent_descriptions, get_available_agents
+
+    agents_dict = get_available_agents()
+    descriptions = get_agent_descriptions()
+
+    return [
+        {
+            "name": name,
+            "display_name": display_name,
+            "description": descriptions.get(name, "No description"),
+        }
+        for name, display_name in agents_dict.items()
+    ]

code_puppy/api/routers/commands.py

@@ -0,0 +1,198 @@
+"""Commands API endpoints for slash command execution and autocomplete.
+
+This router provides REST endpoints for:
+- Listing all available slash commands
+- Getting info about specific commands
+- Executing slash commands
+- Autocomplete suggestions for partial commands
+"""
+
+from typing import Any, List, Optional
+
+from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel
+
+router = APIRouter()
+
+
+# =============================================================================
+# Pydantic Models
+# =============================================================================
+
+
+class CommandInfo(BaseModel):
+    """Information about a registered command."""
+
+    name: str
+    description: str
+    usage: str
+    aliases: List[str] = []
+    category: str = "core"
+    detailed_help: Optional[str] = None
+
+
+class CommandExecuteRequest(BaseModel):
+    """Request to execute a slash command."""
+
+    command: str  # Full command string, e.g., "/set model=gpt-4o"
+
+
+class CommandExecuteResponse(BaseModel):
+    """Response from executing a slash command."""
+
+    success: bool
+    result: Any = None
+    error: Optional[str] = None
+
+
+class AutocompleteRequest(BaseModel):
+    """Request for command autocomplete."""
+
+    partial: str  # Partial command string, e.g., "/se" or "/set mo"
+
+
+class AutocompleteResponse(BaseModel):
+    """Response with autocomplete suggestions."""
+
+    suggestions: List[str]
+
+
+# =============================================================================
+# Endpoints
+# =============================================================================
+
+
+@router.get("/")
+async def list_commands() -> List[CommandInfo]:
+    """List all available slash commands.
+
+    Returns a sorted list of all unique commands (no alias duplicates),
+    with their metadata including name, description, usage, aliases,
+    category, and detailed help.
+
+    Returns:
+        List[CommandInfo]: Sorted list of command information.
+    """
+    from code_puppy.command_line.command_registry import get_unique_commands
+
+    commands = []
+    for cmd in get_unique_commands():
+        commands.append(
+            CommandInfo(
+                name=cmd.name,
+                description=cmd.description,
+                usage=cmd.usage,
+                aliases=cmd.aliases,
+                category=cmd.category,
+                detailed_help=cmd.detailed_help,
+            )
+        )
+    return sorted(commands, key=lambda c: c.name)
+
+
+@router.get("/{name}")
+async def get_command_info(name: str) -> CommandInfo:
+    """Get detailed info about a specific command.
+
+    Looks up a command by name or alias (case-insensitive).
+
+    Args:
+        name: Command name or alias (without leading /).
+
+    Returns:
+        CommandInfo: Full command information.
+
+    Raises:
+        HTTPException: 404 if command not found.
+    """
+    from code_puppy.command_line.command_registry import get_command
+
+    cmd = get_command(name)
+    if not cmd:
+        raise HTTPException(404, f"Command '/{name}' not found")
+
+    return CommandInfo(
+        name=cmd.name,
+        description=cmd.description,
+        usage=cmd.usage,
+        aliases=cmd.aliases,
+        category=cmd.category,
+        detailed_help=cmd.detailed_help,
+    )
+
+
+@router.post("/execute")
+async def execute_command(request: CommandExecuteRequest) -> CommandExecuteResponse:
+    """Execute a slash command.
+
+    Takes a command string (with or without leading /) and executes it
+    using the command handler.
+
+    Args:
+        request: CommandExecuteRequest with the command to execute.
+
+    Returns:
+        CommandExecuteResponse: Result of command execution.
+    """
+    from code_puppy.command_line.command_handler import handle_command
+
+    command = request.command
+    if not command.startswith("/"):
+        command = "/" + command
+
+    try:
+        result = handle_command(command)
+        return CommandExecuteResponse(success=True, result=result)
+    except Exception as e:
+        return CommandExecuteResponse(success=False, error=str(e))
+
+
+@router.post("/autocomplete")
+async def autocomplete_command(request: AutocompleteRequest) -> AutocompleteResponse:
+    """Get autocomplete suggestions for a partial command.
+
+    Provides intelligent autocomplete based on partial input:
+    - Empty input: returns all command names
+    - Partial command name: returns matching commands and aliases
+    - Complete command with args: returns usage hint
+
+    Args:
+        request: AutocompleteRequest with partial command string.
+
+    Returns:
+        AutocompleteResponse: List of autocomplete suggestions.
+    """
+    from code_puppy.command_line.command_registry import (
+        get_command,
+        get_unique_commands,
+    )
+
+    partial = request.partial.lstrip("/")
+
+    # If empty, return all command names
+    if not partial:
+        suggestions = [f"/{cmd.name}" for cmd in get_unique_commands()]
+        return AutocompleteResponse(suggestions=sorted(suggestions))
+
+    # Split into command name and args
+    parts = partial.split(maxsplit=1)
+    cmd_partial = parts[0].lower()
+
+    # If just the command name (no space yet), suggest matching commands
+    if len(parts) == 1:
+        suggestions = []
+        for cmd in get_unique_commands():
+            if cmd.name.startswith(cmd_partial):
+                suggestions.append(f"/{cmd.name}")
+            for alias in cmd.aliases:
+                if alias.startswith(cmd_partial):
+                    suggestions.append(f"/{alias}")
+        return AutocompleteResponse(suggestions=sorted(set(suggestions)))
+
+    # Command name complete, suggest based on command type
+    # (For now, just return the command usage as a hint)
+    cmd = get_command(cmd_partial)
+    if cmd:
+        return AutocompleteResponse(suggestions=[cmd.usage])
+
+    return AutocompleteResponse(suggestions=[])