tracc/src/tracc.rs

use super::layout;
use super::listview::ListView;
use super::timesheet::TimeSheet;
use super::todolist::TodoList;
use std::default::Default;
use std::io::{self, Write};
use termion::event::Key;
use termion::input::TermRead;
use tui::backend::TermionBackend;
use tui::widgets::*;

type Terminal = tui::Terminal<TermionBackend<termion::raw::RawTerminal<io::Stdout>>>;
const JSON_PATH_TIME: &str = "tracc_time.json";
const JSON_PATH_TODO: &str = "tracc_todo.json";

pub enum Mode {
    Insert,
    Normal,
}

#[derive(PartialEq)]
enum Focus {
    Top,
    Bottom,
}

pub struct Tracc {
    todos: TodoList,
    times: TimeSheet,
    terminal: Terminal,
    input_mode: Mode,
    focus: Focus,
}

impl Tracc {
    pub fn new(terminal: Terminal) -> Self {
        Self {
            todos: TodoList::open_or_create(JSON_PATH_TODO),
            times: TimeSheet::open_or_create(JSON_PATH_TIME),
            terminal,
            input_mode: Mode::Normal,
            focus: Focus::Bottom,
        }
    }

    pub fn run(&mut self) -> Result<(), io::Error> {
        macro_rules! with_focused {
            ($action: expr $(, $arg: expr)*) => {
                match self.focus {
                    Focus::Top => $action(&mut self.todos, $($arg,)*),
                    Focus::Bottom => $action(&mut self.times, $($arg,)*),
                }
            };
        }

        let mut inputs = io::stdin().keys();
        loop {
            self.refresh()?;
            // I need to find a better way to handle inputs. This is awful.
            let input = inputs.next().unwrap()?;
            match self.input_mode {
                Mode::Normal => match input {
                    Key::Char('q') => break,
                    Key::Char('j') => with_focused!(ListView::selection_down),
                    Key::Char('k') => with_focused!(ListView::selection_up),
                    Key::Char('G') => with_focused!(ListView::selection_first),
                    // gg
                    Key::Char('g') => {
                        if let Some(Ok(Key::Char('g'))) = inputs.next() {
                            with_focused!(ListView::selection_last);
                        }
                    }
                    Key::Char('o') => {
                        with_focused!(ListView::insert, Default::default(), None);
                        self.set_mode(Mode::Insert)?;
                    }
                    Key::Char('a') | Key::Char('A') => self.set_mode(Mode::Insert)?,
                    Key::Char(' ') if self.focus == Focus::Top => self.todos.toggle_current(),
                    // Subtract only 1 minute because the number is truncated to the next multiple
                    // of 5 afterwards, so this is effectively a -5.
                    // See https://git.kageru.moe/kageru/tracc/issues/8
                    Key::Char('-') if self.focus == Focus::Bottom => self.times.shift_current(-1),
                    Key::Char('+') if self.focus == Focus::Bottom => self.times.shift_current(5),
                    // dd
                    Key::Char('d') => {
                        if let Some(Ok(Key::Char('d'))) = inputs.next() {
                            with_focused!(ListView::remove_current);
                        }
                    }
                    // yy
                    Key::Char('y') => {
                        if let Some(Ok(Key::Char('y'))) = inputs.next() {
                            with_focused!(ListView::yank);
                        }
                    }
                    Key::Char('p') => with_focused!(ListView::paste),
                    _ => (),
                },
                Mode::Insert => match input {
                    Key::Char('\n') | Key::Esc => self.set_mode(Mode::Normal)?,
                    Key::Backspace => with_focused!(ListView::backspace),
                    Key::Char(x) => with_focused!(ListView::append_to_current, x),
                    _ => (),
                },
            };
        }
        self.terminal.clear()?;
        persist_state(&self.todos, &self.times);
        Ok(())
    }

    fn set_mode(&mut self, mode: Mode) -> Result<(), io::Error> {
        match mode {
            Mode::Insert => self.terminal.show_cursor()?,
            Mode::Normal => {
                self.todos.normal_mode();
                self.times.normal_mode();
                self.terminal.hide_cursor()?;
                persist_state(&self.todos, &self.times);
            }
        };
        self.input_mode = mode;
        Ok(())
    }

    fn refresh(&mut self) -> Result<(), io::Error> {
        let summary_content = [Text::raw(format!(
            "Sum for today: {}\n{}\n\n{}",
            self.times.sum_as_str(),
            self.times.pause_time(),
            self.times.time_by_tasks()
        ))];
        let mut summary = Paragraph::new(summary_content.iter())
            .wrap(true)
            .block(Block::default().borders(Borders::ALL));
        let todos = self.todos.printable();
        let mut todolist = layout::selectable_list(
            " t r a c c ",
            &todos,
            Some(self.todos.selected).filter(|_| self.focus == Focus::Top),
        );
        let times = self.times.printable();
        let mut timelist = layout::selectable_list(
            " 🕑 ",
            &times,
            Some(self.times.selected).filter(|_| self.focus == Focus::Bottom),
        );

        self.terminal.draw(|mut frame| {
            let chunks = layout::layout(frame.size());
            todolist.render(&mut frame, chunks[0]);
            timelist.render(&mut frame, chunks[1]);
            summary.render(&mut frame, chunks[2]);
        })?;
        Ok(())
    }
}

fn persist_state(todos: &TodoList, times: &TimeSheet) {
    fn write(path: &str, content: String) {
        std::fs::OpenOptions::new()
            .create(true)
            .write(true)
            .truncate(true)
            .open(path)
            .ok()
            .or_else(|| panic!("Can’t save state to JSON. Dumping raw data:\n{}", content))
            .map(|mut f| f.write(content.as_bytes()));
    }
    let todos = serde_json::to_string(&todos.todos).unwrap();
    write(JSON_PATH_TODO, todos);
    let times = serde_json::to_string(&times.times).unwrap();
    write(JSON_PATH_TIME, times);
}