notifier/services/auth.py

from functools import wraps
import aiohttp

from orm.author import Author
from services.db import local_session
from settings import AUTH_URL


from functools import wraps
import aiohttp
from aiohttp.web import HTTPUnauthorized

from settings import AUTH_URL


async def check_auth(req) -> (bool, int | None):
    token = req.headers.get("Authorization")
    if token:
        # Logging the authentication token
        print(f"[services.auth] checking auth token: {token}")
        query_name = "validate_jwt_token"
        opeation = "ValidateToken"
        headers = {
            "Content-Type": "application/json",
        }

        variables = {
            "params": {
                "token_type": "access_token",
                "token": token.encode("utf-8"),
            }
        }

        gql = {
            "query": f"query {opeation}($params: ValidateJWTTokenInput!) {{ {query_name}(params: $params) {{ is_valid claims }} }}",
            "variables": variables,
            "operationName": opeation,
        }
        print(f"[services.auth] Graphql: {gql}")
        try:
            # Asynchronous HTTP request to the authentication server
            async with aiohttp.ClientSession() as session:
                async with session.post(AUTH_URL, json=gql, headers=headers) as response:
                    # Logging the GraphQL response
                    response_text = await response.text()
                    print(f"[services.auth] GraphQL Response: {response_text}")

                    if response.status == 200:
                        # Parsing JSON response
                        data = await response.json()
                        errors = data.get("errors")
                        if errors:
                            print(f"[services.auth] errors: {errors}")
                        else:
                            user_id = data.get("data", {}).get(query_name, {}).get("claims", {}).get("sub")

                            if user_id:
                                # Logging the retrieved user ID
                                print(f"[services.auth] User ID retrieved: {user_id}")
                                return True, user_id
                            else:
                                # Logging when no user ID is found in the response
                                print("[services.auth] No user ID found in the response")
                    else:
                        # Logging when the request to the authentication server fails
                        print(f"[services.auth] Request failed with status: {response.status}")

        except Exception as e:
            # Handling and logging exceptions during authentication check
            print(f"[services.auth] {e}")

    return False, None


def login_required(f):
    @wraps(f)
    async def decorated_function(*args, **kwargs):
        info = args[1]
        context = info.context
        req = context.get("request")
        is_authenticated, user_id = await check_auth(req)
        if not is_authenticated:
            # Raising HTTPUnauthorized exception if the user is not authenticated
            raise HTTPUnauthorized(text="Please, login first")
        else:
            # Добавляем author_id и user_id в контекст
            with local_session() as session:
                author = session.query(Author).filter(Author.user == user_id).first()
                if author:
                    context["author_id"] = author.id
                if user_id:
                    context["user_id"] = user_id

        # Если пользователь аутентифицирован, выполняем резолвер
        return await f(*args, **kwargs)

    return decorated_function