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

mcp_proxy_adapter/examples/custom_commands/start_server_with_registration.py

@@ -0,0 +1,278 @@
+#!/usr/bin/env python3
+"""
+Server startup script with proper registration sequence.
+First starts the server, then registers with proxy after server is ready.
+"""
+
+import asyncio
+import time
+import sys
+import os
+import uvicorn
+import threading
+from pathlib import Path
+
+# Add parent directories to path
+sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', '..'))
+
+from mcp_proxy_adapter import create_app
+from mcp_proxy_adapter.core.logging import get_logger, setup_logging
+from mcp_proxy_adapter.core.settings import (
+    Settings,
+    get_server_host,
+    get_server_port,
+    get_server_debug,
+    get_setting,
+    get_custom_setting_value
+)
+from mcp_proxy_adapter.core.ssl_utils import SSLUtils
+from mcp_proxy_adapter.core.transport_manager import transport_manager
+from mcp_proxy_adapter.core.proxy_registration import register_with_proxy
+from custom_settings_manager import CustomSettingsManager, get_app_name, is_feature_enabled
+
+# Import custom commands and hooks
+from custom_help_command import CustomHelpCommand
+from custom_health_command import CustomHealthCommand
+from data_transform_command import DataTransformCommand
+from intercept_command import InterceptCommand
+from advanced_hooks import register_advanced_hooks
+
+# Import auto-registered commands
+from auto_commands.auto_echo_command import AutoEchoCommand
+from auto_commands.auto_info_command import AutoInfoCommand
+
+# Import manual registration example
+from manual_echo_command import ManualEchoCommand
+
+# Import echo command
+from echo_command import EchoCommand
+
+# Import custom OpenAPI generator
+from custom_openapi_generator import custom_openapi_generator
+
+# Import command registry for manual registration
+from mcp_proxy_adapter.commands.command_registry import registry
+from mcp_proxy_adapter.config import config
+from mcp_proxy_adapter.commands.hooks import register_custom_commands_hook
+
+
+def custom_commands_hook(registry):
+    """Hook function for registering custom commands."""
+    logger = get_logger("custom_commands")
+    logger.info("Registering custom commands via hook...")
+    
+    # Get custom commands configuration from custom settings
+    custom_commands_config = get_custom_setting_value("custom_commands", {})
+    
+    # Register echo command (only if not already registered)
+    if not registry.command_exists("echo"):
+        registry.register_custom(EchoCommand)
+        logger.info("Registered: echo command")
+    else:
+        logger.debug("Command 'echo' is already registered, skipping")
+    
+    # Register custom help command (will override built-in)
+    if custom_commands_config.get("help", {}).get("enabled", True):
+        registry.register_custom(CustomHelpCommand)
+        logger.info("Registered: custom help command")
+    
+    # Register custom health command (will override built-in)
+    if custom_commands_config.get("health", {}).get("enabled", True):
+        registry.register_custom(CustomHealthCommand)
+        logger.info("Registered: custom health command")
+    
+    # Register advanced demonstration commands
+    if custom_commands_config.get("data_transform", {}).get("enabled", True):
+        registry.register_custom(DataTransformCommand)
+        logger.info("Registered: data_transform command")
+    
+    if custom_commands_config.get("intercept", {}).get("enabled", True):
+        registry.register_custom(InterceptCommand)
+        logger.info("Registered: intercept command")
+
+
+def setup_hooks():
+    """Setup hooks for command processing."""
+    logger = get_logger("custom_commands")
+    logger.info("Setting up hooks...")
+    
+    # Register custom commands hook
+    register_custom_commands_hook(custom_commands_hook)
+    
+    # Note: Advanced hooks are not compatible with current API
+    # They will be registered automatically by the command registry
+    logger.info("Basic hooks setup completed")
+
+
+async def initialize_commands():
+    """
+    Initialize commands using the unified system initialization logic.
+    This function is used both at startup and during reload.
+    
+    Returns:
+        Number of commands discovered.
+    """
+    # Use the unified reload method from registry
+    result = await registry.reload_system()
+    return result["total_commands"]
+
+
+async def register_with_proxy_after_startup():
+    """
+    Register with proxy after server is fully started and listening.
+    """
+    logger = get_logger("proxy_registration")
+    
+    # Wait a bit for server to fully start
+    await asyncio.sleep(3)
+    
+    # Get server configuration
+    server_config = config.get("server", {})
+    server_host = server_config.get("host", "0.0.0.0")
+    server_port = server_config.get("port", 8000)
+    
+    # Use localhost for external access if host is 0.0.0.0
+    if server_host == "0.0.0.0":
+        server_host = "localhost"
+    
+    server_url = f"http://{server_host}:{server_port}"
+    
+    logger.info(f"Attempting to register server with proxy at {server_url}")
+    
+    # Attempt registration
+    registration_success = await register_with_proxy(server_url)
+    if registration_success:
+        logger.info("✅ Successfully registered with proxy after server startup")
+    else:
+        logger.warning("⚠️ Failed to register with proxy after server startup")
+
+
+def start_server():
+    """Start the server in a separate thread."""
+    # Initialize settings
+    settings = Settings()
+    server_settings = settings.get_server_settings()
+    logging_settings = settings.get_logging_settings()
+    commands_settings = settings.get_commands_settings()
+    
+    # Setup logging - pass only the level, not the entire dict
+    setup_logging(logging_settings.get('level', 'INFO'))
+    logger = get_logger("server_startup")
+    
+    # Load transport configuration
+    if not transport_manager.load_config(config.config_data):
+        logger.error("Failed to load transport configuration")
+        return
+    
+    # Validate transport configuration
+    if not transport_manager.validate_config():
+        logger.error("Transport configuration validation failed")
+        return
+    
+    # Print server header and description
+    print("=" * 80)
+    print("🔧 ADVANCED MCP PROXY ADAPTER SERVER WITH HOOKS")
+    print("=" * 80)
+    print("📋 Description:")
+    print(f"   {get_app_name()} - Advanced server with custom settings management")
+    print()
+    
+    # Get transport info
+    transport_info = transport_manager.get_transport_info()
+    
+    print("⚙️  Configuration:")
+    print(f"   • Server: {server_settings['host']}:{transport_manager.get_port()}")
+    print(f"   • Transport: {transport_info['type']}")
+    print(f"   • Debug: {server_settings['debug']}")
+    print(f"   • Log Level: {logging_settings.get('level', 'INFO')}")
+    print(f"   • Log Directory: {logging_settings.get('log_dir', './logs')}")
+    print(f"   • Auto Discovery: {commands_settings['auto_discovery']}")
+    print(f"   • SSL Enabled: {transport_info['ssl_enabled']}")
+    if transport_info['ssl_enabled']:
+        ssl_config = transport_info['ssl_config']
+        print(f"   • SSL Cert: {ssl_config.get('cert_file', 'Not specified')}")
+        print(f"   • Client Verification: {ssl_config.get('verify_client', False)}")
+    print()
+    print("🔧 Available Commands:")
+    print("   • help - Custom help command (overrides built-in)")
+    print("   • health - Custom health command (overrides built-in)")
+    print("   • config - Built-in config command")
+    print("   • reload - Built-in reload command")
+    print("   • settings - Built-in settings command")
+    print("   • load - Built-in load command")
+    print("   • unload - Built-in unload command")
+    print("   • plugins - Built-in plugins command")
+    print("   • echo - Custom echo command")
+    print("   • data_transform - Data transformation command")
+    print("   • intercept - Command interception example")
+    print("   • manual_echo - Manually registered echo command")
+    print("   • test - Remote command (loaded from plugin server)")
+    print()
+    print("🎯 Features:")
+    print("   • Advanced JSON-RPC API")
+    print("   • Custom commands with hooks")
+    print("   • Data transformation hooks")
+    print("   • Command interception hooks")
+    print("   • Auto-registration and manual registration")
+    print("   • Custom OpenAPI schema generation")
+    print("   • Configuration-driven settings")
+    print("   • Custom settings management")
+    print("=" * 80)
+    print()
+    
+    logger.info("Starting Advanced Custom Commands MCP Proxy Adapter Server with Hooks...")
+    logger.info(f"Server configuration: {server_settings}")
+    logger.info(f"Logging configuration: {logging_settings}")
+    logger.info(f"Commands configuration: {commands_settings}")
+    logger.info(f"Transport configuration: {transport_info}")
+    
+    # Setup hooks for command processing
+    setup_hooks()
+    
+    # Initialize commands
+    asyncio.run(initialize_commands())
+    
+    # Create application with settings from configuration
+    app = create_app(
+        title=get_app_name(),
+        description="Advanced MCP Proxy Adapter server with custom settings management, demonstrating hook capabilities including data transformation, command interception, conditional processing, and smart interception hooks. Features custom commands with enhanced functionality and comprehensive settings management.",
+        version="2.1.0"
+    )
+    
+    # Get uvicorn configuration from transport manager
+    uvicorn_config = transport_manager.get_uvicorn_config()
+    uvicorn_config["host"] = server_settings['host']
+    uvicorn_config["log_level"] = server_settings['log_level'].lower()
+    
+    logger.info(f"Starting server with uvicorn config: {uvicorn_config}")
+    
+    # Run the server with configuration settings
+    uvicorn.run(
+        app,
+        **uvicorn_config
+    )
+
+
+def main():
+    """Main function to start server and register with proxy."""
+    # Start server in a separate thread
+    server_thread = threading.Thread(target=start_server, daemon=True)
+    server_thread.start()
+    
+    # Wait for server to start
+    time.sleep(5)
+    
+    # Register with proxy after server is ready
+    asyncio.run(register_with_proxy_after_startup())
+    
+    # Keep main thread alive
+    try:
+        while True:
+            time.sleep(1)
+    except KeyboardInterrupt:
+        print("\n🛑 Server shutdown requested")
+        sys.exit(0)
+
+
+if __name__ == "__main__":
+    main() 

