grid_tariffs/

costs.rs

use std::slice::Iter;

use chrono::{DateTime, Datelike};
use serde::Serialize;

use crate::{
    Country, Language, LoadType, Money, Timezone, helpers,
    hours::Hours,
    months::{Month, Months},
};

// TODO: Make CostBuilder!
#[derive(Debug, Clone, Copy, Serialize)]
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
pub enum Cost {
    None,
    /// Cost has not been verified
    Unverified,
    Fixed(Money),
    Fuses(&'static [(u16, Money)]),
    /// Fuse size combined with a yearly energy consumption limit
    FusesYearlyConsumption(&'static [(u16, Option<u32>, Money)]),
    FuseRange(&'static [(u16, u16, Money)]),
}

impl Cost {
    pub const fn fuses(values: &'static [(u16, Money)]) -> Self {
        Self::Fuses(values)
    }

    pub const fn fuse_range(ranges: &'static [(u16, u16, Money)]) -> Self {
        Self::FuseRange(ranges)
    }

    pub const fn fuses_with_yearly_consumption(
        values: &'static [(u16, Option<u32>, Money)],
    ) -> Self {
        Self::FusesYearlyConsumption(values)
    }

    pub const fn fixed(int: i64, fract: u8) -> Self {
        Self::Fixed(Money::new(int, fract))
    }

    pub const fn fixed_yearly(int: i64, fract: u8) -> Self {
        Self::Fixed(Money::new(int, fract).divide_by(12))
    }

    pub const fn fixed_subunit(subunit: f64) -> Self {
        Self::Fixed(Money::new_subunit(subunit))
    }

    pub const fn is_unverified(&self) -> bool {
        matches!(self, Self::Unverified)
    }

    pub const fn divide_by(&self, by: i64) -> Self {
        match self {
            Self::None => Self::None,
            Self::Unverified => Self::Unverified,
            Self::Fixed(money) => Self::Fixed(money.divide_by(by)),
            Self::Fuses(_) => panic!(".divide_by() is unsupported on Cost::Fuses"),
            Self::FusesYearlyConsumption(_) => {
                panic!(".divide_by() is unsupported on Cost::FuseRangeYearlyConsumption")
            }
            Self::FuseRange(_) => panic!(".divide_by() is unsupported on Cost::FuseRange"),
        }
    }

    pub const fn cost_for(&self, fuse_size: u16, yearly_consumption: u32) -> Option<Money> {
        match *self {
            Cost::None => None,
            Cost::Unverified => None,
            Cost::Fixed(money) => Some(money),
            Cost::Fuses(values) => {
                let mut i = 0;
                while i < values.len() {
                    let (fsize, money) = values[i];
                    if fuse_size == fsize {
                        return Some(money);
                    }
                    i += 1;
                }
                None
            }
            Cost::FusesYearlyConsumption(values) => {
                let mut i = 0;
                while i < values.len() {
                    let (fsize, max_consumption, money) = values[i];
                    if fsize == fuse_size {
                        if let Some(max_consumption) = max_consumption {
                            if yearly_consumption <= max_consumption {
                                return Some(money);
                            }
                        } else {
                            return Some(money);
                        }
                    }
                    i += 1;
                }
                None
            }
            Cost::FuseRange(ranges) => {
                let mut i = 0;
                while i < ranges.len() {
                    let (min, max, money) = ranges[i];
                    if fuse_size >= min && fuse_size <= max {
                        return Some(money);
                    }
                    i += 1;
                }
                None
            }
        }
    }

    pub const fn add_vat(&self, country: Country) -> Cost {
        match self {
            Cost::None => Cost::None,
            Cost::Unverified => Cost::Unverified,
            Cost::Fixed(money) => Cost::Fixed(money.add_vat(country)),
            Cost::Fuses(_) => todo!(),
            Cost::FusesYearlyConsumption(_) => todo!(),
            Cost::FuseRange(_) => todo!(),
        }
    }

    pub const fn remove_vat(&self, country: Country) -> Cost {
        match self {
            Cost::None => Cost::None,
            Cost::Unverified => Cost::Unverified,
            Cost::Fixed(money) => Cost::Fixed(money.remove_vat(country)),
            Cost::Fuses(_) => todo!(),
            Cost::FusesYearlyConsumption(_) => todo!(),
            Cost::FuseRange(_) => todo!(),
        }
    }

    pub fn is_yearly_consumption_based(&self, fuse_size: u16) -> bool {
        match self {
            Cost::FusesYearlyConsumption(items) => items
                .iter()
                .filter(|(fsize, _, _)| *fsize == fuse_size)
                .any(|(_, yearly_consumption, _)| yearly_consumption.is_some()),
            _ => false,
        }
    }
}

/// How to match against the different CostPeriod items
#[derive(Debug, Clone, Copy, Serialize, PartialEq, Eq)]
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
pub enum CostPeriodMatching {
    /// Only return the first matching CostPeriod
    First,
    /// Return all CostPeriod items that match
    All,
}

#[derive(Debug, Clone, Copy, Serialize)]
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
pub struct CostPeriods {
    match_method: CostPeriodMatching,
    periods: &'static [CostPeriod],
}

impl CostPeriods {
    // Will return first matching cost period
    pub const fn new_first(periods: &'static [CostPeriod]) -> Self {
        Self {
            match_method: CostPeriodMatching::First,
            periods,
        }
    }

    // Will return all matching cost periods
    pub const fn new_all(periods: &'static [CostPeriod]) -> Self {
        Self {
            match_method: CostPeriodMatching::All,
            periods,
        }
    }

    pub const fn match_method(&self) -> CostPeriodMatching {
        self.match_method
    }

    pub fn iter(&self) -> Iter<'_, CostPeriod> {
        self.periods.iter()
    }

    pub(crate) fn is_yearly_consumption_based(&self, fuse_size: u16) -> bool {
        self.periods
            .iter()
            .any(|cp| cp.is_yearly_consumption_based(fuse_size))
    }

    pub fn matching_periods<Tz: chrono::TimeZone>(
        &self,
        timestamp: DateTime<Tz>,
    ) -> Vec<&CostPeriod>
    where
        DateTime<Tz>: Copy,
    {
        let mut ret = vec![];
        for period in self.periods {
            if period.matches(timestamp) {
                ret.push(period);
                if self.match_method == CostPeriodMatching::First {
                    break;
                }
            }
        }
        ret
    }
}

/// Like CostPeriods, but with costs being simple Money objects
#[derive(Debug, Clone, Serialize)]
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
pub struct CostPeriodsSimple {
    periods: Vec<CostPeriodSimple>,
}

impl CostPeriodsSimple {
    pub(crate) fn new(
        periods: CostPeriods,
        fuse_size: u16,
        yearly_consumption: u32,
        language: Language,
    ) -> Self {
        Self {
            periods: periods
                .periods
                .iter()
                .flat_map(|period| {
                    CostPeriodSimple::new(period, fuse_size, yearly_consumption, language)
                })
                .collect(),
        }
    }
}

#[derive(Debug, Clone, Serialize)]
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
pub struct CostPeriod {
    cost: Cost,
    load: LoadType,
    #[serde(serialize_with = "helpers::skip_nones")]
    include: [Option<Include>; 2],
    #[serde(serialize_with = "helpers::skip_nones")]
    exclude: [Option<Exclude>; 2],
}

/// Like CostPeriod, but with cost being a simple Money object
#[derive(Debug, Clone, Serialize)]
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
pub(crate) struct CostPeriodSimple {
    cost: Money,
    load: LoadType,
    include: Vec<Include>,
    exclude: Vec<Exclude>,
    info: String,
}

impl CostPeriodSimple {
    fn new(
        period: &CostPeriod,
        fuse_size: u16,
        yearly_consumption: u32,
        language: Language,
    ) -> Option<Self> {
        let cost = period.cost().cost_for(fuse_size, yearly_consumption)?;
        Some(
            Self {
                cost,
                load: period.load,
                include: period.include.into_iter().flatten().collect(),
                exclude: period.exclude.into_iter().flatten().collect(),
                info: Default::default(),
            }
            .add_info(language),
        )
    }

    fn add_info(mut self, language: Language) -> Self {
        let mut infos = Vec::new();
        for include in &self.include {
            infos.push(include.translate(language));
        }
        for exclude in &self.exclude {
            infos.push(exclude.translate(language).into());
        }
        self.info = infos.join(", ");
        self
    }
}

impl CostPeriod {
    pub const fn builder() -> CostPeriodBuilder {
        CostPeriodBuilder::new()
    }

    pub const fn cost(&self) -> Cost {
        self.cost
    }

    pub const fn load(&self) -> LoadType {
        self.load
    }

    pub fn matches<Tz: chrono::TimeZone>(&self, timestamp: DateTime<Tz>) -> bool
    where
        DateTime<Tz>: Copy,
    {
        for include in self.include_period_types() {
            if !include.matches(timestamp) {
                return false;
            }
        }

        for exclude in self.exclude_period_types() {
            if exclude.matches(timestamp) {
                return false;
            }
        }
        true
    }

    fn include_period_types(&self) -> Vec<Include> {
        self.include.iter().flatten().copied().collect()
    }

    fn exclude_period_types(&self) -> Vec<Exclude> {
        self.exclude.iter().flatten().copied().collect()
    }

    fn is_yearly_consumption_based(&self, fuse_size: u16) -> bool {
        self.cost.is_yearly_consumption_based(fuse_size)
    }
}

#[derive(Clone)]
pub struct CostPeriodBuilder {
    timezone: Option<Timezone>,
    cost: Cost,
    load: Option<LoadType>,
    include: [Option<Include>; 2],
    exclude: [Option<Exclude>; 2],
}

impl Default for CostPeriodBuilder {
    fn default() -> Self {
        Self::new()
    }
}

impl CostPeriodBuilder {
    pub const fn new() -> Self {
        let builder = Self {
            timezone: None,
            cost: Cost::None,
            load: None,
            include: [None; 2],
            exclude: [None; 2],
        };
        // TODO: Don't hardcode this!
        builder.timezone(Timezone::Stockholm)
    }

    pub const fn build(self) -> CostPeriod {
        CostPeriod {
            cost: self.cost,
            load: self.load.expect("`load` must be specified"),
            include: self.include,
            exclude: self.exclude,
        }
    }

    pub const fn cost(mut self, cost: Cost) -> Self {
        self.cost = cost;
        self
    }

    pub const fn load(mut self, load: LoadType) -> Self {
        self.load = Some(load);
        self
    }

    pub const fn fixed_cost(mut self, int: i64, fract: u8) -> Self {
        self.cost = Cost::fixed(int, fract);
        self
    }

    pub const fn fixed_cost_subunit(mut self, subunit: f64) -> Self {
        self.cost = Cost::fixed_subunit(subunit);
        self
    }

    pub const fn timezone(mut self, timezone: Timezone) -> Self {
        self.timezone = Some(timezone);
        self
    }

    const fn get_timezone(&self) -> Timezone {
        self.timezone.expect("`timezone` must be specified")
    }

    pub const fn include(mut self, period_type: Include) -> Self {
        let mut i = 0;
        while i < self.include.len() {
            if self.include[i].is_some() {
                i += 1;
            } else {
                self.include[i] = Some(period_type);
                return self;
            }
        }
        panic!("Too many includes");
    }

    pub const fn months(self, from: Month, to: Month) -> Self {
        let timezone = self.get_timezone();
        self.include(Include::Months(Months::new(from, to, timezone)))
    }

    pub const fn month(self, month: Month) -> Self {
        self.months(month, month)
    }

    pub const fn hours(self, from: u8, to_inclusive: u8) -> Self {
        let timezone = self.get_timezone();
        self.include(Include::Hours(Hours::new(from, to_inclusive, timezone)))
    }

    const fn exclude(mut self, period_type: Exclude) -> Self {
        let mut i = 0;
        while i < self.exclude.len() {
            if self.exclude[i].is_some() {
                i += 1;
            } else {
                self.exclude[i] = Some(period_type);
                return self;
            }
        }
        panic!("Too many excludes");
    }

    pub const fn exclude_public_holidays(self, country: Country) -> Self {
        let tz = self.get_timezone();
        self.exclude(Exclude::PublicHolidays(country, tz))
    }

    pub const fn exclude_weekends(self) -> Self {
        let tz = self.get_timezone();
        self.exclude(Exclude::Weekends(tz))
    }
}

#[derive(Debug, Clone, Copy, Serialize)]
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
pub enum Include {
    Months(Months),
    Hours(Hours),
}

impl Include {
    fn translate(&self, language: Language) -> String {
        match self {
            Include::Months(months) => months.translate(language),
            Include::Hours(hours) => hours.translate(language),
        }
    }

    fn matches<Tz: chrono::TimeZone>(&self, timestamp: DateTime<Tz>) -> bool {
        match self {
            Include::Months(months) => months.matches(timestamp),
            Include::Hours(hours) => hours.matches(timestamp),
        }
    }
}

#[derive(Debug, Clone, Copy, Serialize)]
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
pub enum Exclude {
    Weekends(Timezone),
    PublicHolidays(Country, Timezone),
}

impl Exclude {
    pub(crate) fn translate(&self, language: Language) -> &'static str {
        match language {
            Language::En => match self {
                Exclude::Weekends(_) => "Weekends",
                Exclude::PublicHolidays(country, _) => match country {
                    Country::SE => "Swedish holidays",
                },
            },
            Language::Sv => match self {
                Exclude::Weekends(_) => "Helg",
                Exclude::PublicHolidays(country, _) => match country {
                    Country::SE => "Svenska helgdagar",
                },
            },
        }
    }

    fn matches<Tz: chrono::TimeZone>(&self, timestamp: DateTime<Tz>) -> bool {
        let tz_timestamp = timestamp.with_timezone(&self.tz());
        match self {
            Exclude::Weekends(_) => (6..=7).contains(&tz_timestamp.weekday().number_from_monday()),
            Exclude::PublicHolidays(country, _) => {
                country.is_public_holiday(tz_timestamp.date_naive())
            }
        }
    }

    const fn tz(&self) -> chrono_tz::Tz {
        match self {
            Exclude::Weekends(timezone) => timezone.to_tz(),
            Exclude::PublicHolidays(_, timezone) => timezone.to_tz(),
        }
    }
}

#[cfg(test)]
mod tests {

