MCP Claude Code

"""Command-line interface for the MCP Claude Code server.""" import argparse import json import os import sys from pathlib import Path from typing import Any, cast from mcp_claude_code.server import ClaudeCodeServer def main() -> None: """Run the CLI for the MCP Claude Code server.""" parser = argparse.ArgumentParser( description="MCP server implementing Claude Code capabilities" ) _ = parser.add_argument( "--transport", choices=["stdio", "sse"], default="stdio", help="Transport protocol to use (default: stdio)", ) _ = parser.add_argument( "--name", default="claude-code", help="Name of the MCP server (default: claude-code)", ) _ = parser.add_argument( "--allow-path", action="append", dest="allowed_paths", help="Add an allowed path (can be specified multiple times)", ) _ = parser.add_argument( "--project", action="append", dest="project_paths", help="Add a project path for prompt generation (can be specified multiple times)", ) _ = parser.add_argument( "--allow-pattern", action="append", dest="allowed_patterns", help="Allow access to files/directories matching this pattern (overrides default exclusions, can be specified multiple times)", ) _ = parser.add_argument( "--agent-model", dest="agent_model", help="Specify the model name in LiteLLM format (e.g., 'openai/gpt-4o', 'anthropic/claude-3-sonnet')", ) _ = parser.add_argument( "--agent-max-tokens", dest="agent_max_tokens", type=int, help="Specify the maximum tokens for agent responses", ) _ = parser.add_argument( "--agent-api-key", dest="agent_api_key", help="Specify the API key for the LLM provider (for development/testing only)", ) _ = parser.add_argument( "--agent-base-url", dest="agent_base_url", help="Specify the base URL for the LLM provider API endpoint (e.g., 'http://localhost:1234/v1')", ) _ = parser.add_argument( "--agent-max-iterations", dest="agent_max_iterations", type=int, default=10, help="Maximum number of iterations for agent (default: 10)", ) _ = parser.add_argument( "--agent-max-tool-uses", dest="agent_max_tool_uses", type=int, default=30, help="Maximum number of total tool uses for agent (default: 30)", ) _ = parser.add_argument( "--enable-agent-tool", dest="enable_agent_tool", action="store_true", default=False, help="Enable the agent tool (disabled by default)", ) _ = parser.add_argument( "--command-timeout", dest="command_timeout", type=float, default=120.0, help="Default timeout for command execution in seconds (default: 120.0)", ) _ = parser.add_argument( "--install", action="store_true", help="Install server configuration in Claude Desktop", ) args = parser.parse_args() # Cast args attributes to appropriate types to avoid 'Any' warnings name: str = cast(str, args.name) install: bool = cast(bool, args.install) transport: str = cast(str, args.transport) agent_model: str | None = cast(str | None, args.agent_model) agent_max_tokens: int | None = cast(int | None, args.agent_max_tokens) agent_api_key: str | None = cast(str | None, args.agent_api_key) agent_base_url: str | None = cast(str | None, args.agent_base_url) agent_max_iterations: int = cast(int, args.agent_max_iterations) agent_max_tool_uses: int = cast(int, args.agent_max_tool_uses) enable_agent_tool: bool = cast(bool, args.enable_agent_tool) command_timeout: float = cast(float, args.command_timeout) allowed_patterns: list[str] = ( cast(list[str], args.allowed_patterns) if args.allowed_patterns else [] ) allowed_paths: list[str] = ( cast(list[str], args.allowed_paths) if args.allowed_paths else [] ) project_paths: list[str] = ( cast(list[str], args.project_paths) if args.project_paths else [] ) if install: install_claude_desktop_config(name, allowed_paths, project_paths) return # If no allowed paths are specified, use the current directory if not allowed_paths: allowed_paths = [os.getcwd()] # Run the server server = ClaudeCodeServer( name=name, allowed_paths=allowed_paths, project_paths=project_paths, agent_model=agent_model, agent_max_tokens=agent_max_tokens, agent_api_key=agent_api_key, agent_base_url=agent_base_url, agent_max_iterations=agent_max_iterations, agent_max_tool_uses=agent_max_tool_uses, enable_agent_tool=enable_agent_tool, command_timeout=command_timeout, allowed_patterns=allowed_patterns, ) # Transport will be automatically cast to Literal['stdio', 'sse'] by the server server.run(transport=transport) def install_claude_desktop_config( name: str = "claude-code", allowed_paths: list[str] | None = None, project_paths: list[str] | None = None, ) -> None: """Install the server configuration in Claude Desktop. Args: name: The name to use for the server in the config allowed_paths: Optional list of paths to allow project_paths: Optional list of project paths for prompt generation """ # Find the Claude Desktop config directory home: Path = Path.home() if sys.platform == "darwin": # macOS config_dir: Path = home / "Library" / "Application Support" / "Claude" elif sys.platform == "win32": # Windows config_dir = Path(os.environ.get("APPDATA", "")) / "Claude" else: # Linux and others config_dir = home / ".config" / "claude" config_file: Path = config_dir / "claude_desktop_config.json" # Create directory if it doesn't exist config_dir.mkdir(parents=True, exist_ok=True) # Get current script path script_path: Path = Path(sys.executable) # Create args array args: list[str] = ["-m", "mcp_claude_code.cli"] # Add allowed paths if specified if allowed_paths: for path in allowed_paths: args.extend(["--allow-path", path]) else: # Allow home directory by default args.extend(["--allow-path", str(home)]) # Add project paths if specified if project_paths: for path in project_paths: args.extend(["--project", path]) # Create config object config: dict[str, Any] = { "mcpServers": {name: {"command": script_path.as_posix(), "args": args}} } # Check if the file already exists if config_file.exists(): try: with open(config_file, "r") as f: existing_config: dict[str, Any] = json.load(f) # Update the existing config if "mcpServers" not in existing_config: existing_config["mcpServers"] = {} existing_config["mcpServers"][name] = config["mcpServers"][name] config = existing_config except Exception as e: print(f"Error reading existing config: {e}") print("Creating new config file.") # Write the config file with open(config_file, mode="w") as f: json.dump(config, f, indent=2) print(f"Successfully installed {name} in Claude Desktop configuration.") print(f"Config file: {config_file}") if allowed_paths: print("\nAllowed paths:") for path in allowed_paths: print(f"- {path}") else: print(f"\nDefault allowed path: {home}") if project_paths: print("\nProject paths:") for path in project_paths: print(f"- {path}") print( "\nYou can modify allowed paths and project paths in the config file directly." ) print("Restart Claude Desktop for changes to take effect.") if __name__ == "__main__": main()