from uuid import UUID

from domain.chat import Chat
from usecase.interface import ChatRepository


class InMemoryChatRepository(ChatRepository):
    def __init__(self) -> None:
        self._by_id: dict[UUID, Chat] = {}

    def get(self, chat_id: UUID) -> Chat | None:
        return self._by_id.get(chat_id)

    def list_by_workspace_id(self, workspace_id: UUID) -> list[Chat]:
        return sorted(
            (c for c in self._by_id.values() if c.workspace_id == workspace_id),
            key=lambda c: (c.created_at, c.chat_id),
        )

    def save(self, chat: Chat) -> None:
        self._by_id[chat.chat_id] = chat

    def delete(self, chat_id: UUID) -> None:
        self._by_id.pop(chat_id, None)