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

from db.models.orders import Order, OrderAddon, OrderStatus


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

    async def create(
        self,
        *,
        user_id: int,
        devices: int,
        duration_days: int,
        total_amount: float,
        balance_amount: float,
        addons: list[str] | None = None,
        status: OrderStatus = OrderStatus.PENDING,
    ) -> Order:
        if not addons:
            addons = []

        order = Order(
            user_id=user_id,
            devices=devices,
            duration_days=duration_days,
            total_amount=total_amount,
            balance_amount=balance_amount,
            status=status,
        )
        self.session.add(order)
        await self.session.flush()

        for addon_id in addons:
            addon = OrderAddon(order_id=order.id, addon_id=addon_id)
            self.session.add(addon)

        await self.session.commit()
        await self.session.refresh(order, attribute_names=["addons"])

        return order

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

        return r.scalar_one_or_none()

    async def get_order_by_user_id(self, user_id: int) -> list[Order]:
        stmt = select(Order).where(Order.user_id == user_id)
        r = await self.session.execute(stmt)

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

    async def get_active_by_user_id(self, user_id: int) -> list[Order]:
        stmt = (
            select(Order).where(Order.user_id == user_id).where(Order.status == OrderStatus.PENDING)
        )
        r = await self.session.execute(stmt)

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

    async def get_paid_unapplied_by_user_id(self, user_id: int) -> list[Order]:
        stmt = (
            select(Order)
            .where(Order.user_id == user_id)
            .where(Order.status == OrderStatus.PAID)
            .where(Order.applied_at.is_(None))
            .order_by(Order.applies_at.asc(), Order.id.asc())
        )
        r = await self.session.execute(stmt)

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