icu_calendar 2.3.0

// This file is part of ICU4X. For terms of use, please see the file
// called LICENSE at the top level of the ICU4X source tree
// (online at: https://github.com/unicode-org/icu4x/blob/main/LICENSE ).

//! 🚧 \[Unstable\] Data provider struct definitions for this ICU4X component.
//!
//! <div class="stab unstable">
//! 🚧 This code is considered unstable; it may change at any time, in breaking or non-breaking ways,
//! including in SemVer minor releases. While the serde representation of data structs is guaranteed
//! to be stable, their Rust representation might not be. Use with caution.
//! </div>
//!
//! Read more about data providers: [`icu_provider`]

// Provider structs must be stable
#![allow(clippy::exhaustive_structs, clippy::exhaustive_enums)]

use core::fmt::Debug;
use core::num::NonZeroU8;

use crate::types::Weekday;
use icu_locale_core::preferences::extensions::unicode::keywords::{
    CalendarAlgorithm, HijriCalendarAlgorithm,
};
use icu_provider::fallback::{LocaleFallbackConfig, LocaleFallbackPriority};
use icu_provider::prelude::*;
use tinystr::TinyAsciiStr;

#[cfg(feature = "compiled_data")]
#[derive(Debug)]
/// Baked data
///
/// <div class="stab unstable">
/// 🚧 This code is considered unstable; it may change at any time, in breaking or non-breaking ways,
/// including in SemVer minor releases. In particular, the `DataProvider` implementations are only
/// guaranteed to match with this version's `*_unstable` providers. Use with caution.
/// </div>
pub struct Baked;

#[cfg(feature = "compiled_data")]
#[allow(unused_imports)]
const _: () = {
    use icu_calendar_data::*;
    pub mod icu {
        pub use crate as calendar;
    }
    make_provider!(Baked);
    impl_calendar_japanese_modern_v1!(Baked);
    impl_calendar_preferred_v1!(Baked);
    impl_calendar_week_v1!(Baked);
};

icu_provider::data_marker!(
    /// Modern Japanese era names
    CalendarJapaneseModernV1,
    "calendar/japanese/modern/v1",
    JapaneseEras,
    is_singleton = true
);

icu_provider::data_marker!(
    /// Week information
    CalendarWeekV1,
    "calendar/week/v1",
    WeekData,
    fallback_config = {
        let mut config = LocaleFallbackConfig::default();
        config.priority = LocaleFallbackPriority::Region;
        config
    },
);

/// Default calendar preferences for a region.
#[derive(Debug, Clone, Copy, PartialEq, Eq, yoke::Yokeable, zerofrom::ZeroFrom)]
#[cfg_attr(feature = "datagen", derive(databake::Bake))]
#[cfg_attr(feature = "datagen", databake(path = icu_calendar::provider))]
pub struct CalendarPreference {
    /// The default calendar to use for this region.
    pub default_algorithm: CalendarAlgorithm,
    /// The default Hijri calendar to use for this region.
    pub default_hijri_algorithm: HijriCalendarAlgorithm,
}

#[cfg(any(feature = "serde", feature = "datagen"))]
#[derive(serde::Serialize, serde::Deserialize)]
struct CalendarPreferenceHuman<'a> {
    #[serde(borrow)]
    default_algorithm: &'a str,
    #[serde(borrow)]
    default_hijri_algorithm: &'a str,
}

#[cfg(feature = "datagen")]
impl serde::Serialize for CalendarPreference {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        if serializer.is_human_readable() {
            let human = CalendarPreferenceHuman {
                default_algorithm: self.default_algorithm.as_str(),
                default_hijri_algorithm: CalendarAlgorithm::Hijri(Some(
                    self.default_hijri_algorithm,
                ))
                .as_str(),
            };
            return human.serialize(serializer);
        }

