from chain_service.database.database import Database
from chain_service.database.models.chain import Chain


class ChainRepository:

    def __init__(self, database: Database):
        self.collection = database.get_collection("chains")

    async def upsert(self, chain: Chain) -> Chain:
        await self.collection.replace_one(
            {"_id": chain.id}, chain.model_dump(by_alias=True), upsert=True
        )

        return chain