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

mcp_proxy_adapter/main.py

@@ -1,174 +1,92 @@
 #!/usr/bin/env python3
 """
-Main entry point for MCP Proxy Adapter.
+MCP Proxy Adapter - Main Entry Point
 
-This module provides the main function for running the MCP Proxy Adapter server.
+Author: Vasiliy Zdanovskiy
+email: vasilyvz@gmail.com
 """
 
-import argparse
-import asyncio
-import uvicorn
 import sys
-import os
+import ssl
+import hypercorn.asyncio
+import hypercorn.config
+import asyncio
+import argparse
 from pathlib import Path
 
-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
-)
-from mcp_proxy_adapter.core.ssl_utils import SSLUtils
+# Add the project root to the path
+sys.path.insert(0, str(Path(__file__).parent.parent))
 
-
-def parse_args():
-    """Parse command line arguments."""
-    parser = argparse.ArgumentParser(description="MCP Proxy Adapter Server")
-    parser.add_argument(
-        "--config", 
-        type=str, 
-        default=None,
-        help="Path to configuration file"
-    )
-    parser.add_argument(
-        "--host",
-        type=str,
-        default=None,
-        help="Host to bind to (overrides config)"
-    )
-    parser.add_argument(
-        "--port",
-        type=int,
-        default=None,
-        help="Port to bind to (overrides config)"
-    )
-    parser.add_argument(
-        "--debug",
-        action="store_true",
-        help="Enable debug mode (overrides config)"
-    )
-    parser.add_argument(
-        "--log-level",
-        type=str,
-        default=None,
-        choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
-        help="Log level (overrides config)"
-    )
-    return parser.parse_args()
+from mcp_proxy_adapter.api.app import create_app
+from mcp_proxy_adapter.config import Config
 
 
 def main():
-    """Run the MCP Proxy Adapter server."""
-    args = parse_args()
+    """Main entry point for the MCP Proxy Adapter."""
+    # Parse command line arguments
+    parser = argparse.ArgumentParser(description="MCP Proxy Adapter Server")
+    parser.add_argument("--config", "-c", type=str, help="Path to configuration file")
+    args = parser.parse_args()
     
-    # Load configuration if specified
+    # Load configuration
     if args.config:
-        config_path = Path(args.config)
-        if config_path.exists():
-            from mcp_proxy_adapter.config import config
-            config.load_from_file(str(config_path))
-            print(f"✅ Loaded configuration from: {config_path}")
-        else:
-            print(f"❌ Configuration file not found: {config_path}")
-            sys.exit(1)
+        config = Config(config_path=args.config)
     else:
-        print("⚠️  No configuration file specified, using defaults")
-    
-    # Setup logging with configuration
-    setup_logging()
-    logger = get_logger("mcp_proxy_adapter")
-    
-    # Get settings from configuration
-    server_settings = Settings.get_server_settings()
-    logging_settings = Settings.get_logging_settings()
-    commands_settings = Settings.get_commands_settings()
-    ssl_settings = Settings.get_custom_setting("ssl", {})
-    security_settings = Settings.get_custom_setting("security", {})
-    
-    # STRICT CONFIGURATION VALIDATION
-    from mcp_proxy_adapter.core.config_validator import ConfigValidator
-    
-    # Get full config for validation
-    full_config = {
-        "server": server_settings,
-        "logging": logging_settings,
-        "commands": commands_settings,
-        "ssl": ssl_settings,
-        "security": security_settings,
-        "auth_enabled": Settings.get_custom_setting("auth_enabled", False),
-        "roles": Settings.get_custom_setting("roles", {})
-    }
-    
-    # Validate configuration
-    validator = ConfigValidator(full_config)
-    if not validator.validate_all():
-        logger.critical("CRITICAL SECURITY ERROR: Configuration validation failed")
-        validator.print_validation_report()
-        logger.critical("Server startup blocked for security reasons.")
-        logger.critical("Please fix configuration errors or disable security features.")
-        sys.exit(1)
+        config = Config()
+
+
+    # Create application
+    app = create_app(app_config=config)
     
-    logger.info("Configuration validation passed")
+    # Get server configuration
+    host = config.get("server.host", "0.0.0.0")
+    port = config.get("server.port", 8000)
     
