from app.config import settings
from app.const import BANCHOBOT_ID
from app.database.statistics import UserStatistics
from app.database.user import User
from app.dependencies.database import with_db
from app.log import logger
from app.models.score import GameMode

from sqlalchemy import exists
from sqlmodel import select


async def create_rx_statistics():
    async with with_db() as session:
        users = (await session.exec(select(User.id))).all()
        total_users = len(users)
        logger.info(f"Ensuring RX/AP statistics exist for {total_users} users")
        rx_created = 0
        ap_created = 0
        for i in users:
            if i == BANCHOBOT_ID:
                continue

            if settings.enable_rx:
                for mode in (
                    GameMode.OSURX,
                    GameMode.TAIKORX,
                    GameMode.FRUITSRX,
                ):
                    is_exist = (
                        await session.exec(
                            select(exists()).where(
                                UserStatistics.user_id == i,
                                UserStatistics.mode == mode,
                            )
                        )
                    ).first()
                    if not is_exist:
                        statistics_rx = UserStatistics(mode=mode, user_id=i)
                        session.add(statistics_rx)
                        rx_created += 1
            if settings.enable_ap:
                is_exist = (
                    await session.exec(
                        select(exists()).where(
                            UserStatistics.user_id == i,
                            UserStatistics.mode == GameMode.OSUAP,
                        )
                    )
                ).first()
                if not is_exist:
                    statistics_ap = UserStatistics(mode=GameMode.OSUAP, user_id=i)
                    session.add(statistics_ap)
                    ap_created += 1
        await session.commit()
        if rx_created or ap_created:
            logger.success(
                f"Created {rx_created} RX statistics rows and {ap_created} AP statistics rows during backfill"
            )