OpenSlides/server/openslides/chat/views.py

from django.conf import settings
from django.contrib.auth.models import AnonymousUser
from rest_framework.utils.serializer_helpers import ReturnDict

from openslides.utils.auth import has_perm
from openslides.utils.autoupdate import (
    disable_history,
    inform_changed_data,
    inform_deleted_data,
)
from openslides.utils.rest_api import (
    CreateModelMixin,
    DestroyModelMixin,
    GenericViewSet,
    ListModelMixin,
    ModelViewSet,
    Response,
    RetrieveModelMixin,
    detail_route,
    status,
)

from .access_permissions import ChatGroupAccessPermissions, ChatMessageAccessPermissions
from .models import ChatGroup, ChatMessage


ENABLE_CHAT = getattr(settings, "ENABLE_CHAT", False)


class ChatGroupViewSet(ModelViewSet):
    """
    API endpoint for chat groups.

    There are the following views: metadata, list, retrieve, create,
    partial_update, update, destroy and clear.
    """

    access_permissions = ChatGroupAccessPermissions()
    queryset = ChatGroup.objects.all()

    def check_view_permissions(self):
        """
        Returns True if the user has required permissions.
        """
        if self.action in ("list", "retrieve"):
            result = True
        else:
            result = has_perm(self.request.user, "chat.can_manage")

        return result and ENABLE_CHAT

    def update(self, *args, **kwargs):
        response = super().update(*args, **kwargs)
        # Update all affected chatmessages to update their `read_groups_id`  and
        # `write_groups_id` field, which is taken from the updated chatgroup.
        inform_changed_data(ChatMessage.objects.filter(chatgroup=self.get_object()))
        return response

    @detail_route(methods=["POST"])
    def clear(self, request, *args, **kwargs):
        """
        Deletes all chat messages of the group.
        """
        messages = self.get_object().messages.all()
        messages_id = [message.id for message in messages]
        messages.delete()
        collection = ChatMessage.get_collection_string()
        inform_deleted_data((collection, id) for id in messages_id)
        return Response()


class ChatMessageViewSet(
    ListModelMixin,
    RetrieveModelMixin,
    CreateModelMixin,
    DestroyModelMixin,
    GenericViewSet,
):
    """
    API endpoint for chat groups.

    There are the following views: metadata, list, retrieve, create
    """

    access_permissions = ChatMessageAccessPermissions()
    queryset = ChatMessage.objects.all()

    def check_view_permissions(self):
        # The permissions are checked in the view.
        return ENABLE_CHAT and not isinstance(self.request.user, AnonymousUser)

    def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)

        if not serializer.validated_data["chatgroup"].can_write(self.request.user):
            self.permission_denied(self.request)

        # Do not use the serializer.save since it will put the model in the history.
        validated_data = {
            **serializer.validated_data,
            "username": self.request.user.short_name(),
            "user_id": self.request.user.id,
        }
        chatmessage = ChatMessage(**validated_data)
        chatmessage.save(disable_history=True)

        return Response(
            ReturnDict(id=chatmessage.id, serializer=serializer),
            status=status.HTTP_201_CREATED,
        )

    def destroy(self, request, *args, **kwargs):
        if (
            not has_perm(self.request.user, "chat.can_manage")
            and self.get_object().user_id != self.request.user.id
        ):
            self.permission_denied(request)

        disable_history()

        return super().destroy(request, *args, **kwargs)