    use super::*;
    use crate::money::Money;
    use crate::months::Month::*;
    use crate::{Stockholm, Utc};

    #[test]
    fn cost_for_none() {
        const NONE_COST: Cost = Cost::None;
        assert_eq!(NONE_COST.cost_for(16, 0), None);
        assert_eq!(NONE_COST.cost_for(25, 5000), None);
    }

    #[test]
    fn cost_for_unverified() {
        const UNVERIFIED_COST: Cost = Cost::Unverified;
        assert_eq!(UNVERIFIED_COST.cost_for(16, 0), None);
        assert_eq!(UNVERIFIED_COST.cost_for(25, 5000), None);
    }

    #[test]
    fn cost_for_fixed() {
        const FIXED_COST: Cost = Cost::Fixed(Money::new(100, 50));
        // Fixed cost should return the same value regardless of fuse size or consumption
        assert_eq!(FIXED_COST.cost_for(16, 0), Some(Money::new(100, 50)));
        assert_eq!(FIXED_COST.cost_for(25, 5000), Some(Money::new(100, 50)));
        assert_eq!(FIXED_COST.cost_for(63, 10000), Some(Money::new(100, 50)));
    }

    #[test]
    fn cost_for_fuses_exact_match() {
        const FUSES_COST: Cost = Cost::fuses(&[
            (16, Money::new(50, 0)),
            (25, Money::new(75, 0)),
            (35, Money::new(100, 0)),
            (50, Money::new(150, 0)),
        ]);

        // Test exact matches
        assert_eq!(FUSES_COST.cost_for(16, 0), Some(Money::new(50, 0)));
        assert_eq!(FUSES_COST.cost_for(25, 0), Some(Money::new(75, 0)));
        assert_eq!(FUSES_COST.cost_for(35, 0), Some(Money::new(100, 0)));
        assert_eq!(FUSES_COST.cost_for(50, 0), Some(Money::new(150, 0)));

        // Yearly consumption should not affect the result
        assert_eq!(FUSES_COST.cost_for(25, 500000), Some(Money::new(75, 0)));
    }

