import logging
from datetime import datetime
from typing import Optional

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

from app.database.crud.transaction import get_user_total_spent_kopeks
from app.database.models import PromoGroup, User

logger = logging.getLogger(__name__)


async def _get_best_group_for_spending(
    db: AsyncSession,
    total_spent_kopeks: int,
) -> Optional[PromoGroup]:
    if total_spent_kopeks <= 0:
        return None

    result = await db.execute(
        select(PromoGroup)
        .where(PromoGroup.auto_assign_total_spent_kopeks.is_not(None))
        .where(PromoGroup.auto_assign_total_spent_kopeks > 0)
        .order_by(PromoGroup.auto_assign_total_spent_kopeks.desc(), PromoGroup.id.desc())
    )
    groups = result.scalars().all()

    for group in groups:
        threshold = group.auto_assign_total_spent_kopeks or 0
        if threshold and total_spent_kopeks >= threshold:
            return group

    return None


async def maybe_assign_promo_group_by_total_spent(
    db: AsyncSession,
    user_id: int,
) -> Optional[PromoGroup]:
    user = await db.get(User, user_id)
    if not user:
        logger.debug("Не удалось найти пользователя %s для автовыдачи промогруппы", user_id)
        return None

    total_spent = await get_user_total_spent_kopeks(db, user_id)
    if total_spent <= 0:
        return None

    target_group = await _get_best_group_for_spending(db, total_spent)
    if not target_group:
        return None

    try:
        previous_group_id = user.promo_group_id

        if user.auto_promo_group_assigned and target_group.id == previous_group_id:
            logger.debug(
                "Пользователь %s уже находится в актуальной промогруппе '%s', повторная выдача не требуется",
                user.telegram_id,
                target_group.name,
            )
            return target_group

        user.auto_promo_group_assigned = True
        user.updated_at = datetime.utcnow()

        if target_group.id != previous_group_id:
            user.promo_group_id = target_group.id
            user.promo_group = target_group
            logger.info(
                "🤖 Пользователь %s автоматически переведен в промогруппу '%s' за траты %s ₽",
                user.telegram_id,
                target_group.name,
                total_spent / 100,
            )
        else:
            logger.info(
                "🤖 Пользователь %s уже находится в подходящей промогруппе '%s', отмечаем автоприсвоение",
                user.telegram_id,
                target_group.name,
            )

        await db.commit()
        await db.refresh(user)

        return target_group
    except Exception as exc:
        logger.error(
            "Ошибка при автоматическом назначении промогруппы пользователю %s: %s",
            user_id,
            exc,
        )
        await db.rollback()
        return None