1 files changed, 102 insertions, 0 deletions

diff --git a/src/habit/count.rs b/src/habit/count.rs
new file mode 100644
index 0000000..a0e0aee
--- /dev/null
+++ b/src/habit/count.rs
@@ -0,0 +1,102 @@
+use std::collections::HashMap;
+
+use chrono::NaiveDate;
+use serde::{Deserialize, Serialize};
+
+use crate::habit::traits::Habit;
+use crate::habit::{TrackEvent, ViewMode};
+
+#[derive(Debug, Serialize, Deserialize)]
+pub struct Count {
+    name: String,
+    stats: HashMap<NaiveDate, u32>,
+    goal: u32,
+
+    #[serde(skip)]
+    view_month_offset: u32,
+
+    #[serde(skip)]
+    view_mode: ViewMode,
+}
+
+impl Count {
+    pub fn new(name: impl AsRef<str>, goal: u32) -> Self {
+        return Count {
+            name: name.as_ref().to_owned(),
+            stats: HashMap::new(),
+            goal,
+            view_month_offset: 0,
+            view_mode: ViewMode::Day,
+        };
+    }
+}
+
+impl Habit for Count {
+    type HabitType = u32;
+
+    fn name(&self) -> String {
+        return self.name.clone();
+    }
+    fn set_name(&mut self, n: impl AsRef<str>) {
+        self.name = n.as_ref().to_owned();
+    }
+    fn set_goal(&mut self, g: Self::HabitType) {
+        self.goal = g;
+    }
+    fn get_by_date(&self, date: NaiveDate) -> Option<&Self::HabitType> {
+        self.stats.get(&date)
+    }
+    fn insert_entry(&mut self, date: NaiveDate, val: Self::HabitType) {
+        *self.stats.entry(date).or_insert(val) = val;
+    }
+    fn reached_goal(&self, date: NaiveDate) -> bool {
+        if let Some(val) = self.stats.get(&date) {
+            if val >= &self.goal {
+                return true;
+            }
+        }
+        return false;
+    }
+    fn remaining(&self, date: NaiveDate) -> u32 {
+        if self.reached_goal(date) {
+            return 0;
+        } else {
+            if let Some(val) = self.stats.get(&date) {
+                return self.goal - val;
+            } else {
+                return self.goal;
+            }
+        }
+    }
+    fn goal(&self) -> u32 {
+        return self.goal;
+    }
+    fn modify(&mut self, date: NaiveDate, event: TrackEvent) {
+        if let Some(val) = self.stats.get_mut(&date) {
+            match event {
+                TrackEvent::Increment => *val += 1,
+                TrackEvent::Decrement => {
+                    if *val > 0 {
+                        *val -= 1
+                    } else {
+                        *val = 0
+                    };
+                }
+            }
+        } else {
+            self.insert_entry(date, 1);
+        }
+    }
+    fn set_view_month_offset(&mut self, offset: u32) {
+        self.view_month_offset = offset;
+    }
+    fn view_month_offset(&self) -> u32 {
+        self.view_month_offset
+    }
+    fn set_view_mode(&mut self, mode: ViewMode) {
+        self.view_mode = mode;
+    }
+    fn view_mode(&self) -> ViewMode {
+        self.view_mode
+    }
+}