    #[test]
    fn cost_for_fuses_no_match() {
        const FUSES_COST: Cost = Cost::fuses(&[(16, Money::new(50, 0)), (25, Money::new(75, 0))]);

        // Test non-matching fuse sizes
        assert_eq!(FUSES_COST.cost_for(20, 0), None);
        assert_eq!(FUSES_COST.cost_for(63, 0), None);
    }

    #[test]
    fn cost_for_fuses_yearly_consumption_with_limit() {
        const FUSES_WITH_CONSUMPTION: Cost = Cost::fuses_with_yearly_consumption(&[
            (16, Some(5000), Money::new(50, 0)),
            (16, None, Money::new(75, 0)),
            (25, Some(10000), Money::new(100, 0)),
            (25, None, Money::new(125, 0)),
        ]);

        // 16A fuse with consumption below limit - matches the entry with the limit
        assert_eq!(
            FUSES_WITH_CONSUMPTION.cost_for(16, 3000),
            Some(Money::new(50, 0))
        );

        // 16A fuse with consumption at limit - matches the entry with the limit
        assert_eq!(
            FUSES_WITH_CONSUMPTION.cost_for(16, 5000),
            Some(Money::new(50, 0))
        );

        // 16A fuse with consumption above limit - falls through to entry with no limit
        assert_eq!(
            FUSES_WITH_CONSUMPTION.cost_for(16, 6000),
            Some(Money::new(75, 0))
        );

        // 16A fuse with very high consumption - falls through to entry with no limit
        assert_eq!(
            FUSES_WITH_CONSUMPTION.cost_for(16, 20000),
            Some(Money::new(75, 0))
        );

        // 25A fuse with consumption at limit - matches the entry with 10000 limit
        assert_eq!(
            FUSES_WITH_CONSUMPTION.cost_for(25, 10000),
            Some(Money::new(100, 0))
        );

        // 25A fuse with consumption above limit - falls through to entry with no limit
        assert_eq!(
            FUSES_WITH_CONSUMPTION.cost_for(25, 15000),
            Some(Money::new(125, 0))
        );

        // 25A fuse with consumption below limit - matches the entry with the limit
        assert_eq!(
            FUSES_WITH_CONSUMPTION.cost_for(25, 5000),
            Some(Money::new(100, 0))
        );
    }

