views2

# -*- encoding:utf-8 -*-

from allauth.account.adapter import get_adapter
from allauth.account.forms import EmailAwarePasswordResetTokenGenerator
from allauth.account.models import EmailConfirmationHMAC
from allauth.account.utils import complete_signup, user_pk_to_url_str, user_username

from allauth.utils import build_absolute_uri
from django.contrib.auth.signals import user_logged_in
from django.contrib.auth import (
    login as django_login,
)
from django.conf import settings
from django.contrib.sites.shortcuts import get_current_site
from django.urls import reverse
from django.utils.decorators import method_decorator
from django.views.decorators.debug import sensitive_post_parameters
from rest_framework import status


from rest_framework.response import Response
from rest_framework.permissions import AllowAny, IsAuthenticated
from rest_framework.viewsets import ViewSet
from rest_framework.authtoken.models import Token

from allauth.account import app_settings as allauth_settings

# from apps.auths.messages import AuthMessages
from apps.contrib.api.exceptions import NotFound
from apps.contrib.api.responses import DoneResponse
from apps.auths.api.v1 import codes


from .serializers import (
    LoginSerializer, PasswordResetSerializer,
    PasswordResetConfirmSerializer, VerifyEmailSerializer,
    TokenSerializer, UserSerializer,
)

# RegisterSerializer,


sensitive_post_parameters_m = method_decorator(
    sensitive_post_parameters(
        'password', 'old_password', 'new_password1', 'new_password2'
    )
)

# TokenModel = import_callable(getattr(settings, 'REST_AUTH_TOKEN_MODEL', DefaultTokenModel))
# create_token = import_callable(getattr(settings, 'REST_AUTH_TOKEN_CREATOR', default_create_token))


def get_session(token, user):
    return Response({
        'token': token.key,
        'profile': UserSerializer(user).data,
    }, status=status.HTTP_200_OK)

class TokenAuthViewSet(ViewSet):

    authentication_classes = ()
    permission_classes = [AllowAny]

    def login(self, request):
        """
        Inicio de Sesión
        """
        serializer = LoginSerializer(data=request.data, context={'request': request})
        serializer.is_valid(raise_exception=True)

        user = serializer.validated_data['user']
        token, created = Token.objects.get_or_create(user=user)

        # user_logged_in.send(sender=user.__class__, request=request, user=user)

        # if hasattr(settings, "AUTH_TOKEN_SESSION") and settings.AUTH_TOKEN_SESSION:
        #     django_login(self.request, user)

        return get_session(token, user)

    def logout(self, request):
        """
        Clear all application sessions.
        """
        serializer = TokenSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)

        token = serializer.validated_data["token"]
        token.delete()
        return DoneResponse(**codes.LOGGED_OUT)


    def register(self, request):
        serializer = RegisterSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)

        user = serializer.save(self.request)


        token, created = TokenModel.objects.get_or_create(user=user)
        complete_signup(self.request._request,
                        user, allauth_settings.EMAIL_VERIFICATION, None)


        if allauth_settings.EMAIL_VERIFICATION == allauth_settings.EmailVerificationMethod.MANDATORY:
            return DoneResponse(**codes.CONFIRMATION_EMAIL_SENT)

        return get_session(token, user)

    def confirm_email(self, request):
        serializer = VerifyEmailSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        confirmation = EmailConfirmationHMAC.from_key(serializer.validated_data['key'])
        confirmation.confirm(self.request)
        return DoneResponse(**codes.EMAIL_CONFIRMED)



    def password_reset(self, request):
        """
        Restaurar Contraseña
        """
        serializer = PasswordResetSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        action = serializer.validated_data["action"]

        if "redirect_uri" in serializer.validated_data:
            redirect_uri = serializer.validated_data["redirect_uri"]
            AuthMessages.reset_password(request, action, redirect_uri=redirect_uri)
        else:
            self.get_allauth_reset_password(action.user)

        return DoneResponse(**codes.RESET_PASSWORD_SENT)


    def get_allauth_reset_password(self, user):
        token_generator = EmailAwarePasswordResetTokenGenerator()
        temp_key = token_generator.make_token(user)


        # send the password reset email
        path = reverse("account_reset_password_from_key", kwargs=dict(uidb36=user_pk_to_url_str(user), key=temp_key))
        url = build_absolute_uri(self.request, path)

        context = {"current_site": get_current_site(self.request),
                   "user": user, "password_reset_url": url, "request": self.request}

        if settings.ACCOUNT_AUTHENTICATION_METHOD != 'email':
            context['username'] = user_username(user)
        get_adapter(self.request).send_mail('account/email/password_reset_key',user.email, context)

    def password_reset_confirm(self, request):
        """
        Confirmar la Restauración de Contraseña
        """
        serializer = PasswordResetConfirmSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)

        password = serializer.validated_data["password"]
        action = serializer.validated_data["action"]

        action.user.set_password(password)
        action.delete()
        return DoneResponse(**codes.PASSWORD_UPDATED)