        let default = match self.default_algorithm {
            CalendarAlgorithm::Buddhist => 1u8,
            CalendarAlgorithm::Chinese => 2,
            CalendarAlgorithm::Coptic => 3,
            CalendarAlgorithm::Dangi => 4,
            CalendarAlgorithm::Ethioaa => 5,
            CalendarAlgorithm::Ethiopic => 6,
            CalendarAlgorithm::Gregory => 7,
            CalendarAlgorithm::Hebrew => 8,
            CalendarAlgorithm::Indian => 9,
            CalendarAlgorithm::Hijri(Some(s)) if s == self.default_hijri_algorithm => 10,
            CalendarAlgorithm::Iso8601 => 11,
            CalendarAlgorithm::Japanese => 12,
            CalendarAlgorithm::Persian => 13,
            CalendarAlgorithm::Roc => 14,
            _ => {
                debug_assert!(
                    false,
                    "unknown calendar algorithm: {:?}",
                    self.default_algorithm
                );
                0
            }
        };
        let hijri = match self.default_hijri_algorithm {
            HijriCalendarAlgorithm::Umalqura => 1,
            HijriCalendarAlgorithm::Tbla => 2,
            HijriCalendarAlgorithm::Civil => 3,
            HijriCalendarAlgorithm::Rgsa => 4,
            _ => {
                debug_assert!(
                    false,
                    "unknown hijri algorithm: {:?}",
                    self.default_hijri_algorithm
                );
                0
            }
        };

        (default << 3 | hijri).serialize(serializer)
    }
}

#[cfg(feature = "serde")]
impl<'de> serde::Deserialize<'de> for CalendarPreference {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        if deserializer.is_human_readable() {
            let human = CalendarPreferenceHuman::deserialize(deserializer)?;
            let default_val =
                icu_locale_core::extensions::unicode::Value::try_from_str(human.default_algorithm)
                    .map_err(serde::de::Error::custom)?;
            let default_algorithm =
                CalendarAlgorithm::try_from(&default_val).map_err(serde::de::Error::custom)?;
            let hijri_val = icu_locale_core::extensions::unicode::Value::try_from_str(
                human.default_hijri_algorithm,
            )
            .map_err(serde::de::Error::custom)?;
            let default_hijri_algorithm = match CalendarAlgorithm::try_from(&hijri_val) {
                Ok(CalendarAlgorithm::Hijri(Some(h))) => h,
                _ => return Err(serde::de::Error::custom("expected islamic-* calendar")),
            };
            return Ok(Self {
                default_algorithm,
                default_hijri_algorithm,
            });
        }

        let packed = u8::deserialize(deserializer)?;
        let default_algorithm = packed >> 3;
        let default_hijri_algorithm = packed & 0x07;
        let default_hijri_algorithm = match default_hijri_algorithm {
            1 => HijriCalendarAlgorithm::Umalqura,
            2 => HijriCalendarAlgorithm::Tbla,
            3 => HijriCalendarAlgorithm::Civil,
            4 => HijriCalendarAlgorithm::Rgsa,
            _ => {
                debug_assert!(
                    false,
                    "unknown hijri algorithm discriminant: {default_hijri_algorithm}"
                );
                HijriCalendarAlgorithm::Civil
            }
        };
        let default_algorithm = match default_algorithm {
            1 => CalendarAlgorithm::Buddhist,
            2 => CalendarAlgorithm::Chinese,
            3 => CalendarAlgorithm::Coptic,
            4 => CalendarAlgorithm::Dangi,
            5 => CalendarAlgorithm::Ethioaa,
            6 => CalendarAlgorithm::Ethiopic,
            7 => CalendarAlgorithm::Gregory,
            8 => CalendarAlgorithm::Hebrew,
            9 => CalendarAlgorithm::Indian,
            10 => CalendarAlgorithm::Hijri(Some(default_hijri_algorithm)),
            11 => CalendarAlgorithm::Iso8601,
            12 => CalendarAlgorithm::Japanese,
            13 => CalendarAlgorithm::Persian,
            14 => CalendarAlgorithm::Roc,
            _ => {
                debug_assert!(
                    false,
                    "unknown calendar algorithm discriminant: {default_algorithm}"
                );
                CalendarAlgorithm::Gregory
            }
        };
        Ok(Self {
            default_algorithm,
            default_hijri_algorithm,
        })
    }
}

impl CalendarPreference {
    /// Resolves a [`Option<CalendarAlgorithm>`] to a concrete algorithm against these preferences.
    pub fn resolve(&self, unresolved: Option<CalendarAlgorithm>) -> CalendarAlgorithm {
        match unresolved {
            Some(CalendarAlgorithm::Hijri(None)) => {
                CalendarAlgorithm::Hijri(Some(self.default_hijri_algorithm))
            }
            Some(a) => a,
            None => self.default_algorithm,
        }
    }
}

