cbh/api/message/views.py

from cbh.api.chat.dto import Paging
from cbh.api.message import message_router
from cbh.api.message.db_requests import create_message_obj, get_all_chat_messages_obj
from cbh.api.message.model import MessageModel
from cbh.api.message.schemas import CreateMessageRequest, AllMessagesResponse
from cbh.core.wrappers import CbhResponseWrapper


@message_router.get(
    '/{chatId}/all', response_model_by_alias=False, response_model=CbhResponseWrapper[AllMessagesResponse]
)
async def get_all_chat_messages(chatId: str):
    messages, _ = await get_all_chat_messages_obj(chatId)
    response = AllMessagesResponse(
        paging=Paging(pageSize=len(messages), pageIndex=0, totalCount=len(messages)),
        data=messages
    )
    return CbhResponseWrapper(data=response)


@message_router.post('/{chatId}', response_model_by_alias=False, response_model=CbhResponseWrapper[MessageModel])
async def create_message(chatId: str, message: CreateMessageRequest):
    message = await create_message_obj(chatId, message)
    return CbhResponseWrapper(data=message)