"""用户可添加抖音账号数量限制（限制功能已移除，保留接口兼容）。"""

from __future__ import annotations

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

from models.models import Account, User
from .roles import is_admin

UNLIMITED_ACCOUNTS = -1


def normalize_max_accounts(value: int | None, role: str = "operator") -> int:
    if is_admin(role):
        return UNLIMITED_ACCOUNTS
    if value is None:
        return 3
    try:
        parsed = int(value)
    except (TypeError, ValueError):
        return 3
    if parsed < 0:
        return UNLIMITED_ACCOUNTS
    return parsed


def account_limit_for_user(user: User) -> int | None:
    """账号数量/并发限制已移除，始终不限制。"""
    return None


async def count_user_accounts(db: AsyncSession, user_id: int) -> int:
    result = await db.execute(
        select(func.count()).select_from(Account).where(Account.owner_id == user_id)
    )
    return int(result.scalar() or 0)


async def count_user_account_breakdown(db: AsyncSession, user_id: int) -> dict[str, int]:
    """统计用户名下托管账号：总数 / 可用 / 额度停用。"""
    total = await count_user_accounts(db, user_id)
    if total <= 0:
        return {"total": 0, "active": 0, "disabled": 0}
    disabled_result = await db.execute(
        select(func.count())
        .select_from(Account)
        .where(Account.owner_id == user_id, Account.quota_disabled.is_(True))
    )
    disabled = int(disabled_result.scalar() or 0)
    active = max(0, total - disabled)
    return {"total": total, "active": active, "disabled": disabled}


async def ensure_can_add_account(db: AsyncSession, user: User) -> None:
    """账号数量限制已移除，任何用户可添加任意数量账号。"""
    return