""" ASGI application for Yargı MCP Server This module provides ASGI/HTTP access to the Yargı MCP server, allowing it to be deployed as a web service with FastAPI wrapper for Stripe webhook integration. Usage: uvicorn asgi_app:app --host 0.0.0.0 --port 8000 """ import os import jwt import time import logging from datetime import datetime, timedelta from fastapi import FastAPI, Request, HTTPException, Query from fastapi.responses import JSONResponse, HTMLResponse from fastapi.exception_handlers import http_exception_handler from starlette.middleware import Middleware from starlette.middleware.cors import CORSMiddleware from starlette.responses import Response # Import the fully configured MCP app with all tools from mcp_server_main import app as mcp_server # Import Stripe webhook router from stripe_webhook import router as stripe_router # Import MCP Auth HTTP adapter from mcp_auth_http_adapter import router as mcp_auth_router # OAuth configuration from environment variables CLERK_ISSUER = os.getenv("CLERK_ISSUER", "https://accounts.yargimcp.com") BASE_URL = os.getenv("BASE_URL", "https://yargimcp.com") JWT_SECRET = os.getenv("JWT_SECRET_KEY", "your-secret-key-here") # Setup logging logger = logging.getLogger(__name__) # Configure CORS middleware cors_origins = os.getenv("ALLOWED_ORIGINS", "*").split(",") custom_middleware = [ Middleware( CORSMiddleware, allow_origins=cors_origins, allow_credentials=True, allow_methods=["GET", "POST", "OPTIONS"], allow_headers=["Content-Type", "Authorization", "X-Request-ID"], ), ] # Create MCP Starlette sub-application first mcp_app = mcp_server.http_app( path="/", middleware=custom_middleware ) # Create FastAPI wrapper application with MCP app's lifespan app = FastAPI( title="Yargı MCP Server", description="MCP server for Turkish legal databases with OAuth authentication", version="0.1.0", middleware=custom_middleware, lifespan=mcp_app.lifespan # Critical: Get lifespan from mcp_app, not mcp_server ) # Add Stripe webhook router to FastAPI app.include_router(stripe_router, prefix="/api") # Add MCP Auth HTTP adapter to FastAPI (replaces old OAuth router) app.include_router(mcp_auth_router) # Custom 401 exception handler for MCP spec compliance @app.exception_handler(401) async def custom_401_handler(request: Request, exc: HTTPException): """Custom 401 handler that adds WWW-Authenticate header as required by MCP spec""" response = await http_exception_handler(request, exc) # Add WWW-Authenticate header pointing to protected resource metadata # as required by RFC 9728 Section 5.1 and MCP Authorization spec response.headers["WWW-Authenticate"] = ( 'Bearer ' 'error="invalid_token", ' 'error_description="The access token is missing or invalid", ' f'resource="{BASE_URL}/.well-known/oauth-protected-resource"' ) return response # Mount MCP app as sub-application at /mcp-server to avoid path conflicts app.mount("/mcp-server", mcp_app) # Add custom route to handle /mcp requests and forward to mounted app @app.api_route("/mcp", methods=["GET", "POST", "OPTIONS"]) @app.api_route("/mcp/", methods=["GET", "POST", "OPTIONS"]) async def mcp_protocol_handler(request: Request): """Handle MCP protocol requests by forwarding to mounted app""" # Optional: Validate Bearer JWT tokens for direct API access auth_header = request.headers.get("Authorization") if auth_header and auth_header.startswith("Bearer "): token = auth_header.split(" ")[1] try: # Validate custom JWT token (for direct API access) user_payload = validate_mcp_token(token) logger.info(f"Bearer JWT token validated for user: {user_payload.get('user_id')}") # Add user info to request state request.state.user_id = user_payload["user_id"] request.state.token_scopes = user_payload.get("scopes", ["read", "search"]) except HTTPException as e: logger.warning(f"Bearer token validation failed: {e.detail}") # Don't fail here - let MCP Auth Toolkit handle it pass # Forward the request to the mounted MCP app async def receive(): return await request.receive() # Create new scope for the mounted app scope = request.scope.copy() scope["path"] = "/" # Root path for mounted app scope["path_info"] = "/" # Capture the response response_parts = {"status": 200, "headers": [], "body": b""} async def send(message): if message["type"] == "http.response.start": response_parts["status"] = message["status"] response_parts["headers"] = message["headers"] elif message["type"] == "http.response.body": response_parts["body"] += message.get("body", b"") # Call the mounted MCP app await mcp_app(scope, receive, send) # Return the response from starlette.responses import Response # Convert ASGI headers to dict headers = {} for name, value in response_parts["headers"]: headers[name.decode()] = value.decode() return Response( content=response_parts["body"], status_code=response_parts["status"], headers=headers ) # Add custom route to handle /sse requests for Server-Sent Events @app.api_route("/sse", methods=["GET", "POST", "OPTIONS"]) @app.api_route("/sse/", methods=["GET", "POST", "OPTIONS"]) async def sse_protocol_handler(request: Request): """Handle SSE MCP protocol requests by forwarding to mounted SSE app""" # Optional: Validate Bearer JWT tokens for direct API access auth_header = request.headers.get("Authorization") if auth_header and auth_header.startswith("Bearer "): token = auth_header.split(" ")[1] try: # Validate custom JWT token (for direct API access) user_payload = validate_mcp_token(token) logger.info(f"SSE Bearer JWT token validated for user: {user_payload.get('user_id')}") # Add user info to request state request.state.user_id = user_payload["user_id"] request.state.token_scopes = user_payload.get("scopes", ["read", "search"]) except HTTPException as e: logger.warning(f"SSE Bearer token validation failed: {e.detail}") # Don't fail here - let MCP Auth Toolkit handle it pass # Forward the request to the mounted SSE app async def receive(): return await request.receive() # Create new scope for the mounted app scope = request.scope.copy() scope["path"] = "/" # Root path for mounted app scope["path_info"] = "/" # Capture the response response_parts = {"status": 200, "headers": [], "body": b""} async def send(message): if message["type"] == "http.response.start": response_parts["status"] = message["status"] response_parts["headers"] = message["headers"] elif message["type"] == "http.response.body": response_parts["body"] += message.get("body", b"") # Call the main MCP app (same as HTTP endpoint) await mcp_app(scope, receive, send) # Return the response from starlette.responses import Response # Convert ASGI headers to dict and add SSE headers headers = {} for name, value in response_parts["headers"]: headers[name.decode()] = value.decode() # Add SSE-specific headers headers["Content-Type"] = "text/event-stream" headers["Cache-Control"] = "no-cache" headers["Connection"] = "keep-alive" headers["Access-Control-Allow-Origin"] = "*" return Response( content=response_parts["body"], status_code=response_parts["status"], headers=headers ) # FastAPI health check endpoint @app.get("/health") async def health_check(): """Health check endpoint for monitoring""" return JSONResponse({ "status": "healthy", "service": "Yargı MCP Server", "version": "0.1.0", "tools_count": len(mcp_server._tool_manager._tools), "auth_enabled": os.getenv("ENABLE_AUTH", "false").lower() == "true" }) # FastAPI root endpoint @app.get("/") async def root(): """Root endpoint with service information""" return JSONResponse({ "service": "Yargı MCP Server", "description": "MCP server for Turkish legal databases with OAuth authentication", "endpoints": { "mcp": "/mcp", "sse": "/sse", "health": "/health", "status": "/status", "stripe_webhook": "/api/stripe/webhook", "oauth_login": "/auth/login", "oauth_callback": "/auth/callback", "oauth_google": "/auth/google/login", "user_info": "/auth/user" }, "transports": { "http": "/mcp", "sse": "/sse" }, "supported_databases": [ "Yargıtay (Court of Cassation)", "Danıştay (Council of State)", "Emsal (Precedent)", "Uyuşmazlık Mahkemesi (Court of Jurisdictional Disputes)", "Anayasa Mahkemesi (Constitutional Court)", "Kamu İhale Kurulu (Public Procurement Authority)", "Rekabet Kurumu (Competition Authority)", "Sayıştay (Court of Accounts)", "Bedesten API (Multiple courts)" ], "authentication": { "enabled": os.getenv("ENABLE_AUTH", "false").lower() == "true", "type": "OAuth 2.0 via Clerk", "issuer": os.getenv("CLERK_ISSUER", "https://clerk.accounts.dev"), "providers": ["google"], "flow": "authorization_code" } }) # OAuth 2.0 Authorization Server Metadata proxy (for MCP clients that can't reach Clerk directly) @app.get("/.well-known/oauth-authorization-server") async def oauth_authorization_server(): """OAuth 2.0 Authorization Server Metadata proxy to Clerk""" return JSONResponse({ "issuer": CLERK_ISSUER, "authorization_endpoint": f"{BASE_URL}/auth/login", "token_endpoint": f"{BASE_URL}/auth/callback", "jwks_uri": f"{CLERK_ISSUER}/.well-known/jwks.json", "response_types_supported": ["code"], "grant_types_supported": ["authorization_code", "refresh_token"], "token_endpoint_auth_methods_supported": ["client_secret_basic", "none"], "scopes_supported": ["read", "search", "openid", "profile", "email"], "subject_types_supported": ["public"], "id_token_signing_alg_values_supported": ["RS256"], "claims_supported": ["sub", "iss", "aud", "exp", "iat", "email", "name"], "code_challenge_methods_supported": ["S256"], "service_documentation": f"{BASE_URL}/mcp", "registration_endpoint": f"{BASE_URL}/auth/register", "resource_documentation": f"{BASE_URL}/mcp" }) # MCP endpoint info for GET requests (ChatGPT compatibility) @app.get("/mcp") async def mcp_info(): """MCP endpoint information for discovery""" return JSONResponse({ "mcp_server": True, "name": "Yargı MCP Server", "version": "0.1.0", "description": "MCP server for Turkish legal databases", "protocol": "mcp/1.0", "transport": ["http", "sse"], "authentication_required": True, "authentication": { "type": "oauth2", "authorization_url": "https://yargimcp.com/sign-in?redirect_url=https://api.yargimcp.com/auth/mcp-callback", "token_url": f"{BASE_URL}/auth/mcp-token", "scopes": ["read", "search"], "provider": "clerk" }, "endpoints": { "mcp_protocol": "/mcp", "sse_protocol": "/sse", "discovery": "/mcp/discovery", "well_known": "/.well-known/mcp", "health": "/health", "oauth_login": "/auth/login" }, "capabilities": { "tools": True, "resources": True, "prompts": False }, "tools_count": len(mcp_server._tool_manager._tools), "usage": { "note": "This is an MCP server. Use POST to /mcp/ with proper MCP protocol headers.", "headers_required": [ "Content-Type: application/json", "Accept: application/json, text/event-stream", "Authorization: Bearer ", "X-Session-ID: " ] } }) # OAuth 2.0 Protected Resource Metadata (RFC 9728) - MCP Spec Required @app.get("/.well-known/oauth-protected-resource") async def oauth_protected_resource(): """OAuth 2.0 Protected Resource Metadata as required by MCP spec""" return JSONResponse({ "resource": BASE_URL, "authorization_servers": [ BASE_URL ], "scopes_supported": ["read", "search"], "bearer_methods_supported": ["header"], "resource_documentation": f"{BASE_URL}/mcp", "resource_policy_uri": f"{BASE_URL}/privacy" }) # Standard well-known discovery endpoint @app.get("/.well-known/mcp") async def well_known_mcp(): """Standard MCP discovery endpoint""" return JSONResponse({ "mcp_server": { "name": "Yargı MCP Server", "version": "0.1.0", "endpoint": f"{BASE_URL}/mcp", "authentication": { "type": "oauth2", "authorization_url": f"{BASE_URL}/auth/login", "scopes": ["read", "search"] }, "capabilities": ["tools", "resources"], "tools_count": len(mcp_server._tool_manager._tools) } }) # MCP Discovery endpoint for ChatGPT integration @app.get("/mcp/discovery") async def mcp_discovery(): """MCP Discovery endpoint for ChatGPT and other MCP clients""" return JSONResponse({ "name": "Yargı MCP Server", "description": "MCP server for Turkish legal databases", "version": "0.1.0", "protocol": "mcp", "transport": "http", "endpoint": "/mcp", "authentication": { "type": "oauth2", "authorization_url": "/auth/login", "token_url": "/auth/callback", "scopes": ["read", "search"], "provider": "clerk" }, "capabilities": { "tools": True, "resources": True, "prompts": False }, "tools_count": len(mcp_server._tool_manager._tools), "contact": { "url": BASE_URL, "email": "support@yargi-mcp.dev" } }) # FastAPI status endpoint @app.get("/status") async def status(): """Status endpoint with detailed information""" tools = [] for tool in mcp_server._tool_manager._tools.values(): tools.append({ "name": tool.name, "description": tool.description[:100] + "..." if len(tool.description) > 100 else tool.description }) return JSONResponse({ "status": "operational", "tools": tools, "total_tools": len(tools), "transport": "streamable_http", "architecture": "FastAPI wrapper + MCP Starlette sub-app", "auth_status": "enabled" if os.getenv("ENABLE_AUTH", "false").lower() == "true" else "disabled" }) # MCP Token Generation and Validation def generate_mcp_token(user_id: str, expires_in: int = 3600) -> str: """Generate MCP access token for authenticated user""" payload = { "user_id": user_id, "iat": int(time.time()), "exp": int(time.time()) + expires_in, "iss": BASE_URL, "aud": "mcp-client", "scopes": ["read", "search"] } return jwt.encode(payload, JWT_SECRET, algorithm="HS256") def validate_mcp_token(token: str) -> dict: """Validate MCP access token and return user info""" try: payload = jwt.decode(token, JWT_SECRET, algorithms=["HS256"]) return payload except jwt.ExpiredSignatureError: raise HTTPException(status_code=401, detail="Token expired") except jwt.InvalidTokenError: raise HTTPException(status_code=401, detail="Invalid token") async def validate_clerk_session(request: Request, clerk_token: str = None) -> str: """Validate Clerk session from cookies or JWT token and return user_id""" logger.info(f"Validating Clerk session - token provided: {bool(clerk_token)}") try: # Try to import Clerk SDK from clerk_backend_api import Clerk clerk = Clerk(bearer_auth=os.getenv("CLERK_SECRET_KEY")) # Try JWT token first (from URL parameter) if clerk_token: logger.info("Validating Clerk JWT token from URL parameter") try: # Verify JWT token with Clerk jwt_claims = clerk.jwt_templates.verify_token(clerk_token) user_id = jwt_claims.get("sub") if user_id: logger.info(f"JWT token validation successful - user_id: {user_id}") return user_id else: logger.error("JWT token validation failed - no user_id in claims") except Exception as e: logger.error(f"JWT token validation failed: {str(e)}") # Fall through to cookie validation # Fallback to cookie validation logger.info("Attempting cookie-based session validation") clerk_session = request.cookies.get("__session") if not clerk_session: logger.error("No Clerk session cookie found") raise HTTPException(status_code=401, detail="No Clerk session found") # Validate session with Clerk session = clerk.sessions.verify_session(clerk_session) logger.info(f"Cookie session validation successful - user_id: {session.user_id}") return session.user_id except ImportError: # Fallback for development without Clerk SDK logger.warning("Clerk SDK not available - using development fallback") return "dev_user_123" except Exception as e: logger.error(f"Session validation failed: {str(e)}") raise HTTPException(status_code=401, detail=f"Session validation failed: {str(e)}") # MCP OAuth Callback Endpoint @app.get("/auth/mcp-callback") async def mcp_oauth_callback(request: Request, clerk_token: str = Query(None)): """Handle OAuth callback for MCP token generation""" logger.info(f"MCP OAuth callback - clerk_token provided: {bool(clerk_token)}") try: # Validate Clerk session with JWT token support user_id = await validate_clerk_session(request, clerk_token) logger.info(f"User authenticated successfully - user_id: {user_id}") # Generate MCP token mcp_token = generate_mcp_token(user_id) logger.info("MCP token generated successfully") # Return success response return HTMLResponse(f""" MCP Connection Successful

