"""Federal Reserve Federal Funds Rate Model."""

# pylint: disable=unused-argument

from datetime import datetime
from typing import Any

from openbb_core.provider.abstract.fetcher import Fetcher
from openbb_core.provider.standard_models.federal_funds_rate import (
    FederalFundsRateData,
    FederalFundsRateQueryParams,
)
from openbb_core.provider.utils.errors import EmptyDataError
from pydantic import Field, field_validator


class FederalReserveFederalFundsRateQueryParams(FederalFundsRateQueryParams):
    """FederalReserve FED Query."""


class FederalReserveFederalFundsRateData(FederalFundsRateData):
    """FederalReserve FED Data."""

    __alias_dict__ = {
        "date": "effectiveDate",
        "rate": "percentRate",
        "target_range_upper": "targetRateTo",
        "target_range_lower": "targetRateFrom",
        "percentile_1": "percentPercentile1",
        "percentile_25": "percentPercentile25",
        "percentile_75": "percentPercentile75",
        "percentile_99": "percentPercentile99",
        "volume": "volumeInBillions",
        "intraday_low": "intraDayLow",
        "intraday_high": "intraDayHigh",
        "standard_deviation": "stdDeviation",
        "revision_indicator": "revisionIndicator",
    }

    intraday_low: float | None = Field(
        default=None,
        description="Intraday low. This field is only present for data before 2016.",
    )
    intraday_high: float | None = Field(
        default=None,
        description="Intraday high. This field is only present for data before 2016.",
    )
    standard_deviation: float | None = Field(
        default=None,
        description="Standard deviation. This field is only present for data before 2016.",
    )
    revision_indicator: str | None = Field(
        default=None,
        description="Indicates a revision of the data for that date.",
    )

    @field_validator("revision_indicator", mode="before", check_fields=False)
    @classmethod
    def validate_revision_indicator(cls, v):
        """Validate revision indicator."""
        return None if v in ("", "''") else v

    @field_validator(
        "rate",
        "target_range_upper",
        "target_range_lower",
        "percentile_1",
        "percentile_25",
        "percentile_75",
        "percentile_99",
        "intraday_high",
        "intraday_low",
        mode="before",
        check_fields=False,
    )
    @classmethod
    def normalize_percent(cls, v):
        """Normalize percent."""
        if v is not None:
            return v / 100 if v != 0 else 0
        return None


class FederalReserveFederalFundsRateFetcher(
    Fetcher[
        FederalReserveFederalFundsRateQueryParams,
        list[FederalReserveFederalFundsRateData],
    ]
):
    """Federal Reserve Federal Funds Fetcher."""

    @staticmethod
    def transform_query(
        params: dict[str, Any],
    ) -> FederalReserveFederalFundsRateQueryParams:
        """Transform query."""
        transformed_params = params.copy()
        now = datetime.now().date()
        if params.get("start_date") is None:
            transformed_params["start_date"] = datetime(2016, 3, 1).date()
        if params.get("end_date") is None:
            transformed_params["end_date"] = now

        return FederalReserveFederalFundsRateQueryParams(**transformed_params)

    @staticmethod
    async def aextract_data(
        query: FederalReserveFederalFundsRateQueryParams,
        credentials: dict[str, str] | None,
        **kwargs: Any,
    ) -> list[dict]:
        """Extract the raw data."""
        # pylint: disable=import-outside-toplevel
        from openbb_core.provider.utils.helpers import amake_request

        url = (
            "https://markets.newyorkfed.org/api/rates/unsecured/effr/search.json?"
            + f"startDate={query.start_date}&endDate={query.end_date}"
        )
        results: list[dict] = []
        response = await amake_request(url, **kwargs)  # type: ignore
        if response.get("refRates"):  # type: ignore
            results = response["refRates"]  # type: ignore
        if not results:
            raise EmptyDataError()
        return results

    @staticmethod
    def transform_data(
        query: FederalReserveFederalFundsRateQueryParams,
        data: list[dict],
        **kwargs: Any,
    ) -> list[FederalReserveFederalFundsRateData]:
        """Transform data."""
        results: list[FederalReserveFederalFundsRateData] = []
        for d in data.copy():
            _ = d.pop("type", None)
            _ = d.pop("footnoteId", None)
            results.append(FederalReserveFederalFundsRateData.model_validate(d))

        return sorted(results, key=lambda x: x.date)
