from __future__ import annotations

from typing import TYPE_CHECKING

from django.core.exceptions import ValidationError

import strawberry_django

from .utils import get_current_user

if TYPE_CHECKING:
    from strawberry.types import Info

    from strawberry_django.utils.typing import UserType


def resolve_current_user(info: Info) -> UserType:
    user = get_current_user(info)

    if not getattr(user, "is_authenticated", False):
        raise ValidationError("User is not logged in.")

    return user


def current_user():
    return strawberry_django.field(resolver=resolve_current_user)