✅ MCP Connection Successful!

Your Yargı MCP integration is now active.

Access Token:
{mcp_token}

You can now close this window and return to your MCP client.

""") except HTTPException as e: logger.error(f"MCP OAuth callback failed: {e.detail}") return HTMLResponse(f""" MCP Connection Failed

❌ MCP Connection Failed

{e.detail}

Debug Info:
Clerk Token: {'✅ Provided' if clerk_token else '❌ Missing'}
Error: {e.detail}
Status: {e.status_code}

Please try again or contact support.

Return to Sign In """, status_code=e.status_code) except Exception as e: logger.error(f"Unexpected error in MCP OAuth callback: {str(e)}") return HTMLResponse(f""" MCP Connection Error

❌ Unexpected Error

An unexpected error occurred during authentication.

Error: {str(e)}

Return to Sign In """, status_code=500) # MCP Token Endpoint (for OAuth2 compatibility) @app.post("/auth/mcp-token") async def mcp_token_endpoint(request: Request): """OAuth2 token endpoint for MCP clients""" try: # For simplicity, we'll handle this as a redirect from callback # In a full OAuth2 implementation, this would handle authorization codes user_id = await validate_clerk_session(request) mcp_token = generate_mcp_token(user_id) return JSONResponse({ "access_token": mcp_token, "token_type": "Bearer", "expires_in": 3600, "scope": "read search" }) except HTTPException as e: return JSONResponse( status_code=e.status_code, content={"error": "invalid_request", "error_description": e.detail} ) # Note: SSE endpoint uses the same mcp_app as HTTP endpoint # No separate SSE app needed - the difference is in headers and response handling # Export for uvicorn __all__ = ["app"]