icu_provider::data_marker!(
    /// The preferred calendar algorithm for a region.
    CalendarPreferredV1,
    "calendar/preferred/v1",
    CalendarPreference,
    fallback_config = {
        let mut config = LocaleFallbackConfig::default();
        config.priority = LocaleFallbackPriority::Region;
        config
    },
);

icu_provider::data_struct!(
    CalendarPreference,
    #[cfg(feature = "datagen")]
);

#[cfg(feature = "datagen")]
/// The latest minimum set of markers required by this component.
pub const MARKERS: &[DataMarkerInfo] = &[
    CalendarJapaneseModernV1::INFO,
    CalendarWeekV1::INFO,
    CalendarPreferredV1::INFO,
];

/// The date at which an era started
///
/// The order of fields in this struct is important!
///
/// <div class="stab unstable">
/// 🚧 This code is considered unstable; it may change at any time, in breaking or non-breaking ways,
/// including in SemVer minor releases. While the serde representation of data structs is guaranteed
/// to be stable, their Rust representation might not be. Use with caution.
/// </div>
#[zerovec::make_ule(EraStartDateULE)]
#[derive(
    Copy, Clone, PartialEq, PartialOrd, Eq, Ord, Hash, Debug, yoke::Yokeable, zerofrom::ZeroFrom,
)]
#[cfg_attr(feature = "datagen", derive(serde::Serialize, databake::Bake))]
#[cfg_attr(feature = "datagen", databake(path = icu_calendar::provider))]
#[cfg_attr(feature = "serde", derive(serde::Deserialize))]
#[cfg_attr(not(feature = "alloc"), zerovec::skip_derive(ZeroMapKV))]
pub struct EraStartDate {
    /// The Gregorian year the era started in
    pub year: i32,
    /// The Gregorian month the era started in
    pub month: u8,
    /// The Gregorian day the era started in
    pub day: u8,
}

/// A data structure containing the necessary era data for constructing a
/// [`Japanese`](crate::cal::Japanese) calendar object
///
/// <div class="stab unstable">
/// 🚧 This code is considered unstable; it may change at any time, in breaking or non-breaking ways,
/// including in SemVer minor releases. While the serde representation of data structs is guaranteed
/// to be stable, their Rust representation might not be. Use with caution.
/// </div>
#[derive(Debug, PartialEq, Clone, Copy, Default, yoke::Yokeable, zerofrom::ZeroFrom)]
pub struct JapaneseEras(Option<PackedEra>);

icu_provider::data_struct!(
    JapaneseEras,
    #[cfg(feature = "datagen")]
);

impl JapaneseEras {
    /// Creates a new [`JapaneseEras`] without a post-Reiwa era.
    pub const fn up_to_reiwa() -> Self {
        Self(None)
    }

    /// Creates a new [`JapaneseEras`] with the last post-Reiwa era's start date, code, and index.
    pub const fn with_last_era(
        start: EraStartDate,
        code: TinyAsciiStr<8>,
        index: u8,
    ) -> Option<Self> {
        if start.year < 2026 {
            return Some(Self::up_to_reiwa());
        }
        let Some(packed) = PackedEra::pack(start, code, index) else {
            return None;
        };
        Some(Self(Some(packed)))
    }

    /// Returns all post-Meji eras, in chronological order.
    pub fn eras(self) -> impl DoubleEndedIterator<Item = (EraStartDate, TinyAsciiStr<8>, u8)> {
        use tinystr::tinystr;
        [
            (
                EraStartDate {
                    year: 1868,
                    month: 10,
                    day: 23,
                },
                tinystr!(8, "meiji"),
                2,
            ),
            (
                EraStartDate {
                    year: 1912,
                    month: 7,
                    day: 30,
                },
                tinystr!(8, "taisho"),
                3,
            ),
            (
                EraStartDate {
                    year: 1926,
                    month: 12,
                    day: 25,
                },
                tinystr!(8, "showa"),
                4,
            ),
            (
                EraStartDate {
                    year: 1989,
                    month: 1,
                    day: 8,
                },
                tinystr!(8, "heisei"),
                5,
            ),
            (
                EraStartDate {
                    year: 2019,
                    month: 5,
                    day: 1,
                },
                tinystr!(8, "reiwa"),
                6,
            ),
        ]
        .into_iter()
        .chain(self.0.map(PackedEra::unpack))
    }
}

