Add Matcher trait for indicators

1 files changed, 37 insertions, 0 deletions

diff --git a/src/matcher.rs b/src/matcher.rs
new file mode 100644
index 0000000..1482b68
--- /dev/null
+++ b/src/matcher.rs
@@ -0,0 +1,37 @@
+
+use chrono::Datelike;
+
+use error::KairosError as KE;
+use error::KairosErrorKind as KEK;
+use error::Result;
+use indicator::Day;
+use indicator::Month;
+use 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(KE::from_kind(KEK::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(KE::from_kind(KEK::ArgumentErrorNotAMoment(tt.name())))
+    }
+
+}
+
+