from mcp.server import MCPServer
from fastapi import FastAPI
import logging

logger = logging.getLogger(__name__)

_mcp_server = None

def init_mcp_server(app: FastAPI):
    """Initialize MCP server"""
    global _mcp_server
    try:
        _mcp_server = MCPServer(app)
        logger.info("✓ MCP Server initialized successfully")
        return _mcp_server
    except Exception as e:
        logger.error(f"✗ Failed to initialize MCP server: {str(e)}")
        raise

def get_mcp_server() -> MCPServer:
    """Get MCP server instance"""
    if _mcp_server is None:
        raise RuntimeError("MCP server not initialized")
    return _mcp_server