Spaces:
Running
Running
File size: 1,301 Bytes
3959209 341332a 3959209 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | """Provider registry for FastMCP auth providers."""
from __future__ import annotations
from collections.abc import Callable
from typing import TYPE_CHECKING, TypeVar
if TYPE_CHECKING:
from fastmcp.server.auth import AuthProvider
# Type variable for auth providers
T = TypeVar("T", bound="AuthProvider")
# Provider Registry
_PROVIDER_REGISTRY: dict[str, type[AuthProvider]] = {}
def register_provider(name: str) -> Callable[[type[T]], type[T]]:
"""Decorator to register an auth provider with a given name.
Args:
name: The name to register the provider under (e.g., 'AUTHKIT')
Returns:
The decorated class
Example:
@register_provider('AUTHKIT')
class AuthKitProvider(AuthProvider):
...
"""
def decorator(cls: type[T]) -> type[T]:
_PROVIDER_REGISTRY[name.upper()] = cls
return cls
return decorator
def get_registered_provider(name: str) -> type[AuthProvider]:
"""Get a registered provider by name.
Args:
name: The provider name (case-insensitive)
Returns:
The provider class if found, None otherwise
"""
if name.upper() in _PROVIDER_REGISTRY:
return _PROVIDER_REGISTRY[name.upper()]
raise ValueError(f"Provider {name!r} has not been registered.")
|