"""
Middleware module for payload-guard.
Provides web framework integrations.
"""

from .fastapi import PayloadGuardMiddleware as FastAPIPayloadGuardMiddleware
from .fastapi import guard_middleware as fastapi_guard_middleware
from .fastapi import route_guard as fastapi_route_guard

from .flask import PayloadGuardMiddleware as FlaskPayloadGuardMiddleware
from .flask import guard_middleware as flask_guard_middleware
from .flask import route_guard as flask_route_guard
from .flask import get_sanitized_body as flask_get_sanitized_body

from .django import PayloadGuardMiddleware as DjangoPayloadGuardMiddleware
from .django import guard_middleware as django_guard_middleware
from .django import route_guard as django_route_guard
from .django import get_sanitized_body as django_get_sanitized_body

__all__ = [
    # FastAPI
    "FastAPIPayloadGuardMiddleware",
    "fastapi_guard_middleware",
    "fastapi_route_guard",
    # Flask
    "FlaskPayloadGuardMiddleware",
    "flask_guard_middleware",
    "flask_route_guard",
    "flask_get_sanitized_body",
    # Django
    "DjangoPayloadGuardMiddleware",
    "django_guard_middleware",
    "django_route_guard",
    "django_get_sanitized_body",
]
