# -*- coding: utf-8 -*-
#
# ITerativ GmbH
# http://www.iterativ.ch/
#
# Copyright (c) 2019 ITerativ GmbH. All rights reserved.
#
# Created on 2019-10-08
# @author: chrigu <christian.cueni@iterativ.ch>
import graphene
from graphene import relay

from core.views import SetPasswordView
from registration.models import License
from registration.serializers import RegistrationSerializer
from users.models import User, Role, UserRole, SchoolClass


class PublicFieldError(graphene.ObjectType):
    code = graphene.String()


class MutationError(graphene.ObjectType):
    field = graphene.String()
    errors = graphene.List(PublicFieldError)


class Registration(relay.ClientIDMutation):
    class Input:
        firstname_input = graphene.String()
        lastname_input = graphene.String()
        email_input = graphene.String()
        license_key_input = graphene.String()

    success = graphene.Boolean()
    errors = graphene.List(MutationError)  # todo: change for consistency

    @classmethod
    def mutate_and_get_payload(cls, root, info, **kwargs):
        first_name = kwargs.get('firstname_input')
        last_name = kwargs.get('lastname_input')
        email = kwargs.get('email_input')
        license_key = kwargs.get('license_key_input')
        registration_data = {
            'first_name': first_name,
            'last_name': last_name,
            'email': email,
            'license_key': license_key,
        }

        serializer = RegistrationSerializer(data=registration_data)

        if serializer.is_valid():
            user = User.objects.create_user_with_random_password(serializer.data['first_name'],
                                                                 serializer.data['last_name'],
                                                                 serializer.data['email'])
            sb_license = License.objects.create(licensee=user, license_type=serializer.context['license_type'])

            if sb_license.license_type.is_teacher_license():
                teacher_role = Role.objects.get(key=Role.objects.TEACHER_KEY)
                UserRole.objects.get_or_create(user=user, role=teacher_role)
                default_class_name = SchoolClass.generate_default_group_name()
                default_class = SchoolClass.objects.create(name=default_class_name)
                user.school_classes.add(default_class)
            else:
                student_role = Role.objects.get(key=Role.objects.STUDENT_KEY)
                UserRole.objects.get_or_create(user=user, role=student_role)

            password_reset_view = SetPasswordView()
            password_reset_view.request = info.context
            form = password_reset_view.form_class({'email': user.email})

            if not form.is_valid():
                return cls(success=False, errors=form.errors)

            password_reset_view.form_valid(form)

            return cls(success=True)

        errors = []
        for key, value in serializer.errors.items():
            error = MutationError(field=key, errors=[])
            for field_error in serializer.errors[key]:
                error.errors.append(PublicFieldError(code=field_error.code))

            errors.append(error)

        return cls(success=False, errors=errors)


class RegistrationMutations:
    registration = Registration.Field()