#[cfg(feature = "datagen")]
impl databake::Bake for JapaneseEras {
    fn bake(&self, ctx: &databake::CrateEnv) -> databake::TokenStream {
        ctx.insert("icu_calendar");
        if let Some((era_start_date, code, idx)) = self.eras().last()
            && code != "reiwa"
        {
            let era_start_date = era_start_date.bake(ctx);
            let code = code.bake(ctx);
            let idx = idx.bake(ctx);
            databake::quote! {
                icu_calendar::provider::JapaneseEras::with_last_era(#era_start_date, #code, #idx).unwrap()
            }
        } else {
            databake::quote! {
                icu_calendar::provider::JapaneseEras::up_to_reiwa()
            }
        }
    }
}

#[cfg(feature = "datagen")]
impl databake::BakeSize for JapaneseEras {
    fn borrows_size(&self) -> usize {
        0
    }
}

#[cfg(feature = "datagen")]
impl serde::Serialize for JapaneseEras {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        use zerovec::ZeroVec;

        #[derive(serde::Serialize)]
        struct Raw<'data> {
            pub dates_to_eras: ZeroVec<'data, (EraStartDate, TinyAsciiStr<16>)>,
        }

        let mut dates_to_eras = self
            .eras()
            .map(|(start, code, _)| (start, code.resize()))
            .collect::<ZeroVec<_>>();

        while let Some((start, code, index)) = self.eras().last()
            && dates_to_eras.len() + 2 <= index as usize
        {
            use zerovec::ule::AsULE;

            dates_to_eras.with_mut(|v| v.push((start, code.resize()).to_unaligned()));
        }

        Raw { dates_to_eras }.serialize(serializer)
    }
}

#[cfg(feature = "serde")]
impl<'de> serde::Deserialize<'de> for JapaneseEras {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        use serde::de::Error;
        use zerovec::ZeroVec;

        #[derive(serde::Deserialize)]
        struct Raw<'data> {
            #[serde(borrow)]
            pub dates_to_eras: ZeroVec<'data, (EraStartDate, TinyAsciiStr<16>)>,
        }

        let Raw { dates_to_eras } = Raw::deserialize(deserializer)?;

        let Some((start, code)) = dates_to_eras.last() else {
            return Err(D::Error::custom("At least one era is required"));
        };

        if Self::up_to_reiwa()
            .eras()
            .map(|(start, code, ..)| (start, code.resize()))
            .ne(dates_to_eras
                .iter()
                .take(Self::up_to_reiwa().eras().count()))
        {
            return Err(D::Error::custom(
                "Invalid era data: Meiji through Reiwa must be present and in order",
            ));
        }

        Self::with_last_era(
            start,
            code.resize(),
            dates_to_eras.as_slice().len() as u8 + 1,
        )
        .ok_or_else(|| D::Error::custom("Invalid era"))
    }
}

/// A type to represent a modern (post 2026, 8-byte code) era.
///
/// <div class="stab unstable">
/// 🚧 This code is considered unstable; it may change at any time, in breaking or non-breaking ways,
/// including in SemVer minor releases. While the serde representation of data structs is guaranteed
/// to be stable, their Rust representation might not be. Use with caution.
/// </div>
#[derive(PartialEq, Copy, Clone, yoke::Yokeable, zerofrom::ZeroFrom)]
struct PackedEra(NonZeroU8 /* give a niche */, [u8; 6]);

impl Debug for PackedEra {
    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
        let (start, code, index) = self.unpack();
        f.debug_struct("PackedEra")
            .field("start", &start)
            .field("code", &code)
            .field("index", &index)
            .field("<packed>", &(self.0, self.1))
            .finish()
    }
}

