from typing import Optional

from pydantic import Field, field_validator

from akkudoktoreos.config.configabc import SettingsBaseModel
from akkudoktoreos.prediction.feedintariffabc import FeedInTariffProvider
from akkudoktoreos.prediction.feedintarifffixed import FeedInTariffFixedCommonSettings
from akkudoktoreos.prediction.feedintariffimport import FeedInTariffImportCommonSettings
from akkudoktoreos.prediction.prediction import get_prediction

prediction_eos = get_prediction()

# Valid feedintariff providers
feedintariff_providers = [
    provider.provider_id()
    for provider in prediction_eos.providers
    if isinstance(provider, FeedInTariffProvider)
]


class FeedInTariffCommonProviderSettings(SettingsBaseModel):
    """Feed In Tariff Prediction Provider Configuration."""

    FeedInTariffFixed: Optional[FeedInTariffFixedCommonSettings] = Field(
        default=None, description="FeedInTariffFixed settings", examples=[None]
    )
    FeedInTariffImport: Optional[FeedInTariffImportCommonSettings] = Field(
        default=None, description="FeedInTariffImport settings", examples=[None]
    )


class FeedInTariffCommonSettings(SettingsBaseModel):
    """Feed In Tariff Prediction Configuration."""

    provider: Optional[str] = Field(
        default=None,
        description="Feed in tariff provider id of provider to be used.",
        examples=["FeedInTariffFixed", "FeedInTarifImport"],
    )

    provider_settings: FeedInTariffCommonProviderSettings = Field(
        default_factory=FeedInTariffCommonProviderSettings,
        description="Provider settings",
        examples=[
            # Example 1: Empty/default settings (all providers None)
            {
                "FeedInTariffFixed": None,
                "FeedInTariffImport": None,
            },
        ],
    )

    # Validators
    @field_validator("provider", mode="after")
    @classmethod
    def validate_provider(cls, value: Optional[str]) -> Optional[str]:
        if value is None or value in feedintariff_providers:
            return value
        raise ValueError(
            f"Provider '{value}' is not a valid feed in tariff provider: {feedintariff_providers}."
        )