Fix error with permissions in project chats

Author: yakser

chats/permissions.py

@@ -0,0 +1,11 @@
+from rest_framework.permissions import BasePermission
+
+
+class IsProjectChatMember(BasePermission):
+    def has_object_permission(self, request, view, obj):
+        collaborators = [
+            collaborator.user for collaborator in obj.project.collaborator_set.all()
+        ]
+        if request.user in collaborators:
+            return True
+        return False

chats/serializers.py

@@ -43,7 +43,6 @@ class Meta:
         fields = [
             "id",
             "users",
-            "messages",
         ]
 
 
@@ -82,7 +81,6 @@ class Meta:
             "id",
             "name",
             "image_address",
-            "messages",
             "users",
         ]
 

chats/views.py

@@ -12,6 +12,7 @@
 
 from chats.models import ProjectChat, DirectChat
 from chats.pagination import MessageListPagination
+from chats.permissions import IsProjectChatMember
 from chats.serializers import (
     DirectChatListSerializer,
     DirectChatMessageListSerializer,
@@ -47,7 +48,7 @@ def get_queryset(self):
 class ProjectChatDetail(RetrieveAPIView):
     queryset = ProjectChat.objects.all()
     serializer_class = ProjectChatDetailSerializer
-    permission_classes = [IsAuthenticated]
+    permission_classes = [IsProjectChatMember]
 
 
 class DirectChatDetail(RetrieveAPIView):