@musashishao/agent-kit 1.2.2 → 1.3.0

package/.agent/scripts/setup_host.py

@@ -0,0 +1,557 @@
+#!/usr/bin/env python3
+"""
+AI Host Auto-Configuration
+
+Automatically detects and configures AI hosts (Claude Desktop, Cursor, VS Code)
+to use the Agent Kit MCP Gateway.
+
+Usage:
+    python setup_host.py --project-root /path/to/project
+    python setup_host.py --list  # List detected hosts only
+"""
+
+import os
+import sys
+import json
+import shutil
+import platform
+import argparse
+import subprocess
+from pathlib import Path
+from typing import Dict, List, Optional, Tuple
+
+
+# ============================================================================
+# Constants
+# ============================================================================
+
+SUPPORTED_HOSTS = {
+    "claude_desktop": {
+        "name": "Claude Desktop",
+        "config_paths": {
+            "Darwin": "~/Library/Application Support/Claude/claude_desktop_config.json",
+            "Windows": "%APPDATA%/Claude/claude_desktop_config.json",
+            "Linux": "~/.config/Claude/claude_desktop_config.json",
+        },
+        "detection": {
+            "Darwin": ["/Applications/Claude.app"],
+            "Windows": ["%LOCALAPPDATA%/Programs/Claude/Claude.exe"],
+            "Linux": [],
+        },
+    },
+    "cursor": {
+        "name": "Cursor",
+        "config_paths": {
+            "Darwin": ".cursor/mcp.json",  # Project-local
+            "Windows": ".cursor/mcp.json",
+            "Linux": ".cursor/mcp.json",
+        },
+        "detection": {
+            "Darwin": ["/Applications/Cursor.app"],
+            "Windows": ["%LOCALAPPDATA%/Programs/Cursor/Cursor.exe"],
+            "Linux": ["/usr/bin/cursor", "~/.local/bin/cursor"],
+        },
+    },
+    "vscode": {
+        "name": "VS Code",
+        "config_paths": {
+            "Darwin": ".vscode/settings.json",  # Project-local
+            "Windows": ".vscode/settings.json",
+            "Linux": ".vscode/settings.json",
+        },
+        "detection": {
+            "Darwin": ["/Applications/Visual Studio Code.app"],
+            "Windows": ["%LOCALAPPDATA%/Programs/Microsoft VS Code/Code.exe"],
+            "Linux": ["/usr/bin/code", "/snap/bin/code"],
+        },
+    },
+    "windsurf": {
+        "name": "Windsurf",
+        "config_paths": {
+            "Darwin": ".windsurf/mcp.json",
+            "Windows": ".windsurf/mcp.json",
+            "Linux": ".windsurf/mcp.json",
+        },
+        "detection": {
+            "Darwin": ["/Applications/Windsurf.app"],
+            "Windows": [],
+            "Linux": [],
+        },
+    },
+}
+
+
+# ============================================================================
+# Utility Functions
+# ============================================================================
+
+def get_os() -> str:
+    """Get current operating system."""
+    return platform.system()
+
+
+def expand_path(path: str) -> Path:
+    """Expand environment variables and user home in path."""
+    expanded = os.path.expandvars(os.path.expanduser(path))
+    return Path(expanded)
+
+
+def is_host_installed(host_id: str) -> bool:
+    """Check if an AI host is installed."""
+    host = SUPPORTED_HOSTS.get(host_id)
+    if not host:
+        return False
+    
+    os_name = get_os()
+    detection_paths = host["detection"].get(os_name, [])
+    
+    for path in detection_paths:
+        expanded = expand_path(path)
+        if expanded.exists():
+            return True
+    
+    return False
+
+
+def detect_installed_hosts() -> List[str]:
+    """Detect which AI hosts are installed on this system."""
+    installed = []
+    for host_id in SUPPORTED_HOSTS:
+        if is_host_installed(host_id):
+            installed.append(host_id)
+    return installed
+
+
+def get_config_path(host_id: str, project_root: Path) -> Optional[Path]:
+    """Get the config file path for a host."""
+    host = SUPPORTED_HOSTS.get(host_id)
+    if not host:
+        return None
+    
+    os_name = get_os()
+    config_path = host["config_paths"].get(os_name)
+    if not config_path:
+        return None
+    
+    # Check if it's a project-local path (starts with .)
+    if config_path.startswith("."):
+        return project_root / config_path
+    else:
+        return expand_path(config_path)
+
+
+def backup_config(config_path: Path) -> Optional[Path]:
+    """Create a backup of existing config file."""
+    if not config_path.exists():
+        return None
+    
+    backup_path = config_path.with_suffix(config_path.suffix + ".backup")
+    shutil.copy2(config_path, backup_path)
+    return backup_path
+
+
+def load_json_config(path: Path) -> Dict:
+    """Load JSON config file, return empty dict if not exists."""
+    if not path.exists():
+        return {}
+    
+    try:
+        with open(path, "r", encoding="utf-8") as f:
+            return json.load(f)
+    except json.JSONDecodeError:
+        return {}
+
+
+def save_json_config(path: Path, config: Dict) -> bool:
+    """Save config to JSON file."""
+    try:
+        path.parent.mkdir(parents=True, exist_ok=True)
+        with open(path, "w", encoding="utf-8") as f:
+            json.dump(config, f, indent=2)
+        return True
+    except Exception as e:
+        print(f"Error saving config: {e}")
+        return False
+
+
+# ============================================================================
+# Configuration Functions
+# ============================================================================
+
+def generate_mcp_config(
+    project_root: Path,
+    mcp_server_path: Path,
+    kit_path: Path
+) -> Dict:
+    """Generate MCP server configuration."""
+    return {
+        "command": "node",
+        "args": [str(mcp_server_path)],
+        "env": {
+            "PROJECT_ROOT": str(project_root),
+            "AGENT_KIT_PATH": str(kit_path),
+        },
+    }
+
+
+def configure_claude_desktop(
+    project_root: Path,
+    mcp_server_path: Path,
+    kit_path: Path,
+    dry_run: bool = False
+) -> Tuple[bool, str]:
+    """Configure Claude Desktop to use Agent Kit MCP."""
+    config_path = get_config_path("claude_desktop", project_root)
+    if not config_path:
+        return False, "Could not determine Claude Desktop config path"
+    
+    # Load existing config
+    config = load_json_config(config_path)
+    
+    # Initialize mcpServers if not exists
+    if "mcpServers" not in config:
+        config["mcpServers"] = {}
+    
+    # Check if already configured
+    if "agent-kit" in config["mcpServers"]:
+        existing = config["mcpServers"]["agent-kit"]
+        if existing.get("env", {}).get("PROJECT_ROOT") == str(project_root):
+            return True, "Already configured for this project"
+    
+    # Add our server config
+    config["mcpServers"]["agent-kit"] = generate_mcp_config(
+        project_root, mcp_server_path, kit_path
+    )
+    
+    if dry_run:
+        return True, f"Would update: {config_path}"
+    
+    # Backup and save
+    backup = backup_config(config_path)
+    if backup:
+        print(f"  Backed up to: {backup}")
+    
+    if save_json_config(config_path, config):
+        return True, f"Configured: {config_path}"
+    else:
+        return False, "Failed to save config"
+
+
+def configure_cursor(
+    project_root: Path,
+    mcp_server_path: Path,
+    kit_path: Path,
+    dry_run: bool = False
+) -> Tuple[bool, str]:
+    """Configure Cursor to use Agent Kit MCP (project-local)."""
+    config_path = get_config_path("cursor", project_root)
+    if not config_path:
+        return False, "Could not determine Cursor config path"
+    
+    # For Cursor, use relative paths since it's project-local
+    relative_mcp_path = ".agent/mcp-gateway/dist/index.js"
+    
+    config = load_json_config(config_path)
+    
+    if "mcpServers" not in config:
+        config["mcpServers"] = {}
+    
+    config["mcpServers"]["agent-kit"] = {
+        "command": "node",
+        "args": [relative_mcp_path],
+        "env": {
+            "PROJECT_ROOT": ".",
+        },
+    }
+    
+    if dry_run:
+        return True, f"Would create: {config_path}"
+    
+    if save_json_config(config_path, config):
+        return True, f"Created: {config_path}"
+    else:
+        return False, "Failed to save config"
+
+
+def configure_vscode(
+    project_root: Path,
+    mcp_server_path: Path,
+    kit_path: Path,
+    dry_run: bool = False
+) -> Tuple[bool, str]:
+    """Configure VS Code settings for MCP (if MCP extension is installed)."""
+    config_path = get_config_path("vscode", project_root)
+    if not config_path:
+        return False, "Could not determine VS Code config path"
+    
+    config = load_json_config(config_path)
+    
+    # Add MCP config (format may vary by extension)
+    if "mcp.servers" not in config:
+        config["mcp.servers"] = {}
+    
+    config["mcp.servers"]["agent-kit"] = {
+        "command": "node",
+        "args": ["${workspaceFolder}/.agent/mcp-gateway/dist/index.js"],
+        "env": {
+            "PROJECT_ROOT": "${workspaceFolder}",
+        },
+    }
+    
+    if dry_run:
+        return True, f"Would update: {config_path}"
+    
+    if save_json_config(config_path, config):
+        return True, f"Updated: {config_path}"
+    else:
+        return False, "Failed to save config"
+
+
+def configure_windsurf(
+    project_root: Path,
+    mcp_server_path: Path,
+    kit_path: Path,
+    dry_run: bool = False
+) -> Tuple[bool, str]:
+    """Configure Windsurf to use Agent Kit MCP (project-local)."""
+    config_path = get_config_path("windsurf", project_root)
+    if not config_path:
+        return False, "Could not determine Windsurf config path"
+    
+    relative_mcp_path = ".agent/mcp-gateway/dist/index.js"
+    
+    config = load_json_config(config_path)
+    
+    if "mcpServers" not in config:
+        config["mcpServers"] = {}
+    
+    config["mcpServers"]["agent-kit"] = {
+        "command": "node",
+        "args": [relative_mcp_path],
+        "env": {
+            "PROJECT_ROOT": ".",
+        },
+    }
+    
+    if dry_run:
+        return True, f"Would create: {config_path}"
+    
+    if save_json_config(config_path, config):
+        return True, f"Created: {config_path}"
+    else:
+        return False, "Failed to save config"
+
+
+# ============================================================================
+# Clipboard Helper
+# ============================================================================
+
+def copy_to_clipboard(text: str) -> bool:
+    """Copy text to system clipboard."""
+    os_name = get_os()
+    
+    try:
+        if os_name == "Darwin":
+            process = subprocess.Popen(
+                ["pbcopy"],
+                stdin=subprocess.PIPE,
+            )
+            process.communicate(text.encode("utf-8"))
+            return process.returncode == 0
+        elif os_name == "Windows":
+            process = subprocess.Popen(
+                ["clip"],
+                stdin=subprocess.PIPE,
+            )
+            process.communicate(text.encode("utf-16"))
+            return process.returncode == 0
+        elif os_name == "Linux":
+            # Try xclip first, then xsel
+            for cmd in [["xclip", "-selection", "clipboard"], ["xsel", "--clipboard", "--input"]]:
+                try:
+                    process = subprocess.Popen(cmd, stdin=subprocess.PIPE)
+                    process.communicate(text.encode("utf-8"))
+                    if process.returncode == 0:
+                        return True
+                except FileNotFoundError:
+                    continue
+            return False
+    except Exception:
+        return False
+    
+    return False
+
+
+# ============================================================================
+# Main Setup Function
+# ============================================================================
+
+def setup_hosts(
+    project_root: Path,
+    hosts: Optional[List[str]] = None,
+    dry_run: bool = False,
+    force: bool = False
+) -> Dict[str, Tuple[bool, str]]:
+    """
+    Configure all detected AI hosts.
+    
+    Args:
+        project_root: Path to the project root
+        hosts: Specific hosts to configure (None = all detected)
+        dry_run: If True, only show what would be done
+        force: If True, reconfigure even if already set up
+    
+    Returns:
+        Dict mapping host_id to (success, message)
+    """
+    results = {}
+    
+    # Determine MCP server path
+    mcp_server_path = project_root / ".agent" / "mcp-gateway" / "dist" / "index.js"
+    kit_path = project_root / ".agent"
+    
+    # Check if MCP server is built
+    if not mcp_server_path.exists():
+        print("⚠️  MCP Gateway not built. Building now...")
+        build_result = subprocess.run(
+            ["npm", "run", "build"],
+            cwd=str(project_root / ".agent" / "mcp-gateway"),
+            capture_output=True,
+            text=True,
+        )
+        if build_result.returncode != 0:
+            return {"error": (False, "Failed to build MCP Gateway")}
+    
+    # Determine which hosts to configure
+    if hosts:
+        target_hosts = hosts
+    else:
+        target_hosts = detect_installed_hosts()
+        # Always include project-local configs
+        target_hosts.extend(["cursor", "vscode", "windsurf"])
+        target_hosts = list(set(target_hosts))
+    
+    # Configure each host
+    configurators = {
+        "claude_desktop": configure_claude_desktop,
+        "cursor": configure_cursor,
+        "vscode": configure_vscode,
+        "windsurf": configure_windsurf,
+    }
+    
+    for host_id in target_hosts:
+        if host_id not in configurators:
+            results[host_id] = (False, f"Unknown host: {host_id}")
+            continue
+        
+        print(f"\n🔧 Configuring {SUPPORTED_HOSTS[host_id]['name']}...")
+        success, message = configurators[host_id](
+            project_root, mcp_server_path, kit_path, dry_run
+        )
+        results[host_id] = (success, message)
+        
+        status = "✅" if success else "❌"
+        print(f"  {status} {message}")
+    
+    return results
+
+
+# ============================================================================
+# CLI Entry Point
+# ============================================================================
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Auto-configure AI hosts for Agent Kit MCP"
+    )
+    parser.add_argument(
+        "--project-root",
+        default=".",
+        help="Project root directory",
+    )
+    parser.add_argument(
+        "--list",
+        action="store_true",
+        help="List detected AI hosts only",
+    )
+    parser.add_argument(
+        "--dry-run",
+        action="store_true",
+        help="Show what would be done without making changes",
+    )
+    parser.add_argument(
+        "--hosts",
+        nargs="+",
+        choices=list(SUPPORTED_HOSTS.keys()),
+        help="Specific hosts to configure",
+    )
+    parser.add_argument(
+        "--copy-config",
+        action="store_true",
+        help="Copy config to clipboard for manual setup",
+    )
+    
+    args = parser.parse_args()
+    project_root = Path(args.project_root).resolve()
+    
+    print("🤖 Agent Kit - AI Host Auto-Configuration")
+    print(f"📁 Project: {project_root}")
+    print(f"💻 OS: {get_os()}")
+    
+    # List mode
+    if args.list:
+        print("\n📋 Detected AI Hosts:")
+        installed = detect_installed_hosts()
+        for host_id, host in SUPPORTED_HOSTS.items():
+            status = "✅ Installed" if host_id in installed else "❌ Not found"
+            print(f"  - {host['name']}: {status}")
+        return 0
+    
+    # Copy config to clipboard mode
+    if args.copy_config:
+        mcp_server_path = project_root / ".agent" / "mcp-gateway" / "dist" / "index.js"
+        config = {
+            "mcpServers": {
+                "agent-kit": {
+                    "command": "node",
+                    "args": [str(mcp_server_path)],
+                    "env": {
+                        "PROJECT_ROOT": str(project_root),
+                    },
+                }
+            }
+        }
+        config_json = json.dumps(config, indent=2)
+        
+        if copy_to_clipboard(config_json):
+            print("\n✅ Config copied to clipboard!")
+            print("\n📋 Paste this into your AI host's MCP configuration:")
+            print(config_json)
+        else:
+            print("\n⚠️  Could not copy to clipboard. Here's the config:")
+            print(config_json)
+        return 0
+    
+    # Normal configuration mode
+    results = setup_hosts(
+        project_root,
+        hosts=args.hosts,
+        dry_run=args.dry_run,
+    )
+    
+    # Summary
+    print("\n" + "=" * 50)
+    print("📊 Configuration Summary:")
+    success_count = sum(1 for success, _ in results.values() if success)
+    total_count = len(results)
+    print(f"  ✅ {success_count}/{total_count} hosts configured successfully")
+    
+    if success_count > 0:
+        print("\n🎉 Setup complete! Your AI hosts are now connected to Agent Kit.")
+        print("   Restart your AI host to apply changes.")
+    
+    return 0 if success_count == total_count else 1
+
+
+if __name__ == "__main__":
+    sys.exit(main())