mcp_proxy_adapter/examples/custom_commands/test_openapi.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+"""
+Simple test server to get OpenAPI schema.
+"""
+
+import uvicorn
+from fastapi import FastAPI
+from mcp_proxy_adapter.custom_openapi import custom_openapi_with_fallback
+
+# Create FastAPI app
+app = FastAPI(
+    title="Test MCP Proxy Server",
+    description="Test server for OpenAPI schema",
+    version="1.0.0"
+)
+
+# Set custom OpenAPI generator
+app.openapi = lambda: custom_openapi_with_fallback(app)
+
+# Add OpenAPI endpoint
+@app.get("/openapi.json")
+async def get_openapi_schema():
+    """Returns OpenAPI schema."""
+    return custom_openapi_with_fallback(app)
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8001) 

mcp_proxy_adapter/examples/custom_commands/test_registry.py

@@ -0,0 +1,23 @@
+#!/usr/bin/env python3
+"""
+Test script to understand registry issues.
+"""
+
+import sys
+import os
+sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', '..'))
+
+print("1. Importing registry directly...")
+from mcp_proxy_adapter.commands.command_registry import registry
+print(f"   Registry type: {type(registry)}")
+print(f"   Registry has reload_system: {hasattr(registry, 'reload_system')}")
+print(f"   Registry methods: {[m for m in dir(registry) if not m.startswith('_')]}")
+
+print("2. Testing reload_system...")
+try:
+    result = registry.reload_system()
+    print(f"   ✅ reload_system worked: {result}")
+except Exception as e:
+    print(f"   ❌ reload_system failed: {e}")
+    import traceback
+    traceback.print_exc() 

