code-puppy 0.0.97__py3-none-any.whl → 0.0.119__py3-none-any.whl

code_puppy/command_line/command_handler.py

@@ -0,0 +1,359 @@
+import os
+
+from code_puppy.command_line.model_picker_completion import (
+    load_model_names,
+    update_model_in_input,
+)
+from code_puppy.command_line.motd import print_motd
+from code_puppy.command_line.utils import make_directory_table
+from code_puppy.config import get_config_keys
+from code_puppy.tools.tools_content import tools_content
+
+COMMANDS_HELP = """
+[bold magenta]Commands Help[/bold magenta]
+/help, /h             Show this help message
+/cd <dir>             Change directory or show directories
+
+/exit, /quit          Exit interactive mode
+/generate-pr-description [@dir]  Generate comprehensive PR description
+/m <model>            Set active model
+/motd                 Show the latest message of the day (MOTD)
+/show                 Show puppy config key-values
+/compact              Summarize and compact current chat history
+/dump_context <name>  Save current message history to file
+/load_context <name>  Load message history from file
+/set                  Set puppy config key-values (e.g., /set yolo_mode true)
+/tools                Show available tools and capabilities
+/<unknown>            Show unknown command warning
+"""
+
+
+def handle_command(command: str):
+    from code_puppy.messaging import emit_error, emit_info, emit_success, emit_warning
+
+    """
+    Handle commands prefixed with '/'.
+
+    Args:
+        command: The command string to handle
+
+    Returns:
+        True if the command was handled, False if not, or a string to be processed as user input
+    """
+    command = command.strip()
+
+    if command.strip().startswith("/motd"):
+        print_motd(force=True)
+        return True
+
+    if command.strip().startswith("/compact"):
+        from code_puppy.message_history_processor import (
+            estimate_tokens_for_message,
+            summarize_messages,
+        )
+        from code_puppy.messaging import (
+            emit_error,
+            emit_info,
+            emit_success,
+            emit_warning,
+        )
+        from code_puppy.state_management import get_message_history, set_message_history
+
+        try:
+            history = get_message_history()
+            if not history:
+                emit_warning("No history to compact yet. Ask me something first!")
+                return True
+
+            before_tokens = sum(estimate_tokens_for_message(m) for m in history)
+            emit_info(
+                f"🤔 Compacting {len(history)} messages... (~{before_tokens} tokens)"
+            )
+
+            compacted, _ = summarize_messages(history, with_protection=False)
+            if not compacted:
+                emit_error("Summarization failed. History unchanged.")
+                return True
+
+            set_message_history(compacted)
+
+            after_tokens = sum(estimate_tokens_for_message(m) for m in compacted)
+            reduction_pct = (
+                ((before_tokens - after_tokens) / before_tokens * 100)
+                if before_tokens > 0
+                else 0
+            )
+            emit_success(
+                f"✨ Done! History: {len(history)} → {len(compacted)} messages\n"
+                f"🏦 Tokens: {before_tokens:,} → {after_tokens:,} ({reduction_pct:.1f}% reduction)"
+            )
+            return True
+        except Exception as e:
+            emit_error(f"/compact error: {e}")
+            return True
+
+    if command.startswith("/cd"):
+        tokens = command.split()
+        if len(tokens) == 1:
+            try:
+                table = make_directory_table()
+                emit_info(table)
+            except Exception as e:
+                emit_error(f"Error listing directory: {e}")
+            return True
+        elif len(tokens) == 2:
+            dirname = tokens[1]
+            target = os.path.expanduser(dirname)
+            if not os.path.isabs(target):
+                target = os.path.join(os.getcwd(), target)
+            if os.path.isdir(target):
+                os.chdir(target)
+                emit_success(f"Changed directory to: {target}")
+            else:
+                emit_error(f"Not a directory: {dirname}")
+            return True
+
+    if command.strip().startswith("/show"):
+        from code_puppy.command_line.model_picker_completion import get_active_model
+        from code_puppy.config import (
+            get_owner_name,
+            get_protected_token_count,
+            get_puppy_name,
+            get_summarization_threshold,
+            get_yolo_mode,
+        )
+
+        puppy_name = get_puppy_name()
+        owner_name = get_owner_name()
+        model = get_active_model()
+        yolo_mode = get_yolo_mode()
+        protected_tokens = get_protected_token_count()
+        summary_threshold = get_summarization_threshold()
+
+        status_msg = f"""[bold magenta]🐶 Puppy Status[/bold magenta]
+
+[bold]puppy_name:[/bold]            [cyan]{puppy_name}[/cyan]
+[bold]owner_name:[/bold]            [cyan]{owner_name}[/cyan]
+[bold]model:[/bold]                 [green]{model}[/green]
+[bold]YOLO_MODE:[/bold]             {"[red]ON[/red]" if yolo_mode else "[yellow]off[/yellow]"}
+[bold]protected_tokens:[/bold]      [cyan]{protected_tokens:,}[/cyan] recent tokens preserved
+[bold]summary_threshold:[/bold]     [cyan]{summary_threshold:.1%}[/cyan] context usage triggers summarization
+
+"""
+        emit_info(status_msg)
+        return True
+
+    if command.startswith("/set"):
+        # Syntax: /set KEY=VALUE or /set KEY VALUE
+        from code_puppy.config import set_config_value
+
+        tokens = command.split(None, 2)
+        argstr = command[len("/set") :].strip()
+        key = None
+        value = None
+        if "=" in argstr:
+            key, value = argstr.split("=", 1)
+            key = key.strip()
+            value = value.strip()
+        elif len(tokens) >= 3:
+            key = tokens[1]
+            value = tokens[2]
+        elif len(tokens) == 2:
+            key = tokens[1]
+            value = ""
+        else:
+            emit_warning(
+                f"Usage: /set KEY=VALUE or /set KEY VALUE\nConfig keys: {', '.join(get_config_keys())}"
+            )
+            return True
+        if key:
+            set_config_value(key, value)
+            emit_success(f'🌶 Set {key} = "{value}" in puppy.cfg!')
+        else:
+            emit_error("You must supply a key.")
+        return True
+
+    if command.startswith("/tools"):
+        # Display the tools_content.py file content with markdown formatting
+        from rich.markdown import Markdown
+
+        markdown_content = Markdown(tools_content)
+        emit_info(markdown_content)
+        return True
+
+    if command.startswith("/m"):
+        # Try setting model and show confirmation
+        new_input = update_model_in_input(command)
+        if new_input is not None:
+            from code_puppy.agent import get_code_generation_agent
+            from code_puppy.command_line.model_picker_completion import get_active_model
+
+            model = get_active_model()
+            # Make sure this is called for the test
+            get_code_generation_agent(force_reload=True)
+            emit_success(f"Active model set and loaded: {model}")
+            return True
+        # If no model matched, show available models
+        model_names = load_model_names()
+        emit_warning("Usage: /m <model-name>")
+        emit_warning(f"Available models: {', '.join(model_names)}")
+        return True
+    if command in ("/help", "/h"):
+        emit_info(COMMANDS_HELP)
+        return True
+
+    if command.startswith("/generate-pr-description"):
+        # Parse directory argument (e.g., /generate-pr-description @some/dir)
+        tokens = command.split()
+        directory_context = ""
+        for t in tokens:
+            if t.startswith("@"):
+                directory_context = f" Please work in the directory: {t[1:]}"
+                break
+
+        # Hard-coded prompt from user requirements
+        pr_prompt = f"""Generate a comprehensive PR description for my current branch changes. Follow these steps:
+
+ 1 Discover the changes: Use git CLI to find the base branch (usually main/master/develop) and get the list of changed files, commits, and diffs.
+ 2 Analyze the code: Read and analyze all modified files to understand:
+    • What functionality was added/changed/removed
+    • The technical approach and implementation details
+    • Any architectural or design pattern changes
+    • Dependencies added/removed/updated
+ 3 Generate a structured PR description with these sections:
+    • Title: Concise, descriptive title (50 chars max)
+    • Summary: Brief overview of what this PR accomplishes
+    • Changes Made: Detailed bullet points of specific changes
+    • Technical Details: Implementation approach, design decisions, patterns used
+    • Files Modified: List of key files with brief description of changes
+    • Testing: What was tested and how (if applicable)
+    • Breaking Changes: Any breaking changes (if applicable)
+    • Additional Notes: Any other relevant information
+ 4 Create a markdown file: Generate a PR_DESCRIPTION.md file with proper GitHub markdown formatting that I can directly copy-paste into GitHub's PR
+   description field. Use proper markdown syntax with headers, bullet points, code blocks, and formatting.
+ 5 Make it review-ready: Ensure the description helps reviewers understand the context, approach, and impact of the changes.
+6. If you have Github MCP, or gh cli is installed and authenticated then find the PR for the branch we analyzed and update the PR description there and then delete the PR_DESCRIPTION.md file. (If you have a better name (title) for the PR, go ahead and update the title too.{directory_context}"""
+
+        # Return the prompt to be processed by the main chat system
+        return pr_prompt
+
+    if command.startswith("/dump_context"):
+        import json
+        import pickle
+        from datetime import datetime
+        from pathlib import Path
+
+        from code_puppy.config import CONFIG_DIR
+        from code_puppy.message_history_processor import estimate_tokens_for_message
+        from code_puppy.state_management import get_message_history
+
+        tokens = command.split()
+        if len(tokens) != 2:
+            emit_warning("Usage: /dump_context <session_name>")
+            return True
+
+        session_name = tokens[1]
+        history = get_message_history()
+
+        if not history:
+            emit_warning("No message history to dump!")
+            return True
+
+        # Create contexts directory inside CONFIG_DIR if it doesn't exist
+        contexts_dir = Path(CONFIG_DIR) / "contexts"
+        contexts_dir.mkdir(parents=True, exist_ok=True)
+
+        try:
+            # Save as pickle for exact preservation
+            pickle_file = contexts_dir / f"{session_name}.pkl"
+            with open(pickle_file, "wb") as f:
+                pickle.dump(history, f)
+
+            # Also save metadata as JSON for readability
+            meta_file = contexts_dir / f"{session_name}_meta.json"
+            metadata = {
+                "session_name": session_name,
+                "timestamp": datetime.now().isoformat(),
+                "message_count": len(history),
+                "total_tokens": sum(estimate_tokens_for_message(m) for m in history),
+                "file_path": str(pickle_file),
+            }
+
+            with open(meta_file, "w") as f:
+                json.dump(metadata, f, indent=2)
+
+            emit_success(
+                f"✅ Context saved: {len(history)} messages ({metadata['total_tokens']} tokens)\n"
+                f"📁 Files: {pickle_file}, {meta_file}"
+            )
+            return True
+
+        except Exception as e:
+            emit_error(f"Failed to dump context: {e}")
+            return True
+
+    if command.startswith("/load_context"):
+        import pickle
+        from pathlib import Path
+
+        from code_puppy.config import CONFIG_DIR
+        from code_puppy.message_history_processor import estimate_tokens_for_message
+        from code_puppy.state_management import set_message_history
+
+        tokens = command.split()
+        if len(tokens) != 2:
+            emit_warning("Usage: /load_context <session_name>")
+            return True
+
+        session_name = tokens[1]
+        contexts_dir = Path(CONFIG_DIR) / "contexts"
+        pickle_file = contexts_dir / f"{session_name}.pkl"
+
+        if not pickle_file.exists():
+            emit_error(f"Context file not found: {pickle_file}")
+            # List available contexts
+            available = list(contexts_dir.glob("*.pkl"))
+            if available:
+                names = [f.stem for f in available]
+                emit_info(f"Available contexts: {', '.join(names)}")
+            return True
+
+        try:
+            with open(pickle_file, "rb") as f:
+                history = pickle.load(f)
+
+            set_message_history(history)
+            total_tokens = sum(estimate_tokens_for_message(m) for m in history)
+
+            emit_success(
+                f"✅ Context loaded: {len(history)} messages ({total_tokens} tokens)\n"
+                f"📁 From: {pickle_file}"
+            )
+            return True
+
+        except Exception as e:
+            emit_error(f"Failed to load context: {e}")
+            return True
+
+    if command in ("/exit", "/quit"):
+        emit_success("Goodbye!")
+        # Signal to the main app that we want to exit
+        # The actual exit handling is done in main.py
+        return True
+    if command.startswith("/"):
+        name = command[1:].split()[0] if len(command) > 1 else ""
+        if name:
+            emit_warning(
+                f"Unknown command: {command}\n[dim]Type /help for options.[/dim]"
+            )
+        else:
+            # Show current model ONLY here
+            from code_puppy.command_line.model_picker_completion import get_active_model
+
+            current_model = get_active_model()
+            emit_info(
+                f"[bold green]Current Model:[/bold green] [cyan]{current_model}[/cyan]"
+            )
+        return True
+
+    return False

