apm-cli 0.1.0__py3-none-any.whl

apm_cli/__init__.py

@@ -0,0 +1,5 @@
+"""APM-CLI package."""
+
+from .version import get_version
+
+__version__ = get_version()

apm_cli/adapters/__init__.py

@@ -0,0 +1 @@
+"""Adapters package."""

apm_cli/adapters/client/__init__.py

@@ -0,0 +1 @@
+"""Client adapters package."""

apm_cli/adapters/client/base.py

@@ -0,0 +1,36 @@
+"""Base adapter interface for MCP clients."""
+
+from abc import ABC, abstractmethod
+
+
+class MCPClientAdapter(ABC):
+    """Base adapter for MCP clients."""
+
+    @abstractmethod
+    def get_config_path(self):
+        """Get the path to the MCP configuration file."""
+        pass
+
+    @abstractmethod
+    def update_config(self, config_updates):
+        """Update the MCP configuration."""
+        pass
+
+    @abstractmethod
+    def get_current_config(self):
+        """Get the current MCP configuration."""
+        pass
+
+    @abstractmethod
+    def configure_mcp_server(self, server_url, server_name=None, enabled=True):
+        """Configure an MCP server in the client configuration.
+
+        Args:
+            server_url (str): URL of the MCP server.
+            server_name (str, optional): Name of the server. Defaults to None.
+            enabled (bool, optional): Whether to enable the server. Defaults to True.
+
+        Returns:
+            bool: True if successful, False otherwise.
+        """
+        pass

apm_cli/adapters/client/vscode.py

