bdays 0.1.4

Business Days calendars for Rustaceans.
Documentation 
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

use crate::HolidayCalendar;
use ::chrono::{Datelike, NaiveDate};

/// Holiday Calendars for Brazil.
pub mod brazil;

/// Holiday Calendars for the United States.
pub mod us;

/// Holiday Calendars for Germany.
pub mod de;

/// The `WeekendsOnly` holiday calendar always returns `false` for method `is_holiday`.
/// So `is_bday` method returns `false` only for weekend dates.
pub struct WeekendsOnly;

impl<T: Datelike + Copy + PartialOrd> HolidayCalendar<T> for WeekendsOnly {
    fn is_holiday(&self, _date: T) -> bool {
        false
    }

    fn bdays(&self, d0: T, d1: T) -> i32 {
        if d0 == d1 {
            0
        } else {
            let from: NaiveDate;
            let to: NaiveDate;
            let swapped: bool;

            if d1 < d0 {
                from = NaiveDate::from_num_days_from_ce_opt(d1.num_days_from_ce()).unwrap();
                to = NaiveDate::from_num_days_from_ce_opt(d0.num_days_from_ce()).unwrap();
                swapped = true;
            } else {
                from = NaiveDate::from_num_days_from_ce_opt(d0.num_days_from_ce()).unwrap();
                to = NaiveDate::from_num_days_from_ce_opt(d1.num_days_from_ce()).unwrap();
                swapped = false;
            }

            let mut result: i32 = 0;
            let days = to.num_days_from_ce() - from.num_days_from_ce();
            let whole_weeks = days / 7;
            result += whole_weeks * 5;

            let mut current_date =
                NaiveDate::from_num_days_from_ce_opt(from.num_days_from_ce() + whole_weeks * 7)
                    .unwrap();

            if current_date < to {
                let mut day_of_week = current_date.weekday();

                while current_date < to {
                    if day_of_week.number_from_monday() < 6 {
                        result += 1;
                    }

                    current_date =
                        NaiveDate::from_num_days_from_ce_opt(current_date.num_days_from_ce() + 1)
                            .unwrap();
                    day_of_week = day_of_week.succ();
                }
            }

            if swapped {
                result = -result;
            }

            result
        }
    }
}