code_puppy/command_line/load_context_completion.py

@@ -0,0 +1,59 @@
+from pathlib import Path
+
+from prompt_toolkit.completion import Completer, Completion
+
+from code_puppy.config import CONFIG_DIR
+
+
+class LoadContextCompleter(Completer):
+    def __init__(self, trigger: str = "/load_context"):
+        self.trigger = trigger
+
+    def get_completions(self, document, complete_event):
+        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):
+            return
+
+        # Determine the part of the text that is relevant for this completer
+        actual_trigger_pos = text_before_cursor.find(self.trigger)
+        effective_input = text_before_cursor[actual_trigger_pos:]
+
+        tokens = effective_input.split()
+
+        # Case 1: Input is exactly the trigger (e.g., "/load_context") and nothing more
+        if (
+            len(tokens) == 1
+            and tokens[0] == self.trigger
+            and not effective_input.endswith(" ")
+        ):
+            yield Completion(
+                text=self.trigger + " ",
+                start_position=-len(tokens[0]),
+                display=self.trigger + " ",
+                display_meta="load saved context",
+            )
+            return
+
+        # Case 2: Input is trigger + space or trigger + partial session name
+        session_filter = ""
+        if len(tokens) > 1:  # e.g., ["/load_context", "partial"]
+            session_filter = tokens[1]
+
+        # Get available context files
+        try:
+            contexts_dir = Path(CONFIG_DIR) / "contexts"
+            if contexts_dir.exists():
+                for pkl_file in contexts_dir.glob("*.pkl"):
+                    session_name = pkl_file.stem  # removes .pkl extension
+                    if session_name.startswith(session_filter):
+                        yield Completion(
+                            session_name,
+                            start_position=-len(session_filter),
+                            display=session_name,
+                            display_meta="saved context session",
+                        )
+        except Exception:
+            # Silently ignore errors (e.g., permission issues, non-existent dir)
+            pass

