| """ |
| `dw doctor` command. |
| Performs system health checks, MCP discovery, and environment diagnostics. |
| """ |
| from __future__ import annotations |
|
|
| import sys |
| from pathlib import Path |
| import typer |
| from rich.table import Table |
| from rich.panel import Panel |
|
|
| from diffweave.cli.config import load_config, get_config_path |
| from diffweave.cli.output import print_success, print_error, print_warning, print_json, console |
| from diffweave.mcp.client import DiffWeaveMCPClient |
|
|
| REQUIRED_TOOLS = [ |
| "list_workspaces", |
| "create_workspace", |
| "upload_document", |
| "list_documents", |
| "get_workflow_status", |
| "list_workflows", |
| "list_pending_proposals", |
| "approve_proposal", |
| "reject_proposal", |
| "archive_proposal", |
| "restore_proposal", |
| "batch_review_proposals", |
| "get_semantic_diff", |
| "validate_proposals", |
| "get_knowledge_graph", |
| "list_knowledge", |
| "get_knowledge_item", |
| "search_knowledge", |
| "list_rules", |
| "create_rule", |
| "enable_rule", |
| "disable_rule", |
| "delete_rule", |
| "get_dashboard_stats", |
| "get_activity_feed", |
| ] |
|
|
|
|
| def doctor_command( |
| as_json: bool = typer.Option(False, "--json", help="Output machine-readable JSON"), |
| ): |
| """ |
| Run diagnostic checks on DiffWeave environment, MCP server, and workspace. |
| """ |
| checks = [] |
|
|
| |
| config = load_config() |
| cfg_path = get_config_path() |
| if config and config.get("workspace_id"): |
| checks.append({ |
| "check": "Local Workspace Configuration (.diffweave)", |
| "status": "PASS", |
| "details": f"Bound to '{config.get('workspace_name')}' ({config.get('workspace_id')[:8]}...)", |
| }) |
| else: |
| checks.append({ |
| "check": "Local Workspace Configuration (.diffweave)", |
| "status": "WARN", |
| "details": "No local workspace initialized. Run 'dw init' to bind one.", |
| }) |
|
|
| |
| client = DiffWeaveMCPClient() |
| tools = [] |
| try: |
| backend_str = str(client.backend_dir) |
| if backend_str not in sys.path: |
| sys.path.insert(0, backend_str) |
| from mcp_server import list_tools |
| import asyncio |
| discovered_tools = asyncio.run(list_tools()) |
| tools = [t.name for t in discovered_tools] |
| checks.append({ |
| "check": "DocWeave MCP Server Connectivity", |
| "status": "PASS", |
| "details": f"Successfully connected. Found {len(tools)} tools registered.", |
| }) |
| except Exception as e: |
| checks.append({ |
| "check": "DocWeave MCP Server Connectivity", |
| "status": "FAIL", |
| "details": f"Failed to connect or import server: {e}", |
| }) |
|
|
| |
| if tools: |
| missing = [t for t in REQUIRED_TOOLS if t not in tools] |
| if not missing: |
| checks.append({ |
| "check": "Required MCP Tools Coverage", |
| "status": "PASS", |
| "details": f"All {len(REQUIRED_TOOLS)} core DiffWeave operations are available.", |
| }) |
| else: |
| checks.append({ |
| "check": "Required MCP Tools Coverage", |
| "status": "FAIL", |
| "details": f"Missing tools: {', '.join(missing)}", |
| }) |
|
|
| |
| if config.get("workspace_id") and tools: |
| try: |
| stats = client.call_tool_sync("get_dashboard_stats", {"workspace_id": config.get("workspace_id")}) |
| checks.append({ |
| "check": "Workspace State & Database Verification", |
| "status": "PASS", |
| "details": f"Workspace verified. Documents: {stats.get('total_documents')}, Knowledge: {stats.get('knowledge_items')}", |
| }) |
| except Exception as e: |
| checks.append({ |
| "check": "Workspace State & Database Verification", |
| "status": "FAIL", |
| "details": f"Failed to query workspace state: {e}", |
| }) |
|
|
| all_passed = all(c["status"] == "PASS" for c in checks) |
|
|
| if as_json: |
| print_json({"healthy": all_passed, "checks": checks}) |
| return |
|
|
| table = Table(title="DiffWeave Doctor Diagnostic Report", header_style="bold blue") |
| table.add_column("System Subsystem", style="bold white") |
| table.add_column("Status", style="bold") |
| table.add_column("Details", style="white") |
|
|
| for c in checks: |
| st = c["status"] |
| if st == "PASS": |
| st_text = "[green][PASS][/green]" |
| elif st == "WARN": |
| st_text = "[yellow][WARN][/yellow]" |
| else: |
| st_text = "[bold red][FAIL][/bold red]" |
| table.add_row(c["check"], st_text, c["details"]) |
|
|
| console.print(table) |
| if all_passed: |
| console.print("\n[bold green][OK] All systems nominal. DiffWeave is fully operational![/bold green]") |
| else: |
| console.print("\n[bold yellow][!] Some checks require attention. See details above.[/bold yellow]") |
|
|
