1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
//! A selection of pre-defined holidays provided for convenience

pub mod global;
pub mod united_states;

use crate::Holiday;
use std::str::FromStr;

#[macro_export]
/// Macro to create Holiday
macro_rules! holiday {
    ($name:expr, $month:expr, $day:expr) => {
        Holiday {
            name: $name,
            date: HolidayDate::FixedDate(DayOfMonth {
                day: $day,
                month: $month,
            }),
        }
    };

    ($name:expr, $nth:expr, $weekday:expr, $month:expr) => {
        Holiday {
            name: $name,
            date: HolidayDate::NthDate(NthWeekdayOfMonth {
                nth: $nth,
                weekday: $weekday,
                month: $month,
            }),
        }
    };
}

#[macro_export]
/// Macro to create a `pub const Holiday`
macro_rules! holiday_const {
    ($(#[$attr:meta])* $var:ident, $name:expr, $month:expr, $day:expr) => {
        $(#[$attr])*
        pub const $var: Holiday<&str> = holiday!($name, $month, $day);
    };

    ($(#[$attr:meta])* $var:ident, $name:expr, $nth:expr, $weekday:expr, $month:expr) => {
        $(#[$attr])*
        pub const $var: Holiday<&str> = holiday!($name, $nth, $weekday, $month);
    };
}

impl FromStr for Holiday<&str> {
    type Err = std::io::Error;
    fn from_str(s: &str) -> Result<Self, Self::Err> {
        use united_states::*;
        use global::*;
        Ok(match s
            .to_lowercase()
            .replace("\'", "")
            .trim_start_matches("the")
            .trim_end_matches(" day")
            .trim()
        {
            "martin luther king jr." => MLKJ_DAY,
            "groundhog" => GROUNDHOG_DAY,
            "superbowl sunday" | "superbowl" => SUPERBOWL_SUNDAY,
            "presidents" => PRESIDENTS_DAY,
            "valentines" => VALENTINES_DAY,
            "daylight saving time starts" => DST_START,
            "april fools" => APRIL_FOOLS_DAY,
            "kentucky derby" => KENTUCKY_DERBY,
            "memorial" => MEMORIAL_DAY,
            "mothers" => MOTHERS_DAY,
            "flag" => FLAG_DAY,
            "independence" | "july 4th" | "july fourth" | "fourth of july" => INDEPENDENCE_DAY,
            "fathers" => FATHERS_DAY,
            "labor" => LABOR_DAY,
            "halloween" => HALLOWEEN,
            "columbus" => COLUMBUS_DAY,
            "veterans" => VETERANS_DAY,
            "daylight saving time ends" => DST_END,
            "thanksgiving" => THANKSGIVING,
            "christmas eve" => CHRISTMAS_EVE,
            "christmas" => CHRISTMAS,
            _ => return Err(std::io::Error::from(std::io::ErrorKind::InvalidInput))
        })
    }
}