code_puppy/command_line/model_picker_completion.py

@@ -1,4 +1,3 @@
-import json
 import os
 from typing import Iterable, Optional
 
@@ -8,17 +7,13 @@ from prompt_toolkit.document import Document
 from prompt_toolkit.history import FileHistory
 
 from code_puppy.config import get_model_name, set_model_name
-
-MODELS_JSON_PATH = os.environ.get("MODELS_JSON_PATH")
-if not MODELS_JSON_PATH:
-    MODELS_JSON_PATH = os.path.join(os.path.dirname(__file__), "..", "models.json")
-    MODELS_JSON_PATH = os.path.abspath(MODELS_JSON_PATH)
+from code_puppy.model_factory import ModelFactory
 
 
 def load_model_names():
-    with open(MODELS_JSON_PATH, "r") as f:
-        models = json.load(f)
-    return list(models.keys())
+    """Load model names from the config that's fetched from the endpoint."""
+    models_config = ModelFactory.load_config()
+    return list(models_config.keys())
 
 
 def get_active_model():
@@ -31,11 +26,9 @@ def get_active_model():
 
 def set_active_model(model_name: str):
     """
-    Sets the active model name by updating both config (for persistence)
-    and env (for process lifetime override).
+    Sets the active model name by updating the config (for persistence).
     """
     set_model_name(model_name)
