import graphene
from graphene import relay
from rest_framework.exceptions import PermissionDenied

from api.utils import get_object
from books.schema.inputs import UserGroupBlockVisibility
from core.utils import set_visible_for, set_hidden_for
from objectives.inputs import AddObjectiveArgument
from objectives.models import ObjectiveProgressStatus, Objective, ObjectiveGroup
from objectives.schema import ObjectiveNode, ObjectiveGroupNode


class UpdateObjectiveProgress(relay.ClientIDMutation):
    class Input:
        id = graphene.ID(required=True, description="The ID of the objective")
        done = graphene.Boolean(required=True)

    errors = graphene.List(graphene.String)
    objective = graphene.Field(ObjectiveNode)

    @classmethod
    def mutate_and_get_payload(cls, root, info, **kwargs):
        objective_id = kwargs.get('id')
        done = kwargs.get('done')
        objective = get_object(Objective, objective_id)  # info.context.user = django user
        objective_progress, created = ObjectiveProgressStatus.objects.get_or_create(
            objective=objective,
            user=info.context.user
        )

        objective_progress.done = done
        objective_progress.save()

        return cls(objective=objective)


class UpdateObjectiveVisibility(relay.ClientIDMutation):
    class Input:
        id = graphene.ID(required=True, description='The ID of the objective')
        visibility = graphene.List(UserGroupBlockVisibility)

    objective = graphene.Field(ObjectiveNode)

    @classmethod
    def mutate_and_get_payload(cls, root, info, **kwargs):
        objective_id = kwargs.get('id')
        visibility_list = kwargs.get('visibility')
        objective = get_object(Objective, objective_id)  # info.context.user = django user

        if visibility_list is not None:
            if objective.owner is not None:
                set_visible_for(objective, visibility_list)
            else:
                set_hidden_for(objective, visibility_list)

                objective.save()

        return cls(objective=objective)


class AddObjective(relay.ClientIDMutation):
    class Input:
        objective = graphene.Argument(AddObjectiveArgument)

    objective = graphene.Field(ObjectiveNode)

    @classmethod
    def mutate_and_get_payload(cls, root, info, **kwargs):
        owner = info.context.user
        if not owner.has_perm('users.can_manage_school_class_content'):
            raise PermissionDenied('Missing permissions')

        objective_data = kwargs.get('objective')
        objective_group_id = objective_data.get('objective_group')
        text = objective_data.get('text')

        objective_group = get_object(ObjectiveGroup, objective_group_id)

        objective = Objective.objects.create(text=text, owner=owner, group=objective_group)

        return cls(objective=objective)


class DeleteObjective(relay.ClientIDMutation):
    class Input:
        id = graphene.ID(required=True)

    success = graphene.Boolean()

    @classmethod
    def mutate_and_get_payload(cls, root, info, **kwargs):
        id = kwargs.get('id')

        objective = get_object(Objective, id)
        user = info.context.user

        if not user.has_perm('users.can_manage_school_class_content'):
            raise PermissionDenied('Missing permissions')
        if objective.owner.pk != user.pk:
            raise PermissionDenied('Permission denied: Not owner')

        objective.delete()
        return cls(success=True)


class ObjectiveMutations:
    update_objective_progress = UpdateObjectiveProgress.Field()
    update_objective_visibility = UpdateObjectiveVisibility.Field()
    add_objective = AddObjective.Field()
    delete_objective = DeleteObjective.Field()