implement goals

1 files changed, 14 insertions, 1 deletions

diff --git a/src/habit.rs b/src/habit.rs
index 4f05217..88e1cb6 100644
--- a/src/habit.rs
+++ b/src/habit.rs
@@ -6,16 +6,18 @@ use chrono::NaiveDate;
 pub struct Habit<T> {
     name: String,
     stats: HashMap<NaiveDate, T>,
+    goal: T,
 }
 
 impl<T> Habit<T>
 where
     T: Copy,
 {
-    pub fn new(name: &str) -> Habit<T> {
+    pub fn new(name: &str, goal: T) -> Habit<T> {
         return Habit {
             name: name.to_owned(),
             stats: HashMap::new(),
+            goal,
         };
     }
 
@@ -40,6 +42,9 @@ impl Habit<bool> {
             self.insert_entry(date, true);
         }
     }
+    pub fn reached_goal(&self, date: NaiveDate) -> bool {
+        *self.get_by_date(date).unwrap_or(&false)
+    }
 }
 
 impl Habit<u32> {
@@ -60,4 +65,12 @@ impl Habit<u32> {
     pub fn set(&mut self, date: NaiveDate, val: u32) {
         *self.stats.entry(date).or_insert(val) = val;
     }
+    pub fn reached_goal(&self, date: NaiveDate) -> bool {
+        if let Some(v) = self.get_by_date(date) {
+            if *v >= self.goal {
+                return true;
+            }
+        }
+        return false;
+    }
 }