mcp-proxy-adapter 4.1.0__py3-none-any.whl → 6.0.0__py3-none-any.whl

mcp_proxy_adapter/commands/unload_command.py

@@ -0,0 +1,158 @@
+"""
+Module with unload command implementation.
+"""
+
+from typing import Dict, Any, Optional, List
+
+from mcp_proxy_adapter.commands.base import Command
+from mcp_proxy_adapter.commands.result import CommandResult, SuccessResult
+from mcp_proxy_adapter.commands.command_registry import registry
+
+
+class UnloadResult(SuccessResult):
+    """
+    Result of the unload command execution.
+    """
+    
+    def __init__(self, success: bool, command_name: str, message: str, error: Optional[str] = None):
+        """
+        Initialize unload command result.
+        
+        Args:
+            success: Whether unloading was successful
+            command_name: Name of the command that was unloaded
+            message: Result message
+            error: Error message if unloading failed
+        """
+        data = {
+            "success": success,
+            "command_name": command_name
+        }
+        if error:
+            data["error"] = error
+            
+        super().__init__(data=data, message=message)
+    
+    @classmethod
+    def get_schema(cls) -> Dict[str, Any]:
+        """
+        Get JSON schema for result validation.
+        
+        Returns:
+            Dict[str, Any]: JSON schema
+        """
+        return {
+            "type": "object",
+            "properties": {
+                "data": {
+                    "type": "object",
+                    "properties": {
+                        "success": {"type": "boolean"},
+                        "command_name": {"type": "string"},
+                        "error": {"type": "string"}
+                    },
+                    "required": ["success", "command_name"]
+                }
+            },
+            "required": ["data"]
+        }
+
+
+class UnloadCommand(Command):
+    """
+    Command that unloads loaded commands from registry.
+    
+    This command allows removal of dynamically loaded commands from the command registry.
+    Only commands that were loaded via the 'load' command or from the commands directory
+    can be unloaded. Built-in commands and custom commands registered with higher priority
+    cannot be unloaded using this command.
+    
+    When a command is unloaded:
+    - The command class is removed from the loaded commands registry
+    - Any command instances are also removed
+    - The command becomes unavailable for execution
+    - Built-in and custom commands with the same name remain unaffected
+    
+    This is useful for:
+    - Removing outdated or problematic commands
+    - Managing memory usage by unloading unused commands
+    - Testing different versions of commands
+    - Cleaning up temporary commands loaded for testing
+    
+    Note: Unloading a command does not affect other commands and does not require
+    a system restart. The command can be reloaded later if needed.
+    """
+    
+    name = "unload"
+    result_class = UnloadResult
+    
+    async def execute(self, command_name: str, **kwargs) -> UnloadResult:
+        """
+        Execute unload command.
+        
+        Args:
+            command_name: Name of the command to unload
+            **kwargs: Additional parameters
+            
+        Returns:
+            UnloadResult: Unload command result
+        """
+        # Unload command from registry
+        result = registry.unload_command(command_name)
+        
+        return UnloadResult(
+            success=result.get("success", False),
+            command_name=result.get("command_name", command_name),
+            message=result.get("message", "Unknown result"),
+            error=result.get("error")
+        )
+    
+    @classmethod
+    def get_schema(cls) -> Dict[str, Any]:
+        """
+        Get JSON schema for command parameters.
+        
+        Returns:
+            Dict[str, Any]: JSON schema
+        """
+        return {
+            "type": "object",
+            "properties": {
+                "command_name": {
+                    "type": "string",
+                    "description": "Name of the command to unload (must be a loaded command)"
+                }
+            },
+            "required": ["command_name"]
+        }
+    
+    @classmethod
+    def _generate_examples(cls, params: Dict[str, Dict[str, Any]]) -> List[Dict[str, Any]]:
+        """
+        Generate custom examples for unload command.
+        
+        Args:
+            params: Information about command parameters
+            
+        Returns:
+            List of examples
+        """
+        examples = [
+            {
+                "command": cls.name,
+                "params": {"command_name": "test_command"},
+                "description": "Unload a previously loaded test command"
+            },
+            {
+                "command": cls.name,
+                "params": {"command_name": "remote_command"},
+                "description": "Unload a command that was loaded from URL"
+            },
+            {
+                "command": cls.name,
+                "params": {"command_name": "custom_command"},
+                "description": "Unload a custom command loaded from local file"
+            }
+        ]
+        
+        return examples 