-    # Override settings with command line arguments
-    if args.host:
-        server_settings['host'] = args.host
-    if args.port:
-        server_settings['port'] = args.port
-    if args.debug:
-        server_settings['debug'] = True
-    if args.log_level:
-        logging_settings['level'] = args.log_level
-        server_settings['log_level'] = args.log_level
+    # Get SSL configuration
+    ssl_enabled = config.get("ssl.enabled", False)
+    ssl_cert_file = config.get("ssl.cert_file")
+    ssl_key_file = config.get("ssl.key_file")
+    ssl_ca_cert = config.get("ssl.ca_cert")
+    verify_client = config.get("ssl.verify_client", False)
     
-    # Print server header and description
-    print("=" * 80)
-    print("🚀 MCP PROXY ADAPTER SERVER")
-    print("=" * 80)
-    print("📋 Configuration:")
-    print(f"   • Server: {server_settings['host']}:{server_settings['port']}")
-    print(f"   • Debug: {server_settings['debug']}")
-    print(f"   • Log Level: {logging_settings['level']}")
-    print(f"   • Auto Discovery: {commands_settings['auto_discovery']}")
-    print(f"   • SSL Enabled: {ssl_settings.get('enabled', False)}")
-    print(f"   • Security Enabled: {security_settings.get('enabled', False)}")
-    if ssl_settings.get('enabled', False):
-        print(f"   • SSL Mode: {ssl_settings.get('mode', 'https_only')}")
-    if security_settings.get('enabled', False):
-        print(f"   • Security Framework: {security_settings.get('framework', 'built-in')}")
-    print("=" * 80)
-    print()
+    print(f"🚀 Starting MCP Proxy Adapter")
+    print(f"🌐 Server: {host}:{port}")
+    if ssl_enabled:
+        print(f"🔐 SSL: Enabled")
+        print(f"   Certificate: {ssl_cert_file}")
+        print(f"   Key: {ssl_key_file}")
+        if ssl_ca_cert:
+            print(f"   CA: {ssl_ca_cert}")
+        print(f"   Client verification: {verify_client}")
+    print("=" * 50)
     
-    logger.info("Starting MCP Proxy Adapter Server...")
-    logger.info(f"Server configuration: {server_settings}")
-    logger.info(f"Security configuration: {security_settings}")
+    # Configure hypercorn
+    config_hypercorn = hypercorn.config.Config()
+    config_hypercorn.bind = [f"{host}:{port}"]
     
-    try:
-        # Create application
-        app = create_app(
-            title="MCP Proxy Adapter Server",
-            description="Model Context Protocol Proxy Adapter with Security Framework",
-            version="1.0.0"
-        )
+    if ssl_enabled and ssl_cert_file and ssl_key_file:
+        config_hypercorn.certfile = ssl_cert_file
+        config_hypercorn.keyfile = ssl_key_file
         
-        # Get SSL configuration for uvicorn
-        uvicorn_ssl_config = SSLUtils.get_ssl_config_for_uvicorn(ssl_settings)
+        if ssl_ca_cert:
+            config_hypercorn.ca_certs = ssl_ca_cert
         
-        # Run the server
-        uvicorn.run(
-            app,
-            host=server_settings['host'],
-            port=server_settings['port'],
-            log_level=server_settings.get('log_level', 'info').lower(),
-            **uvicorn_ssl_config
-        )
+        if verify_client:
+            # For mTLS, require client certificates
+            config_hypercorn.set_cert_reqs(ssl.CERT_REQUIRED)
+            config_hypercorn.verify_mode = ssl.CERT_REQUIRED
+            print("🔐 mTLS: Client certificate verification enabled")
+        else:
+            # For regular HTTPS without client verification
+            config_hypercorn.set_cert_reqs(ssl.CERT_NONE)
+            config_hypercorn.verify_mode = ssl.CERT_NONE
         
-    except Exception as e:
-        logger.error(f"Failed to start server: {e}")
-        sys.exit(1)
+        print(f"🔐 Starting HTTPS server with hypercorn...")
+    else:
+        print(f"🌐 Starting HTTP server with hypercorn...")
+    
+    # Run the server
+    asyncio.run(hypercorn.asyncio.serve(app, config_hypercorn))
 
 
 if __name__ == "__main__":