"""额度查询聚合服务"""

from __future__ import annotations

from app import database as db
from app.providers import ProviderRegistry
from app.providers.base import QuotaInfo


async def query_plan_quota(plan_id: str) -> list[dict]:
    """查询指定 Plan 的所有 QuotaRule 状态，对 api_sync 类型尝试实时查询"""
    plan = await db.get_plan(plan_id)
    if not plan:
        return []

    rules = await db.list_quota_rules(plan_id)
    result = []

    for r in rules:
        info = {
            "rule_id": r["id"],
            "rule_name": r["rule_name"],
            "quota_total": r["quota_total"],
            "quota_used": r["quota_used"],
            "quota_remaining": r["quota_total"] - r["quota_used"],
            "quota_unit": r["quota_unit"],
            "refresh_type": r["refresh_type"],
            "next_refresh_at": r.get("next_refresh_at"),
        }

        # api_sync 类型尝试实时查询
        if r["refresh_type"] == "api_sync":
            provider = ProviderRegistry.get(plan["provider_name"])
            if provider:
                try:
                    qi = await provider.query_quota(plan)
                    if qi:
                        info["quota_remaining"] = qi.quota_remaining
                        info["quota_used"] = qi.quota_used
                        info["api_raw"] = qi.raw
                except Exception:
                    pass

        result.append(info)

    return result