mcp_proxy_adapter/config.py

@@ -53,8 +53,105 @@ class Config:
             },
             "commands": {
                 "auto_discovery": True,
-                "discovery_path": "mcp_proxy_adapter.commands",
-                "custom_commands_path": None
+                "commands_directory": "./commands",  # Path to directory with command files
+                "catalog_directory": "./catalog",  # Path to command catalog directory
+                "plugin_servers": [],  # List of plugin server URLs
+                "auto_install_dependencies": True  # Automatically install plugin dependencies
+            },
+            "ssl": {
+                "enabled": False,
+                "mode": "https_only",
+                "cert_file": None,
+                "key_file": None,
+                "ca_cert": None,
+                "verify_client": False,
+                "client_cert_required": False,
+                "cipher_suites": ["TLS_AES_256_GCM_SHA384", "TLS_CHACHA20_POLY1305_SHA256"],
+                "min_tls_version": "1.2",
+                "max_tls_version": "1.3",
+                "token_auth": {
+                    "enabled": False,
+                    "header_name": "Authorization",
+                    "token_prefix": "Bearer",
+                    "tokens_file": "tokens.json",
+                    "token_expiry": 3600,
+                    "jwt_secret": "",
+                    "jwt_algorithm": "HS256"
+                }
+            },
+            "roles": {
+                "enabled": True,
+                "config_file": "schemas/roles_schema.json",
+                "default_policy": {
+                    "deny_by_default": True,
+                    "require_role_match": True,
+                    "case_sensitive": False,
+                    "allow_wildcard": True
+                },
+                "auto_load": True,
+                "validation_enabled": True
+            },
+            "transport": {
+                "type": "http",
+                "port": None,
+                "ssl": {
+                    "enabled": False,
+                    "cert_file": None,
+                    "key_file": None,
+                    "ca_cert": None,
+                    "verify_client": False,
+                    "client_cert_required": False
+                }
+            },
+            "proxy_registration": {
+                "enabled": False,
+                "proxy_url": "http://localhost:3004",
+                "server_id": "mcp_proxy_adapter",
+                "server_name": "MCP Proxy Adapter",
+                "description": "JSON-RPC API for interacting with MCP Proxy",
+                "registration_timeout": 30,
+                "retry_attempts": 3,
+                "retry_delay": 5,
+                "auto_register_on_startup": True,
+                "auto_unregister_on_shutdown": True
+            },
+            "debug": {
+                "enabled": False,
+                "level": "WARNING"
+            },
+            "security": {
+                "framework": "mcp_security_framework",
+                "enabled": True,
+                "auth": {
+                    "enabled": True,
+                    "methods": ["api_key"],
+                    "api_keys": {},
+                    "jwt_secret": "",
+                    "jwt_algorithm": "HS256"
+                },
+                "ssl": {
+                    "enabled": False,
+                    "cert_file": None,
+                    "key_file": None,
+                    "ca_cert": None,
+                    "min_tls_version": "TLSv1.2",
+                    "verify_client": False,
+                    "client_cert_required": False
+                },
+                "permissions": {
+                    "enabled": True,
+                    "roles_file": "roles.json",
+                    "default_role": "user",
+                    "deny_by_default": True
+                },
+                "rate_limit": {
+                    "enabled": True,
+                    "requests_per_minute": 60,
+                    "requests_per_hour": 1000,
+                    "burst_limit": 10,
+                    "by_ip": True,
+                    "by_user": True
+                }
             }
         }