add GoalKinds

 - this allows for changing habit goals easily
 - easier to add new habits to dijo in newer versions

1 files changed, 4 insertions, 0 deletions

diff --git a/src/habit/count.rs b/src/habit/count.rs
index 09fd399..75b51cc 100644
--- a/src/habit/count.rs
+++ b/src/habit/count.rs
@@ -4,6 +4,7 @@ use std::default::Default;
 use chrono::NaiveDate;
 use serde::{Deserialize, Serialize};
 
+use crate::command::GoalKind;
 use crate::habit::prelude::default_auto;
 use crate::habit::traits::Habit;
 use crate::habit::{InnerData, TrackEvent};
@@ -42,6 +43,9 @@ impl Habit for Count {
     fn set_name(&mut self, n: impl AsRef<str>) {
         self.name = n.as_ref().to_owned();
     }
+    fn kind(&self) -> GoalKind {
+        GoalKind::Count(self.goal)
+    }
     fn set_goal(&mut self, g: Self::HabitType) {
         self.goal = g;
     }