code-muse 0.0.1__py3-none-any.whl

code_muse/command_line/utils.py

@@ -0,0 +1,105 @@
+import os
+from pathlib import Path
+
+from rich.table import Table
+
+
+def list_directory(path: str = None) -> tuple[list[str], list[str]]:
+    """
+    Returns (dirs, files) for the specified path, splitting out directories and files.
+    """
+    if path is None:
+        path = os.getcwd()
+    entries = []
+    try:
+        entries = [e for e in os.listdir(path)]
+    except Exception as e:
+        raise RuntimeError(f"Error listing directory: {e}") from e
+    dirs = [e for e in entries if (Path(path) / e).is_dir()]
+    files = [e for e in entries if not (Path(path) / e).is_dir()]
+    return dirs, files
+
+
+def make_directory_table(path: str = None) -> Table:
+    """
+    Returns a rich.Table object containing the directory listing.
+    """
+    if path is None:
+        path = os.getcwd()
+    dirs, files = list_directory(path)
+    table = Table(
+        title=f"\U0001f4c1 [bold blue]Current directory:[/bold blue] [cyan]{path}[/cyan]"
+    )
+    table.add_column("Type", style="dim", width=8)
+    table.add_column("Name", style="bold")
+    for d in sorted(dirs):
+        table.add_row("[green]dir[/green]", f"[cyan]{d}[/cyan]")
+    for f in sorted(files):
+        table.add_row("[yellow]file[/yellow]", f"{f}")
+    return table
+
+
+def _reset_windows_console() -> None:
+    """Reset Windows console to normal input mode.
+
+    After a prompt_toolkit Application exits on Windows, the console can be
+    left in a weird state where Enter doesn't work properly. This resets it.
+    """
+    import sys
+
+    if sys.platform != "win32":
+        return
+
+    try:
+        import ctypes
+
+        kernel32 = ctypes.windll.kernel32
+        # Get handle to stdin
+        STD_INPUT_HANDLE = -10
+        handle = kernel32.GetStdHandle(STD_INPUT_HANDLE)
+
+        # Enable line input and echo (normal console mode)
+        # ENABLE_LINE_INPUT = 0x0002
+        # ENABLE_ECHO_INPUT = 0x0004
+        # ENABLE_PROCESSED_INPUT = 0x0001
+        NORMAL_MODE = 0x0007  # Line input + echo + processed
+        kernel32.SetConsoleMode(handle, NORMAL_MODE)
+    except Exception:
+        pass  # Silently ignore errors - this is best-effort
+
+
+def safe_input(prompt_text: str = "") -> str:
+    """Cross-platform safe input that works after prompt_toolkit Applications.
+
+    On Windows, raw input() can fail after a prompt_toolkit Application exits
+    because the terminal can be left in a weird state. This function resets
+    the Windows console mode before calling input().
+
+    Args:
+        prompt_text: The prompt to display to the user
+
+    Returns:
+        The user's input string (stripped)
+
+    Raises:
+        KeyboardInterrupt: If user presses Ctrl+C
+        EOFError: If user presses Ctrl+D/Ctrl+Z
+    """
+    # Reset Windows console to normal mode before reading input
+    _reset_windows_console()
+
+    # Drain the message queue so any preceding emit_info/emit_warning text
+    # actually paints to the terminal BEFORE we put up our prompt. Without
+    # this, the async render queue can lose the race against input() and
+    # the user sees a bare prompt with no context above it.
+    try:
+        from code_muse.messaging.message_queue import get_global_queue
+
+        get_global_queue().drain(timeout=1.0)
+    except Exception:
+        # Never let a drain failure block input. Fall through.
+        pass
+
+    # Use standard input() - now that console is reset, it should work
+    result = input(prompt_text)
+    return result.strip() if result else ""

code_muse/command_line/wiggum_state.py

@@ -0,0 +1,77 @@
+"""Wiggum loop state management.
+
+This module tracks the state for the /wiggum command, which causes
+the agent to automatically re-run the same prompt after completing,
+like Chief Wiggum chasing donuts in circles. 🍩
+
+Usage:
+    /wiggum <prompt>  - Start looping with the given prompt
+    Ctrl+C            - Stop the wiggum loop
+"""
+
+from dataclasses import dataclass
+
+
+@dataclass
+class WiggumState:
+    """State container for wiggum loop mode."""
+
+    active: bool = False
+    prompt: str | None = None
+    loop_count: int = 0
+
+    def start(self, prompt: str) -> None:
+        """Start wiggum mode with the given prompt."""
+        self.active = True
+        self.prompt = prompt
+        self.loop_count = 0
+
+    def stop(self) -> None:
+        """Stop wiggum mode."""
+        self.active = False
+        self.prompt = None
+        self.loop_count = 0
+
+    def increment(self) -> int:
+        """Increment and return the loop count."""
+        self.loop_count += 1
+        return self.loop_count
+
+
+# Global singleton for wiggum state
+_wiggum_state = WiggumState()
+
+
+def get_wiggum_state() -> WiggumState:
+    """Get the global wiggum state."""
+    return _wiggum_state
+
+
+def is_wiggum_active() -> bool:
+    """Check if wiggum mode is currently active."""
+    return _wiggum_state.active
+
+
+def get_wiggum_prompt() -> str | None:
+    """Get the current wiggum prompt, if active."""
+    return _wiggum_state.prompt if _wiggum_state.active else None
+
+
+def start_wiggum(prompt: str) -> None:
+    """Start wiggum mode with the given prompt."""
+    _wiggum_state.start(prompt)
+
+
+def stop_wiggum() -> None:
+    """Stop wiggum mode."""
+    _wiggum_state.stop()
+
+
+def increment_wiggum_count() -> int:
+    """Increment wiggum loop count and return the new value."""
+    return _wiggum_state.increment()
+
+
+def get_wiggum_count() -> int:
+    """Get the current wiggum loop count."""
+    return _wiggum_state.loop_count