Merge pull request #215 from PROCOLLAB-github/feature/unread-messages-route

Unread messages view

Author: VeryBigSad

chats/urls.py

@@ -3,6 +3,7 @@
 from chats.views import (
     DirectChatList,
     DirectChatMessageList,
+    HasChatUnreadsView,
     ProjectChatList,
     ProjectChatMessageList,
     ProjectChatDetail,
@@ -38,4 +39,9 @@
         ProjectChatFileList.as_view(),
         name="project-chat-files",
     ),
+    path(
+        "has-unreads/",
+        HasChatUnreadsView.as_view(),
+        name="has-chat-unreads",
+    ),
 ]

chats/views.py

@@ -2,11 +2,13 @@
 
 from rest_framework import status
 from rest_framework.generics import (
+    GenericAPIView,
     ListAPIView,
     ListCreateAPIView,
     RetrieveAPIView,
 )
-
+from drf_yasg.utils import swagger_auto_schema
+from drf_yasg import openapi
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.response import Response
 
@@ -190,3 +192,37 @@ def get_queryset(self):
             return get_all_files(messages)
         except ProjectChat.DoesNotExist:
             return UserFile.objects.none()
+
+
+class HasChatUnreadsView(GenericAPIView):
+    """Returns True if user has unread messages"""
+
+    permission_classes = [IsAuthenticated]
+
+    @swagger_auto_schema(
+        responses={
+            200: openapi.Response(
+                description="Gives you True or False whether user has unread messages or not",
+                schema=openapi.Schema(
+                    type=openapi.TYPE_OBJECT,
+                    properties={
+                        "has_unreads": openapi.Schema(
+                            type=openapi.TYPE_BOOLEAN,
+                            description="True if user has unread messages",
+                        )
+                    },
+                ),
+            )
+        }
+    )
+    def get(self, request, *args, **kwargs):
+        user = request.user
+        # get all user chats
+        direct_messages = user.direct_chats.all().messages.all()
+        project_messages = user.get_project_chats().messages.all()
+
+        has_direct_messages_unread = direct_messages.filter(is_read=False).exists()
+        has_project_messages_unread = project_messages.filter(is_read=False).exists()
+        return Response(
+            {"has_unreads": has_direct_messages_unread or has_project_messages_unread}
+        )

users/models.py

@@ -134,7 +134,9 @@ def calculate_ordering_score(self) -> int:
         return score
 
     def get_project_chats(self) -> list:
-        collaborations = self.collaborations.all()
+        collaborations = self.collaborations.prefetch_related(
+            "project__project_chats"
+        ).all()
         projects = []
         for collaboration in collaborations:
             projects.extend(list(collaboration.project.project_chats.all()))