mcp_proxy_adapter/examples/custom_commands/test_simple.py

@@ -0,0 +1,19 @@
+#!/usr/bin/env python3
+"""
+Simple test in custom_commands directory.
+"""
+
+import sys
+import os
+sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', '..'))
+
+print("1. Importing registry directly...")
+from mcp_proxy_adapter.commands.command_registry import registry
+print(f"   Registry has reload_system: {hasattr(registry, 'reload_system')}")
+
+print("2. Testing reload_system...")
+try:
+    result = registry.reload_system()
+    print(f"   ✅ reload_system worked: {result}")
+except Exception as e:
+    print(f"   ❌ reload_system failed: {e}") 

mcp_proxy_adapter/examples/custom_project_example/README.md

@@ -0,0 +1,103 @@
+# Пример пользовательского проекта с командами
+
+Этот пример показывает, как создать свой проект с командами и настроить `discovery_path` для их автоматического обнаружения.
+
+## Структура проекта
+
+```
+myproject/
+├── config.json
+├── main.py
+├── myproject/
+│   ├── __init__.py
+│   └── commands/
+│       ├── __init__.py
+│       ├── echo_command.py
+│       └── info_command.py
+└── README.md
+```
+
+## Конфигурация
+
+В файле `config.json` указываем путь к нашим командам:
+
+```json
+{
+  "server": {
+    "host": "127.0.0.1",
+    "port": 8001,
+    "debug": true,
+    "log_level": "DEBUG"
+  },
+  "logging": {
+    "level": "DEBUG",
+    "log_dir": "./logs",
+    "log_file": "myproject.log"
+  },
+  "commands": {
+    "auto_discovery": true,
+    "discovery_path": "myproject.commands"
+  }
+}
+```
+
+## Команды
+
+### echo_command.py
+
+```python
+from mcp_proxy_adapter.commands.base import Command
+from mcp_proxy_adapter.commands.result import CommandResult
+
+class EchoCommand(Command):
+    """Echo command that returns the input text."""
+    
+    name = "echo"
+    
+    def execute(self, text: str) -> CommandResult:
+        return CommandResult(success=True, data={"echo": text})
+    
+    @classmethod
+    def get_param_info(cls) -> dict:
+        return {
+            "text": {
+                "type": "string",
+                "description": "Text to echo",
+                "required": True
+            }
+        }
+```
+
+### info_command.py
+
+```python
+from mcp_proxy_adapter.commands.base import Command
+from mcp_proxy_adapter.commands.result import CommandResult
+
+class InfoCommand(Command):
+    """Info command that returns project information."""
+    
+    name = "info"
+    
+    def execute(self) -> CommandResult:
+        return CommandResult(
+            success=True, 
+            data={
+                "project": "myproject",
+                "version": "1.0.0",
+                "description": "Example project with custom commands"
+            }
+        )
+    
+    @classmethod
+    def get_param_info(cls) -> dict:
+        return {}
+```
+
+## Запуск
+
+```bash
+python main.py
+```
+
+При запуске сервис автоматически обнаружит команды из пакета `myproject.commands` благодаря настройке `discovery_path` в конфигурации. 

