Merge branch 'master' of https://github.com/yoms/dijo into fix/duplicate-habits

1 files changed, 18 insertions, 1 deletions

diff --git a/src/app/impl_self.rs b/src/app/impl_self.rs
index 1ed19e6..1dfe268 100644
--- a/src/app/impl_self.rs
+++ b/src/app/impl_self.rs
@@ -33,8 +33,25 @@ impl App {
         };
     }
 
+    pub fn list_habit(&self) -> Vec<String> {
+        let habits_names = self.habits.iter().map(|x| x.name()).collect::<Vec<_>>();
+        return habits_names;
+    }
+
     pub fn add_habit(&mut self, h: Box<dyn HabitWrapper>) {
-        self.habits.push(h);
+        if self
+            .habits
+            .iter()
+            .filter(|hab| hab.name() == h.name())
+            .count()
+            > 0
+        {
+            self.message.set_kind(MessageKind::Error);
+            self.message
+                .set_message(format!("Habit `{}` allready exist", h.name()))
+        } else {
+            self.habits.push(h);
+        }
     }
 
     pub fn list_habits(&self) -> Vec<String> {