    #[test]
    fn cost_for_fuses_yearly_consumption_no_limit() {
        const FUSES_NO_LIMIT: Cost = Cost::fuses_with_yearly_consumption(&[
            (16, None, Money::new(50, 0)),
            (25, None, Money::new(75, 0)),
        ]);

        // Should match regardless of consumption when limit is None
        assert_eq!(FUSES_NO_LIMIT.cost_for(16, 0), Some(Money::new(50, 0)));
        assert_eq!(FUSES_NO_LIMIT.cost_for(16, 1000), Some(Money::new(50, 0)));
        assert_eq!(FUSES_NO_LIMIT.cost_for(16, 50000), Some(Money::new(50, 0)));
        assert_eq!(FUSES_NO_LIMIT.cost_for(25, 100000), Some(Money::new(75, 0)));
    }

    #[test]
    fn cost_for_fuses_yearly_consumption_no_fuse_match() {
        const FUSES_WITH_CONSUMPTION: Cost = Cost::fuses_with_yearly_consumption(&[
            (16, Some(5000), Money::new(50, 0)),
            (25, Some(10000), Money::new(100, 0)),
        ]);

        // Non-matching fuse size
        assert_eq!(FUSES_WITH_CONSUMPTION.cost_for(35, 5000), None);
        assert_eq!(FUSES_WITH_CONSUMPTION.cost_for(50, 10000), None);
    }

    #[test]
    fn cost_for_fuses_yearly_consumption_max_limit_no_fallback() {
        const FUSES_ONLY_LIMITS: Cost = Cost::fuses_with_yearly_consumption(&[
            (16, Some(5000), Money::new(50, 0)),
            (25, Some(10000), Money::new(100, 0)),
        ]);

        // Matching fuse size with consumption at or below limit - should match
        assert_eq!(FUSES_ONLY_LIMITS.cost_for(16, 0), Some(Money::new(50, 0)));
        assert_eq!(
            FUSES_ONLY_LIMITS.cost_for(16, 3000),
            Some(Money::new(50, 0))
        );
        assert_eq!(
            FUSES_ONLY_LIMITS.cost_for(16, 4999),
            Some(Money::new(50, 0))
        );
        assert_eq!(
            FUSES_ONLY_LIMITS.cost_for(16, 5000),
            Some(Money::new(50, 0))
        );
        assert_eq!(
            FUSES_ONLY_LIMITS.cost_for(25, 9999),
            Some(Money::new(100, 0))
        );
        assert_eq!(
            FUSES_ONLY_LIMITS.cost_for(25, 10000),
            Some(Money::new(100, 0))
        );

        // Above limit with no fallback - should return None
        assert_eq!(FUSES_ONLY_LIMITS.cost_for(16, 5001), None);
        assert_eq!(FUSES_ONLY_LIMITS.cost_for(16, 10000), None);
        assert_eq!(FUSES_ONLY_LIMITS.cost_for(25, 10001), None);
        assert_eq!(FUSES_ONLY_LIMITS.cost_for(25, 20000), None);
    }

    #[test]
    fn cost_for_fuse_range_within_range() {
        const FUSE_BASED: Cost = Cost::fuse_range(&[
            (16, 35, Money::new(54, 0)),
            (35, u16::MAX, Money::new(108, 50)),
        ]);

        // Test values below the first range
        assert_eq!(FUSE_BASED.cost_for(10, 0), None);
        assert_eq!(FUSE_BASED.cost_for(15, 0), None);

        // Test values within the first range
        assert_eq!(FUSE_BASED.cost_for(16, 0), Some(Money::new(54, 0)));
        assert_eq!(FUSE_BASED.cost_for(25, 0), Some(Money::new(54, 0)));
        assert_eq!(FUSE_BASED.cost_for(35, 0), Some(Money::new(54, 0)));

        // Test values within the second range
        assert_eq!(FUSE_BASED.cost_for(36, 0), Some(Money::new(108, 50)));
        assert_eq!(FUSE_BASED.cost_for(50, 0), Some(Money::new(108, 50)));
        assert_eq!(FUSE_BASED.cost_for(200, 0), Some(Money::new(108, 50)));
        assert_eq!(FUSE_BASED.cost_for(u16::MAX, 0), Some(Money::new(108, 50)));
    }

    #[test]
    fn cost_for_fuse_range_multiple_ranges() {
        const MULTI_RANGE: Cost = Cost::fuse_range(&[
            (1, 15, Money::new(20, 0)),
            (16, 35, Money::new(50, 0)),
            (36, 63, Money::new(100, 0)),
            (64, u16::MAX, Money::new(200, 0)),
        ]);

        // Test each range
        assert_eq!(MULTI_RANGE.cost_for(10, 0), Some(Money::new(20, 0)));
        assert_eq!(MULTI_RANGE.cost_for(25, 0), Some(Money::new(50, 0)));
        assert_eq!(MULTI_RANGE.cost_for(50, 0), Some(Money::new(100, 0)));
        assert_eq!(MULTI_RANGE.cost_for(100, 0), Some(Money::new(200, 0)));

        // Yearly consumption should not affect range-based costs
        assert_eq!(MULTI_RANGE.cost_for(25, 10000), Some(Money::new(50, 0)));
    }

