inbox/resolvers/messages.py

import json
import logging
import time

from models.chat import Message
from services.auth import login_required
from services.presence import notify_message
from services.rediscache import redis
from services.schema import mutation

logger = logging.getLogger("[resolvers.messages] ")
logger.setLevel(logging.DEBUG)


@mutation.field("create_message")
@login_required
async def create_message(_, info, chat_id: str, body: str, reply_to=None):
    """Создание сообщения с телом :body для чата :chat_id с возможным ответом на :reply_to"""
    author_id = info.context["author_id"]

    # Получение данных чата из Redis
    chat_data = await redis.execute("GET", f"chats/{chat_id}")
    logger.debug(f"chat data: {chat_data}")

    # Если данных чата нет, возвращаем ошибку
    if not chat_data:
        return {"error": "chat is not exist"}
    elif isinstance(chat_data, str):
        # Преобразование данных чата из строки JSON в словарь
        chat_dict = json.loads(chat_data)
        chat_id = chat_dict["id"]

        # Получение ID следующего сообщения
        message_id = await redis.execute(
            "GET", f"chats/{chat_dict['id']}/next_message_id"
        )
        if isinstance(message_id, str) or isinstance(message_id, int):
            message_id = int(message_id) if message_id else 0
            # Создание нового сообщения
            new_message: Message = {
                "chat_id": chat_id,
                "id": message_id,
                "created_by": author_id,
                "body": body,
                "created_at": int(time.time()),
                "updated_at": None,
                "reply_to": None,
            }

            # Если есть ответ, добавляем его в сообщение
            if reply_to:
                new_message["reply_to"] = reply_to

            # Обновление времени последнего обновления чата
            chat_dict["updated_at"] = new_message["created_at"]

            # Запись обновленных данных чата обратно в Redis
            await redis.execute("SET", f"chats/{chat_id}", json.dumps(chat_dict))
            logger.debug(f"creating message {new_message}")

            # Запись нового сообщения в Redis
            await redis.execute(
                "SET",
                f"chats/{chat_id}/messages/{message_id}",
                json.dumps(new_message),
            )

            # Добавление ID нового сообщения в список ID сообщений чата
            await redis.execute(
                "LPUSH", f"chats/{chat_id}/message_ids", str(message_id)
            )

            # Обновление ID следующего сообщения
            await redis.execute(
                "SET", f"chats/{chat_id}/next_message_id", str(message_id + 1)
            )

            # Добавление нового сообщения в список непрочитанных сообщений для каждого участника чата
            members = chat_dict["members"]
            for member_id in members:
                await redis.execute(
                    "LPUSH",
                    f"chats/{chat_dict['id']}/unread/{member_id}",
                    str(message_id),
                )

            # Отправка уведомления о новом сообщении
            new_message["chat_id"] = chat_id
            await notify_message(new_message, "create")

            return {"message": new_message, "error": None}
    return {"error": "cannot create message"}


@mutation.field("update_message")
@login_required
async def update_message(_, info, message):
    author_id = info.context["author_id"]
    chat_id = message.get("chat_id")
    chat_str = ""
    if chat_id:
        chat_str = await redis.execute("GET", f"chats/{chat_id}")
        if not chat_str:
            return {"error": "chat not exist"}

    message_id = message.get("id")
    body = message.get("body")

    if message_id:
        message = await redis.execute("GET", f"chats/{chat_id}/messages/{message_id}")
        if isinstance(message, str):
            message = json.loads(message)
            if message["created_by"] != author_id:
                return {"error": "access denied"}

            if body:
                message["body"] = body
                message["updated_at"] = int(time.time())

            await redis.execute(
                "SET", f"chats/{chat_id}/messages/{message_id}", json.dumps(message)
            )

            # Отправка уведомления
            message["chat_id"] = chat_id
            await notify_message(message, "update")

            return {"message": message, "error": None}

    return {"message": message, "error": "cannot update"}


@mutation.field("delete_message")
@login_required
async def delete_message(_, info, chat_id: str, message_id: int):
    author_id = info.context["author_id"]

    chat_str = await redis.execute("GET", f"chats/{chat_id}")
    if isinstance(chat_str, str):
        chat = json.loads(chat_str)
        message_data = await redis.execute(
            "GET", f"chats/{chat_id}/messages/{str(message_id)}"
        )
        if isinstance(message_data, str):
            message: Message = json.loads(message_data)
            if message["created_by"] != author_id:
                return {"error": "access denied"}

            await redis.execute(
                "LREM", f"chats/{chat_id}/message_ids", 0, str(message_id)
            )
            await redis.execute("DEL", f"chats/{chat_id}/messages/{str(message_id)}")

            members = chat["members"]
            for member_id in members:
                await redis.execute(
                    "LREM", f"chats/{chat_id}/unread/{member_id}", 0, str(message_id)
                )

            message["chat_id"] = chat_id
            await notify_message(message, "delete")

    return {}


@mutation.field("mark_as_read")
@login_required
async def mark_as_read(_, info, chat_id: str, message_id: int):
    author_id = info.context["author_id"]

    chat_str = await redis.execute("GET", f"chats/{chat_id}")
    if isinstance(chat_str, str):
        chat = json.loads(chat_str)
        members = set(chat["members"])
        if author_id not in members:
            return {"error": "access denied"}

        await redis.execute(
            "LREM", f"chats/{chat_id}/unread/{author_id}", 0, str(message_id)
        )

        message_data = await redis.execute(
            "GET", f"chats/{chat_id}/messages/{str(message_id)}"
        )
        if isinstance(message_data, str):
            message: Message = json.loads(message_data)

            await notify_message(message, "seen")

    return {"error": None}