"""
Number entity definitions for Tibber Prices configuration overrides.

These number entities allow runtime configuration of Best Price and Peak Price
period calculation settings. They are disabled by default - users can enable
individual entities to override specific settings at runtime.

When enabled, the entity value takes precedence over the options flow setting.
When disabled (default), the options flow setting is used.
"""

from __future__ import annotations

from dataclasses import dataclass

from homeassistant.components.number import (
    NumberEntityDescription,
    NumberMode,
)
from homeassistant.const import PERCENTAGE, EntityCategory


@dataclass(frozen=True, kw_only=True)
class TibberPricesNumberEntityDescription(NumberEntityDescription):
    """Describes a Tibber Prices number entity for config overrides."""

    # The config key this entity overrides (matches CONF_* constants)
    config_key: str
    # The section in options where this setting is stored (e.g., "flexibility_settings")
    config_section: str
    # Whether this is for best_price (False) or peak_price (True)
    is_peak_price: bool = False
    # Default value from const.py
    default_value: float | int = 0


# ============================================================================
# BEST PRICE PERIOD CONFIGURATION OVERRIDES
# ============================================================================

BEST_PRICE_NUMBER_ENTITIES = (
    TibberPricesNumberEntityDescription(
        key="best_price_flex_override",
        translation_key="best_price_flex_override",
        icon="mdi:arrow-down-bold-circle",
        entity_category=EntityCategory.CONFIG,
        entity_registry_enabled_default=False,
        native_min_value=0,
        native_max_value=50,
        native_step=1,
        native_unit_of_measurement=PERCENTAGE,
        mode=NumberMode.SLIDER,
        config_key="best_price_flex",
        config_section="flexibility_settings",
        is_peak_price=False,
        default_value=15,  # DEFAULT_BEST_PRICE_FLEX
    ),
    TibberPricesNumberEntityDescription(
        key="best_price_min_distance_override",
        translation_key="best_price_min_distance_override",
        icon="mdi:arrow-down-bold-circle",
        entity_category=EntityCategory.CONFIG,
        entity_registry_enabled_default=False,
        native_min_value=-50,
        native_max_value=0,
        native_step=1,
        native_unit_of_measurement=PERCENTAGE,
        mode=NumberMode.SLIDER,
        config_key="best_price_min_distance_from_avg",
        config_section="flexibility_settings",
        is_peak_price=False,
        default_value=-5,  # DEFAULT_BEST_PRICE_MIN_DISTANCE_FROM_AVG
    ),
    TibberPricesNumberEntityDescription(
        key="best_price_min_period_length_override",
        translation_key="best_price_min_period_length_override",
        icon="mdi:arrow-down-bold-circle",
        entity_category=EntityCategory.CONFIG,
        entity_registry_enabled_default=False,
        native_min_value=15,
        native_max_value=180,
        native_step=15,
        native_unit_of_measurement="min",
        mode=NumberMode.SLIDER,
        config_key="best_price_min_period_length",
        config_section="period_settings",
        is_peak_price=False,
        default_value=60,  # DEFAULT_BEST_PRICE_MIN_PERIOD_LENGTH
    ),
    TibberPricesNumberEntityDescription(
        key="best_price_min_periods_override",
        translation_key="best_price_min_periods_override",
        icon="mdi:arrow-down-bold-circle",
        entity_category=EntityCategory.CONFIG,
        entity_registry_enabled_default=False,
        native_min_value=1,
        native_max_value=10,
        native_step=1,
        mode=NumberMode.SLIDER,
        config_key="min_periods_best",
        config_section="relaxation_and_target_periods",
        is_peak_price=False,
        default_value=2,  # DEFAULT_MIN_PERIODS_BEST
    ),
    TibberPricesNumberEntityDescription(
        key="best_price_relaxation_attempts_override",
        translation_key="best_price_relaxation_attempts_override",
        icon="mdi:arrow-down-bold-circle",
        entity_category=EntityCategory.CONFIG,
        entity_registry_enabled_default=False,
        native_min_value=1,
        native_max_value=12,
        native_step=1,
        mode=NumberMode.SLIDER,
        config_key="relaxation_attempts_best",
        config_section="relaxation_and_target_periods",
        is_peak_price=False,
        default_value=11,  # DEFAULT_RELAXATION_ATTEMPTS_BEST
    ),
    TibberPricesNumberEntityDescription(
        key="best_price_gap_count_override",
        translation_key="best_price_gap_count_override",
        icon="mdi:arrow-down-bold-circle",
        entity_category=EntityCategory.CONFIG,
        entity_registry_enabled_default=False,
        native_min_value=0,
        native_max_value=8,
        native_step=1,
        mode=NumberMode.SLIDER,
        config_key="best_price_max_level_gap_count",
        config_section="period_settings",
        is_peak_price=False,
        default_value=1,  # DEFAULT_BEST_PRICE_MAX_LEVEL_GAP_COUNT
    ),
)