@@ -0,0 +1,301 @@
+"""VSCode implementation of MCP client adapter.
+
+This adapter implements the VSCode-specific handling of MCP server configuration,
+following the official documentation at:
+https://code.visualstudio.com/docs/copilot/chat/mcp-servers
+"""
+
+import json
+import os
+from pathlib import Path
+from .base import MCPClientAdapter
+from ...registry.client import SimpleRegistryClient
+from ...registry.integration import RegistryIntegration
+
+
+class VSCodeClientAdapter(MCPClientAdapter):
+    """VSCode implementation of MCP client adapter.
+    
+    This adapter handles VSCode-specific configuration for MCP servers using
+    a repository-level .vscode/mcp.json file, following the format specified
+    in the VSCode documentation.
+    """
+    
+    def __init__(self, registry_url=None):
+        """Initialize the VSCode client adapter.
+        
+        Args:
+            registry_url (str, optional): URL of the MCP registry.
+                If not provided, uses the MCP_REGISTRY_URL environment variable
+                or falls back to the default demo registry.
+        """
+        self.registry_client = SimpleRegistryClient(registry_url)
+        self.registry_integration = RegistryIntegration(registry_url)
+    
+    def get_config_path(self):
+        """Get the path to the VSCode MCP configuration file in the repository.
+        
+        Returns:
+            str: Path to the .vscode/mcp.json file.
+        """
+        # Use the current working directory as the repository root
+        repo_root = Path(os.getcwd())
+        
+        # Path to .vscode/mcp.json in the repository
+        vscode_dir = repo_root / ".vscode"
+        mcp_config_path = vscode_dir / "mcp.json"
+        
+        # Create the .vscode directory if it doesn't exist
+        try:
+            if not vscode_dir.exists():
+                vscode_dir.mkdir(parents=True, exist_ok=True)
+        except Exception as e:
+            print(f"Warning: Could not create .vscode directory: {e}")
+            
+        return str(mcp_config_path)
+    
+    def update_config(self, config_updates):
+        """Update the VSCode MCP configuration with new values.
+        
+        Args:
+            config_updates (dict): Dictionary of settings to update.
+            
+        Returns:
+            bool: True if successful, False otherwise.
+        """
+        config_path = self.get_config_path()
+        
+        try:
+            # Read existing config or create a new one
+            try:
+                with open(config_path, "r", encoding="utf-8") as f:
+                    config = json.load(f)
+            except (FileNotFoundError, json.JSONDecodeError):
+                config = {}
+            
+            # Update config with new values or remove entries set to None
+            for key, value in config_updates.items():
+                if value is None:
+                    # Remove the entry if it exists
+                    if key in config:
+                        del config[key]
+                else:
+                    config[key] = value
+                
+            # Write the updated config
+            with open(config_path, "w", encoding="utf-8") as f:
+                json.dump(config, f, indent=2)
+                
+            return True
+        except Exception as e:
+            print(f"Error updating VSCode MCP configuration: {e}")
+            return False
+    
+    def get_current_config(self):
+        """Get the current VSCode MCP configuration.
+        
+        Returns:
+            dict: Current VSCode MCP configuration from the local .vscode/mcp.json file.
+        """
+        config_path = self.get_config_path()
+        
+        try:
+            try:
+                with open(config_path, "r", encoding="utf-8") as f:
+                    return json.load(f)
+            except (FileNotFoundError, json.JSONDecodeError):
+                return {}
+        except Exception as e:
+            print(f"Error reading VSCode MCP configuration: {e}")
+            return {}
+    
+    def configure_mcp_server(self, server_url, server_name=None, enabled=True):
+        """Configure an MCP server in VSCode configuration.
+        
+        This method follows the VSCode documentation for MCP server configuration format:
+        https://code.visualstudio.com/docs/copilot/chat/mcp-servers#_configuration-format
+        
+        Args:
+            server_url (str): URL or identifier of the MCP server.
+            server_name (str, optional): Name of the server. Defaults to None.
+            enabled (bool, optional): Ignored parameter, kept for API compatibility.
+            
+        Returns:
+            bool: True if successful, False otherwise.
+        """
+        if not server_url:
+            print("Error: server_url cannot be empty")
+            return False
+            
+        if not server_name:
+            server_name = server_url
+            
+        try:
+            # Use enhanced lookup with multiple strategies
+            server_info = self.registry_client.find_server_by_reference(server_url)
+            
+            # Fail if server is not found in registry - security requirement
+            if not server_info:
+                raise ValueError(f"Failed to retrieve server details for '{server_url}'. Server not found in registry.")
+            
+            # Format server configuration and get input variables if any
+            server_config, input_vars = self._format_server_config(server_info)
+            
+            config = self.get_current_config()
+            
+            # Make sure we have the servers object
+            if "servers" not in config:
+                config["servers"] = {}
+                
+            # Add input variables if any
+            if input_vars:
+                if "inputs" not in config:
+                    config["inputs"] = []
+                # Merge with existing inputs, avoiding duplicates by id
+                existing_input_ids = [input_var.get("id") for input_var in config.get("inputs", [])]
+                for input_var in input_vars:
+                    if input_var.get("id") not in existing_input_ids:
+                        config["inputs"].append(input_var)
+                
+            # Add the server configuration
+            config["servers"][server_name] = server_config
+                
+            # Update the configuration
+            return self.update_config(config)
+            
+        except ValueError as ve:
+            # Re-raise ValueError to indicate missing server details
+            raise ve
+        except Exception as e:
+            print(f"Error configuring MCP server: {e}")
+            return False
+    
+    def _format_server_config(self, server_info):
+        """Format server details into VSCode mcp.json compatible format.
+        
+        Args:
+            server_info (dict): Server information from registry.
+            
+        Returns:
+            tuple: (server_config, input_vars) where:
+                - server_config is the formatted server configuration for mcp.json
+                - input_vars is a list of input variable definitions
+        """
+        # Initialize the base config structure
+        server_config = {}
+        input_vars = []
+        
+        # Check for packages information
+        if "packages" in server_info and server_info["packages"]:
+            package = server_info["packages"][0]
+            runtime_hint = package.get("runtime_hint", "")
+            
+            # Handle npm packages
+            if runtime_hint == "npx" or "npm" in package.get("registry_name", "").lower():
+                # Get args directly from runtime_arguments
+                args = []
+                if "runtime_arguments" in package and package["runtime_arguments"]:
+                    for arg in package["runtime_arguments"]:
+                        if arg.get("is_required", False) and arg.get("value_hint"):
+                            args.append(arg.get("value_hint"))
+                
+                # Fallback if no runtime_arguments are provided
+                if not args and package.get("name"):
+                    args = [package.get("name")]
+                
+                server_config = {
+                    "type": "stdio",
+                    "command": "npx",
+                    "args": args
+                }
+            
+            # Handle docker packages
+            elif runtime_hint == "docker":
+                # Get args directly from runtime_arguments
+                args = []
+                if "runtime_arguments" in package and package["runtime_arguments"]:
+                    for arg in package["runtime_arguments"]:
+                        if arg.get("is_required", False) and arg.get("value_hint"):
+                            args.append(arg.get("value_hint"))
+                
+                # Fallback if no runtime_arguments are provided - use standard docker run command
+                if not args:
+                    args = ["run", "-i", "--rm", package.get("name")]
+                
+                server_config = {
+                    "type": "stdio",
+                    "command": "docker",
+                    "args": args
+                }
+            
+            # Handle Python packages
+            elif runtime_hint in ["uvx", "pip", "python"] or "python" in runtime_hint or package.get("registry_name", "").lower() == "pypi":
+                # Determine the command based on runtime_hint
+                if runtime_hint == "uvx":
+                    command = "uvx"
+                elif "python" in runtime_hint:
+                    # Use the specified Python path if it's a full path, otherwise default to python3
+                    command = "python3" if runtime_hint in ["python", "pip"] else runtime_hint
+                else:
+                    command = "python3"
+                
+                # Get args directly from runtime_arguments
+                args = []
+                if "runtime_arguments" in package and package["runtime_arguments"]:
+                    for arg in package["runtime_arguments"]:
+                        if arg.get("is_required", False) and arg.get("value_hint"):
+                            args.append(arg.get("value_hint"))
+                
+                # Fallback if no runtime_arguments are provided
+                if not args:
+                    if runtime_hint == "uvx":
+                        module_name = package.get("name", "").replace("mcp-server-", "")
+                        args = [f"mcp-server-{module_name}"]
+                    else:
+                        module_name = package.get("name", "").replace("mcp-server-", "").replace("-", "_")
+                        args = ["-m", f"mcp_server_{module_name}"]
+                
+                server_config = {
+                    "type": "stdio",
+                    "command": command,
+                    "args": args
+                }
+            
+            # Add environment variables if present
+            if "environment_variables" in package and package["environment_variables"]:
+                server_config["env"] = {}
+                for env_var in package["environment_variables"]:
+                    if "name" in env_var:
+                        # Convert variable name to lowercase and replace underscores with hyphens for VS Code convention
+                        input_var_name = env_var["name"].lower().replace("_", "-")
+                        
+                        # Create the input variable reference
+                        server_config["env"][env_var["name"]] = f"${{input:{input_var_name}}}"
+                        
+                        # Create the input variable definition
+                        input_var_def = {
+                            "type": "promptString",
+                            "id": input_var_name,
+                            "description": env_var.get("description", f"{env_var['name']} for MCP server"),
+                            "password": True  # Default to True for security
+                        }
+                        input_vars.append(input_var_def)
+            
+        # If no server config was created from packages, check for other server types
+        if not server_config:
+            # Check for SSE endpoints
+            if "sse_endpoint" in server_info:
+                server_config = {
+                    "type": "sse",
+                    "url": server_info["sse_endpoint"],
+                    "headers": server_info.get("sse_headers", {})
+                }
+            # Default fallback
+            else:
+                server_config = {
+                    "type": "stdio",
+                    "command": "uvx",
+                    "args": [f"mcp-server-{server_info.get('name', '')}"]
+                }
+        
+        return server_config, input_vars