    #[test]
    fn include_matches_hours() {
        let include = Include::Hours(Hours::new(6, 22, Stockholm));
        let timestamp_match = Stockholm.dt(2025, 1, 15, 14, 0, 0);
        let timestamp_no_match = Stockholm.dt(2025, 1, 15, 23, 0, 0);

        assert!(include.matches(timestamp_match));
        assert!(!include.matches(timestamp_no_match));
    }

    #[test]
    fn include_matches_months() {
        let include = Include::Months(Months::new(November, March, Stockholm));
        let timestamp_match = Stockholm.dt(2025, 1, 15, 12, 0, 0);
        let timestamp_no_match = Stockholm.dt(2025, 7, 15, 12, 0, 0);

        assert!(include.matches(timestamp_match));
        assert!(!include.matches(timestamp_no_match));
    }

    #[test]
    fn exclude_matches_weekends_saturday() {
        let exclude = Exclude::Weekends(Stockholm);
        // January 4, 2025 is a Saturday
        let timestamp = Stockholm.dt(2025, 1, 4, 12, 0, 0);
        assert!(exclude.matches(timestamp));
    }

    #[test]
    fn exclude_matches_weekends_sunday() {
        let exclude = Exclude::Weekends(Stockholm);
        // January 5, 2025 is a Sunday
        let timestamp = Stockholm.dt(2025, 1, 5, 12, 0, 0);
        assert!(exclude.matches(timestamp));
    }

    #[test]
    fn exclude_does_not_match_weekday() {
        let exclude = Exclude::Weekends(Stockholm);
        // January 6, 2025 is a Monday
        let timestamp = Stockholm.dt(2025, 1, 6, 12, 0, 0);
        assert!(!exclude.matches(timestamp));
    }

    #[test]
    fn exclude_matches_swedish_new_year() {
        let exclude = Exclude::PublicHolidays(Country::SE, Stockholm);
        // January 1 is a Swedish holiday
        let timestamp = Stockholm.dt(2025, 1, 1, 12, 0, 0);
        assert!(exclude.matches(timestamp));
    }

    #[test]
    fn exclude_does_not_match_non_holiday() {
        let exclude = Exclude::PublicHolidays(Country::SE, Stockholm);
        // January 2, 2025 is not a Swedish holiday
        let timestamp = Stockholm.dt(2025, 1, 2, 12, 0, 0);
        assert!(!exclude.matches(timestamp));
    }

    #[test]
    fn cost_period_matches_with_single_include() {
        let period = CostPeriod::builder()
            .load(LoadType::High)
            .fixed_cost(10, 0)
            .hours(6, 22)
            .build();

        let timestamp_match = Stockholm.dt(2025, 1, 15, 14, 0, 0);
        let timestamp_no_match = Stockholm.dt(2025, 1, 15, 23, 0, 0);

        assert!(period.matches(timestamp_match));
        assert!(!period.matches(timestamp_no_match));
    }

    #[test]
    fn cost_period_matches_with_multiple_includes() {
        let period = CostPeriod::builder()
            .load(LoadType::High)
            .fixed_cost(10, 0)
            .hours(6, 22)
            .months(November, March)
            .build();

        // Winter daytime - should match
        let timestamp_match = Stockholm.dt(2025, 1, 15, 14, 0, 0);
        // Winter nighttime - should not match (wrong hours)
        let timestamp_wrong_hours = Stockholm.dt(2025, 1, 15, 23, 0, 0);
        // Summer daytime - should not match (wrong months)
        let timestamp_wrong_months = Stockholm.dt(2025, 7, 15, 14, 0, 0);

        assert!(period.matches(timestamp_match));
        assert!(!period.matches(timestamp_wrong_hours));
        assert!(!period.matches(timestamp_wrong_months));
    }

    #[test]
    fn cost_period_matches_with_exclude_weekends() {
        let period = CostPeriod::builder()
            .load(LoadType::High)
            .fixed_cost(10, 0)
            .hours(6, 22)
            .exclude_weekends()
            .build();

        println!("Excludes: {:?}", period.exclude_period_types());
        println!("Includes: {:?}", period.include_period_types());

        // Monday daytime - should match
        let timestamp_weekday = Stockholm.dt(2025, 1, 6, 14, 0, 0);
        // Saturday daytime - should not match (excluded)
        let timestamp_saturday = Stockholm.dt(2025, 1, 4, 14, 0, 0);

        assert!(period.matches(timestamp_weekday));
        assert!(!period.matches(timestamp_saturday));
    }

    #[test]
    fn cost_period_matches_with_exclude_holidays() {
        let period = CostPeriod::builder()
            .load(LoadType::High)
            .fixed_cost(10, 0)
            .hours(6, 22)
            .exclude_public_holidays(Country::SE)
            .build();

        // Regular weekday - should match
        let timestamp_regular = Stockholm.dt(2025, 1, 2, 14, 0, 0);
        // New Year's Day - should not match (excluded)
        let timestamp_holiday = Stockholm.dt(2025, 1, 1, 14, 0, 0);

        assert!(period.matches(timestamp_regular));
        assert!(!period.matches(timestamp_holiday));
    }

