import logging

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

from app.database.models import Squad


logger = logging.getLogger(__name__)


async def get_squad_by_uuid(db: AsyncSession, uuid: str) -> Squad | None:
    result = await db.execute(select(Squad).where(Squad.uuid == uuid))
    return result.scalar_one_or_none()


async def get_available_squads(db: AsyncSession) -> list[Squad]:
    result = await db.execute(select(Squad).where(Squad.is_available == True))
    return result.scalars().all()


async def create_squad(
    db: AsyncSession, uuid: str, name: str, country_code: str = None, price_kopeks: int = 0, description: str = None
) -> Squad:
    squad = Squad(uuid=uuid, name=name, country_code=country_code, price_kopeks=price_kopeks, description=description)

    db.add(squad)
    await db.commit()
    await db.refresh(squad)

    logger.info(f'✅ Создан сквад: {name}')
    return squad


async def update_squad(db: AsyncSession, squad: Squad, **kwargs) -> Squad:
    for field, value in kwargs.items():
        if hasattr(squad, field):
            setattr(squad, field, value)

    await db.commit()
    await db.refresh(squad)

    return squad