# ============================================================================
# PEAK PRICE PERIOD CONFIGURATION OVERRIDES
# ============================================================================

PEAK_PRICE_NUMBER_ENTITIES = (
    TibberPricesNumberEntityDescription(
        key="peak_price_flex_override",
        translation_key="peak_price_flex_override",
        icon="mdi:arrow-up-bold-circle",
        entity_category=EntityCategory.CONFIG,
        entity_registry_enabled_default=False,
        native_min_value=-50,
        native_max_value=0,
        native_step=1,
        native_unit_of_measurement=PERCENTAGE,
        mode=NumberMode.SLIDER,
        config_key="peak_price_flex",
        config_section="flexibility_settings",
        is_peak_price=True,
        default_value=-20,  # DEFAULT_PEAK_PRICE_FLEX
    ),
    TibberPricesNumberEntityDescription(
        key="peak_price_min_distance_override",
        translation_key="peak_price_min_distance_override",
        icon="mdi:arrow-up-bold-circle",
        entity_category=EntityCategory.CONFIG,
        entity_registry_enabled_default=False,
        native_min_value=0,
        native_max_value=50,
        native_step=1,
        native_unit_of_measurement=PERCENTAGE,
        mode=NumberMode.SLIDER,
        config_key="peak_price_min_distance_from_avg",
        config_section="flexibility_settings",
        is_peak_price=True,
        default_value=5,  # DEFAULT_PEAK_PRICE_MIN_DISTANCE_FROM_AVG
    ),
    TibberPricesNumberEntityDescription(
        key="peak_price_min_period_length_override",
        translation_key="peak_price_min_period_length_override",
        icon="mdi:arrow-up-bold-circle",
        entity_category=EntityCategory.CONFIG,
        entity_registry_enabled_default=False,
        native_min_value=15,
        native_max_value=180,
        native_step=15,
        native_unit_of_measurement="min",
        mode=NumberMode.SLIDER,
        config_key="peak_price_min_period_length",
        config_section="period_settings",
        is_peak_price=True,
        default_value=30,  # DEFAULT_PEAK_PRICE_MIN_PERIOD_LENGTH
    ),
    TibberPricesNumberEntityDescription(
        key="peak_price_min_periods_override",
        translation_key="peak_price_min_periods_override",
        icon="mdi:arrow-up-bold-circle",
        entity_category=EntityCategory.CONFIG,
        entity_registry_enabled_default=False,
        native_min_value=1,
        native_max_value=10,
        native_step=1,
        mode=NumberMode.SLIDER,
        config_key="min_periods_peak",
        config_section="relaxation_and_target_periods",
        is_peak_price=True,
        default_value=2,  # DEFAULT_MIN_PERIODS_PEAK
    ),
    TibberPricesNumberEntityDescription(
        key="peak_price_relaxation_attempts_override",
        translation_key="peak_price_relaxation_attempts_override",
        icon="mdi:arrow-up-bold-circle",
        entity_category=EntityCategory.CONFIG,
        entity_registry_enabled_default=False,
        native_min_value=1,
        native_max_value=12,
        native_step=1,
        mode=NumberMode.SLIDER,
        config_key="relaxation_attempts_peak",
        config_section="relaxation_and_target_periods",
        is_peak_price=True,
        default_value=11,  # DEFAULT_RELAXATION_ATTEMPTS_PEAK
    ),
    TibberPricesNumberEntityDescription(
        key="peak_price_gap_count_override",
        translation_key="peak_price_gap_count_override",
        icon="mdi:arrow-up-bold-circle",
        entity_category=EntityCategory.CONFIG,
        entity_registry_enabled_default=False,
        native_min_value=0,
        native_max_value=8,
        native_step=1,
        mode=NumberMode.SLIDER,
        config_key="peak_price_max_level_gap_count",
        config_section="period_settings",
        is_peak_price=True,
        default_value=1,  # DEFAULT_PEAK_PRICE_MAX_LEVEL_GAP_COUNT
    ),
)

# All number entity descriptions combined
NUMBER_ENTITY_DESCRIPTIONS = BEST_PRICE_NUMBER_ENTITIES + PEAK_PRICE_NUMBER_ENTITIES