1 files changed, 169 insertions, 0 deletions

diff --git a/src/app.rs b/src/app.rs
new file mode 100644
index 0000000..c43a43e
--- /dev/null
+++ b/src/app.rs
@@ -0,0 +1,169 @@
+use std::f64;
+
+use cursive::direction::{Absolute, Direction};
+use cursive::event::{Event, EventResult, Key};
+use cursive::view::View;
+use cursive::{Printer, Vec2};
+
+use crate::views::habitview::HabitView;
+
+#[derive(PartialEq)]
+pub enum ViewMode {
+    Day,
+    Month,
+    Year,
+}
+
+pub struct App {
+    habits: Vec<HabitView>,
+    view_mode: ViewMode,
+    focus: usize,
+    grid_width: usize,
+
+    padding: usize,
+}
+
+impl App {
+    pub fn new() -> Self {
+        return App {
+            habits: vec![],
+            view_mode: ViewMode::Day,
+            focus: 0,
+            grid_width: 3,
+            padding: 12,
+        };
+    }
+
+    pub fn add_habit(mut self, h: HabitView) -> Self {
+        self.habits.push(h);
+        return self;
+    }
+
+    pub fn set_mode(mut self, set_mode: ViewMode) -> Self {
+        if set_mode != self.view_mode {
+            self.view_mode = set_mode
+        }
+        return self;
+    }
+
+    fn set_focus(&mut self, d: Absolute) {
+        match d {
+            Absolute::Right => {
+                if self.focus != self.habits.len() - 1 {
+                    self.focus += 1;
+                }
+            }
+            Absolute::Left => {
+                if self.focus != 0 {
+                    self.focus -= 1;
+                }
+            }
+            Absolute::Down => {
+                if self.focus + self.grid_width < self.habits.len() - 1 {
+                    self.focus += self.grid_width;
+                } else {
+                    self.focus = self.habits.len() - 1;
+                }
+            }
+            Absolute::Up => {
+                if self.focus as isize - self.grid_width as isize >= 0 {
+                    self.focus -= self.grid_width;
+                } else {
+                    self.focus = 0;
+                }
+            }
+            Absolute::None => {}
+        }
+    }
+
+    fn status(&self) -> String {
+        let remaining = self.habits.iter().map(|h| h.remaining()).sum::<u32>();
+        let total = self.habits.iter().map(|h| h.total()).sum::<u32>();
+        let completed = total - remaining;
+
+        return format!("{}/{} ", completed, total);
+    }
+    fn max_size(&self) -> Vec2 {
+        let grid_width = self.grid_width;
+        let width = {
+            if self.habits.len() > 0 {
+                grid_width * self.habits[0].get_size().x
+            } else {
+                0
+            }
+        };
+        let height = {
+            if self.habits.len() > 0 {
+                (self.habits[0].get_size().y as f64
+                    * (self.habits.len() as f64 / grid_width as f64).ceil())
+                    as usize
+            } else {
+                0
+            }
+        };
+        Vec2::new(width, height)
+    }
+}
+
+impl View for App {
+    fn draw(&self, printer: &Printer) {
+        let grid_width = self.grid_width;
+        let mut offset = Vec2::zero();
+        for (idx, i) in self.habits.iter().enumerate() {
+            if idx >= grid_width && idx % grid_width == 0 {
+                offset = offset.map_y(|y| y + i.get_size().y).map_x(|_| 0);
+            }
+            i.draw(&printer.offset(offset).focused(self.focus == idx));
+            offset = offset.map_x(|x| x + i.get_size().x);
+        }
+        offset = offset.map_x(|_| 0).map_y(|_| self.max_size().y - 2);
+        printer.print(offset, &self.status());
+    }
+
+    fn required_size(&mut self, _: Vec2) -> Vec2 {
+        let grid_width = self.grid_width;
+        let width = {
+            if self.habits.len() > 0 {
+                grid_width * self.habits[0].get_size().x
+            } else {
+                0
+            }
+        };
+        let height = {
+            if self.habits.len() > 0 {
+                (self.habits[0].get_size().y as f64
+                    * (self.habits.len() as f64 / grid_width as f64).ceil())
+                    as usize
+            } else {
+                0
+            }
+        };
+        Vec2::new(width, height)
+    }
+
+    fn take_focus(&mut self, _: Direction) -> bool {
+        false
+    }
+
+    fn on_event(&mut self, e: Event) -> EventResult {
+        match e {
+            Event::Key(Key::Right) | Event::Key(Key::Tab) | Event::Char('l') => {
+                self.set_focus(Absolute::Right);
+                return EventResult::Consumed(None);
+            }
+            Event::Key(Key::Left) | Event::Shift(Key::Tab) | Event::Char('h') => {
+                self.set_focus(Absolute::Left);
+                return EventResult::Consumed(None);
+            }
+            Event::Key(Key::Up) | Event::Char('k') => {
+                self.set_focus(Absolute::Up);
+                return EventResult::Consumed(None);
+            }
+            Event::Key(Key::Down) | Event::Char('j') => {
+                self.set_focus(Absolute::Down);
+                return EventResult::Consumed(None);
+            }
+            _ => self.habits[self.focus].on_event(e),
+        }
+    }
+}