feat: update IsAdminPermission on the external endpoints

chats/apps/api/v1/external/permissions.py

@@ -1,15 +1,82 @@
+from django.core.exceptions import ObjectDoesNotExist
 from rest_framework import permissions
 
 from chats.apps.projects.models import ProjectPermission
 
 
 class IsAdminPermission(permissions.BasePermission):
     def has_permission(self, request, view):  # pragma: no cover
-        auth_header = request.META.get("HTTP_AUTHORIZATION")
+        if view.action in ["list", "create"]:
+            try:
+                permission = request.auth
+                project = permission.project
+
+                validation = ValidatePermissionRequest(
+                    request_data=request.data or request.query_params, project=project
+                )
+
+                return validation.is_valid
+            except (AttributeError, IndexError, ProjectPermission.DoesNotExist):
+                return False
+
+        return super().has_permission(request, view)
+
+    def has_object_permission(self, request, view, obj):
+        """
+        Return `True` if permission is granted, `False` otherwise.
+        """
+        try:
+            permission = request.auth
+            project = obj.project
+        except ProjectPermission.DoesNotExist:
+            return False
+        return permission.project == project
+
+
+LEVEL_NAME_MAPPING = {
+    "project_uuid": "project",
+    "project": "project",
+    "sector_uuid": "pk",
+    "sector": "pk",
+    "queue_uuid": "queues",
+    "queue": "queues",
+    "room__uuid": "queues__rooms",
+    "room": "queues__rooms",
+}
+
+
+class ValidatePermissionRequest:
+    def __init__(self, request_data, project) -> None:
+        self.project = project
+        self.data = request_data
+        self.queryset = {}
+
+        # Get the instance type and pk
+        for key in [
+            "project",
+            "sector",
+            "queue",
+            "project_uuid",
+            "sector_uuid",
+            "queue_uuid",
+            "room__uuid",
+            "room",
+        ]:
+            self.level_name = LEVEL_NAME_MAPPING[key]
+            self.level_id = self.data.get(key, None)
+            if self.level_name != "project" and self.level_id is not None:
+                self.queryset = {self.level_name: self.level_id}
+                break
+            elif self.level_name == "project" and self.level_id is not None:
+                break
+
+    @property
+    def is_valid(self):
         try:
-            auth_token = auth_header.split()[1]
-            return (
-                True if ProjectPermission.objects.get(pk=auth_token, role=1) else False
-            )
-        except (AttributeError, IndexError, ProjectPermission.DoesNotExist):
+            if self.level_name == "project":
+                return str(self.project.pk) == self.level_id
+            if self.queryset != {}:
+                return self.project.sectors.filter(**self.queryset).exists()
+        except ObjectDoesNotExist:
             return False
+        return False

chats/apps/api/v1/external/rooms/viewsets.py

@@ -197,9 +197,6 @@ def partial_update(self, request, pk=None):
 
 class CustomFieldsUserExternalViewSet(viewsets.ViewSet):
     serializer_class = RoomFlowSerializer
-    permission_classes = [
-        IsAdminPermission,
-    ]
     authentication_classes = [ProjectAdminAuthentication]
 
     def partial_update(self, request, pk=None):
@@ -210,7 +207,7 @@ def partial_update(self, request, pk=None):
             return Response(
                 {"Detail": "No contact id on the request"}, status.HTTP_400_BAD_REQUEST
             )
-        request_permission = self.request.auth
+        request_permission = request.auth
         project = request_permission.project
 
         response = FlowRESTClient().create_contact(

chats/apps/dashboard/models.py

@@ -23,3 +23,7 @@ class Meta:
 
     def __str__(self):
         return self.room.queue.name
+
+    @property
+    def project(self):
+        return self.room.project

chats/apps/msgs/models.py

@@ -87,6 +87,10 @@ def notify_room(self, action: str, callback: bool = False):
                 headers={"content-type": "application/json"},
             )
 
+    @property
+    def project(self):
+        return self.room.project
+
 
 class MessageMedia(BaseModel):
     message = models.ForeignKey(
@@ -146,3 +150,7 @@ def callback(self):
     def notify_room(self, *args, **kwargs):
         """ """
         self.message.notify_room(*args, **kwargs)
+
+    @property
+    def project(self):
+        return self.message.project

chats/apps/projects/models.py

@@ -351,3 +351,7 @@ class Meta:
 
     def __str__(self):
         return self.receiver_type + ": " + self.external_id
+
+    @property
+    def project(self):
+        return self.flow_start.project

chats/apps/queues/models.py

@@ -86,6 +86,10 @@ def set_queue_authorization(self, user, role: int):
         )
         return sector_auth
 
+    @property
+    def project(self):
+        return self.sector.project
+
 
 class QueueAuthorization(BaseModel):
     ROLE_NOT_SETTED = 0
@@ -132,6 +136,10 @@ def get_permission(self, user):
     def sector(self):
         return self.queue.sector
 
+    @property
+    def project(self):
+        return self.queue.project
+
     @property
     def is_agent(self):
         return self.role == self.ROLE_AGENT

chats/apps/rooms/models.py

@@ -103,6 +103,10 @@ def get_is_waiting(self):
         check_flowstarts = self.flowstarts.filter(is_deleted=False).exists()
         return check_messages or check_flowstarts
 
+    @property
+    def project(self):
+        return self.queue.project
+
     @property
     def last_contact_message(self):
         return (

chats/apps/sectors/models.py

@@ -250,6 +250,10 @@ def notify_user(self, action):
             action=f"sector_authorization.{action}",
         )
 
+    @property
+    def project(self):
+        return self.sector.project
+
 
 class SectorTag(BaseModel):
     name = models.CharField(_("Name"), max_length=120)
@@ -276,3 +280,7 @@ class Meta:
 
     def __str__(self):
         return self.name
+
+    @property
+    def project(self):
+        return self.sector.project