add

Author: jzhu

code_assistant_manager/cli/prompts_commands.py

@@ -8,7 +8,12 @@
 import typer
 
 from code_assistant_manager.menu.base import Colors
-from code_assistant_manager.prompts import PROMPT_FILE_PATHS, Prompt, PromptManager
+from code_assistant_manager.prompts import (
+    PROMPT_FILE_PATHS,
+    Prompt,
+    PromptManager,
+    get_prompt_file_path,
+)
 
 logger = logging.getLogger(__name__)
 
@@ -19,6 +24,7 @@
 
 # Valid app types
 VALID_APP_TYPES = ["claude", "codex", "gemini"]
+VALID_LEVELS = ["user", "project"]
 
 
 def _get_prompt_manager() -> PromptManager:
@@ -264,8 +270,20 @@ def sync_prompts(
         "-a",
         help="App type to sync to (claude, codex, gemini). Required if prompt_id is specified.",
     ),
+    level: str = typer.Option(
+        "user",
+        "--level",
+        "-l",
+        help="Sync level: 'user' (~/.claude/) or 'project' (current directory)",
+    ),
 ):
     """Sync prompts to app files. Can sync a specific prompt or all active prompts."""
+    if level not in VALID_LEVELS:
+        typer.echo(
+            f"{Colors.RED}✗ Invalid level: {level}. Valid: {', '.join(VALID_LEVELS)}{Colors.RESET}"
+        )
+        raise typer.Exit(1)
+
     manager = _get_prompt_manager()
 
     if prompt_id:
@@ -287,21 +305,37 @@ def sync_prompts(
             typer.echo(f"{Colors.RED}✗ Prompt not found: {prompt_id}{Colors.RESET}")
             raise typer.Exit(1)
 
+        file_path = get_prompt_file_path(app_type, level)
+        if not file_path:
+            typer.echo(f"{Colors.RED}✗ Unknown app type: {app_type}{Colors.RESET}")
+            raise typer.Exit(1)
+
         try:
-            manager._sync_prompt_to_file(prompt.content, app_type)
-            typer.echo(f"{Colors.GREEN}✓ Synced '{prompt_id}' to {app_type}{Colors.RESET}")
-            file_path = PROMPT_FILE_PATHS.get(app_type)
+            # Ensure parent directory exists
+            file_path.parent.mkdir(parents=True, exist_ok=True)
+            file_path.write_text(prompt.content, encoding="utf-8")
+            typer.echo(
+                f"{Colors.GREEN}✓ Synced '{prompt_id}' to {app_type} ({level}){Colors.RESET}"
+            )
             typer.echo(f"  {Colors.CYAN}File:{Colors.RESET} {file_path}")
         except Exception as e:
             typer.echo(f"{Colors.RED}✗ Error: {e}{Colors.RESET}")
             raise typer.Exit(1)
     else:
         # Sync all active prompts
+        if level == "project":
+            typer.echo(
+                f"{Colors.RED}✗ --level project requires specifying a prompt ID{Colors.RESET}"
+            )
+            raise typer.Exit(1)
+
         results = manager.sync_all()
 
         for app, synced_prompt_id in results.items():
             if synced_prompt_id:
-                typer.echo(f"{Colors.GREEN}✓ {app}: synced ({synced_prompt_id}){Colors.RESET}")
+                typer.echo(
+                    f"{Colors.GREEN}✓ {app}: synced ({synced_prompt_id}){Colors.RESET}"
+                )
             else:
                 typer.echo(f"{Colors.YELLOW}○ {app}: no active prompt{Colors.RESET}")
 
@@ -409,6 +443,12 @@ def unsync_prompt(
         "-a",
         help="App type to unsync (claude, codex, gemini)",
     ),
+    level: str = typer.Option(
+        "user",
+        "--level",
+        "-l",
+        help="Unsync level: 'user' (~/.claude/) or 'project' (current directory)",
+    ),
     force: bool = typer.Option(False, "--force", "-f", help="Skip confirmation"),
 ):
     """Clear/unsync the prompt file for an app (removes CLAUDE.md, AGENTS.md, or GEMINI.md content)."""
@@ -418,13 +458,21 @@ def unsync_prompt(
         )
         raise typer.Exit(1)
 
