codegraph-cli 2.1.0__py3-none-any.whl → 2.1.2__py3-none-any.whl

codegraph_cli/cli_setup.py

@@ -305,10 +305,10 @@ def set_llm(
     """Quickly switch LLM provider without full setup wizard.
     
     Examples:
-        cg set-llm groq -k YOUR_API_KEY
-        cg set-llm gemini -k YOUR_API_KEY -m gemini-2.0-flash
-        cg set-llm openrouter -k YOUR_API_KEY -m google/gemini-2.0-flash-exp:free
-        cg set-llm ollama -m qwen2.5-coder:7b
+        cg config set-llm groq -k YOUR_API_KEY
+        cg config set-llm gemini -k YOUR_API_KEY -m gemini-2.0-flash
+        cg config set-llm openrouter -k YOUR_API_KEY -m google/gemini-2.0-flash-exp:free
+        cg config set-llm ollama -m qwen2.5-coder:7b
     """
     provider = provider.lower().strip()
     
@@ -467,9 +467,9 @@ def show_llm():
     typer.echo("")
     typer.echo(typer.style("  Quick Commands", bold=True))
     typer.echo(typer.style("  ─────────────────────────────────────────", dim=True))
-    typer.echo(f"  {typer.style('cg setup', fg=typer.colors.YELLOW)}              Full interactive wizard")
-    typer.echo(f"  {typer.style('cg set-llm <name>', fg=typer.colors.YELLOW)}     Quick switch provider")
-    typer.echo(f"  {typer.style('cg unset-llm', fg=typer.colors.YELLOW)}          Reset / clear config")
+    typer.echo(f"  {typer.style('cg config setup', fg=typer.colors.YELLOW)}              Full interactive wizard")
+    typer.echo(f"  {typer.style('cg config set-llm <name>', fg=typer.colors.YELLOW)}     Quick switch provider")
+    typer.echo(f"  {typer.style('cg config unset-llm', fg=typer.colors.YELLOW)}          Reset / clear config")
     typer.echo("")
 
 
@@ -550,9 +550,9 @@ def set_embedding(
         qodo-1.5b   ~6.2 GB   Best quality, code-optimized
 
     Examples:
-        cg set-embedding minilm
-        cg set-embedding jina-code
-        cg set-embedding hash
+        cg config set-embedding minilm
+        cg config set-embedding jina-code
+        cg config set-embedding hash
     """
     model = model.lower().strip()
 
@@ -619,8 +619,8 @@ def show_embedding():
     typer.echo("")
     typer.echo(typer.style("  Quick Commands", bold=True))
     typer.echo(typer.style("  ─────────────────────────────────────────", dim=True))
-    typer.echo(f"  {typer.style('cg set-embedding <model>', fg=typer.colors.YELLOW)}   Switch model")
-    typer.echo(f"  {typer.style('cg unset-embedding', fg=typer.colors.YELLOW)}         Reset to hash")
+    typer.echo(f"  {typer.style('cg config set-embedding <model>', fg=typer.colors.YELLOW)}   Switch model")
+    typer.echo(f"  {typer.style('cg config unset-embedding', fg=typer.colors.YELLOW)}         Reset to hash")
     typer.echo("")
 
 

codegraph_cli/cli_suggestions.py

@@ -0,0 +1,90 @@
+"""Contextual next-step suggestions after command completion."""
+
+from __future__ import annotations
+
+from rich.console import Console
+from rich.panel import Panel
+
+console = Console()
+
+NEXT_STEPS: dict[str, list[str]] = {
+    "index": [
+        "[cyan]cg search[/cyan] 'your query'             Search code semantically",
+        "[cyan]cg chat start[/cyan]                      Ask questions about your code",
+        "[cyan]cg impact[/cyan] 'symbol'                 See what depends on a function",
+    ],
+    "search": [
+        "[cyan]cg chat start[/cyan]                      Discuss results with AI",
+        "[cyan]cg v2 review[/cyan] <file>                Get AI code review",
+        "[cyan]cg graph[/cyan] 'symbol'                  Visualize dependencies",
+    ],
+    "generate": [
+        "[cyan]cg v2 review[/cyan] <file>                Review generated code",
+        "[cyan]cg v2 test unit[/cyan] <symbol>           Generate tests for new code",
+        "[cyan]cg export-graph[/cyan]                    Visualize changes",
+    ],
+    "diagnose": [
+        "[cyan]cg v2 test unit[/cyan] <symbol>           Add tests for fixed code",
+        "[cyan]cg v2 review[/cyan] <file>                Review the fixes",
+    ],
+    "chat": [
+        "[cyan]cg search[/cyan] 'query'                  Find specific code",
+        "[cyan]cg v2 generate[/cyan] 'description'       Generate new code",
+    ],
+    "review": [
+        "[cyan]cg v2 diagnose fix[/cyan] <path>          Auto-fix detected issues",
+        "[cyan]cg v2 refactor rename[/cyan] <old> <new>  Rename symbols safely",
+        "[cyan]cg v2 test unit[/cyan] <symbol>           Generate tests",
+    ],
+    "refactor": [
+        "[cyan]cg v2 review[/cyan] <file>                Review refactored code",
+        "[cyan]cg v2 test unit[/cyan] <symbol>           Add tests after refactoring",
+    ],
+    "test": [
+        "[cyan]cg v2 review[/cyan] <file>                Review generated tests",
+        "[cyan]cg chat start[/cyan]                      Discuss test strategy with AI",
+    ],
+    "impact": [
+        "[cyan]cg graph[/cyan] 'symbol'                  Visualize dependency graph",
+        "[cyan]cg search[/cyan] 'related query'          Find related code",
+        "[cyan]cg chat start[/cyan]                      Deep-dive with AI",
+    ],
+    "quickstart": [
+        "[cyan]cg search[/cyan] 'your query'             Search code semantically",
+        "[cyan]cg chat start[/cyan]                      Chat with AI about your code",
+        "[cyan]cg v2 generate[/cyan] 'description'       Generate new code",
+    ],
+}
+
+
+def show_next_steps(command_name: str) -> None:
+    """Show contextual next steps after command completion.
+
+    Args:
+        command_name: The name of the command that just completed.
+    """
+    steps = NEXT_STEPS.get(command_name)
+    if not steps:
+        return
+
+    console.print(
+        "\n"
+        + Panel.fit(
+            "\n".join([f"  {step}" for step in steps]),
+            title="[bold cyan]💡 Next steps:[/bold cyan]",
+            border_style="cyan",
+            padding=(0, 1),
+        ).markup  # type: ignore[attr-defined]
+        if False
+        else ""
+    )
+    # Use direct Panel printing to avoid markup issues
+    console.print()
+    console.print(
+        Panel(
+            "\n".join([f"  {step}" for step in steps]),
+            title="[bold cyan]💡 Next steps:[/bold cyan]",
+            border_style="cyan",
+            padding=(0, 1),
+        )
+    )

codegraph_cli/cli_test.py

@@ -45,7 +45,12 @@ def generate_unit_tests(
     symbol: str = typer.Argument(..., help="Function name to generate tests for"),
     output: Optional[str] = typer.Option(None, "--output", "-o", help="Output test file path"),
 ):
-    """Generate unit tests for a function."""
+    """🧪 Generate unit tests for a function.
+
+    Example:
+      cg v2 test unit "calculate_total"
+      cg v2 test unit "UserService.authenticate" --output tests/test_auth.py
+    """
     pm = ProjectManager()
     agent = _get_testgen_agent(pm)
     
@@ -106,7 +111,12 @@ def generate_integration_tests(
     flow: str = typer.Argument(..., help="User flow description"),
     output: Optional[str] = typer.Option(None, "--output", "-o", help="Output test file path"),
 ):
-    """Generate integration tests for a user flow."""
+    """🧪 Generate integration tests for a user flow.
+
+    Example:
+      cg v2 test integration "user login and token refresh"
+      cg v2 test integration "payment processing" --output tests/test_payments.py
+    """
     pm = ProjectManager()
     agent = _get_testgen_agent(pm)
     
@@ -155,7 +165,11 @@ def generate_integration_tests(
 def show_coverage_prediction(
     symbol: str = typer.Argument(..., help="Function to analyze"),
 ):
-    """Show predicted coverage impact of generating tests."""
+    """📊 Show predicted coverage impact of generating tests.
+
+    Example:
+      cg v2 test coverage "process_payment"
+    """
     pm = ProjectManager()
     agent = _get_testgen_agent(pm)
     

codegraph_cli/cli_tui.py

@@ -0,0 +1,210 @@
+"""Interactive TUI menu for CodeGraph CLI."""
+
+from __future__ import annotations
+
+import typer
+from rich.console import Console
+from rich.panel import Panel
+from rich.prompt import Prompt
+
+console = Console()
+
+
+def show_interactive_menu() -> None:
+    """Display the interactive TUI menu when no command is provided."""
+    console.print()
+    console.print(
+        Panel.fit(
+            "[bold cyan]🧠 CodeGraph CLI[/bold cyan]\n"
+            "[dim]AI-powered code intelligence & multi-agent assistant[/dim]",
+            border_style="cyan",
+        )
+    )
+
+    while True:
+        console.print("\n[bold]What would you like to do?[/bold]\n")
+
+        choices = [
+            "1. 🔍 Search my codebase",
+            "2. 💬 Chat with AI about my code",
+            "3. ✨ Generate new code",
+            "4. 📊 Analyze code impact",
+            "5. 🔧 Review and improve code",
+            "6. ⚙️  Configure settings",
+            "7. 📚 Learn more / tutorial",
+            "8. 🏥 Project health dashboard",
+            "0. Exit",
+        ]
+        for choice in choices:
+            console.print(f"  {choice}")
+
+        selection = Prompt.ask(
+            "\nChoice",
+            choices=["0", "1", "2", "3", "4", "5", "6", "7", "8"],
+            default="0",
+        )
+
+        if selection == "0":
+            console.print("[cyan]Goodbye![/cyan]")
+            break
+
+        elif selection == "1":
+            query = Prompt.ask("Search query")
+            if query.strip():
+                _run_search(query.strip())
+
+        elif selection == "2":
+            _run_chat()
+
+        elif selection == "3":
+            desc = Prompt.ask("What code to generate")
+            if desc.strip():
+                _run_generate(desc.strip())
+
+        elif selection == "4":
+            symbol = Prompt.ask("Symbol to analyze")
+            if symbol.strip():
+                _run_impact(symbol.strip())
+
+        elif selection == "5":
+            path = Prompt.ask("File path to review")
+            if path.strip():
+                _run_review(path.strip())
+
+        elif selection == "6":
+            _run_setup()
+
+        elif selection == "7":
+            _show_tutorial()
+
+        elif selection == "8":
+            _run_health()
+
+
+def _run_search(query: str) -> None:
+    """Run search from TUI."""
+    try:
+        from .storage import ProjectManager, GraphStore
+        from .orchestrator import MCPOrchestrator
+
+        pm = ProjectManager()
+        project = pm.get_current_project()
+        if not project:
+            console.print("[red]✗[/red] No project loaded. Run 'cg index <path>' first.")
+            return
+        store = GraphStore(pm.project_dir(project))
+        orchestrator = MCPOrchestrator(store)
+        results = orchestrator.search(query, top_k=5)
+        if not results:
+            console.print("[yellow]No results found.[/yellow]")
+        else:
+            for item in results:
+                console.print(
+                    f"  [{item.node_type}] {item.qualname}  "
+                    f"[dim]score={item.score:.3f}[/dim]"
+                )
+                console.print(f"    [dim]{item.file_path}:{item.start_line}-{item.end_line}[/dim]")
+        store.close()
+    except Exception as e:
+        console.print(f"[red]Error: {e}[/red]")
+
+
+def _run_chat() -> None:
+    """Start chat from TUI."""
+    try:
+        # Import and invoke via typer to handle all the setup
+        console.print("[cyan]Starting chat... (use /exit to return)[/cyan]")
+        from .cli_chat import start_chat
+        start_chat()
+    except (typer.Exit, SystemExit):
+        pass
+    except Exception as e:
+        console.print(f"[red]Error: {e}[/red]")
+
+
+def _run_generate(description: str) -> None:
+    """Run code generation from TUI."""
+    try:
+        from .cli_v2 import generate_code
+        generate_code(description)
+    except (typer.Exit, SystemExit):
+        pass
+    except Exception as e:
+        console.print(f"[red]Error: {e}[/red]")
+
+
+def _run_impact(symbol: str) -> None:
+    """Run impact analysis from TUI."""
+    try:
+        from .storage import ProjectManager, GraphStore
+        from .orchestrator import MCPOrchestrator
+
+        pm = ProjectManager()
+        project = pm.get_current_project()
+        if not project:
+            console.print("[red]✗[/red] No project loaded.")
+            return
+        store = GraphStore(pm.project_dir(project))
+        orchestrator = MCPOrchestrator(store)
+        report = orchestrator.impact(symbol, hops=2)
+        console.print(f"Root: {report.root}")
+        if report.impacted:
+            console.print("Impacted symbols:")
+            for imp in report.impacted:
+                console.print(f"  • {imp}")
+        console.print(f"\n{report.explanation}")
+        store.close()
+    except Exception as e:
+        console.print(f"[red]Error: {e}[/red]")
+
+
+def _run_review(path: str) -> None:
+    """Run code review from TUI."""
+    try:
+        from .cli_v2 import review_code
+        review_code(path)
+    except (typer.Exit, SystemExit):
+        pass
+    except Exception as e:
+        console.print(f"[red]Error: {e}[/red]")
+
+
+def _run_setup() -> None:
+    """Run setup wizard from TUI."""
+    try:
+        from .cli_setup import setup
+        setup()
+    except (typer.Exit, SystemExit):
+        pass
+    except Exception as e:
+        console.print(f"[red]Error: {e}[/red]")
+
+
+def _run_health() -> None:
+    """Run health dashboard from TUI."""
+    try:
+        from .cli_health import health_dashboard
+        health_dashboard()
+    except (typer.Exit, SystemExit):
+        pass
+    except Exception as e:
+        console.print(f"[red]Error: {e}[/red]")
+
+
+def _show_tutorial() -> None:
+    """Show interactive tutorial."""
+    console.print(
+        Panel(
+            "📚 [bold]Quick Tutorial[/bold]\n\n"
+            "1. [cyan]Index your project[/cyan]:  cg index ./my-project\n"
+            "2. [cyan]Search code[/cyan]:          cg search 'authentication'\n"
+            "3. [cyan]Chat with AI[/cyan]:         cg chat start\n"
+            "4. [cyan]Generate code[/cyan]:        cg v2 generate 'add API endpoint'\n"
+            "5. [cyan]Impact analysis[/cyan]:      cg impact 'symbol_name'\n"
+            "6. [cyan]Code review[/cyan]:          cg v2 review path/to/file.py\n"
+            "7. [cyan]Project health[/cyan]:       cg health dashboard\n\n"
+            "For full reference:  cg cheatsheet\n"
+            "For documentation:   cg learn",
+            border_style="blue",
+        )
+    )

codegraph_cli/cli_v2.py

@@ -56,7 +56,13 @@ def generate_code(
     llm_api_key: Optional[str] = typer.Option(config.LLM_API_KEY, help="API key for LLM"),
     llm_model: str = typer.Option(config.LLM_MODEL, help="LLM model"),
 ):
-    """Generate code from natural language description (v2.0 experimental)."""
+    """✨ Generate code from natural language description (v2.0 experimental).
+
+    Example:
+      cg v2 generate 'add REST API endpoint for users'
+      cg v2 generate 'add login function' --file auth.py
+      cg v2 generate 'create data model' --output models/ --auto-apply
+    """
     pm = ProjectManager()
     agent = _get_codegen_agent(pm)
     
@@ -116,7 +122,11 @@ def generate_code(
 def rollback_changes(
     backup_id: str = typer.Argument(..., help="Backup ID to rollback to"),
 ):
-    """Rollback to a previous backup."""
+    """⏪ Rollback to a previous backup.
+
+    Example:
+      cg v2 rollback backup_20240101_120000
+    """
     diff_engine = DiffEngine()
     
     typer.echo(f"🔄 Rolling back to backup: {backup_id}")
@@ -130,7 +140,11 @@ def rollback_changes(
 
 @v2_app.command("list-backups")
 def list_backups():
-    """List all available backups."""
+    """📦 List all available backups.
+
+    Example:
+      cg v2 list-backups
+    """
     diff_engine = DiffEngine()
     backups = diff_engine.list_backups()
     
@@ -156,7 +170,13 @@ def review_code(
     use_llm: bool = typer.Option(False, "--llm", help="Use LLM for deeper analysis"),
     show_fixes: bool = typer.Option(False, "--fix", help="Show auto-fix suggestions"),
 ):
-    """Run AI-powered code review on a file."""
+    """🔍 Run AI-powered code review on a file.
+
+    Example:
+      cg v2 review src/auth.py
+      cg v2 review src/models.py --check security --llm
+      cg v2 review src/api.py --fix --verbose
+    """
     from .bug_detector import BugDetector
     from .security_scanner import SecurityScanner
     from .performance_analyzer import PerformanceAnalyzer

codegraph_cli/cli_watch.py

@@ -0,0 +1,158 @@
+"""Watch mode for auto-reindexing on file changes."""
+
+from __future__ import annotations
+
+import time
+from pathlib import Path
+
+import typer
+from rich.console import Console
+
+console = Console()
+
+watch_app = typer.Typer(help="👀 Watch mode for auto-reindexing")
+
+# Supported code file extensions
+WATCHED_EXTENSIONS = {
+    ".py", ".js", ".ts", ".tsx", ".jsx",
+    ".go", ".rs", ".java", ".kt", ".kts",
+    ".rb", ".php", ".ex", ".exs",
+    ".c", ".cpp", ".h", ".hpp", ".cs",
+    ".swift", ".dart",
+}
+
+
+class CodeChangeHandler:
+    """Handle file system events and trigger re-indexing."""
+
+    def __init__(self, reindex_callback, debounce_seconds: float = 2.0):
+        from watchdog.events import FileSystemEventHandler
+        self._base_class = FileSystemEventHandler
+        self.reindex_callback = reindex_callback
+        self.last_reindex = 0.0
+        self.debounce_seconds = debounce_seconds
+        self._pending_files: set[str] = set()
+
+    def dispatch(self, event):
+        """Route events to handler methods."""
+        if event.is_directory:
+            return
+        if hasattr(event, "src_path"):
+            self._handle_change(event.src_path)
+
+    def _handle_change(self, src_path: str):
+        file_path = Path(src_path)
+        if file_path.suffix not in WATCHED_EXTENSIONS:
+            return
+
+        # Skip hidden/temp files
+        if any(part.startswith(".") for part in file_path.parts):
+            return
+
+        now = time.time()
+        self._pending_files.add(str(file_path))
+
+        if now - self.last_reindex < self.debounce_seconds:
+            return
+
+        # Flush pending
+        files = list(self._pending_files)
+        self._pending_files.clear()
+        self.last_reindex = now
+
+        for f in files:
+            self.reindex_callback(Path(f))
+
+
+@watch_app.command("start")
+def watch(
+    path: str = typer.Argument(".", help="Path to watch for changes."),
+    interval: float = typer.Option(2.0, "--interval", "-i", help="Debounce interval in seconds."),
+    full_reindex: bool = typer.Option(False, "--full", help="Re-index entire project on each change."),
+):
+    """👀 Watch mode — auto-reindex on file changes.
+
+    Monitors your project directory for file changes and automatically
+    re-indexes modified files to keep the code graph up to date.
+
+    Example:
+      cg watch
+      cg watch ./src --interval 5
+      cg watch --full
+    """
+    try:
+        from watchdog.observers import Observer
+        from watchdog.events import FileSystemEventHandler
+    except ImportError:
+        console.print("[red]✗[/red] watchdog is not installed.")
+        console.print("[dim]Install with: pip install watchdog[/dim]")
+        raise typer.Exit(1)
+
+    watch_path = Path(path).resolve()
+    if not watch_path.exists():
+        console.print(f"[red]✗[/red] Path not found: {path}")
+        raise typer.Exit(1)
+
+    from .storage import ProjectManager, GraphStore
+    from .orchestrator import MCPOrchestrator
+
+    pm = ProjectManager()
+    project = pm.get_current_project()
+    if not project:
+        console.print("[red]✗[/red] No project loaded. Run 'cg index <path>' first.")
+        raise typer.Exit(1)
+
+    project_dir = pm.project_dir(project)
+    reindex_count = 0
+
+    def reindex_file(file_path: Path):
+        nonlocal reindex_count
+        try:
+            if full_reindex:
+                store = GraphStore(project_dir)
+                orchestrator = MCPOrchestrator(store)
+                stats = orchestrator.index(watch_path)
+                store.close()
+                console.print(
+                    f"  [green]✓[/green] Full re-index: {stats['nodes']} nodes, {stats['edges']} edges"
+                )
+            else:
+                # Incremental: re-index the single changed file
+                store = GraphStore(project_dir)
+                orchestrator = MCPOrchestrator(store)
+                # For now, do a full re-index (single-file not yet supported in orchestrator)
+                stats = orchestrator.index(watch_path)
+                store.close()
+                console.print(f"  [green]✓[/green] Re-indexed ({file_path.name} changed)")
+            reindex_count += 1
+        except Exception as e:
+            console.print(f"  [red]✗[/red] Re-index failed: {e}")
+
+    console.print(f"\n[bold green]👀 Watching[/bold green] [cyan]{watch_path}[/cyan] for changes...")
+    console.print(f"[dim]  Debounce:  {interval}s")
+    console.print(f"  Mode:      {'full re-index' if full_reindex else 'incremental'}")
+    console.print(f"  Project:   {project}")
+    console.print(f"  Press Ctrl+C to stop[/dim]\n")
+
+    handler = CodeChangeHandler(reindex_file, debounce_seconds=interval)
+
+    # Wrap as proper watchdog handler
+    class WatchdogAdapter(FileSystemEventHandler):
+        def on_modified(self, event):
+            handler.dispatch(event)
+
+        def on_created(self, event):
+            handler.dispatch(event)
+
+    observer = Observer()
+    observer.schedule(WatchdogAdapter(), str(watch_path), recursive=True)
+    observer.start()
+
+    try:
+        while True:
+            time.sleep(1)
+    except KeyboardInterrupt:
+        observer.stop()
+        console.print(f"\n[yellow]Stopped watching.[/yellow] Re-indexed {reindex_count} time(s).")
+
+    observer.join()