from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession

from db.models import Addon


class AddonsRepository:
    def __init__(self, session: AsyncSession) -> None:
        self.session = session

    async def get_all(self) -> list[Addon]:
        stmt = select(Addon)
        r = await self.session.execute(stmt)

        return list(r.scalars().all())

    async def get_by_id(self, id: str) -> Addon | None:
        stmt = select(Addon).where(Addon.id == id)
        r = await self.session.execute(stmt)

        return r.scalar_one_or_none()