-    os.environ["MODEL_NAME"] = model_name.strip()
     # Reload agent globally
     try:
         from code_puppy.agent import reload_code_generation_agent
@@ -47,11 +40,11 @@ def set_active_model(model_name: str):
 
 class ModelNameCompleter(Completer):
     """
-    A completer that triggers on '~m' to show available models from models.json.
-    Only '~m' (not just '~') will trigger the dropdown.
+    A completer that triggers on '/m' to show available models from models.json.
+    Only '/m' (not just '/') will trigger the dropdown.
     """
 
-    def __init__(self, trigger: str = "~m"):
+    def __init__(self, trigger: str = "/m"):
         self.trigger = trigger
         self.model_names = load_model_names()
 
@@ -77,23 +70,23 @@ class ModelNameCompleter(Completer):
 
 
 def update_model_in_input(text: str) -> Optional[str]:
-    # If input starts with ~m and a model name, set model and strip it out
+    # If input starts with /m and a model name, set model and strip it out
     content = text.strip()
-    if content.startswith("~m"):
+    if content.startswith("/m"):
         rest = content[2:].strip()
         for model in load_model_names():
             if rest == model:
                 set_active_model(model)
-                # Remove ~mmodel from the input
-                idx = text.find("~m" + model)
+                # Remove /mmodel from the input
+                idx = text.find("/m" + model)
                 if idx != -1:
-                    new_text = (text[:idx] + text[idx + len("~m" + model) :]).strip()
+                    new_text = (text[:idx] + text[idx + len("/m" + model) :]).strip()
                     return new_text
     return None
 
 
 async def get_input_with_model_completion(
-    prompt_str: str = ">>> ", trigger: str = "~m", history_file: Optional[str] = None
+    prompt_str: str = ">>> ", trigger: str = "/m", history_file: Optional[str] = None
 ) -> str:
     history = FileHistory(os.path.expanduser(history_file)) if history_file else None
     session = PromptSession(

code_puppy/command_line/motd.py

@@ -1,35 +1,25 @@
 """
-MOTD (Message of the Day) feature for code-puppy.
-Stores seen versions in ~/.puppy_cfg/motd.txt.
+🐶 MOTD (Message of the Day) feature for code-puppy! 🐕
+Stores seen versions in ~/.code_puppy/motd.txt - woof woof! 🐾
 """
 
 import os
 
-MOTD_VERSION = "20250817"
-MOTD_MESSAGE = """
-
-🐾  Happy Sunday, Aug 17, 2025!
-
-Biscuit the code puppy learned two new tricks!
-Major paws-ups:
-1. On-the-fly summarization: when your model's context hits 90%,
-   Biscuit auto-summarizes older messages to keep you cruising. No sweat, no tokens spilled.
-2. AGENT.md support: ship your project rules and style guide,
-   and Biscuit will obey them like the good pup he is.
-
-• Use ~m to swap models mid-session.
-• YOLO_MODE=true skips command confirmations (danger, zoomies!).
-• Keep files under 600 lines; split big ones like a responsible hooman.
-• DRY code, happy pup.
-
-Today's vibe: sniff context, summarize smartly, obey AGENT.md, and ship.
-Run ~motd anytime you need more puppy hype!
+from code_puppy.config import CONFIG_DIR
+from code_puppy.messaging import emit_info
 
+MOTD_VERSION = "2025-08-24"
+MOTD_MESSAGE = """🐕‍🦺
+🐾```
+# 🐶🎉🐕 WOOF WOOF! AUGUST 24th 🐕🎉🐶
+40k Downloads! Woot!
+Thanks for your support!
+-Mike
 """
-MOTD_TRACK_FILE = os.path.expanduser("~/.puppy_cfg/motd.txt")
+MOTD_TRACK_FILE = os.path.join(CONFIG_DIR, "motd.txt")
 
 
-def has_seen_motd(version: str) -> bool:
+def has_seen_motd(version: str) -> bool:  # 🐕 Check if puppy has seen this MOTD!
     if not os.path.exists(MOTD_TRACK_FILE):
         return False
     with open(MOTD_TRACK_FILE, "r") as f:
@@ -37,15 +27,41 @@ def has_seen_motd(version: str) -> bool:
     return version in seen_versions
 
 
-def mark_motd_seen(version: str):
+def mark_motd_seen(version: str):  # 🐶 Mark MOTD as seen by this good puppy!
+    # Create directory if it doesn't exist 🏠🐕
     os.makedirs(os.path.dirname(MOTD_TRACK_FILE), exist_ok=True)
-    with open(MOTD_TRACK_FILE, "a") as f:
-        f.write(f"{version}\n")
 
+    # Check if the version is already in the file 📋🐶
+    seen_versions = set()
+    if os.path.exists(MOTD_TRACK_FILE):
+        with open(MOTD_TRACK_FILE, "r") as f:
+            seen_versions = {line.strip() for line in f if line.strip()}
+
+    # Only add the version if it's not already there 📝🐕‍🦺
+    if version not in seen_versions:
+        with open(MOTD_TRACK_FILE, "a") as f:
+            f.write(f"{version}\n")
 
-def print_motd(console, force: bool = False) -> bool:
+
+def print_motd(
+    console=None, force: bool = False
+) -> bool:  # 🐶 Print exciting puppy MOTD!
+    """
+    🐕 Print the message of the day to the user - woof woof! 🐕
+
+    Args:
+        console: Optional console object (for backward compatibility) 🖥️🐶
+        force: Whether to force printing even if the MOTD has been seen 💪🐕‍🦺
+
+    Returns:
+        True if the MOTD was printed, False otherwise 🐾
+    """
     if force or not has_seen_motd(MOTD_VERSION):
-        console.print(MOTD_MESSAGE)
+        # Create a Rich Markdown object for proper rendering 🎨🐶
+        from rich.markdown import Markdown
+
+        markdown_content = Markdown(MOTD_MESSAGE)
+        emit_info(markdown_content)
         mark_motd_seen(MOTD_VERSION)
         return True
     return False