apm_cli/adapters/package_manager/__init__.py

@@ -0,0 +1 @@
+"""Package manager adapters package."""

apm_cli/adapters/package_manager/base.py

@@ -0,0 +1,27 @@
+"""Base adapter interface for MCP package managers."""
+
+from abc import ABC, abstractmethod
+
+
+class MCPPackageManagerAdapter(ABC):
+    """Base adapter for MCP package managers."""
+    
+    @abstractmethod
+    def install(self, package_name, version=None):
+        """Install an MCP package."""
+        pass
+    
+    @abstractmethod
+    def uninstall(self, package_name):
+        """Uninstall an MCP package."""
+        pass
+    
+    @abstractmethod
+    def list_installed(self):
+        """List all installed MCP packages."""
+        pass
+    
+    @abstractmethod
+    def search(self, query):
+        """Search for MCP packages."""
+        pass

apm_cli/adapters/package_manager/default_manager.py

@@ -0,0 +1,123 @@
+"""Implementation of the default MCP package manager."""
+
+from .base import MCPPackageManagerAdapter
+from ...config import get_default_client
+from ...registry.integration import RegistryIntegration
+
+
+class DefaultMCPPackageManager(MCPPackageManagerAdapter):
+    """Implementation of the default MCP package manager."""
+    
+    def install(self, package_name, version=None):
+        """Install an MCP package.
+        
+        Args:
+            package_name (str): Name of the package to install.
+            version (str, optional): Version of the package to install.
+        
+        Returns:
+            bool: True if successful, False otherwise.
+        """
+
+        try:
+            # Import here to avoid circular import
+            from ...factory import ClientFactory
+            
+            client_type = get_default_client()
+            client_adapter = ClientFactory.create_client(client_type)
+            
+            # For VSCode, configure MCP server in mcp.json
+            result = client_adapter.configure_mcp_server(package_name, package_name, True)
+            
+            if result:
+                print(f"Successfully installed {package_name}")
+            return result
+        except Exception as e:
+            print(f"Error installing package {package_name}: {e}")
+            return False
+    
+    def uninstall(self, package_name):
+        """Uninstall an MCP package.
+        
+        Args:
+            package_name (str): Name of the package to uninstall.
+        
+        Returns:
+            bool: True if successful, False otherwise.
+        """
+        
+        try:
+            # Import here to avoid circular import
+            from ...factory import ClientFactory
+            
+            client_type = get_default_client()
+            client_adapter = ClientFactory.create_client(client_type)
+            config = client_adapter.get_current_config()
+            
+            # For VSCode, remove the server from mcp.json
+            if "servers" in config and package_name in config["servers"]:
+                servers = config["servers"]
+                servers.pop(package_name, None)
+                result = client_adapter.update_config({"servers": servers})
+                
+                if result:
+                    print(f"Successfully uninstalled {package_name}")
+                return result
+            else:
+                print(f"Package {package_name} not found in configuration")
+                return False
+                
+        except Exception as e:
+            print(f"Error uninstalling package {package_name}: {e}")
+            return False
+    
+    def list_installed(self):
+        """List all installed MCP packages.
+        
+        Returns:
+            list: List of installed packages.
+        """
+        
+        try:
+            # Import here to avoid circular import
+            from ...factory import ClientFactory
+            
+            # Get client type from configuration (default is vscode)
+            client_type = get_default_client()
+            
+            # Create client adapter
+            client_adapter = ClientFactory.create_client(client_type)
+            
+            # Get config from local .vscode/mcp.json file
+            config = client_adapter.get_current_config()
+            
+            # Extract server names from the config
+            servers = config.get("servers", {})
+            
+            # Return the list of server names
+            return list(servers.keys())
+        except Exception as e:
+            print(f"Error retrieving installed MCP servers: {e}")
+            return []
+    
+    def search(self, query):
+        """Search for MCP packages.
+        
+        Args:
+            query (str): Search query.
+        
+        Returns:
+            list: List of packages matching the query.
+        """
+        
+        try:
+            # Use the registry integration to search for packages
+            registry = RegistryIntegration()
+            packages = registry.search_packages(query)
+            
+            # Return the list of package IDs/names
+            return [pkg.get("id", pkg.get("name", "Unknown")) for pkg in packages] if packages else []
+            
+        except Exception as e:
+            print(f"Error searching for packages: {e}")
+            return []