impl PackedEra {
    /// Construct a `PackedEra` from a tuple
    /// This supports start dates until 2086-12-31, indices 7 and 8, and lower ASCII alphabetic codes
    const fn pack(start: EraStartDate, code: TinyAsciiStr<8>, index: u8) -> Option<Self> {
        // 5 bits
        if !(0 < start.day && start.day < 1 << 5) {
            return None;
        }
        // 4 bits
        if !(0 < start.month && start.month < 1 << 4) {
            return None;
        }
        // 6 bits
        if !(2026 <= start.year && start.year < 2026 + (1 << 6)) {
            return None;
        }
        // 1 bit
        if !(7 <= index && index < 7 + (1 << 1)) {
            return None;
        }
        // 40 bits
        if !code.is_ascii_alphabetic_lowercase() {
            return None;
        }

        let mut packed = 0;

        let &[c1, c2, c3, c4, c5, c6, c7, c8] = code.all_bytes();
        packed = (packed << 5) | c1.saturating_sub(b'a' - 1) as u64;
        packed = (packed << 5) | c2.saturating_sub(b'a' - 1) as u64;
        packed = (packed << 5) | c3.saturating_sub(b'a' - 1) as u64;
        packed = (packed << 5) | c4.saturating_sub(b'a' - 1) as u64;
        packed = (packed << 5) | c5.saturating_sub(b'a' - 1) as u64;
        packed = (packed << 5) | c6.saturating_sub(b'a' - 1) as u64;
        packed = (packed << 5) | c7.saturating_sub(b'a' - 1) as u64;
        packed = (packed << 5) | c8.saturating_sub(b'a' - 1) as u64;

        packed = (packed << 1) | (index as u64 - 7);
        packed = (packed << 6) | (start.year as u64 - 2026);
        packed = (packed << 4) | start.month as u64;
        packed = (packed << 5) | start.day as u64;

        let [a, b, c, d, e, f, g, z] = packed.to_le_bytes();
        debug_assert!(z == 0);
        let Some(a) = NonZeroU8::new(a) else {
            return None;
        };
        Some(Self(a, [b, c, d, e, f, g]))
    }

    /// Convert a `PackedEra` into a tuple of start date, code, and
    /// [`era_index`](crate::types::EraYear::era_index)
    const fn unpack(self) -> (EraStartDate, TinyAsciiStr<8>, u8) {
        let Self(a, [b, c, d, e, f, g]) = self;
        let mut packed = u64::from_le_bytes([a.get(), b, c, d, e, f, g, 0]);

        let (day, month, year, index);
        (packed, day) = ((packed >> 5), (packed & 0b1_1111) as u8);
        (packed, month) = ((packed >> 4), (packed & 0b1111) as u8);
        (packed, year) = ((packed >> 6), (packed & 0b11_1111) as i32 + 2026);
        (packed, index) = ((packed >> 1), (packed & 0b1) as u8 + 7);

        let (c1, c2, c3, c4, c5, c6, c7, c8);
        (packed, c8) = ((packed >> 5), (packed & 0b1_1111) as u8 + (b'a' - 1));
        (packed, c7) = ((packed >> 5), (packed & 0b1_1111) as u8 + (b'a' - 1));
        (packed, c6) = ((packed >> 5), (packed & 0b1_1111) as u8 + (b'a' - 1));
        (packed, c5) = ((packed >> 5), (packed & 0b1_1111) as u8 + (b'a' - 1));
        (packed, c4) = ((packed >> 5), (packed & 0b1_1111) as u8 + (b'a' - 1));
        (packed, c3) = ((packed >> 5), (packed & 0b1_1111) as u8 + (b'a' - 1));
        (packed, c2) = ((packed >> 5), (packed & 0b1_1111) as u8 + (b'a' - 1));
        (packed, c1) = ((packed >> 5), (packed & 0b1_1111) as u8 + (b'a' - 1));

        debug_assert!(packed == 0);

        (
            EraStartDate { year, month, day },
            unsafe {
                TinyAsciiStr::from_utf8_unchecked([
                    if c1 < b'a' { 0 } else { c1 },
                    if c2 < b'a' { 0 } else { c2 },
                    if c3 < b'a' { 0 } else { c3 },
                    if c4 < b'a' { 0 } else { c4 },
                    if c5 < b'a' { 0 } else { c5 },
                    if c6 < b'a' { 0 } else { c6 },
                    if c7 < b'a' { 0 } else { c7 },
                    if c8 < b'a' { 0 } else { c8 },
                ])
            },
            index,
        )
    }
}

#[test]
fn japanese_eras_size() {
    assert_eq!(size_of::<PackedEra>(), 7);
    assert_eq!(size_of::<Option<PackedEra>>(), 7);
    assert_eq!(size_of::<crate::cal::Japanese>(), 7);
    assert_eq!(size_of::<crate::cal::AnyCalendar>(), 8);
}

