install.py

#!/usr/bin/env python3
"""Control Tower Lite — One-command MCP server registration.

Usage:
  python install.py           # Install for current user
  python install.py --global  # Install globally
  python install.py --remove  # Remove registration
"""

import json
import os
import sys
from pathlib import Path

MCP_SERVER_NAME = "control-tower-lite"
CLAUDE_CONFIG = Path.home() / ".claude.json"
SCRIPT_DIR = Path(__file__).parent.resolve()

def get_config():
    """Read existing claude.json or return empty structure."""
    if CLAUDE_CONFIG.exists():
        try:
            return json.loads(CLAUDE_CONFIG.read_text())
        except json.JSONDecodeError:
            return {}
    return {}

def save_config(config):
    """Write claude.json."""
    CLAUDE_CONFIG.write_text(json.dumps(config, indent=2))

def install(global_install=False):
    """Register the MCP server in claude.json."""
    config = get_config()
    if "mcpServers" not in config:
        config["mcpServers"] = {}

    python_path = sys.executable
    server_path = str(SCRIPT_DIR / "mcp_server.py")

    config["mcpServers"][MCP_SERVER_NAME] = {
        "command": python_path,
        "args": [server_path],
        "env": {}
    }

    save_config(config)
    print(f"✓ Registered '{MCP_SERVER_NAME}' in {CLAUDE_CONFIG}")
    print(f"  Python: {python_path}")
    print(f"  Server: {server_path}")
    print()
    print("Next steps:")
    print("  1. Start the broker:  python launch.py")
    print("  2. Open Claude Code in a new terminal")
    print("  3. Use the 'join' tool to register as architect or worker")

def remove():
    """Remove MCP server registration."""
    config = get_config()
    if "mcpServers" in config and MCP_SERVER_NAME in config["mcpServers"]:
        del config["mcpServers"][MCP_SERVER_NAME]
        save_config(config)
        print(f"✓ Removed '{MCP_SERVER_NAME}' from {CLAUDE_CONFIG}")
    else:
        print(f"'{MCP_SERVER_NAME}' not found in {CLAUDE_CONFIG}")

def main():
    args = sys.argv[1:]
    if "--remove" in args:
        remove()
    elif "--help" in args or "-h" in args:
        print(__doc__)
    else:
        install(global_install="--global" in args)

if __name__ == "__main__":
    main()