import graphene
from graphene_django import DjangoObjectType
from apps.users.models import User


class UserType(DjangoObjectType):
    """GraphQL type for User."""

    class Meta:
        model = User
        fields = '__all__'


class Query(graphene.ObjectType):
    """Root GraphQL Query."""

    all_users = graphene.List(UserType)
    me = graphene.Field(UserType)

    def resolve_all_users(root, info):
        """Resolve all users query."""
        return User.objects.all()

    def resolve_me(root, info):
        """Resolve current user query."""
        if info.context.user.is_authenticated:
            return info.context.user
        return None


schema = graphene.Schema(query=Query)