1 files changed, 32 insertions, 26 deletions

diff --git a/src/app.rs b/src/app.rs
index 82096e1..f8797fc 100644
--- a/src/app.rs
+++ b/src/app.rs
@@ -9,26 +9,13 @@ use cursive::{Printer, Vec2};
 
 use chrono::{Local, NaiveDate};
 
-use crate::habit::{Bit, Count, Habit, HabitWrapper};
+use crate::habit::{Bit, Count, Habit, HabitWrapper, ViewMode};
 use crate::utils;
 use crate::Command;
 use crate::CONFIGURATION;
 
 use serde::{Deserialize, Serialize};
 
-#[derive(PartialEq, Serialize, Deserialize)]
-pub enum ViewMode {
-    Day,
-    Month,
-    Year,
-}
-
-impl std::default::Default for ViewMode {
-    fn default() -> Self {
-        ViewMode::Month
-    }
-}
-
 struct StatusLine(String, String);
 
 #[derive(Serialize, Deserialize)]
@@ -41,9 +28,6 @@ pub struct App {
     focus: usize,
 
     #[serde(skip)]
-    view_mode: ViewMode,
-
-    #[serde(skip)]
     view_month_offset: u32,
 }
 
@@ -51,7 +35,6 @@ impl App {
     pub fn new() -> Self {
         return App {
             habits: vec![],
-            view_mode: ViewMode::Day,
             focus: 0,
             view_month_offset: 0,
         };
@@ -65,9 +48,9 @@ impl App {
         self.habits.retain(|h| h.get_name() != name);
     }
 
-    pub fn set_mode(&mut self, set_mode: ViewMode) {
-        if set_mode != self.view_mode {
-            self.view_mode = set_mode;
+    pub fn set_mode(&mut self, mode: ViewMode) {
+        if !self.habits.is_empty() {
+            self.habits[self.focus].set_view_mode(mode);
         }
     }
 
@@ -128,7 +111,7 @@ impl App {
     fn status(&self) -> StatusLine {
         let today = chrono::Local::now().naive_utc().date();
         let remaining = self.habits.iter().map(|h| h.remaining(today)).sum::<u32>();
-        let total = self.habits.iter().map(|h| h.total()).sum::<u32>();
+        let total = self.habits.iter().map(|h| h.goal()).sum::<u32>();
         let completed = total - remaining;
 
         let timestamp = if self.view_month_offset == 0 {
@@ -141,7 +124,7 @@ impl App {
             let months = self.view_month_offset;
             format!(
                 "{:>width$}",
-                format!("{} months ago", self.view_month_offset),
+                format!("{} months ago", months),
                 width = CONFIGURATION.view_width * CONFIGURATION.grid_width
             )
         };
@@ -233,8 +216,8 @@ impl View for App {
         }
 
         offset = offset.map_x(|_| 0).map_y(|_| self.max_size().y - 2);
-        printer.print(offset, &self.status().1); // right
-        printer.print(offset, &self.status().0); // left
+        printer.print(offset, &self.status().1); // right status
+        printer.print(offset, &self.status().0); // left status
     }
 
     fn required_size(&mut self, _: Vec2) -> Vec2 {
@@ -243,7 +226,7 @@ impl View for App {
         let view_height = CONFIGURATION.view_height;
         let width = {
             if self.habits.len() > 0 {
-                grid_width * view_width
+                grid_width * (view_width + 2)
             } else {
                 0
             }
@@ -300,6 +283,29 @@ impl View for App {
                 self.save_state();
                 return EventResult::with_cb(|s| s.quit());
             }
+            Event::Char('v') => {
+                if self.habits.is_empty() {
+                    return EventResult::Consumed(None);
+                }
+                if self.habits[self.focus].view_mode() == ViewMode::Week {
+                    self.set_mode(ViewMode::Day)
+                } else {
+                    self.set_mode(ViewMode::Week)
+                }
+                return EventResult::Consumed(None);
+            }
+            Event::Char('V') => {
+                for habit in self.habits.iter_mut() {
+                    habit.set_view_mode(ViewMode::Week);
+                }
+                return EventResult::Consumed(None);
+            }
+            Event::Key(Key::Esc) => {
+                for habit in self.habits.iter_mut() {
+                    habit.set_view_mode(ViewMode::Day);
+                }
+                return EventResult::Consumed(None);
+            }
 
             /* We want sifting to be an app level function,
              * that later trickles down into each habit