    #[test]
    fn cost_period_matches_complex_scenario() {
        // Winter high load period: Nov-Mar, 6-22, excluding weekends and holidays
        let period = CostPeriod::builder()
            .load(LoadType::High)
            .fixed_cost(10, 0)
            .months(November, March)
            .hours(6, 22)
            .exclude_weekends()
            .exclude_public_holidays(Country::SE)
            .build();

        // Winter weekday daytime (not holiday) - should match
        let timestamp_match = Stockholm.dt(2025, 1, 15, 14, 0, 0);

        // Winter weekday nighttime - should not match (wrong hours)
        let timestamp_wrong_hours = Stockholm.dt(2025, 1, 15, 23, 0, 0);

        // Winter Saturday daytime - should not match (weekend)
        let timestamp_weekend = Stockholm.dt(2025, 1, 4, 14, 0, 0);

        // New Year's Day (holiday) - should not match
        let timestamp_holiday = Stockholm.dt(2025, 1, 1, 14, 0, 0);

        // Summer weekday daytime - should not match (wrong months)
        let timestamp_summer = Stockholm.dt(2025, 7, 15, 14, 0, 0);

        assert!(period.matches(timestamp_match));
        assert!(!period.matches(timestamp_wrong_hours));
        assert!(!period.matches(timestamp_weekend));
        assert!(!period.matches(timestamp_holiday));
        assert!(!period.matches(timestamp_summer));
    }

    #[test]
    fn cost_period_matches_base_load() {
        // Base load period with no restrictions
        let period = CostPeriod::builder()
            .load(LoadType::Base)
            .fixed_cost(5, 0)
            .build();

        // Should match any time
        let timestamp1 = Stockholm.dt(2025, 1, 1, 0, 0, 0);
        let timestamp2 = Stockholm.dt(2025, 7, 15, 23, 59, 59);
        let timestamp3 = Stockholm.dt(2025, 1, 4, 12, 0, 0);

        assert!(period.matches(timestamp1));
        assert!(period.matches(timestamp2));
        assert!(period.matches(timestamp3));
    }

    #[test]
    fn include_matches_hours_wraparound() {
        // Night hours crossing midnight: 22:00 to 05:59
        let include = Include::Hours(Hours::new(22, 5, Stockholm));

        // Should match late evening
        let timestamp_evening = Stockholm.dt(2025, 1, 15, 22, 0, 0);
        assert!(include.matches(timestamp_evening));

        // Should match midnight
        let timestamp_midnight = Stockholm.dt(2025, 1, 15, 0, 0, 0);
        assert!(include.matches(timestamp_midnight));

        // Should match early morning
        let timestamp_morning = Stockholm.dt(2025, 1, 15, 5, 30, 0);
        assert!(include.matches(timestamp_morning));

        // Should not match daytime
        let timestamp_day = Stockholm.dt(2025, 1, 15, 14, 0, 0);
        assert!(!include.matches(timestamp_day));

        // Should not match just after the range
        let timestamp_after = Stockholm.dt(2025, 1, 15, 6, 0, 0);
        assert!(!include.matches(timestamp_after));

        // Should not match just before the range
        let timestamp_before = Stockholm.dt(2025, 1, 15, 21, 59, 59);
        assert!(!include.matches(timestamp_before));
    }

    #[test]
    fn include_matches_months_wraparound() {
        // Winter months crossing year boundary: November to March
        let include = Include::Months(Months::new(November, March, Stockholm));

        // Should match November (start)
        let timestamp_nov = Stockholm.dt(2025, 11, 15, 12, 0, 0);
        assert!(include.matches(timestamp_nov));

        // Should match December
        let timestamp_dec = Stockholm.dt(2025, 12, 15, 12, 0, 0);
        assert!(include.matches(timestamp_dec));

        // Should match January
        let timestamp_jan = Stockholm.dt(2025, 1, 15, 12, 0, 0);
        assert!(include.matches(timestamp_jan));

        // Should match March (end)
        let timestamp_mar = Stockholm.dt(2025, 3, 15, 12, 0, 0);
        assert!(include.matches(timestamp_mar));

        // Should not match summer months
        let timestamp_jul = Stockholm.dt(2025, 7, 15, 12, 0, 0);
        assert!(!include.matches(timestamp_jul));

        // Should not match October (just before)
        let timestamp_oct = Stockholm.dt(2025, 10, 31, 23, 59, 59);
        assert!(!include.matches(timestamp_oct));

        // Should not match April (just after)
        let timestamp_apr = Stockholm.dt(2025, 4, 1, 0, 0, 0);
        assert!(!include.matches(timestamp_apr));
    }

    #[test]
    fn cost_period_matches_hours_wraparound() {
        // Night period: 22:00 to 05:59
        let period = CostPeriod::builder()
            .load(LoadType::Low)
            .fixed_cost(5, 0)
            .hours(22, 5)
            .build();

        let timestamp_match_evening = Stockholm.dt(2025, 1, 15, 23, 0, 0);
        let timestamp_match_morning = Stockholm.dt(2025, 1, 15, 3, 0, 0);
        let timestamp_no_match = Stockholm.dt(2025, 1, 15, 14, 0, 0);

        assert!(period.matches(timestamp_match_evening));
        assert!(period.matches(timestamp_match_morning));
        assert!(!period.matches(timestamp_no_match));
    }

    #[test]
    fn cost_period_matches_with_both_excludes() {
        let period = CostPeriod::builder()
            .load(LoadType::High)
            .fixed_cost(10, 0)
            .hours(6, 22)
            .exclude_weekends()
            .exclude_public_holidays(Country::SE)
            .build();

        // Regular weekday - should match
        let weekday = Stockholm.dt(2025, 1, 2, 14, 0, 0);
        assert!(period.matches(weekday));

        // Weekend - should not match
        let saturday = Stockholm.dt(2025, 1, 4, 14, 0, 0);
        assert!(!period.matches(saturday));

        // Holiday (New Year) - should not match
        let holiday = Stockholm.dt(2025, 1, 1, 14, 0, 0);
        assert!(!period.matches(holiday));

        // Weekday but wrong hours - should not match
        let wrong_hours = Stockholm.dt(2025, 1, 2, 23, 0, 0);
        assert!(!period.matches(wrong_hours));
    }

    #[test]
    fn exclude_matches_friday_is_not_weekend() {
        let exclude = Exclude::Weekends(Stockholm);
        // January 3, 2025 is a Friday
        let friday = Stockholm.dt(2025, 1, 3, 12, 0, 0);
        assert!(!exclude.matches(friday));
    }

