summaryrefslogtreecommitdiffstats
path: root/src/matcher.rs
blob: 965a08c1a88ef2de55ee3d7fddacce35694350b6 (plain)
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
use chrono::Datelike;
#[cfg(feature = "with-filters")]
use filters::filter::*;

use crate::error::Error;
use crate::error::Result;
use crate::indicator::Day;
use crate::indicator::Month;
use crate::timetype::TimeType;

/// A trait to extend indicator::* to be able to match them with a TimeType object
pub trait Matcher {
    fn matches(&self, tt: &TimeType) -> Result<bool>;
}

impl Matcher for Day {
    fn matches(&self, tt: &TimeType) -> Result<bool> {
        let this: chrono::Weekday = self.clone().into();
        tt.get_moment()
            .map(|mom| this == mom.weekday())
            .ok_or(Error::ArgumentErrorNotAMoment(tt.name()))
    }
}

impl Matcher for Month {
    fn matches(&self, tt: &TimeType) -> Result<bool> {
        let this: u32 = self.clone().into();
        tt.get_moment()
            .map(|mom| this == mom.month())
            .ok_or(Error::ArgumentErrorNotAMoment(tt.name()))
    }
}

#[cfg(feature = "with-filters")]
impl<F> Matcher for F
where
    F: Filter<TimeType>,
{
    fn matches(&self, tt: &TimeType) -> Result<bool> {
        Ok(self.filter(tt))
    }
}