handle cursor events and entry

1 files changed, 12 insertions, 0 deletions

diff --git a/src/habit/count.rs b/src/habit/count.rs
index d351758..b14354c 100644
--- a/src/habit/count.rs
+++ b/src/habit/count.rs
@@ -1,8 +1,10 @@
 use std::collections::HashMap;
 
 use chrono::NaiveDate;
+use cursive::direction::Absolute;
 use serde::{Deserialize, Serialize};
 
+use crate::app::Cursor;
 use crate::habit::prelude::default_auto;
 use crate::habit::traits::Habit;
 use crate::habit::{TrackEvent, ViewMode};
@@ -20,6 +22,9 @@ pub struct Count {
     view_month_offset: u32,
 
     #[serde(skip)]
+    cursor: Cursor,
+
+    #[serde(skip)]
     view_mode: ViewMode,
 }
 
@@ -31,6 +36,7 @@ impl Count {
             goal,
             auto,
             view_month_offset: 0,
+            cursor: Cursor::new(),
             view_mode: ViewMode::Day,
         };
     }
@@ -101,6 +107,12 @@ impl Habit for Count {
     fn view_month_offset(&self) -> u32 {
         self.view_month_offset
     }
+    fn move_cursor(&mut self, d: Absolute) {
+        self.cursor.do_move(d);
+    }
+    fn cursor(&self) -> Cursor {
+        self.cursor
+    }
     fn set_view_mode(&mut self, mode: ViewMode) {
         self.view_mode = mode;
     }