import os
from beanie import init_beanie
from motor.motor_asyncio import AsyncIOMotorClient

from src.models import User, Session, File, VectorStoreRecordId, Conversation, Message


class DatabaseConfig:
    def __init__(self):

        self.MONGO_URI = os.getenv("MONGO_DB_URI")
        self.MONGO_DB_NAME = os.getenv("MONGO_DB_NAME")
        self.client = AsyncIOMotorClient(self.MONGO_URI, uuidRepresentation="standard")
        self.db = self.client[self.MONGO_DB_NAME]

    async def init_beanie(self):

        await init_beanie(
            database=self.db,
            document_models=[
                User,
                Session,
                Conversation,
                Message,
                File,
                VectorStoreRecordId,
            ],
        )