code-puppy 0.0.214__py3-none-any.whl → 0.0.366__py3-none-any.whl

code_puppy/command_line/prompt_toolkit_completion.py

@@ -8,6 +8,7 @@
 # BOLD = '\033[1m'
 import asyncio
 import os
+import sys
 from typing import Optional
 
 from prompt_toolkit import PromptSession
@@ -26,13 +27,19 @@ from code_puppy.command_line.attachments import (
     _detect_path_tokens,
     _tokenise,
 )
+from code_puppy.command_line.clipboard import (
+    capture_clipboard_image_to_pending,
+    has_image_in_clipboard,
+)
+from code_puppy.command_line.command_registry import get_unique_commands
 from code_puppy.command_line.file_path_completion import FilePathCompleter
 from code_puppy.command_line.load_context_completion import LoadContextCompleter
+from code_puppy.command_line.mcp_completion import MCPCompleter
 from code_puppy.command_line.model_picker_completion import (
     ModelNameCompleter,
     get_active_model,
-    update_model_in_input,
 )
+from code_puppy.command_line.pin_command_completion import PinCompleter, UnpinCompleter
 from code_puppy.command_line.utils import list_directory
 from code_puppy.config import (
     COMMAND_HISTORY_FILE,
@@ -42,65 +49,119 @@ from code_puppy.config import (
 )
 
 
+def _sanitize_for_encoding(text: str) -> str:
+    """Remove or replace characters that can't be safely encoded.
+
+    This handles:
+    - Lone surrogate characters (U+D800-U+DFFF) which are invalid in UTF-8
+    - Other problematic Unicode sequences from Windows copy-paste
+
+    Args:
+        text: The string to sanitize
+
+    Returns:
+        A cleaned string safe for UTF-8 encoding
+    """
+    # First, try to encode as UTF-8 to catch any problematic characters
+    try:
+        text.encode("utf-8")
+        return text  # String is already valid UTF-8
+    except UnicodeEncodeError:
+        pass
+
+    # Replace surrogates and other problematic characters
+    # Use 'surrogatepass' to encode surrogates, then decode with 'replace' to clean them
+    try:
+        # Encode allowing surrogates, then decode replacing invalid sequences
+        cleaned = text.encode("utf-8", errors="surrogatepass").decode(
+            "utf-8", errors="replace"
+        )
+        return cleaned
+    except (UnicodeEncodeError, UnicodeDecodeError):
+        # Last resort: filter out all non-BMP and surrogate characters
+        return "".join(
+            char
+            for char in text
+            if ord(char) < 0xD800 or (ord(char) > 0xDFFF and ord(char) < 0x10000)
+        )
+
+
+class SafeFileHistory(FileHistory):
+    """A FileHistory that handles encoding errors gracefully on Windows.
+
+    Windows terminals and copy-paste operations can introduce invalid
+    Unicode surrogate characters that cause UTF-8 encoding to fail.
+    This class sanitizes history entries before writing them to disk.
+    """
+
+    def store_string(self, string: str) -> None:
+        """Store a string in the history, sanitizing it first."""
+        sanitized = _sanitize_for_encoding(string)
+        try:
+            super().store_string(sanitized)
+        except (UnicodeEncodeError, UnicodeDecodeError, OSError) as e:
+            # If we still can't write, log the error but don't crash
+            # This can happen with particularly malformed input
+            # Note: Using sys.stderr here intentionally - this is a low-level
+            # warning that shouldn't use the messaging system
+            sys.stderr.write(f"Warning: Could not save to command history: {e}\n")
+
+
 class SetCompleter(Completer):
     def __init__(self, trigger: str = "/set"):
         self.trigger = trigger
 
     def get_completions(self, document, complete_event):
+        cursor_position = document.cursor_position
         text_before_cursor = document.text_before_cursor
         stripped_text_for_trigger_check = text_before_cursor.lstrip()
 
-        if not stripped_text_for_trigger_check.startswith(self.trigger):
+        # If user types just /set (no space), suggest adding a space
+        if stripped_text_for_trigger_check == self.trigger:
+            from prompt_toolkit.formatted_text import FormattedText
+
+            yield Completion(
+                self.trigger + " ",
+                start_position=-len(self.trigger),
+                display=self.trigger + " ",
+                display_meta=FormattedText(
+                    [("class:set-completer-meta", "set config key")]
+                ),
+            )
+            return
+
+        # Require a space after /set before showing completions
+        if not stripped_text_for_trigger_check.startswith(self.trigger + " "):
             return
 
         # Determine the part of the text that is relevant for this completer
         # This handles cases like "  /set foo" where the trigger isn't at the start of the string
         actual_trigger_pos = text_before_cursor.find(self.trigger)
-        effective_input = text_before_cursor[
-            actual_trigger_pos:
-        ]  # e.g., "/set keypart" or "/set "
-
-        tokens = effective_input.split()
-
-        # Case 1: Input is exactly the trigger (e.g., "/set") and nothing more (not even a trailing space on effective_input).
-        # Suggest adding a space.
-        if (
-            len(tokens) == 1
-            and tokens[0] == self.trigger
-            and not effective_input.endswith(" ")
-        ):
-            yield Completion(
-                text=self.trigger + " ",  # Text to insert
-                start_position=-len(tokens[0]),  # Replace the trigger itself
-                display=self.trigger + " ",  # Visual display
-                display_meta="set config key",
-            )
-            return
 
-        # Case 2: Input is trigger + space (e.g., "/set ") or trigger + partial key (e.g., "/set partial")
-        base_to_complete = ""
-        if len(tokens) > 1:  # e.g., ["/set", "partialkey"]
-            base_to_complete = tokens[1]
-        # If len(tokens) == 1, it implies effective_input was like "/set ", so base_to_complete remains ""
-        # This means we list all keys.
+        # Extract the input after /set and space (up to cursor)
+        trigger_end = actual_trigger_pos + len(self.trigger) + 1  # +1 for the space
+        text_after_trigger = text_before_cursor[trigger_end:cursor_position].lstrip()
+        start_position = -(len(text_after_trigger))
 
         # --- SPECIAL HANDLING FOR 'model' KEY ---
-        if base_to_complete == "model":
+        if text_after_trigger == "model":
             # Don't return any completions -- let ModelNameCompleter handle it
             return
-        for key in get_config_keys():
+
+        # Get config keys and sort them alphabetically for consistent display
+        config_keys = sorted(get_config_keys())
+
+        for key in config_keys:
             if key == "model" or key == "puppy_token":
                 continue  # exclude 'model' and 'puppy_token' from regular /set completions
-            if key.startswith(base_to_complete):
+            if key.startswith(text_after_trigger):
                 prev_value = get_value(key)
                 value_part = f" = {prev_value}" if prev_value is not None else " = "
                 completion_text = f"{key}{value_part}"
 
                 yield Completion(
                     completion_text,
-                    start_position=-len(
-                        base_to_complete
-                    ),  # Correctly replace only the typed part of the key
+                    start_position=start_position,
                     display_meta="",
                 )
 
@@ -229,23 +290,43 @@ class CDCompleter(Completer):
         self.trigger = trigger
 
     def get_completions(self, document, complete_event):
-        text = document.text_before_cursor
-        if not text.strip().startswith(self.trigger):
+        text_before_cursor = document.text_before_cursor
+        stripped_text = text_before_cursor.lstrip()
+
+        # Require a space after /cd before showing completions (consistency with other completers)
+        if not stripped_text.startswith(self.trigger + " "):
             return
-        tokens = text.strip().split()
-        if len(tokens) == 1:
-            base = ""
-        else:
-            base = tokens[1]
+
+        # Extract the directory path after /cd and space (up to cursor)
+        trigger_pos = text_before_cursor.find(self.trigger)
+        trigger_end = trigger_pos + len(self.trigger) + 1  # +1 for the space
+        dir_path = text_before_cursor[trigger_end:].lstrip()
+        start_position = -(len(dir_path))
+
         try:
-            prefix = os.path.expanduser(base)
+            prefix = os.path.expanduser(dir_path)
             part = os.path.dirname(prefix) if os.path.dirname(prefix) else "."
             dirs, _ = list_directory(part)
-            dirnames = [d for d in dirs if d.startswith(os.path.basename(base))]
-            base_dir = os.path.dirname(base)
+            dirnames = [d for d in dirs if d.startswith(os.path.basename(prefix))]
+            base_dir = os.path.dirname(prefix)
+
+            # Preserve the user's original prefix (e.g., ~/ or relative paths)
+            # Extract what the user originally typed (with ~ or ./ preserved)
+            if dir_path.startswith("~"):
+                # User typed something with ~, preserve it
+                user_prefix = "~" + os.sep
+                # For suggestion, we replace the expanded base_dir back with ~/
+                original_prefix = dir_path.rstrip(os.sep)
+            else:
+                user_prefix = None
+                original_prefix = None
+
             for d in dirnames:
                 # Build the completion text so we keep the already-typed directory parts.
-                if base_dir and base_dir != ".":
+                if user_prefix and original_prefix:
+                    # Restore ~ prefix
+                    suggestion = user_prefix + d + os.sep
+                elif base_dir and base_dir != ".":
                     suggestion = os.path.join(base_dir, d)
                 else:
                     suggestion = d
@@ -253,7 +334,7 @@ class CDCompleter(Completer):
                 suggestion = suggestion.rstrip(os.sep) + os.sep
                 yield Completion(
                     suggestion,
-                    start_position=-len(base),
+                    start_position=start_position,
                     display=d + os.sep,
                     display_meta="Directory",
                 )
@@ -262,6 +343,174 @@ class CDCompleter(Completer):
             pass
 
 
+class AgentCompleter(Completer):
+    """
+    A completer that triggers on '/agent' to show available agents.
+
+    Usage: /agent <agent-name>
+    """
+
+    def __init__(self, trigger: str = "/agent"):
+        self.trigger = trigger
+
+    def get_completions(self, document, complete_event):
+        cursor_position = document.cursor_position
+        text_before_cursor = document.text_before_cursor
+        stripped_text = text_before_cursor.lstrip()
+
+        # Require a space after /agent before showing completions
+        if not stripped_text.startswith(self.trigger + " "):
+            return
+
+        # Extract the input after /agent and space (up to cursor)
+        trigger_pos = text_before_cursor.find(self.trigger)
+        trigger_end = trigger_pos + len(self.trigger) + 1  # +1 for the space
+        text_after_trigger = text_before_cursor[trigger_end:cursor_position].lstrip()
+        start_position = -(len(text_after_trigger))
+
+        # Load all available agent names
+        try:
+            from code_puppy.command_line.pin_command_completion import load_agent_names
+
+            agent_names = load_agent_names()
+        except Exception:
+            # If agent loading fails, return no completions
+            return
+
+        # Filter and yield agent completions
+        try:
+            from code_puppy.command_line.pin_command_completion import (
+                _get_agent_display_meta,
+            )
+        except ImportError:
+            _get_agent_display_meta = lambda x: "default"  # noqa: E731
+
+        for agent_name in agent_names:
+            if agent_name.lower().startswith(text_after_trigger.lower()):
+                yield Completion(
+                    agent_name,
+                    start_position=start_position,
+                    display=agent_name,
+                    display_meta=_get_agent_display_meta(agent_name),
+                )
+
+
+class SlashCompleter(Completer):
+    """
+    A completer that triggers on '/' at the beginning of the line
+    to show all available slash commands.
+    """
+
+    def get_completions(self, document, complete_event):
+        text_before_cursor = document.text_before_cursor
+        stripped_text = text_before_cursor.lstrip()
+
+        # Only trigger if '/' is the first non-whitespace character
+        if not stripped_text.startswith("/"):
+            return
+
+        # Get the text after the initial slash
+        if len(stripped_text) == 1:
+            # User just typed '/', show all commands
+            partial = ""
+            start_position = 0  # Don't replace anything, just insert at cursor
+        else:
+            # User is typing a command after the slash
+            partial = stripped_text[1:]  # text after '/'
+            start_position = -(len(partial))  # Replace what was typed after '/'
+
+        # Load all available commands
+        try:
+            commands = get_unique_commands()
+        except Exception:
+            # If command loading fails, return no completions
+            return
+
+        # Collect all primary commands and their aliases for proper alphabetical sorting
+        all_completions = []
+
+        # Convert partial to lowercase for case-insensitive matching
+        partial_lower = partial.lower()
+
+        for cmd in commands:
+            # Add primary command (case-insensitive matching)
+            if cmd.name.lower().startswith(partial_lower):
+                all_completions.append(
+                    {
+                        "text": cmd.name,
+                        "display": f"/{cmd.name}",
+                        "meta": cmd.description,
+                        "sort_key": cmd.name.lower(),  # Case-insensitive sort
+                    }
+                )
+
+            # Add all aliases (case-insensitive matching)
+            for alias in cmd.aliases:
+                if alias.lower().startswith(partial_lower):
+                    all_completions.append(
+                        {
+                            "text": alias,
+                            "display": f"/{alias} (alias for /{cmd.name})",
+                            "meta": cmd.description,
+                            "sort_key": alias.lower(),  # Sort by alias name, not primary command
+                        }
+                    )
+
+        # Also include custom commands from plugins (like claude-code-auth)
+        try:
+            from code_puppy import callbacks, plugins
+
+            # Ensure plugins are loaded so custom commands are registered
+            plugins.load_plugin_callbacks()
+            custom_help_results = callbacks.on_custom_command_help()
+            for res in custom_help_results:
+                if not res:
+                    continue
+                # Format 1: List of tuples (command_name, description)
+                if isinstance(res, list):
+                    for item in res:
+                        if isinstance(item, tuple) and len(item) == 2:
+                            cmd_name = str(item[0])
+                            description = str(item[1])
+                            if cmd_name.lower().startswith(partial_lower):
+                                all_completions.append(
+                                    {
+                                        "text": cmd_name,
+                                        "display": f"/{cmd_name}",
+                                        "meta": description,
+                                        "sort_key": cmd_name.lower(),
+                                    }
+                                )
+                # Format 2: Single tuple (command_name, description)
+                elif isinstance(res, tuple) and len(res) == 2:
+                    cmd_name = str(res[0])
+                    description = str(res[1])
+                    if cmd_name.lower().startswith(partial_lower):
+                        all_completions.append(
+                            {
+                                "text": cmd_name,
+                                "display": f"/{cmd_name}",
+                                "meta": description,
+                                "sort_key": cmd_name.lower(),
+                            }
+                        )
+        except Exception:
+            # If custom command loading fails, continue with registered commands only
+            pass
+
+        # Sort all completions alphabetically
+        all_completions.sort(key=lambda x: x["sort_key"])
+
+        # Yield the sorted completions
+        for completion in all_completions:
+            yield Completion(
+                completion["text"],
+                start_position=start_position,
+                display=completion["display"],
+                display_meta=completion["meta"],
+            )
+
+
 def get_prompt_with_active_model(base: str = ">>> "):
     from code_puppy.agents.agent_manager import get_current_agent
 
@@ -310,14 +559,22 @@ def get_prompt_with_active_model(base: str = ">>> "):
 async def get_input_with_combined_completion(
     prompt_str=">>> ", history_file: Optional[str] = None
 ) -> str:
-    history = FileHistory(history_file) if history_file else None
+    # Use SafeFileHistory to handle encoding errors gracefully on Windows
+    history = SafeFileHistory(history_file) if history_file else None
     completer = merge_completers(
         [
             FilePathCompleter(symbol="@"),
             ModelNameCompleter(trigger="/model"),
+            ModelNameCompleter(trigger="/m"),
             CDCompleter(trigger="/cd"),
             SetCompleter(trigger="/set"),
             LoadContextCompleter(trigger="/load_context"),
+            PinCompleter(trigger="/pin_model"),
+            UnpinCompleter(trigger="/unpin"),
+            AgentCompleter(trigger="/agent"),
+            AgentCompleter(trigger="/a"),
+            MCPCompleter(trigger="/mcp"),
+            SlashCompleter(),
         ]
     )
     # Add custom key bindings and multiline toggle
@@ -326,20 +583,47 @@ async def get_input_with_combined_completion(
     # Multiline mode state
     multiline = {"enabled": False}
 
+    # Ctrl+X keybinding - exit with KeyboardInterrupt for shell command cancellation
+    @bindings.add(Keys.ControlX)
+    def _(event):
+        try:
+            event.app.exit(exception=KeyboardInterrupt)
+        except Exception:
+            # Ignore "Return value already set" errors when exit was already called
+            # This happens when user presses multiple exit keys in quick succession
+            pass
+
+    # Escape keybinding - exit with KeyboardInterrupt
+    @bindings.add(Keys.Escape)
+    def _(event):
+        try:
+            event.app.exit(exception=KeyboardInterrupt)
+        except Exception:
+            # Ignore "Return value already set" errors when exit was already called
+            pass
+
+    # NOTE: We intentionally do NOT override Ctrl+C here.
+    # prompt_toolkit's default Ctrl+C handler properly resets the terminal state on Windows.
+    # Overriding it with event.app.exit(exception=KeyboardInterrupt) can leave the terminal
+    # in a bad state where characters cannot be typed. Let prompt_toolkit handle Ctrl+C natively.
+
     # Toggle multiline with Alt+M
     @bindings.add(Keys.Escape, "m")
     def _(event):
         multiline["enabled"] = not multiline["enabled"]
         status = "ON" if multiline["enabled"] else "OFF"
         # Print status for user feedback (version-agnostic)
-        print(f"[multiline] {status}", flush=True)
+        # Note: Using sys.stdout here for immediate feedback during input
+        sys.stdout.write(f"[multiline] {status}\n")
+        sys.stdout.flush()
 
     # Also toggle multiline with F2 (more reliable across platforms)
     @bindings.add("f2")
     def _(event):
         multiline["enabled"] = not multiline["enabled"]
         status = "ON" if multiline["enabled"] else "OFF"
-        print(f"[multiline] {status}", flush=True)
+        sys.stdout.write(f"[multiline] {status}\n")
+        sys.stdout.flush()
 
     # Newline insert bindings — robust and explicit
     # Ctrl+J (line feed) works in virtually all terminals; mark eager so it wins
@@ -364,10 +648,125 @@ async def get_input_with_combined_completion(
         else:
             event.current_buffer.validate_and_handle()
 
-    @bindings.add(Keys.Escape)
-    def _(event):
-        """Cancel the current prompt when the user presses the ESC key alone."""
-        event.app.exit(exception=KeyboardInterrupt)
+    # Handle bracketed paste - smart detection for text vs images.
+    # Most terminals (Windows included!) send Ctrl+V through bracketed paste.
+    # - If there's meaningful text content → paste as text (drag-and-drop file paths, copied text)
+    # - If text is empty/whitespace → check for clipboard image (image paste on Windows)
+    @bindings.add(Keys.BracketedPaste)
+    def handle_bracketed_paste(event):
+        """Handle bracketed paste - smart text vs image detection."""
+        pasted_data = event.data
+
+        # If we have meaningful text content, paste it (don't check for images)
+        # This handles drag-and-drop file paths and normal text paste
+        if pasted_data and pasted_data.strip():
+            # Normalize Windows line endings to Unix style
+            sanitized_data = pasted_data.replace("\r\n", "\n").replace("\r", "\n")
+            event.app.current_buffer.insert_text(sanitized_data)
+            return
+
+        # No meaningful text - check if clipboard has an image (Windows image paste!)
+        try:
+            if has_image_in_clipboard():
+                placeholder = capture_clipboard_image_to_pending()
+                if placeholder:
+                    event.app.current_buffer.insert_text(placeholder + " ")
+                    event.app.output.bell()
+                    return
+        except Exception:
+            pass
+
+        # Fallback: if there was whitespace-only data, paste it
+        if pasted_data:
+            sanitized_data = pasted_data.replace("\r\n", "\n").replace("\r", "\n")
+            event.app.current_buffer.insert_text(sanitized_data)
+
+    # Fallback Ctrl+V for terminals without bracketed paste support
+    @bindings.add("c-v", eager=True)
+    def handle_smart_paste(event):
+        """Handle Ctrl+V - auto-detect image vs text in clipboard."""
+        try:
+            # Check for image first
+            if has_image_in_clipboard():
+                placeholder = capture_clipboard_image_to_pending()
+                if placeholder:
+                    event.app.current_buffer.insert_text(placeholder + " ")
+                    # The placeholder itself is visible feedback - no need for extra output
+                    # Use bell for audible feedback (works in most terminals)
+                    event.app.output.bell()
+                    return  # Don't also paste text
+        except Exception:
+            pass  # Fall through to text paste on any error
+
+        # No image (or error) - do normal text paste
+        # prompt_toolkit doesn't have built-in paste, so we handle it manually
+        try:
+            import platform
+            import subprocess
+
+            text = None
+            system = platform.system()
+
+            if system == "Darwin":  # macOS
+                result = subprocess.run(
+                    ["pbpaste"], capture_output=True, text=True, timeout=2
+                )
+                if result.returncode == 0:
+                    text = result.stdout
+            elif system == "Windows":
+                # Windows - use powershell
+                result = subprocess.run(
+                    ["powershell", "-command", "Get-Clipboard"],
+                    capture_output=True,
+                    text=True,
+                    timeout=2,
+                )
+                if result.returncode == 0:
+                    text = result.stdout
+            else:  # Linux
+                # Try xclip first, then xsel
+                for cmd in [
+                    ["xclip", "-selection", "clipboard", "-o"],
+                    ["xsel", "--clipboard", "--output"],
+                ]:
+                    try:
+                        result = subprocess.run(
+                            cmd, capture_output=True, text=True, timeout=2
+                        )
+                        if result.returncode == 0:
+                            text = result.stdout
+                            break
+                    except FileNotFoundError:
+                        continue
+
+            if text:
+                # Normalize Windows line endings to Unix style
+                text = text.replace("\r\n", "\n").replace("\r", "\n")
+                # Strip trailing newline that clipboard tools often add
+                text = text.rstrip("\n")
+                event.app.current_buffer.insert_text(text)
+        except Exception:
+            pass  # Silently fail if text paste doesn't work
+
+    # F3 - dedicated image paste (shows error if no image)
+    @bindings.add("f3")
+    def handle_image_paste_f3(event):
+        """Handle F3 - paste image from clipboard (image-only, shows error if none)."""
+        try:
+            if has_image_in_clipboard():
+                placeholder = capture_clipboard_image_to_pending()
+                if placeholder:
+                    event.app.current_buffer.insert_text(placeholder + " ")
+                    # The placeholder itself is visible feedback
+                    # Use bell for audible feedback (works in most terminals)
+                    event.app.output.bell()
+            else:
+                # Insert a transient message that user can delete
+                event.app.current_buffer.insert_text("[⚠️ no image in clipboard] ")
+                event.app.output.bell()
+        except Exception:
+            event.app.current_buffer.insert_text("[❌ clipboard error] ")
+            event.app.output.bell()
 
     session = PromptSession(
         completer=completer,
@@ -385,19 +784,20 @@ async def get_input_with_combined_completion(
         {
             # Keys must AVOID the 'class:' prefix – that prefix is used only when
             # tagging tokens in `FormattedText`. See prompt_toolkit docs.
-            "puppy": "bold magenta",
-            "owner": "bold white",
-            "agent": "bold blue",
-            "model": "bold cyan",
-            "cwd": "bold green",
-            "arrow": "bold yellow",
-            "attachment-placeholder": "italic cyan",
+            "puppy": "bold ansibrightcyan",
+            "owner": "bold ansibrightblue",
+            "agent": "bold ansibrightblue",
+            "model": "bold ansibrightcyan",
+            "cwd": "bold ansibrightgreen",
+            "arrow": "bold ansibrightblue",
+            "attachment-placeholder": "italic ansicyan",
         }
     )
     text = await session.prompt_async(prompt_str, style=style)
-    possibly_stripped = update_model_in_input(text)
-    if possibly_stripped is not None:
-        return possibly_stripped
+    # NOTE: We used to call update_model_in_input(text) here to handle /model and /m
+    # commands at the prompt level, but that prevented the command handler from running
+    # and emitting success messages. Now we let all /model commands fall through to
+    # the command handler in main.py for consistent handling.
     return text