from django.contrib.auth import get_user_model
from django.contrib.auth.mixins import LoginRequiredMixin
from django.contrib.messages.views import SuccessMessageMixin
from django.urls import reverse
from django.utils.translation import gettext_lazy as _
from django.views.generic import DetailView, RedirectView, UpdateView
from rest_framework_simplejwt.views import TokenObtainPairView
from rest_framework.response import Response
from verify_trusted.users.api.serializers.token import UserTokenObtainPairSerializer
from verify_trusted.users.models import *
User = get_user_model()


class UserDetailView(LoginRequiredMixin, DetailView):
    model = User
    slug_field = "username"
    slug_url_kwarg = "username"


user_detail_view = UserDetailView.as_view()


class UserUpdateView(LoginRequiredMixin, SuccessMessageMixin, UpdateView):
    model = User
    fields = ["name"]
    success_message = _("Information successfully updated")

    def get_success_url(self):
        return self.request.user.get_absolute_url()  # type: ignore [union-attr]

    def get_object(self):
        return self.request.user


user_update_view = UserUpdateView.as_view()


class UserRedirectView(LoginRequiredMixin, RedirectView):
    permanent = False

    def get_redirect_url(self):
        return reverse("users:detail", kwargs={"username": self.request.user.username})


user_redirect_view = UserRedirectView.as_view()



from rest_framework.response import Response
from rest_framework.views import APIView

class UserTokenObtainPairView(TokenObtainPairView):
    serializer_class = UserTokenObtainPairSerializer
    def post(self, request, *args, **kwargs):
        shopify_url = request.data.get('shopify_url')
        request_data = request.data.copy()
        request_data.pop('shopify_url', None)

        serializer = self.get_serializer(data=request_data)
        serializer.is_valid(raise_exception=True)
    
        if shopify_url:
            token_data = serializer.validated_data
            user_data = token_data['user']
            user_id = user_data['id']
            profile = Profile.objects.get(user=user_id)
            profile.shopify_url = shopify_url
            profile.save()

        response = super().post(request, *args, **kwargs)
        return response


user_token_obtain_pair_view = UserTokenObtainPairView.as_view()