app/backend/controllers/chats.py

from app.backend.models.chats import get_chats_by_user_id, get_chat_by_id, refresh_title, add_new_chat
from app.backend.models.messages import get_messages_by_chat_id, Message
from app.backend.models.users import User, get_user_chats
from app.backend.controllers.utils import get_group_title
from app.settings import settings, logger
from datetime import datetime, timedelta
from app.settings import BASE_DIR
from fastapi import HTTPException
from uuid import uuid4
import aiofiles.os
import asyncio
import os


async def create_new_chat(title: str | None, user: User) -> dict:
    if settings.debug:
        await logger.info("START Creating Chat")
    try:
        chat_id = str(uuid4())
        await add_new_chat(id=chat_id, title=title, user=user)
        try:
            path_to_chat = os.path.join(
                BASE_DIR,
                "chats_storage",
                f"user_id={user.id}",
                f"chat_id={chat_id}",
                "documents",
            )

            await aiofiles.os.makedirs(path_to_chat, exist_ok=True)
        except Exception:
            raise HTTPException(500, "error while creating chat folders")

        return {"url": f"/chats/id={chat_id}", "chat_id": chat_id}
    except Exception as exception:
        raise exception
    finally:
        if settings.debug:
            await logger.info("END Creating Chat")


async def dump_messages_dict(messages: list[Message], dst: dict) -> None:
    history = []
    loop = asyncio.get_running_loop()

    def process_message(message):
        return {"role": message.sender, "content": message.content}

    for message in messages:
        history.append(await loop.run_in_executor(None, process_message, message))

    dst.update({"history": history})


async def get_chat_with_messages(id: str) -> dict:
    response = {"chat_id": id}

    chat = await get_chat_by_id(id=id)
    if chat is None:
        raise HTTPException(418, f"Invalid chat id. Chat with id={id} does not exists!")

    messages = await get_messages_by_chat_id(id=id)
    await dump_messages_dict(messages, response)

    return response


async def create_dict_from_chat(chat) -> dict:
    return {"id": chat.id, "title": chat.title}


async def list_user_chats(user_id: str) -> list[dict]:
    current_date = datetime.now()

    today = []
    last_week = []
    last_month = []
    later = []

    groups = [today, last_week, last_month, later]

    chats = await get_chats_by_user_id(user_id)
    loop = asyncio.get_event_loop()

    def categorize_chat(chat):
        if current_date - timedelta(days=1) <= chat.created_at:
            today.append(chat)
        elif current_date - timedelta(weeks=1) <= chat.created_at:
            last_week.append(chat)
        elif current_date - timedelta(weeks=4) <= chat.created_at:
            last_month.append(chat)
        else:
            later.append(chat)

    await asyncio.gather(*[loop.run_in_executor(None, categorize_chat, chat) for chat in chats])

    result = []

    for id, group in enumerate(groups):
        if len(group):
            result.append(
                {"title": await get_group_title(id=id), "chats": [await create_dict_from_chat(chat) for chat in group]}
            )

    return result


async def verify_ownership_rights(user: User, chat_id: str) -> bool:
    return chat_id in [chat.id for chat in await get_user_chats(user.id)]

async def update_title(chat_id: str) -> bool:
    return await refresh_title(chat_id)