metadata.py

"""
Handler for OAuth 2.0 Authorization Server Metadata.

Corresponds to TypeScript file: src/server/auth/handlers/metadata.ts
"""

from typing import Callable

from starlette.requests import Request
from starlette.responses import JSONResponse, Response

from mcp.shared.auth import OAuthMetadata


def create_metadata_handler(metadata: OAuthMetadata) -> Callable:
    """
    Create a handler for OAuth 2.0 Authorization Server Metadata.

    Corresponds to metadataHandler in src/server/auth/handlers/metadata.ts

    Args:
        metadata: The metadata to return in the response

    Returns:
        A Starlette endpoint handler function
    """

    async def metadata_handler(request: Request) -> Response:
        """
        Handler for the OAuth 2.0 Authorization Server Metadata endpoint.

        Args:
            request: The Starlette request

        Returns:
            JSON response with the authorization server metadata
        """
        # Convert metadata to dict and remove any None values
        metadata_dict = metadata.model_dump()
        clean_metadata = {k: v for k, v in metadata_dict.items() if v is not None}

        return JSONResponse(
            content=clean_metadata,
            headers={"Cache-Control": "public, max-age=3600"},  # Cache for 1 hour
        )

    return metadata_handler