from mindmap.serializers import MindmapSerializer, MindmapListSerializer
from mindmap.models import Mindmap
from rest_framework.response import Response
from rest_framework import status
from rest_framework.views import APIView
from rest_framework.exceptions import NotFound
from config.authentication import JWTAuthenticationWithUserCheck, IsAuthenticatedAndOwner

class MindmapListView(APIView):
    authentication_classes = [JWTAuthenticationWithUserCheck]
    permission_classes = [IsAuthenticatedAndOwner]

    def get(self, request):
        mindmaps = Mindmap.objects.filter(user=request.user.publicIdentifier)
        serializer = MindmapListSerializer(mindmaps, many=True)
        return Response(serializer.data)

    def post(self, request):
        # Ajout automatique de l'utilisateur aux données
        data = request.data.copy()
        data['user'] = request.user.publicIdentifier
        
        serializer = MindmapSerializer(data=data)
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.data, status=status.HTTP_201_CREATED)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)


class MindmapDetailView(APIView):
    authentication_classes = [JWTAuthenticationWithUserCheck]
    permission_classes = [IsAuthenticatedAndOwner]

    def get_object(self, pid):
        try:
            return Mindmap.objects.get(publicIdentifier=pid)
        except Mindmap.DoesNotExist:
            raise NotFound({"error": "Mindmap not found"})

    def get(self, request, pid):
        mindmap = self.get_object(pid)
        self.check_object_permissions(request, mindmap)
        serializer = MindmapSerializer(mindmap)
        return Response(serializer.data)

    def put(self, request, pid):
        mindmap = self.get_object(pid)
        self.check_object_permissions(request, mindmap)
        
        data = request.data.copy()
        data['user'] = request.user.publicIdentifier
        
        serializer = MindmapSerializer(mindmap, data=data)
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.data)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

    def delete(self, request, pid):
        mindmap = self.get_object(pid)
        self.check_object_permissions(request, mindmap)
        mindmap.delete()
        return Response(status=status.HTTP_204_NO_CONTENT)