-    file_path = PROMPT_FILE_PATHS.get(app_type)
+    if level not in VALID_LEVELS:
+        typer.echo(
+            f"{Colors.RED}✗ Invalid level: {level}. Valid: {', '.join(VALID_LEVELS)}{Colors.RESET}"
+        )
+        raise typer.Exit(1)
+
+    file_path = get_prompt_file_path(app_type, level)
     if not file_path:
         typer.echo(f"{Colors.RED}✗ Unknown app type: {app_type}{Colors.RESET}")
         raise typer.Exit(1)
 
     if not file_path.exists():
-        typer.echo(f"{Colors.YELLOW}Prompt file does not exist: {file_path}{Colors.RESET}")
+        typer.echo(
+            f"{Colors.YELLOW}Prompt file does not exist: {file_path}{Colors.RESET}"
+        )
         return
 
     if not force:
@@ -434,53 +482,74 @@ def unsync_prompt(
         file_path.write_text("", encoding="utf-8")
         typer.echo(f"{Colors.GREEN}✓ Cleared prompt file: {file_path}{Colors.RESET}")
 
-        # Deactivate any active prompt for this app
-        manager = _get_prompt_manager()
-        prompts = manager.get_all()
-        for prompt in prompts.values():
-            if prompt.enabled and prompt.app_type == app_type:
-                prompt.enabled = False
-                manager.update(prompt)
-                typer.echo(f"  {Colors.CYAN}Deactivated:{Colors.RESET} {prompt.id}")
+        # Deactivate any active prompt for this app (only for user level)
+        if level == "user":
+            manager = _get_prompt_manager()
+            prompts = manager.get_all()
+            for prompt in prompts.values():
+                if prompt.enabled and prompt.app_type == app_type:
+                    prompt.enabled = False
+                    manager.update(prompt)
+                    typer.echo(f"  {Colors.CYAN}Deactivated:{Colors.RESET} {prompt.id}")
     except Exception as e:
         typer.echo(f"{Colors.RED}✗ Error: {e}{Colors.RESET}")
         raise typer.Exit(1)
 
 
 @prompt_app.command("status")
-def show_prompt_status():
+def show_prompt_status(
+    level: str = typer.Option(
+        "all",
+        "--level",
+        "-l",
+        help="Show status for: 'user', 'project', or 'all' (default)",
+    ),
+):
     """Show prompt status for all apps."""
     manager = _get_prompt_manager()
 
-    typer.echo(f"\n{Colors.BOLD}Prompt Status:{Colors.RESET}\n")
+    levels_to_show = ["user", "project"] if level == "all" else [level]
 
-    for app_type in VALID_APP_TYPES:
-        file_path = PROMPT_FILE_PATHS.get(app_type)
-        active_prompt = manager.get_active_prompt(app_type)
-
-        typer.echo(f"{Colors.BOLD}{app_type.capitalize()}:{Colors.RESET}")
-        typer.echo(f"  {Colors.CYAN}File:{Colors.RESET} {file_path}")
-
-        if file_path and file_path.exists():
-            content = file_path.read_text(encoding="utf-8")
-            if content.strip():
-                lines = content.strip().split("\n")
-                preview = lines[0][:50] + "..." if len(lines[0]) > 50 else lines[0]
-                typer.echo(f"  {Colors.GREEN}Content:{Colors.RESET} {preview}")
-                typer.echo(f"  {Colors.CYAN}Lines:{Colors.RESET} {len(lines)}")
-            else:
-                typer.echo(f"  {Colors.YELLOW}Content:{Colors.RESET} (empty)")
-        else:
-            typer.echo(f"  {Colors.YELLOW}Content:{Colors.RESET} (file not found)")
+    if level != "all" and level not in VALID_LEVELS:
+        typer.echo(
+            f"{Colors.RED}✗ Invalid level: {level}. Valid: user, project, all{Colors.RESET}"
+        )
+        raise typer.Exit(1)
 
-        if active_prompt:
-            typer.echo(
-                f"  {Colors.GREEN}Active Prompt:{Colors.RESET} {active_prompt.name} ({active_prompt.id})"
-            )
-        else:
-            typer.echo(f"  {Colors.YELLOW}Active Prompt:{Colors.RESET} None")
+    for lvl in levels_to_show:
+        typer.echo(
+            f"\n{Colors.BOLD}Prompt Status ({lvl.capitalize()} Level):{Colors.RESET}\n"
+        )
 
-        typer.echo()
+        for app_type in VALID_APP_TYPES:
+            file_path = get_prompt_file_path(app_type, lvl)
+
+            typer.echo(f"{Colors.BOLD}{app_type.capitalize()}:{Colors.RESET}")
+            typer.echo(f"  {Colors.CYAN}File:{Colors.RESET} {file_path}")
+
+            if file_path and file_path.exists():
+                content = file_path.read_text(encoding="utf-8")
+                if content.strip():
+                    lines = content.strip().split("\n")
+                    preview = lines[0][:50] + "..." if len(lines[0]) > 50 else lines[0]
+                    typer.echo(f"  {Colors.GREEN}Content:{Colors.RESET} {preview}")
+                    typer.echo(f"  {Colors.CYAN}Lines:{Colors.RESET} {len(lines)}")
+                else:
+                    typer.echo(f"  {Colors.YELLOW}Content:{Colors.RESET} (empty)")
+            else:
+                typer.echo(f"  {Colors.YELLOW}Content:{Colors.RESET} (file not found)")
+
+            # Only show active prompt for user level
+            if lvl == "user":
+                active_prompt = manager.get_active_prompt(app_type)
+                if active_prompt:
+                    typer.echo(
+                        f"  {Colors.GREEN}Active Prompt:{Colors.RESET} {active_prompt.name} ({active_prompt.id})"
+                    )
+                else:
+                    typer.echo(f"  {Colors.YELLOW}Active Prompt:{Colors.RESET} None")
+
+            typer.echo()
 
 
 # Add list shorthand

code_assistant_manager/prompts.py

@@ -1,28 +1,72 @@
 """Prompt management for Code Assistant Manager.
 
 This module provides functionality to manage prompts for AI coding assistants.
-Prompts can be synced to the actual tool config files:
+Prompts can be synced to the actual tool config files at user or project level:
+
+User level (default):
 - Claude: ~/.claude/CLAUDE.md
 - Codex: ~/.codex/AGENTS.md
 - Gemini: ~/.gemini/GEMINI.md
+
+Project level (current directory):
+- Claude: ./CLAUDE.md
+- Codex: ./AGENTS.md
+- Gemini: ./GEMINI.md
 """
 
 import json
 import logging
+import os
 import shutil
 from datetime import datetime
 from pathlib import Path
 from typing import Dict, List, Optional
 
 logger = logging.getLogger(__name__)
 
-# Prompt file paths for each app type
-PROMPT_FILE_PATHS = {
+# User-level prompt file paths for each app type
+USER_PROMPT_FILE_PATHS = {
     "claude": Path.home() / ".claude" / "CLAUDE.md",
     "codex": Path.home() / ".codex" / "AGENTS.md",
     "gemini": Path.home() / ".gemini" / "GEMINI.md",
 }
 
+# Project-level prompt file names for each app type
+PROJECT_PROMPT_FILE_NAMES = {
+    "claude": "CLAUDE.md",
+    "codex": "AGENTS.md",
+    "gemini": "GEMINI.md",
+}
+
+# Keep for backward compatibility
+PROMPT_FILE_PATHS = USER_PROMPT_FILE_PATHS
+
+
+def get_prompt_file_path(
+    app_type: str, level: str = "user", project_dir: Optional[Path] = None
+) -> Optional[Path]:
+    """
+    Get the prompt file path for an app type at the specified level.
+
+    Args:
+        app_type: The app type (claude, codex, gemini)
+        level: Either "user" or "project"
+        project_dir: Project directory (defaults to current working directory)
+
+    Returns:
+        Path to the prompt file, or None if invalid
+    """
+    if level == "user":
+        return USER_PROMPT_FILE_PATHS.get(app_type)
+    elif level == "project":
+        filename = PROJECT_PROMPT_FILE_NAMES.get(app_type)
+        if not filename:
+            return None
+        if project_dir is None:
+            project_dir = Path.cwd()
+        return project_dir / filename
+    return None
+
 
 class Prompt:
     """Represents a prompt configuration."""