mcp_proxy_adapter/examples/custom_project_example/README_EN.md

@@ -0,0 +1,103 @@
+# Custom Project Example with Commands
+
+This example shows how to create your own project with commands and configure `discovery_path` for their automatic discovery.
+
+## Project Structure
+
+```
+myproject/
+├── config.json
+├── main.py
+├── myproject/
+│   ├── __init__.py
+│   └── commands/
+│       ├── __init__.py
+│       ├── echo_command.py
+│       └── info_command.py
+└── README.md
+```
+
+## Configuration
+
+In the `config.json` file, specify the path to our commands:
+
+```json
+{
+  "server": {
+    "host": "127.0.0.1",
+    "port": 8001,
+    "debug": true,
+    "log_level": "DEBUG"
+  },
+  "logging": {
+    "level": "DEBUG",
+    "log_dir": "./logs",
+    "log_file": "myproject.log"
+  },
+  "commands": {
+    "auto_discovery": true,
+    "discovery_path": "myproject.commands"
+  }
+}
+```
+
+## Commands
+
+### echo_command.py
+
+```python
+from mcp_proxy_adapter.commands.base import Command
+from mcp_proxy_adapter.commands.result import CommandResult
+
+class EchoCommand(Command):
+    """Echo command that returns the input text."""
+    
+    name = "echo"
+    
+    def execute(self, text: str) -> CommandResult:
+        return CommandResult(success=True, data={"echo": text})
+    
+    @classmethod
+    def get_param_info(cls) -> dict:
+        return {
+            "text": {
+                "type": "string",
+                "description": "Text to echo",
+                "required": True
+            }
+        }
+```
+
+### info_command.py
+
+```python
+from mcp_proxy_adapter.commands.base import Command
+from mcp_proxy_adapter.commands.result import CommandResult
+
+class InfoCommand(Command):
+    """Info command that returns project information."""
+    
+    name = "info"
+    
+    def execute(self) -> CommandResult:
+        return CommandResult(
+            success=True, 
+            data={
+                "project": "myproject",
+                "version": "1.0.0",
+                "description": "Example project with custom commands"
+            }
+        )
+    
+    @classmethod
+    def get_param_info(cls) -> dict:
+        return {}
+```
+
+## Running
+
+```bash
+python main.py
+```
+
+When starting, the service will automatically discover commands from the `myproject.commands` package thanks to the `discovery_path` configuration setting.