"""Statistics and analytics tools."""

import json
from typing import Annotated

from pydantic import Field

from ..client import client
from ..config import logger


def register_statistics_tools(mcp):

    @mcp.tool()
    async def bunny_get_statistics(
        pullzone_id: Annotated[int | None, Field(default=None, description="Pull Zone ID (optional, omit for account-wide stats)")] = None,
        date_from: Annotated[str | None, Field(default=None, description="Start date in YYYY-MM-DD format (optional)")] = None,
        date_to: Annotated[str | None, Field(default=None, description="End date in YYYY-MM-DD format (optional)")] = None,
    ) -> str:
        """Get bandwidth, cache hit rate, and request statistics."""
        try:
            params = {}
            if pullzone_id is not None:
                params["pullZone"] = pullzone_id
            if date_from:
                params["dateFrom"] = date_from
            if date_to:
                params["dateTo"] = date_to

            data = await client.get("/statistics", params=params or None)

            summary = {
                "TotalBandwidthUsed": data.get("TotalBandwidthUsed"),
                "TotalRequestsServed": data.get("TotalRequestsServed"),
                "CacheHitRate": data.get("CacheHitRate"),
                "AverageOriginResponseTime": data.get("AverageOriginResponseTime"),
                "BandwidthUsedChart": data.get("BandwidthUsedChart"),
                "RequestsServedChart": data.get("RequestsServedChart"),
                "CacheHitRateChart": data.get("CacheHitRateChart"),
                "Error3xxChart": data.get("Error3xxChart"),
                "Error4xxChart": data.get("Error4xxChart"),
                "Error5xxChart": data.get("Error5xxChart"),
            }
            return json.dumps(summary, indent=2)
        except Exception as e:
            logger.error("bunny_get_statistics failed: %s", e)
            return f"Error: {e}"