mcp-vector-search 0.0.3__py3-none-any.whl → 0.4.11__py3-none-any.whl

mcp_vector_search/cli/commands/status.py

@@ -1,8 +1,10 @@
 """Status command for MCP Vector Search CLI."""
 
 import asyncio
+import json
+import subprocess
 from pathlib import Path
-from typing import Dict, Any
+from typing import Any
 
 import typer
 from loguru import logger
@@ -14,13 +16,10 @@ from ...core.indexer import SemanticIndexer
 from ...core.project import ProjectManager
 from ..output import (
     console,
-    print_config,
     print_dependency_status,
     print_error,
-    print_index_stats,
     print_info,
     print_json,
-    print_project_info,
 )
 
 # Create status subcommand app
@@ -30,6 +29,16 @@ status_app = typer.Typer(help="Show project status and statistics")
 @status_app.command()
 def main(
     ctx: typer.Context,
+    project_root: Path | None = typer.Option(
+        None,
+        "--project-root",
+        "-p",
+        help="Project root directory (auto-detected if not specified)",
+        exists=True,
+        file_okay=False,
+        dir_okay=True,
+        readable=True,
+    ),
     verbose: bool = typer.Option(
         False,
         "--verbose",
@@ -41,6 +50,11 @@ def main(
         "--health-check",
         help="Perform health check of all components",
     ),
+    mcp: bool = typer.Option(
+        False,
+        "--mcp",
+        help="Check Claude Code MCP integration status",
+    ),
     json_output: bool = typer.Option(
         False,
         "--json",
@@ -48,25 +62,31 @@ def main(
     ),
 ) -> None:
     """Show project status and indexing statistics.
-    
+
     This command displays comprehensive information about your MCP Vector Search
     project including configuration, indexing status, and system health.
-    
+
     Examples:
         mcp-vector-search status
         mcp-vector-search status --verbose
         mcp-vector-search status --health-check --json
+        mcp-vector-search status --mcp
     """
     try:
-        project_root = ctx.obj.get("project_root") or Path.cwd()
-        
-        asyncio.run(show_status(
-            project_root=project_root,
-            verbose=verbose,
-            health_check=health_check,
-            json_output=json_output,
-        ))
-        
+        # Use provided project_root or current working directory
+        if project_root is None:
+            project_root = Path.cwd()
+
+        asyncio.run(
+            show_status(
+                project_root=project_root,
+                verbose=verbose,
+                health_check=health_check,
+                mcp=mcp,
+                json_output=json_output,
+            )
+        )
+
     except Exception as e:
         logger.error(f"Status check failed: {e}")
         print_error(f"Status check failed: {e}")
@@ -77,49 +97,50 @@ async def show_status(
     project_root: Path,
     verbose: bool = False,
     health_check: bool = False,
+    mcp: bool = False,
     json_output: bool = False,
 ) -> None:
     """Show comprehensive project status."""
     status_data = {}
-    
+
     try:
-        # Check if project is initialized
+        # Check if project is initialized - use the specified project root
         project_manager = ProjectManager(project_root)
-        
+
         if not project_manager.is_initialized():
             if json_output:
                 status_data = {
                     "initialized": False,
                     "project_root": str(project_root),
-                    "error": "Project not initialized"
+                    "error": "Project not initialized",
                 }
                 print_json(status_data)
             else:
                 print_error(f"Project not initialized at {project_root}")
                 print_info("Run 'mcp-vector-search init' to initialize the project")
             return
-        
+
         # Get project information
         project_info = project_manager.get_project_info()
         config = project_manager.load_config()
-        
+
         # Get indexing statistics
         embedding_function, _ = create_embedding_function(config.embedding_model)
         database = ChromaVectorDatabase(
             persist_directory=config.index_path,
             embedding_function=embedding_function,
         )
-        
+
         indexer = SemanticIndexer(
             database=database,
             project_root=project_root,
             file_extensions=config.file_extensions,
         )
-        
+
         async with database:
             index_stats = await indexer.get_indexing_stats()
             db_stats = await database.get_stats()
-        
+
         # Compile status data
         status_data = {
             "project": {
@@ -146,12 +167,17 @@ async def show_status(
                 "last_updated": db_stats.last_updated,
             },
         }
-        
+
         # Add health check if requested
         if health_check:
             health_status = await perform_health_check(project_root, config)
             status_data["health"] = health_status
-        
+
+        # Add MCP integration check if requested
+        if mcp:
+            mcp_status = await check_mcp_integration(project_root)
+            status_data["mcp"] = mcp_status
+
         # Add verbose information
         if verbose:
             status_data["verbose"] = {
@@ -160,13 +186,13 @@ async def show_status(
                 "ignore_patterns": list(indexer.get_ignore_patterns()),
                 "parser_info": index_stats.get("parser_info", {}),
             }
-        
+
         # Output results
         if json_output:
             print_json(status_data)
         else:
-            _display_status(status_data, verbose)
-            
+            _display_status(status_data, verbose, mcp)
+
     except ProjectNotFoundError:
         if json_output:
             print_json({"initialized": False, "error": "Project not initialized"})
@@ -181,45 +207,51 @@ async def show_status(
         raise
 
 
-def _display_status(status_data: Dict[str, Any], verbose: bool) -> None:
+def _display_status(status_data: dict[str, Any], verbose: bool, mcp: bool = False) -> None:
     """Display status in human-readable format."""
     project_data = status_data["project"]
     config_data = status_data["configuration"]
     index_data = status_data["index"]
-    
+
     # Project information
     console.print("[bold blue]Project Information[/bold blue]")
     console.print(f"  Name: {project_data['name']}")
     console.print(f"  Root: {project_data['root_path']}")
-    console.print(f"  Languages: {', '.join(project_data['languages']) if project_data['languages'] else 'None detected'}")
+    console.print(
+        f"  Languages: {', '.join(project_data['languages']) if project_data['languages'] else 'None detected'}"
+    )
     console.print(f"  Indexable Files: {project_data['file_count']}")
     console.print()
-    
+
     # Configuration
     console.print("[bold blue]Configuration[/bold blue]")
     console.print(f"  Embedding Model: {config_data['embedding_model']}")
     console.print(f"  Similarity Threshold: {config_data['similarity_threshold']}")
     console.print(f"  File Extensions: {', '.join(config_data['file_extensions'])}")
-    console.print(f"  Cache Embeddings: {'✓' if config_data['cache_embeddings'] else '✗'}")
+    console.print(
+        f"  Cache Embeddings: {'✓' if config_data['cache_embeddings'] else '✗'}"
+    )
     console.print()
-    
+
     # Index statistics
     console.print("[bold blue]Index Statistics[/bold blue]")
-    console.print(f"  Indexed Files: {index_data['indexed_files']}/{index_data['total_files']}")
+    console.print(
+        f"  Indexed Files: {index_data['indexed_files']}/{index_data['total_files']}"
+    )
     console.print(f"  Total Chunks: {index_data['total_chunks']}")
     console.print(f"  Index Size: {index_data['index_size_mb']:.2f} MB")
-    
-    if index_data['languages']:
+
+    if index_data["languages"]:
         console.print("  Language Distribution:")
-        for lang, count in index_data['languages'].items():
+        for lang, count in index_data["languages"].items():
             console.print(f"    {lang}: {count} chunks")
     console.print()
-    
+
     # Health check results
     if "health" in status_data:
         health_data = status_data["health"]
         console.print("[bold blue]Health Check[/bold blue]")
-        
+
         overall_health = health_data.get("overall", "unknown")
         if overall_health == "healthy":
             console.print("[green]✓ System is healthy[/green]")
@@ -227,7 +259,7 @@ def _display_status(status_data: Dict[str, Any], verbose: bool) -> None:
             console.print("[yellow]⚠ System has warnings[/yellow]")
         else:
             console.print("[red]✗ System has issues[/red]")
-        
+
         for component, status in health_data.get("components", {}).items():
             if status == "ok":
                 console.print(f"  [green]✓[/green] {component}")
@@ -236,42 +268,69 @@ def _display_status(status_data: Dict[str, Any], verbose: bool) -> None:
             else:
                 console.print(f"  [red]✗[/red] {component}")
         console.print()
-    
+
+    # MCP integration status
+    if "mcp" in status_data:
+        mcp_data = status_data["mcp"]
+        console.print("[bold blue]MCP Integration[/bold blue]")
+
+        if mcp_data.get("claude_available"):
+            console.print("[green]✓[/green] Claude Code: Available")
+        else:
+            console.print("[red]✗[/red] Claude Code: Not available")
+
+        server_status = mcp_data.get("server_status", "unknown")
+        server_name = mcp_data.get("server_name", "mcp-vector-search")
+
+        if server_status == "installed":
+            console.print(f"[green]✓[/green] MCP Server '{server_name}': Installed")
+        elif server_status == "not_installed":
+            console.print(f"[red]✗[/red] MCP Server '{server_name}': Not installed")
+        else:
+            console.print(f"[yellow]⚠[/yellow] MCP Server '{server_name}': {server_status}")
+
+        if mcp_data.get("project_config"):
+            console.print("[green]✓[/green] Project Configuration: Found")
+        else:
+            console.print("[red]✗[/red] Project Configuration: Missing")
+
+        console.print()
+
     # Verbose information
     if verbose and "verbose" in status_data:
         verbose_data = status_data["verbose"]
         console.print("[bold blue]Detailed Information[/bold blue]")
         console.print(f"  Config Path: {verbose_data['config_path']}")
         console.print(f"  Index Path: {verbose_data['index_path']}")
-        console.print(f"  Ignore Patterns: {', '.join(verbose_data['ignore_patterns'])}")
+        console.print(
+            f"  Ignore Patterns: {', '.join(verbose_data['ignore_patterns'])}"
+        )
 
 
-async def perform_health_check(project_root: Path, config) -> Dict[str, Any]:
+async def perform_health_check(project_root: Path, config) -> dict[str, Any]:
     """Perform comprehensive health check."""
     health_status = {
         "overall": "healthy",
         "components": {},
         "issues": [],
     }
-    
+
     try:
         # Check dependencies
         deps_ok = check_dependencies()
         health_status["components"]["dependencies"] = "ok" if deps_ok else "error"
         if not deps_ok:
             health_status["issues"].append("Missing dependencies")
-        
+
         # Check configuration
-        config_ok = True
         try:
             # Validate embedding model
             embedding_function, _ = create_embedding_function(config.embedding_model)
             health_status["components"]["embedding_model"] = "ok"
         except Exception as e:
-            config_ok = False
             health_status["components"]["embedding_model"] = "error"
             health_status["issues"].append(f"Embedding model error: {e}")
-        
+
         # Check database
         try:
             database = ChromaVectorDatabase(
@@ -284,7 +343,7 @@ async def perform_health_check(project_root: Path, config) -> Dict[str, Any]:
         except Exception as e:
             health_status["components"]["database"] = "error"
             health_status["issues"].append(f"Database error: {e}")
-        
+
         # Check file system permissions
         try:
             config.index_path.mkdir(parents=True, exist_ok=True)
@@ -295,20 +354,87 @@ async def perform_health_check(project_root: Path, config) -> Dict[str, Any]:
         except Exception as e:
             health_status["components"]["file_permissions"] = "error"
             health_status["issues"].append(f"File permission error: {e}")
-        
+
         # Determine overall health
         if any(status == "error" for status in health_status["components"].values()):
             health_status["overall"] = "error"
-        elif any(status == "warning" for status in health_status["components"].values()):
+        elif any(
+            status == "warning" for status in health_status["components"].values()
+        ):
             health_status["overall"] = "warning"
-        
+
     except Exception as e:
         health_status["overall"] = "error"
         health_status["issues"].append(f"Health check failed: {e}")
-    
+
     return health_status
 
 
+async def check_mcp_integration(project_root: Path, server_name: str = "mcp-vector-search") -> dict[str, Any]:
+    """Check MCP integration status."""
+    mcp_status = {
+        "claude_available": False,
+        "server_status": "unknown",
+        "server_name": server_name,
+        "project_config": False,
+        "issues": [],
+    }
+
+    try:
+        # Import MCP functions from the mcp command module
+        from .mcp import check_claude_code_available, get_claude_command
+
+        # Check if Claude Code is available
+        mcp_status["claude_available"] = check_claude_code_available()
+
+        if mcp_status["claude_available"]:
+            claude_cmd = get_claude_command()
+
+            # Check if MCP server is installed
+            try:
+                result = subprocess.run(
+                    [claude_cmd, "mcp", "get", server_name],
+                    capture_output=True,
+                    text=True,
+                    timeout=10
+                )
+
+                if result.returncode == 0:
+                    mcp_status["server_status"] = "installed"
+                else:
+                    mcp_status["server_status"] = "not_installed"
+                    mcp_status["issues"].append(f"MCP server '{server_name}' not found in Claude Code")
+
+            except subprocess.TimeoutExpired:
+                mcp_status["server_status"] = "timeout"
+                mcp_status["issues"].append("Timeout checking MCP server status")
+            except Exception as e:
+                mcp_status["server_status"] = "error"
+                mcp_status["issues"].append(f"Error checking MCP server: {e}")
+        else:
+            mcp_status["issues"].append("Claude Code not available")
+
+        # Check for project-level .claude.json configuration
+        claude_json_path = project_root / ".claude.json"
+        if claude_json_path.exists():
+            try:
+                with open(claude_json_path, 'r') as f:
+                    config = json.load(f)
+                if config.get("mcpServers", {}).get(server_name):
+                    mcp_status["project_config"] = True
+                else:
+                    mcp_status["issues"].append(f"MCP server '{server_name}' not found in project .claude.json")
+            except Exception as e:
+                mcp_status["issues"].append(f"Error reading project .claude.json: {e}")
+        else:
+            mcp_status["issues"].append("Project .claude.json not found")
+
+    except Exception as e:
+        mcp_status["issues"].append(f"MCP integration check failed: {e}")
+
+    return mcp_status
+
+
 def check_dependencies() -> bool:
     """Check if all required dependencies are available."""
     dependencies = [
@@ -322,9 +448,9 @@ def check_dependencies() -> bool:
         ("watchdog", "Watchdog"),
         ("loguru", "Loguru"),
     ]
-    
+
     all_available = True
-    
+
     for module_name, display_name in dependencies:
         try:
             __import__(module_name)
@@ -332,7 +458,7 @@ def check_dependencies() -> bool:
         except ImportError:
             print_dependency_status(display_name, False)
             all_available = False
-    
+
     return all_available
 
 

mcp_vector_search/cli/commands/watch.py

@@ -4,7 +4,6 @@ import asyncio
 import signal
 import sys
 from pathlib import Path
-from typing import Optional
 
 import typer
 from loguru import logger
@@ -36,7 +35,7 @@ def watch_main(
         dir_okay=True,
         readable=True,
     ),
-    config: Optional[Path] = typer.Option(
+    config: Path | None = typer.Option(
         None,
         "--config",
         "-c",
@@ -54,19 +53,19 @@ def watch_main(
     ),
 ) -> None:
     """Watch for file changes and automatically update the search index.
-    
+
     This command starts a file watcher that monitors your project directory
     for changes to code files. When files are created, modified, or deleted,
     the search index is automatically updated to reflect the changes.
-    
+
     The watcher will:
     - Monitor all files with configured extensions
     - Debounce rapid changes to avoid excessive indexing
     - Update the index incrementally for better performance
     - Ignore common build/cache directories
-    
+
     Press Ctrl+C to stop watching.
-    
+
     Examples:
         mcp-vector-search watch
         mcp-vector-search watch /path/to/project --verbose
@@ -75,7 +74,7 @@ def watch_main(
     if verbose:
         logger.remove()
         logger.add(sys.stderr, level="DEBUG")
-    
+
     try:
         asyncio.run(_watch_async(project_root, config))
     except KeyboardInterrupt:
@@ -85,7 +84,7 @@ def watch_main(
         raise typer.Exit(1)
 
 
-async def _watch_async(project_root: Path, config_path: Optional[Path]) -> None:
+async def _watch_async(project_root: Path, config_path: Path | None) -> None:
     """Async implementation of watch command."""
     # Load project configuration
     try:
@@ -96,17 +95,17 @@ async def _watch_async(project_root: Path, config_path: Optional[Path]) -> None:
                 "Run 'mcp-vector-search init' first."
             )
             raise typer.Exit(1)
-        
+
         config = project_manager.load_config()
         print_info(f"Loaded configuration from {project_root}")
-        
+
     except ProjectNotFoundError:
         print_error(
             f"No MCP Vector Search project found at {project_root}. "
             "Run 'mcp-vector-search init' to initialize."
         )
         raise typer.Exit(1)
-    
+
     # Setup database and indexer
     try:
         embedding_function, _ = create_embedding_function(config.embedding_model)
@@ -114,19 +113,19 @@ async def _watch_async(project_root: Path, config_path: Optional[Path]) -> None:
             persist_directory=config.index_path,
             embedding_function=embedding_function,
         )
-        
+
         indexer = SemanticIndexer(
             database=database,
             project_root=project_root,
             file_extensions=config.file_extensions,
         )
-        
+
         print_info(f"Initialized database at {config.index_path}")
-        
+
     except Exception as e:
         print_error(f"Failed to initialize database: {e}")
         raise typer.Exit(1)
-    
+
     # Start watching
     try:
         async with database:
@@ -136,34 +135,34 @@ async def _watch_async(project_root: Path, config_path: Optional[Path]) -> None:
                 indexer=indexer,
                 database=database,
             )
-            
+
             print_success("🔍 Starting file watcher...")
             print_info(f"📁 Watching: {project_root}")
             print_info(f"📄 Extensions: {', '.join(config.file_extensions)}")
             print_info("Press Ctrl+C to stop watching")
-            
+
             async with watcher:
                 # Set up signal handlers for graceful shutdown
                 stop_event = asyncio.Event()
-                
+
                 def signal_handler():
                     print_info("\n⏹️  Stopping file watcher...")
                     stop_event.set()
-                
+
                 # Handle SIGINT (Ctrl+C) and SIGTERM
                 if sys.platform != "win32":
                     loop = asyncio.get_running_loop()
                     for sig in (signal.SIGINT, signal.SIGTERM):
                         loop.add_signal_handler(sig, signal_handler)
-                
+
                 try:
                     # Wait for stop signal
                     await stop_event.wait()
                 except KeyboardInterrupt:
                     signal_handler()
-                
+
                 print_success("✅ File watcher stopped")
-                
+
     except Exception as e:
         print_error(f"File watching failed: {e}")
         raise typer.Exit(1)
@@ -181,7 +180,7 @@ def watch_status(
     ),
 ) -> None:
     """Check if file watching is enabled for a project.
-    
+
     This command checks the project configuration to see if file watching
     is enabled and provides information about the watch settings.
     """
@@ -190,20 +189,20 @@ def watch_status(
         if not project_manager.is_initialized():
             print_error(f"Project not initialized at {project_root}")
             raise typer.Exit(1)
-        
+
         config = project_manager.load_config()
-        
+
         console.print("\n[bold]File Watch Status[/bold]")
         console.print(f"Project: {project_root}")
         console.print(f"Watch Files: {'✓' if config.watch_files else '✗'}")
-        
+
         if config.watch_files:
             console.print(f"Extensions: {', '.join(config.file_extensions)}")
             print_info("File watching is enabled for this project")
         else:
             print_warning("File watching is disabled for this project")
             print_info("Enable with: mcp-vector-search config set watch_files true")
-            
+
     except ProjectNotFoundError:
         print_error(f"No MCP Vector Search project found at {project_root}")
         raise typer.Exit(1)
@@ -224,7 +223,7 @@ def watch_enable(
     ),
 ) -> None:
     """Enable file watching for a project.
-    
+
     This command enables the watch_files setting in the project configuration.
     After enabling, you can use 'mcp-vector-search watch' to start monitoring.
     """
@@ -233,14 +232,14 @@ def watch_enable(
         if not project_manager.is_initialized():
             print_error(f"Project not initialized at {project_root}")
             raise typer.Exit(1)
-        
+
         config = project_manager.load_config()
         config.watch_files = True
         project_manager.save_config(config)
-        
+
         print_success("✅ File watching enabled")
         print_info("Start watching with: mcp-vector-search watch")
-        
+
     except ProjectNotFoundError:
         print_error(f"No MCP Vector Search project found at {project_root}")
         raise typer.Exit(1)
@@ -261,7 +260,7 @@ def watch_disable(
     ),
 ) -> None:
     """Disable file watching for a project.
-    
+
     This command disables the watch_files setting in the project configuration.
     """
     try:
@@ -269,13 +268,13 @@ def watch_disable(
         if not project_manager.is_initialized():
             print_error(f"Project not initialized at {project_root}")
             raise typer.Exit(1)
-        
+
         config = project_manager.load_config()
         config.watch_files = False
         project_manager.save_config(config)
-        
+
         print_success("✅ File watching disabled")
-        
+
     except ProjectNotFoundError:
         print_error(f"No MCP Vector Search project found at {project_root}")
         raise typer.Exit(1)