1 files changed, 247 insertions, 0 deletions

diff --git a/src/app/impl_self.rs b/src/app/impl_self.rs
new file mode 100644
index 0000000..efed4e0
--- /dev/null
+++ b/src/app/impl_self.rs
@@ -0,0 +1,247 @@
+use std::default::Default;
+use std::f64;
+use std::fs::{File, OpenOptions};
+use std::io::prelude::*;
+use std::path::PathBuf;
+use std::sync::mpsc::channel;
+use std::time::Duration;
+
+use chrono::Local;
+use cursive::direction::Absolute;
+use cursive::Vec2;
+use notify::{watcher, RecursiveMode, Watcher};
+
+use crate::habit::{Bit, Count, HabitWrapper, TrackEvent, ViewMode};
+use crate::utils;
+use crate::Command;
+use crate::CONFIGURATION;
+
+use crate::app::{App, StatusLine};
+
+impl App {
+    pub fn new() -> Self {
+        let (tx, rx) = channel();
+        let mut watcher = watcher(tx, Duration::from_secs(1)).unwrap();
+        watcher
+            .watch(utils::auto_habit_file(), RecursiveMode::Recursive)
+            .unwrap_or_else(|e| {
+                panic!("Unable to start file watcher: {}", e);
+            });
+        return App {
+            habits: vec![],
+            focus: 0,
+            _file_watcher: watcher,
+            file_event_recv: rx,
+            view_month_offset: 0,
+        };
+    }
+
+    pub fn add_habit(&mut self, h: Box<dyn HabitWrapper>) {
+        self.habits.push(h);
+    }
+
+    pub fn delete_by_name(&mut self, name: &str) {
+        self.habits.retain(|h| h.name() != name);
+    }
+
+    pub fn get_mode(&self) -> ViewMode {
+        if self.habits.is_empty() {
+            return ViewMode::Day;
+        }
+        return self.habits[self.focus].view_mode();
+    }
+
+    pub fn set_mode(&mut self, mode: ViewMode) {
+        if !self.habits.is_empty() {
+            self.habits[self.focus].set_view_mode(mode);
+        }
+    }
+
+    pub fn set_view_month_offset(&mut self, offset: u32) {
+        self.view_month_offset = offset;
+        for v in self.habits.iter_mut() {
+            v.set_view_month_offset(offset);
+        }
+    }
+
+    pub fn sift_backward(&mut self) {
+        self.view_month_offset += 1;
+        for v in self.habits.iter_mut() {
+            v.set_view_month_offset(self.view_month_offset);
+        }
+    }
+
+    pub fn sift_forward(&mut self) {
+        if self.view_month_offset > 0 {
+            self.view_month_offset -= 1;
+            for v in self.habits.iter_mut() {
+                v.set_view_month_offset(self.view_month_offset);
+            }
+        }
+    }
+
+    pub fn set_focus(&mut self, d: Absolute) {
+        let grid_width = CONFIGURATION.grid_width;
+        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 + grid_width < self.habits.len() - 1 {
+                    self.focus += grid_width;
+                } else {
+                    self.focus = self.habits.len() - 1;
+                }
+            }
+            Absolute::Up => {
+                if self.focus as isize - grid_width as isize >= 0 {
+                    self.focus -= grid_width;
+                } else {
+                    self.focus = 0;
+                }
+            }
+            Absolute::None => {}
+        }
+    }
+
+    pub 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.goal()).sum::<u32>();
+        let completed = total - remaining;
+
+        let timestamp = if self.view_month_offset == 0 {
+            format!("{}", Local::now().date().format("%d/%b/%y"),)
+        } else {
+            let months = self.view_month_offset;
+            format!("{}", format!("{} months ago", months),)
+        };
+
+        StatusLine {
+            0: format!(
+                "Today: {} completed, {} remaining --{}--",
+                completed,
+                remaining,
+                self.get_mode()
+            ),
+            1: timestamp,
+        }
+    }
+
+    pub fn max_size(&self) -> Vec2 {
+        let grid_width = CONFIGURATION.grid_width;
+        let width = {
+            if self.habits.len() > 0 {
+                grid_width * CONFIGURATION.view_width
+            } else {
+                0
+            }
+        };
+        let height = {
+            if self.habits.len() > 0 {
+                (CONFIGURATION.view_height as f64
+                    * (self.habits.len() as f64 / grid_width as f64).ceil())
+                    as usize
+            } else {
+                0
+            }
+        };
+        Vec2::new(width, height + 2)
+    }
+
+    pub fn load_state() -> Self {
+        let (regular_f, auto_f) = (utils::habit_file(), utils::auto_habit_file());
+        let read_from_file = |file: PathBuf| -> Vec<Box<dyn HabitWrapper>> {
+            if let Ok(ref mut f) = File::open(file) {
+                let mut j = String::new();
+                f.read_to_string(&mut j);
+                return serde_json::from_str(&j).unwrap();
+            } else {
+                return Vec::new();
+            }
+        };
+
+        let mut regular = read_from_file(regular_f);
+        let auto = read_from_file(auto_f);
+        regular.extend(auto);
+        return App {
+            habits: regular,
+            ..Default::default()
+        };
+    }
+
+    // this function does IO
+    // TODO: convert this into non-blocking async function
+    pub fn save_state(&self) {
+        let (regular, auto): (Vec<_>, Vec<_>) = self.habits.iter().partition(|&x| !x.is_auto());
+        let (regular_f, auto_f) = (utils::habit_file(), utils::auto_habit_file());
+
+        let write_to_file = |data: Vec<&Box<dyn HabitWrapper>>, file: PathBuf| {
+            let j = serde_json::to_string_pretty(&data).unwrap();
+            match OpenOptions::new()
+                .write(true)
+                .create(true)
+                .truncate(true)
+                .open(file)
+            {
+                Ok(ref mut f) => f.write_all(j.as_bytes()).unwrap(),
+                Err(_) => panic!("Unable to write!"),
+            };
+        };
+
+        write_to_file(regular, regular_f);
+        write_to_file(auto, auto_f);
+    }
+
+    pub fn parse_command(&mut self, c: Command) {
+        match c {
+            Command::Add(name, goal, auto) => {
+                let kind = if goal == Some(1) { "bit" } else { "count" };
+                if kind == "count" {
+                    self.add_habit(Box::new(Count::new(
+                        name,
+                        goal.unwrap_or(0),
+                        auto.unwrap_or(false),
+                    )));
+                } else if kind == "bit" {
+                    self.add_habit(Box::new(Bit::new(name, auto.unwrap_or(false))));
+                }
+            }
+            Command::Delete(name) => {
+                self.delete_by_name(&name);
+                self.focus = 0;
+            }
+            Command::TrackUp(name) => {
+                let target_habit = self
+                    .habits
+                    .iter_mut()
+                    .find(|x| x.name() == name && x.is_auto());
+                if let Some(h) = target_habit {
+                    h.modify(Local::now().naive_utc().date(), TrackEvent::Increment);
+                }
+            }
+            Command::TrackDown(name) => {
+                let target_habit = self
+                    .habits
+                    .iter_mut()
+                    .find(|x| x.name() == name && x.is_auto());
+                if let Some(h) = target_habit {
+                    h.modify(Local::now().naive_utc().date(), TrackEvent::Decrement);
+                }
+            }
+            Command::Quit => self.save_state(),
+            Command::MonthNext => self.sift_forward(),
+            Command::MonthPrev => self.sift_backward(),
+            _ => {
+                eprintln!("UNKNOWN COMMAND!");
+            }
+        }
+    }
+}