# api/views.py
from rest_framework import generics, viewsets, permissions,status
from django.contrib.auth import get_user_model
from rest_framework_simplejwt.views import TokenObtainPairView
from .serializers import RegisterSerializer, LinkSerializer, LinkGroupSerializer
from .models import Link, LinkGroup
from rest_framework.views import APIView
from rest_framework.response import Response
from django.shortcuts import get_object_or_404
from drf_spectacular.utils import extend_schema, OpenApiParameter


User = get_user_model()

@extend_schema(
    parameters=[
        OpenApiParameter(
            name='id',
            type=int,
            location=OpenApiParameter.PATH
        )
    ]
)

class RegisterView(generics.CreateAPIView):
    queryset = User.objects.all()
    permission_classes = (permissions.AllowAny,)
    serializer_class = RegisterSerializer

class LoginView(TokenObtainPairView):
    permission_classes = (permissions.AllowAny,)

class LinkGroupViewSet(viewsets.ModelViewSet):
    queryset = LinkGroup.objects.all()
    serializer_class = LinkGroupSerializer
    permission_classes = [permissions.IsAuthenticated]

    def get_queryset(self):
        # возвращаем только группы текущего пользователя
        return self.queryset.filter(owner=self.request.user).order_by('order')

class LinkViewSet(viewsets.ModelViewSet):
    serializer_class = LinkSerializer
    permission_classes = (permissions.IsAuthenticated,)

    def get_queryset(self):
        return Link.objects.filter(owner=self.request.user).order_by('order')

    def perform_create(self, serializer):
        serializer.save(owner=self.request.user)

class UserLinksListView(generics.ListAPIView):
    serializer_class = LinkSerializer
    permission_classes = (permissions.AllowAny,)

    def get_queryset(self):
        username = self.kwargs['username']
        return Link.objects.filter(owner__username=username).order_by('order')
    
    
from .serializers import UserSerializer  # нужно завести сериализатор для пользователя

User = get_user_model()

class UserProfileView(generics.RetrieveAPIView):
    """
    Возвращает данные авторизованного пользователя.
    GET /api/auth/user/
    """
    serializer_class = UserSerializer
    permission_classes = [permissions.IsAuthenticated]

    def get_object(self):
        return self.request.user
    
from users.models import User
from links.models import LinkGroup
from .serializers import PublicUserGroupsSerializer

@extend_schema(
    responses=PublicUserGroupsSerializer
)
class PublicUserGroupsView(generics.GenericAPIView):
    """
    Возвращает публичные группы и их ссылки для заданного username.
    GET /api/users/{username}/public/
    """
    serializer_class = PublicUserGroupsSerializer
    permission_classes = [permissions.AllowAny]

    def get(self, request, username):
        user = get_object_or_404(User, username=username)
        groups = LinkGroup.objects.filter(owner=user, is_public=True).prefetch_related('links')
        # Формируем ответ в виде словаря — GenericAPIView сам вызовет нужный сериализатор:
        return Response({
            'username': user.username,
            'groups': groups
        })