    #[test]
    fn exclude_matches_monday_is_not_weekend() {
        let exclude = Exclude::Weekends(Stockholm);
        // January 6, 2025 is a Monday
        let monday = Stockholm.dt(2025, 1, 6, 12, 0, 0);
        assert!(!exclude.matches(monday));
    }

    #[test]
    fn exclude_matches_holiday_midsummer() {
        let exclude = Exclude::PublicHolidays(Country::SE, Stockholm);
        // Midsummer 2025 (June 21)
        let midsummer = Stockholm.dt(2025, 6, 21, 12, 0, 0);
        assert!(exclude.matches(midsummer));
    }

    #[test]
    fn cost_period_matches_month_and_hours() {
        // June with specific hours
        let period = CostPeriod::builder()
            .load(LoadType::Low)
            .fixed_cost(5, 0)
            .month(June)
            .hours(22, 5)
            .build();

        // June during night hours - should match
        let match_june_night = Stockholm.dt(2025, 6, 15, 23, 0, 0);
        assert!(period.matches(match_june_night));

        // June during day hours - should not match
        let june_day = Stockholm.dt(2025, 6, 15, 14, 0, 0);
        assert!(!period.matches(june_day));

        // July during night hours - should not match (wrong month)
        let july_night = Stockholm.dt(2025, 7, 15, 23, 0, 0);
        assert!(!period.matches(july_night));
    }

    #[test]
    fn cost_period_matches_months_and_hours_with_exclude() {
        // Winter high load: Nov-Mar, 6-22, excluding weekends and holidays
        let period = CostPeriod::builder()
            .load(LoadType::High)
            .fixed_cost(15, 0)
            .months(November, March)
            .hours(6, 22)
            .exclude_weekends()
            .exclude_public_holidays(Country::SE)
            .build();

        // Perfect match: winter weekday during day hours
        let perfect = Stockholm.dt(2025, 1, 15, 10, 0, 0);
        assert!(period.matches(perfect));

        // First hour of range
        let first_hour = Stockholm.dt(2025, 1, 15, 6, 0, 0);
        assert!(period.matches(first_hour));

        // Last hour of range
        let last_hour = Stockholm.dt(2025, 1, 15, 22, 59, 59);
        assert!(period.matches(last_hour));

        // Wrong hours (too early)
        let too_early = Stockholm.dt(2025, 1, 15, 5, 59, 59);
        assert!(!period.matches(too_early));

        // Wrong hours (too late)
        let too_late = Stockholm.dt(2025, 1, 15, 23, 0, 0);
        assert!(!period.matches(too_late));

        // Wrong month (summer)
        let summer = Stockholm.dt(2025, 7, 15, 10, 0, 0);
        assert!(!period.matches(summer));

        // Weekend
        let weekend = Stockholm.dt(2025, 1, 4, 10, 0, 0);
        assert!(!period.matches(weekend));
    }

    #[test]
    fn cost_period_matches_base_with_restrictions() {
        // Base load but with hour restrictions
        let period = CostPeriod::builder()
            .load(LoadType::Base)
            .fixed_cost(3, 0)
            .hours(0, 5)
            .build();

        // Should match only during specified hours
        let match_night = Stockholm.dt(2025, 1, 15, 3, 0, 0);
        assert!(period.matches(match_night));

        // Should not match outside hours
        let no_match_day = Stockholm.dt(2025, 1, 15, 14, 0, 0);
        assert!(!period.matches(no_match_day));
    }

    #[test]
    fn cost_period_matches_single_month() {
        let period = CostPeriod::builder()
            .load(LoadType::High)
            .fixed_cost(10, 0)
            .month(December)
            .build();

        // First day of December
        let dec_first = Stockholm.dt(2025, 12, 1, 0, 0, 0);
        assert!(period.matches(dec_first));

        // Last day of December
        let dec_last = Stockholm.dt(2025, 12, 31, 23, 59, 59);
        assert!(period.matches(dec_last));

        // November - should not match
        let nov = Stockholm.dt(2025, 11, 30, 12, 0, 0);
        assert!(!period.matches(nov));

        // January - should not match
        let jan = Stockholm.dt(2025, 1, 1, 12, 0, 0);
        assert!(!period.matches(jan));
    }

    #[test]
    fn cost_period_matches_all_hours() {
        // Full day coverage: 0-23
        let period = CostPeriod::builder()
            .load(LoadType::Low)
            .fixed_cost(5, 0)
            .hours(0, 23)
            .build();

        let midnight = Stockholm.dt(2025, 1, 15, 0, 0, 0);
        let noon = Stockholm.dt(2025, 1, 15, 12, 0, 0);
        let almost_midnight = Stockholm.dt(2025, 1, 15, 23, 59, 59);

        assert!(period.matches(midnight));
        assert!(period.matches(noon));
        assert!(period.matches(almost_midnight));
    }

    #[test]
    fn cost_period_matches_edge_of_month_range() {
        // May to September
        let period = CostPeriod::builder()
            .load(LoadType::Low)
            .fixed_cost(5, 0)
            .months(May, September)
            .build();

        // First second of May
        let may_start = Stockholm.dt(2025, 5, 1, 0, 0, 0);
        assert!(period.matches(may_start));

        // Last second of April - should not match
        let april_end = Stockholm.dt(2025, 4, 30, 23, 59, 59);
        assert!(!period.matches(april_end));

        // Last second of September
        let sept_end = Stockholm.dt(2025, 9, 30, 23, 59, 59);
        assert!(period.matches(sept_end));

        // First second of October - should not match
        let oct_start = Stockholm.dt(2025, 10, 1, 0, 0, 0);
        assert!(!period.matches(oct_start));
    }

    #[test]
    fn include_matches_month_boundary() {
        // Test first and last day of specific month
        let include = Include::Months(Months::new(February, February, Stockholm));

        // First second of February
        let feb_start = Stockholm.dt(2025, 2, 1, 0, 0, 0);
        assert!(include.matches(feb_start));

        // Last second of February
        let feb_end = Stockholm.dt(2025, 2, 28, 23, 59, 59);
        assert!(include.matches(feb_end));

        // Last second of January
        let jan_end = Stockholm.dt(2025, 1, 31, 23, 59, 59);
        assert!(!include.matches(jan_end));

        // First second of March
        let mar_start = Stockholm.dt(2025, 3, 1, 0, 0, 0);
        assert!(!include.matches(mar_start));
    }

