1 files changed, 56 insertions, 8 deletions
diff --git a/src/main.rs b/src/main.rs
index 9d96dfe..b34340e 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -15,11 +15,18 @@ use crossterm::{
 
 use roxmltree::Document;
 
+enum Mode {
+    Help,
+    Nav,
+    Read,
+}
+
 struct Epub {
     container: zip::ZipArchive<File>,
 }
 
 struct Bk {
+    mode: Mode,
     epub: Epub,
     cols: u16,
     chapter: Vec<String>,
@@ -130,6 +137,7 @@ impl Bk {
         let (cols, rows) = terminal::size().unwrap();
         let mut epub = Epub::new(path)?;
         let mut bk = Bk {
+            mode: Mode::Read,
             chapter: Vec::new(),
             chapter_idx,
             toc: epub.get_toc(),
@@ -195,10 +203,25 @@ impl Bk {
             self.pos = (self.chapter.len() / self.rows) * self.rows;
         }
     }
-    fn run(&mut self, event: Event) -> bool {
-        match event {
+    fn run(&mut self, e: Event) -> bool {
+        match self.mode {
+            Mode::Read => return self.run_read(e),
+            Mode::Nav => self.run_nav(e),
+            Mode::Help => self.mode = Mode::Read,
+        }
+        true
+    }
+    fn run_nav(&mut self, e: Event) {
+        match e {
+            _ => self.mode = Mode::Read
+        }
+    }
+    fn run_read(&mut self, e: Event) -> bool {
+        match e {
             Event::Key(e) => match e.code {
-                KeyCode::Char('q') => return true,
+                KeyCode::Char('q') => return false,
+                KeyCode::Tab => self.mode = Mode::Nav,
+                KeyCode::Char('?') => self.mode = Mode::Help,
                 KeyCode::Char('p') => {
                     if self.chapter_idx > 0 {
                         self.get_chapter(self.chapter_idx - 1);
@@ -239,9 +262,36 @@ impl Bk {
                 self.get_chapter(self.chapter_idx);
             }
         }
-        false
+        true
     }
     fn render(&self) {
+        match self.mode {
+            Mode::Read => self.render_read(),
+            Mode::Help => self.render_help(),
+            Mode::Nav => self.render_nav(),
+        }
+    }
+    fn render_nav(&self) {
+        let mut stdout = stdout();
+        queue!(
+            stdout,
+            terminal::Clear(terminal::ClearType::All),
+            cursor::MoveTo(0, 0),
+            Print("TODO: nav")
+            ).unwrap();
+        stdout.flush().unwrap();
+    }
+    fn render_help(&self) {
+        let mut stdout = stdout();
+        queue!(
+            stdout,
+            terminal::Clear(terminal::ClearType::All),
+            cursor::MoveTo(0, 0),
+            Print("TODO: help text")
+            ).unwrap();
+        stdout.flush().unwrap();
+    }
+    fn render_read(&self) {
         let mut stdout = stdout();
         queue!(
             stdout,
@@ -314,11 +364,9 @@ fn main() -> crossterm::Result<()> {
     )?;
     terminal::enable_raw_mode()?;
 
-    loop {
+    bk.render();
+    while bk.run(read()?) {
         bk.render();
-        if bk.run(read()?) {
-            break;
-        }
     }
 
     std::fs::write(