core/auth/handler.py

from ariadne.asgi.handlers import GraphQLHTTPHandler
from starlette.requests import Request
from starlette.responses import JSONResponse

from auth.middleware import auth_middleware
from utils.logger import root_logger as logger


class EnhancedGraphQLHTTPHandler(GraphQLHTTPHandler):
    """
    Улучшенный GraphQL HTTP обработчик с поддержкой cookie и авторизации.

    Расширяет стандартный GraphQLHTTPHandler для:
    1. Создания расширенного контекста запроса с авторизационными данными
    2. Корректной обработки ответов с cookie и headers
    3. Интеграции с AuthMiddleware
    """

    async def get_context_for_request(self, request: Request, data: dict) -> dict:
        """
        Расширяем контекст для GraphQL запросов.

        Добавляет к стандартному контексту:
        - Объект response для установки cookie
        - Интеграцию с AuthMiddleware
        - Расширения для управления авторизацией

        Args:
            request: Starlette Request объект
            data: данные запроса

        Returns:
            dict: контекст с дополнительными данными для авторизации и cookie
        """
        # Получаем стандартный контекст от базового класса
        context = await super().get_context_for_request(request, data)

        # Создаем объект ответа для установки cookie
        response = JSONResponse({})
        context["response"] = response

        # Интегрируем с AuthMiddleware
        auth_middleware.set_context(context)
        context["extensions"] = auth_middleware

        # Добавляем данные авторизации только если они доступны
        # Проверяем наличие данных авторизации в scope
        if hasattr(request, "scope") and isinstance(request.scope, dict) and "auth" in request.scope:
            auth_cred = request.scope.get("auth")
            context["auth"] = auth_cred
            # Безопасно логируем информацию о типе объекта auth
            logger.debug(f"[graphql] Добавлены данные авторизации в контекст из scope: {type(auth_cred).__name__}")

        logger.debug("[graphql] Подготовлен расширенный контекст для запроса")

        return context