#[test]
fn japanese_eras_packing() {
    let start = EraStartDate {
        year: 2086,
        month: 11,
        day: 1,
    };
    let code = tinystr::tinystr!(8, "fuzu");
    let index = 8;

    assert_eq!(
        JapaneseEras::with_last_era(start, code, index)
            .unwrap()
            .eras()
            .last()
            .unwrap(),
        (start, code, index)
    );
    assert_ne!(
        JapaneseEras::up_to_reiwa().eras().last().unwrap(),
        (start, code, index)
    );
}

#[test]
fn japanese_eras_serde_roundtrip() {
    let eras = JapaneseEras::with_last_era(
        EraStartDate {
            year: 2086,
            month: 11,
            day: 1,
        },
        tinystr::tinystr!(8, "fuzu"),
        8,
    )
    .unwrap();

    assert_eq!(
        serde_json::from_str::<JapaneseEras>(&serde_json::to_string(&eras).unwrap()).unwrap(),
        eras
    );

    let eras = JapaneseEras::up_to_reiwa();

    assert_eq!(
        serde_json::from_str::<JapaneseEras>(&serde_json::to_string(&eras).unwrap()).unwrap(),
        eras
    );
}

#[test]
fn japanese_eras_bake() {
    databake::test_bake!(
        JapaneseEras,
        const,
        crate::provider::JapaneseEras::with_last_era(
            crate::provider::EraStartDate {
                year: 2086i32,
                month: 11u8,
                day: 1u8,
            },
            tinystr::tinystr!(8usize, "fuzu"),
            8u8
        )
        .unwrap(),
        icu_calendar
    );
    databake::test_bake!(
        JapaneseEras,
        const,
        crate::provider::JapaneseEras::up_to_reiwa(),
        icu_calendar
    );
}

/// An ICU4X mapping to a subset of CLDR weekData.
/// See CLDR-JSON's weekData.json for more context.
///
/// <div class="stab unstable">
/// 🚧 This code is considered unstable; it may change at any time, in breaking or non-breaking ways,
/// including in SemVer minor releases. While the serde representation of data structs is guaranteed
/// to be stable, their Rust representation might not be. Use with caution.
/// </div>
#[derive(Clone, Copy, Debug, PartialEq, yoke::Yokeable, zerofrom::ZeroFrom)]
#[cfg_attr(feature = "datagen", derive(serde::Serialize, databake::Bake))]
#[cfg_attr(feature = "datagen", databake(path = icu_calendar::provider))]
#[cfg_attr(feature = "serde", derive(serde::Deserialize))]
#[allow(clippy::exhaustive_structs)] // used in data provider
pub struct WeekData {
    /// The first day of a week.
    pub first_weekday: Weekday,
    /// Bitset representing weekdays that are part of the 'weekend', for calendar purposes.
    /// The number of days can be different between locales, and may not be contiguous.
    pub weekend: WeekdaySet,
}

icu_provider::data_struct!(
    WeekData,
    #[cfg(feature = "datagen")]
);

/// Bitset representing weekdays.
//
// This Bitset uses an [u8] to represent the weekend, thus leaving one bit free.
// Each bit represents a day in the following order:
//
//   ┌▷Mon
//   │┌▷Tue
//   ││┌▷Wed
//   │││┌▷Thu
//   ││││ ┌▷Fri
//   ││││ │┌▷Sat
//   ││││ ││┌▷Sun
//   ││││ │││
// 0b0000_1010
//
// Please note that this is not a range, this are the discrete days representing a weekend. Other examples:
// 0b0101_1000 -> Tue, Thu, Fri
// 0b0000_0110 -> Sat, Sun
#[derive(Clone, Copy, Debug, PartialEq)]
pub struct WeekdaySet(u8);

impl WeekdaySet {
    /// Returns whether the set contains the day.
    pub const fn contains(&self, day: Weekday) -> bool {
        self.0 & day.bit_value() != 0
    }
}

impl WeekdaySet {
    /// Creates a new [`WeekdaySet`] using the provided days.
    pub const fn new(days: &[Weekday]) -> Self {
        let mut i = 0;
        let mut w = 0;
        #[expect(clippy::indexing_slicing)]
        while i < days.len() {
            w |= days[i].bit_value();
            i += 1;
        }
        Self(w)
    }
}

impl Weekday {
    /// Defines the bit order used for encoding and reading weekend days.
    const fn bit_value(self) -> u8 {
        match self {
            Weekday::Monday => 1 << 6,
            Weekday::Tuesday => 1 << 5,
            Weekday::Wednesday => 1 << 4,
            Weekday::Thursday => 1 << 3,
            Weekday::Friday => 1 << 2,
            Weekday::Saturday => 1 << 1,
            Weekday::Sunday => 1 << 0,
        }
    }
}