    #[test]
    fn include_matches_hours_exact_boundaries() {
        let include = Include::Hours(Hours::new(6, 22, Stockholm));

        // First second of hour 6
        let start = Stockholm.dt(2025, 1, 15, 6, 0, 0);
        assert!(include.matches(start));

        // Last second of hour 22
        let end = Stockholm.dt(2025, 1, 15, 22, 59, 59);
        assert!(include.matches(end));

        // Last second of hour 5 (just before)
        let before = Stockholm.dt(2025, 1, 15, 5, 59, 59);
        assert!(!include.matches(before));

        // First second of hour 23 (just after)
        let after = Stockholm.dt(2025, 1, 15, 23, 0, 0);
        assert!(!include.matches(after));
    }

    #[test]
    fn exclude_matches_weekends_with_utc_timestamps() {
        let exclude = Exclude::Weekends(Stockholm);

        // Saturday January 4, 2025 at 12:00 Stockholm time
        // = Saturday at 11:00 UTC (Stockholm is UTC+1 in winter)
        let saturday_utc = Utc.dt(2025, 1, 4, 11, 0, 0);
        assert!(exclude.matches(saturday_utc));

        // Sunday January 5, 2025 at 12:00 Stockholm time
        // = Sunday at 11:00 UTC
        let sunday_utc = Utc.dt(2025, 1, 5, 11, 0, 0);
        assert!(exclude.matches(sunday_utc));

        // Monday January 6, 2025 at 12:00 Stockholm time
        // = Monday at 11:00 UTC
        let monday_utc = Utc.dt(2025, 1, 6, 11, 0, 0);
        assert!(!exclude.matches(monday_utc));
    }

    #[test]
    fn exclude_matches_weekends_timezone_boundary() {
        let exclude = Exclude::Weekends(Stockholm);

        // Saturday January 4, 2025 at 00:00 Stockholm time
        // = Friday January 3, 2025 at 23:00 UTC
        // This is tricky: it's Friday in UTC but Saturday in Stockholm
        let friday_utc_saturday_stockholm = Utc.dt(2025, 1, 3, 23, 0, 0);
        assert!(
            exclude.matches(friday_utc_saturday_stockholm),
            "Should match because it's Saturday in Stockholm timezone"
        );

        // Monday January 6, 2025 at 00:00 Stockholm time
        // = Sunday January 5, 2025 at 23:00 UTC
        // This is Sunday in UTC but Monday in Stockholm
        let sunday_utc_monday_stockholm = Utc.dt(2025, 1, 5, 23, 0, 0);
        assert!(
            !exclude.matches(sunday_utc_monday_stockholm),
            "Should not match because it's Monday in Stockholm timezone"
        );

        // Sunday January 5, 2025 at 23:59 Stockholm time
        // = Sunday January 5, 2025 at 22:59 UTC
        let sunday_late_utc = Utc.dt(2025, 1, 5, 22, 59, 0);
        assert!(
            exclude.matches(sunday_late_utc),
            "Should match because it's still Sunday in Stockholm timezone"
        );
    }

    #[test]
    fn exclude_matches_holidays_with_utc_timestamps() {
        let exclude = Exclude::PublicHolidays(Country::SE, Stockholm);

        // New Year's Day 2025 at 12:00 Stockholm time
        // = January 1, 2025 at 11:00 UTC
        let new_year_utc = Utc.dt(2025, 1, 1, 11, 0, 0);
        assert!(exclude.matches(new_year_utc));

        // Regular day: January 2, 2025 at 12:00 Stockholm time
        // = January 2, 2025 at 11:00 UTC
        let regular_day_utc = Utc.dt(2025, 1, 2, 11, 0, 0);
        assert!(!exclude.matches(regular_day_utc));
    }

    #[test]
    fn exclude_matches_holidays_timezone_boundary() {
        let exclude = Exclude::PublicHolidays(Country::SE, Stockholm);

        // New Year's Day 2025 at 00:00 Stockholm time
        // = December 31, 2024 at 23:00 UTC
        // This is Dec 31 in UTC but Jan 1 (holiday) in Stockholm
        let dec31_utc_jan1_stockholm = Utc.dt(2024, 12, 31, 23, 0, 0);
        assert!(
            exclude.matches(dec31_utc_jan1_stockholm),
            "Should match because it's New Year's Day in Stockholm timezone"
        );

        // January 2, 2025 at 00:00 Stockholm time
        // = January 1, 2025 at 23:00 UTC
        // This is Jan 1 (holiday) in UTC but Jan 2 (not holiday) in Stockholm
        let jan1_utc_jan2_stockholm = Utc.dt(2025, 1, 1, 23, 0, 0);
        assert!(
            !exclude.matches(jan1_utc_jan2_stockholm),
            "Should not match because it's January 2 in Stockholm timezone"
        );
    }

    #[test]
    fn exclude_matches_weekends_summer_timezone() {
        let exclude = Exclude::Weekends(Stockholm);

        // Saturday June 7, 2025 at 12:00 Stockholm time (CEST = UTC+2)
        // = Saturday at 10:00 UTC
        let saturday_summer_utc = Utc.dt(2025, 6, 7, 10, 0, 0);
        assert!(exclude.matches(saturday_summer_utc));

        // Saturday June 7, 2025 at 00:00 Stockholm time
        // = Friday June 6, 2025 at 22:00 UTC
        let friday_utc_saturday_stockholm_summer = Utc.dt(2025, 6, 6, 22, 0, 0);
        assert!(
            exclude.matches(friday_utc_saturday_stockholm_summer),
            "Should match because it's Saturday in Stockholm timezone (CEST)"
        );
    }
}