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

import graphene
from django.conf import settings
from django.contrib.auth import authenticate, login
from graphene import relay

from core.hep_client import HepClient, HepClientUnauthorizedException
from registration.models import License
from users.models import MagentoToken, User, Role, UserRole, SchoolClass


class LoginError(graphene.ObjectType):
    field = graphene.String()


class Login(relay.ClientIDMutation):
    class Input:
        username_input = graphene.String()
        password_input = graphene.String()

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

    @classmethod
    def mutate_and_get_payload(cls, root, info, **kwargs):

        username = kwargs.get('username_input')
        password = kwargs.get('password_input')

        if settings.USE_LOCAL_REGISTRATION:
            user = authenticate(username=username, password=password)
            if user is None:
                error = LoginError(field='invalid_credentials')
                return cls(success=False, errors=[error])

        else:
            hep_client = HepClient()

            try:
                token = hep_client.customer_token(username, password)
            except HepClientUnauthorizedException:
                error = LoginError(field='invalid_credentials')
                return cls(success=False, errors=[error])

            try:
                user = User.objects.get(email=username)
            except User.DoesNotExist:
                user = User.objects.create_user_from_hep(token)

            magento_token, created = MagentoToken.objects.get_or_create(user=user)
            magento_token.token = token['token']
            magento_token.save()

            if not hep_client.is_email_verified(username):
                error = LoginError(field='email_not_verified')
                return cls(success=False, errors=[error])

            try:
                license = License.objects.get(licensee=user)
                # Todo how handle invalid license? Cron Job? How to select correct license? Save all licenses? History?
            except License.DoesNotExist:
                product = hep_client.myskillbox_product_for_customer(settings.HEP_ADMIN_TOKEN, user.hep_id)

                if product:
                    license = License.objects.create_license_for_role(user, product['activated'],
                                                                      product['raw'], product['edition'])
                    # todo handle no license case
                else:
                    error = LoginError(field='no_valid_license')
                    return cls(success=False, errors=[error])

                UserRole.objects.create_role_for_user(user, license.for_role.key)

                if license.for_role.key == Role.objects.TEACHER_KEY:
                    SchoolClass.create_default_group_for_teacher(user)

            if not license.is_valid():
                error = LoginError(field='no_valid_license')
                return cls(success=False, errors=[error])



        login(info.context, user)
        return cls(success=True, errors=[])


class UserMutations:
    login = Login.Field()