"""消费查询服务"""

import logging
from decimal import Decimal
from .volcengine_client import get_billing_client

logger = logging.getLogger(__name__)


class BillingService:
    """封装火山引擎 Billing API

    使用 ListSplitBillDetail（分账账单）而非 ListBillDetail（账单明细），
    因为后者的 Project 字段对 Seedance 等按量付费产品显示为 '-'，不准确。
    分账账单能正确按项目归属消费，与火山控制台分账账单页面一致。
    """

    def __init__(self, ak: str, sk: str):
        self.client = get_billing_client(ak, sk)

    def get_spending_by_project(self, bill_period: str, project_name: str = None) -> Decimal:
        """查询指定项目的消费总额（使用分账账单，带分页）"""
        total = Decimal("0")
        offset = 0
        page_size = 100

        while True:
            params = {
                "BillPeriod": bill_period,
                "Limit": str(page_size),
                "Offset": str(offset),
                "GroupTerm": "0",
                "GroupPeriod": "2",
            }
            result = self.client.call("ListSplitBillDetail", params)
            items = result.get("Result", {}).get("List", [])

            for item in items:
                item_project = item.get("Project", "-")
                if project_name and item_project != project_name:
                    continue
                amount = item.get("PayableAmount", "0")
                total += Decimal(str(amount))

            if len(items) < page_size:
                break
            offset += page_size

        return total

    def get_spending_all_projects(self, bill_period: str) -> dict:
        """查询所有项目的消费汇总（返回 {project_name: Decimal}）"""
        by_project = {}
        offset = 0
        page_size = 100

        while True:
            params = {
                "BillPeriod": bill_period,
                "Limit": str(page_size),
                "Offset": str(offset),
                "GroupTerm": "0",
                "GroupPeriod": "2",
            }
            result = self.client.call("ListSplitBillDetail", params)
            items = result.get("Result", {}).get("List", [])

            for item in items:
                project = item.get("Project", "-")
                amount = Decimal(str(item.get("PayableAmount", "0")))
                by_project[project] = by_project.get(project, Decimal("0")) + amount

            if len(items) < page_size:
                break
            offset += page_size

        return by_project

    def get_bill_overview(self, bill_period: str) -> dict:
        """获取账单总览（按产品维度）"""
        result = self.client.call("ListBillOverviewByProd", {
            "BillPeriod": bill_period,
            "Limit": "100",
            "NeedRecordNum": "1",
        })
        return result.get("Result", {})

    def get_balance(self) -> dict:
        """查询主账号余额"""
        result = self.client.call("QueryBalanceAcct")
        return result.get("Result", {})