#[cfg(feature = "datagen")]
impl databake::Bake for WeekdaySet {
    fn bake(&self, ctx: &databake::CrateEnv) -> databake::TokenStream {
        ctx.insert("icu_calendar");
        let days =
            crate::week::WeekdaySetIterator::new(Weekday::Monday, *self).map(|d| d.bake(ctx));
        databake::quote! {
            icu_calendar::provider::WeekdaySet::new(&[#(#days),*])
        }
    }
}

#[cfg(feature = "datagen")]
impl databake::BakeSize for WeekdaySet {
    fn borrows_size(&self) -> usize {
        0
    }
}

#[cfg(feature = "datagen")]
impl serde::Serialize for WeekdaySet {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        if serializer.is_human_readable() {
            use serde::ser::SerializeSeq;

            let mut seq = serializer.serialize_seq(None)?;
            for day in crate::week::WeekdaySetIterator::new(Weekday::Monday, *self) {
                seq.serialize_element(&day)?;
            }
            seq.end()
        } else {
            self.0.serialize(serializer)
        }
    }
}

#[cfg(feature = "serde")]
impl<'de> serde::Deserialize<'de> for WeekdaySet {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        if deserializer.is_human_readable() {
            use core::marker::PhantomData;

            struct Visitor<'de>(PhantomData<&'de ()>);
            impl<'de> serde::de::Visitor<'de> for Visitor<'de> {
                type Value = WeekdaySet;
                fn expecting(&self, f: &mut core::fmt::Formatter) -> core::fmt::Result {
                    core::write!(f, "a sequence of Weekdays")
                }
                fn visit_seq<A: serde::de::SeqAccess<'de>>(
                    self,
                    mut seq: A,
                ) -> Result<Self::Value, A::Error> {
                    let mut set = WeekdaySet::new(&[]);
                    while let Some(day) = seq.next_element::<Weekday>()? {
                        set.0 |= day.bit_value();
                    }
                    Ok(set)
                }
            }
            deserializer.deserialize_seq(Visitor(PhantomData))
        } else {
            u8::deserialize(deserializer).map(Self)
        }
    }
}

#[test]
#[cfg(feature = "datagen")]
fn test_weekdayset_bake() {
    databake::test_bake!(
        WeekdaySet,
        const,
        crate::provider::WeekdaySet::new(&[
            crate::types::Weekday::Monday,
            crate::types::Weekday::Wednesday,
            crate::types::Weekday::Friday
        ]),
        icu_calendar
    );
}

#[test]
fn test_weekdayset_new() {
    use Weekday::*;

    let sat_sun_bitmap = Saturday.bit_value() | Sunday.bit_value();
    let sat_sun_weekend = WeekdaySet::new(&[Saturday, Sunday]);
    assert_eq!(sat_sun_bitmap, sat_sun_weekend.0);

    let fri_sat_bitmap = Friday.bit_value() | Saturday.bit_value();
    let fri_sat_weekend = WeekdaySet::new(&[Friday, Saturday]);
    assert_eq!(fri_sat_bitmap, fri_sat_weekend.0);

    let fri_sun_bitmap = Friday.bit_value() | Sunday.bit_value();
    let fri_sun_weekend = WeekdaySet::new(&[Friday, Sunday]);
    assert_eq!(fri_sun_bitmap, fri_sun_weekend.0);

    let fri_bitmap = Friday.bit_value();
    let fri_weekend = WeekdaySet::new(&[Friday, Friday]);
    assert_eq!(fri_bitmap, fri_weekend.0);

    let sun_mon_bitmap = Sunday.bit_value() | Monday.bit_value();
    let sun_mon_weekend = WeekdaySet::new(&[Sunday, Monday]);
    assert_eq!(sun_mon_bitmap, sun_mon_weekend.0);

    let mon_sun_bitmap = Monday.bit_value() | Sunday.bit_value();
    let mon_sun_weekend = WeekdaySet::new(&[Monday, Sunday]);
    assert_eq!(mon_sun_bitmap, mon_sun_weekend.0);

    let mon_bitmap = Monday.bit_value();
    let mon_weekend = WeekdaySet::new(&[Monday]);
    assert_eq!(mon_bitmap, mon_weekend.0);
}