diff options
| -rw-r--r-- | src/main.rs | 104 |
1 files changed, 88 insertions, 16 deletions
diff --git a/src/main.rs b/src/main.rs index e8f4639..025ad83 100644 --- a/src/main.rs +++ b/src/main.rs @@ -4,10 +4,7 @@ use std::io::{stdout, Error, Read, Write}; use crossterm::{ cursor, - event::{ - read, DisableMouseCapture, EnableMouseCapture, Event, KeyCode, - MouseEvent, - }, + event::{self, Event, KeyCode, MouseEvent}, queue, style::{Attribute, Print}, terminal, @@ -31,6 +28,7 @@ struct Bk { cols: u16, chapter: Vec<String>, chapter_idx: usize, + nav_idx: usize, pos: usize, rows: usize, toc: Vec<(String, String)>, @@ -183,6 +181,7 @@ impl Bk { mode: Mode::Read, chapter: Vec::new(), chapter_idx, + nav_idx: 0, toc: epub.get_toc(), epub, pos, @@ -229,13 +228,13 @@ impl Bk { chapter.pop(); //padding self.chapter = wrap(chapter, self.cols - (self.pad * 2)); self.chapter_idx = idx; + self.pos = 0; } fn scroll_down(&mut self, n: usize) { if self.rows < self.chapter.len() - self.pos { self.pos += n; } else if self.chapter_idx < self.toc.len() - 1 { self.get_chapter(self.chapter_idx + 1); - self.pos = 0; } } fn scroll_up(&mut self, n: usize) { @@ -256,25 +255,60 @@ impl Bk { } fn run_nav(&mut self, e: Event) { match e { - _ => self.mode = Mode::Read, + Event::Key(e) => match e.code { + KeyCode::Tab | KeyCode::Char('q') => self.mode = Mode::Read, + KeyCode::Down | KeyCode::Char('j') => { + if self.nav_idx < self.toc.len() - 1 { + self.nav_idx += 1; + } + } + KeyCode::Up | KeyCode::Char('k') => { + if self.nav_idx > 0 { + self.nav_idx -= 1; + } + } + KeyCode::Enter => { + self.get_chapter(self.nav_idx); + self.mode = Mode::Read; + } + KeyCode::Char('g') => self.nav_idx = 0, + KeyCode::Char('G') => self.nav_idx = self.toc.len() - 1, + _ => () + } + Event::Mouse(e) => match e { + MouseEvent::ScrollDown(_, _, _) => { + if self.nav_idx < self.toc.len() - 1 { + self.nav_idx += 1; + } + } + MouseEvent::ScrollUp(_, _, _) => { + if self.nav_idx > 0 { + self.nav_idx -= 1; + } + } + MouseEvent::Down(event::MouseButton::Left, _, row, _) => { + self.get_chapter(row as usize); + self.mode = Mode::Read; + } + _ => (), + }, + _ => () } } fn run_read(&mut self, e: Event) -> bool { match e { Event::Key(e) => match e.code { KeyCode::Esc | KeyCode::Char('q') => return false, - KeyCode::Tab => self.mode = Mode::Nav, + KeyCode::Tab => self.start_nav(), KeyCode::Char('?') => self.mode = Mode::Help, KeyCode::Char('p') => { if self.chapter_idx > 0 { self.get_chapter(self.chapter_idx - 1); - self.pos = 0; } } KeyCode::Char('n') => { if self.chapter_idx < self.toc.len() - 1 { self.get_chapter(self.chapter_idx + 1); - self.pos = 0; } } KeyCode::End | KeyCode::Char('G') => { @@ -328,6 +362,10 @@ impl Bk { Mode::Nav => self.render_nav(), } } + fn start_nav(&mut self) { + self.nav_idx = self.chapter_idx; + self.mode = Mode::Nav; + } fn render_nav(&self) { let mut stdout = stdout(); queue!( @@ -337,19 +375,52 @@ impl Bk { .unwrap(); let end = std::cmp::min(self.rows, self.toc.len()); for i in 0..end { - queue!(stdout, cursor::MoveTo(0, i as u16), Print(&self.toc[i].0)).unwrap(); + if i == self.nav_idx { + queue!(stdout, + cursor::MoveTo(0, i as u16), + Print(format!("{}{}{}", + Attribute::Reverse, + &self.toc[i].0, + Attribute::Reset))).unwrap(); + } else { + queue!(stdout, + cursor::MoveTo(0, i as u16), + Print(&self.toc[i].0)).unwrap(); + } } + stdout.flush().unwrap(); } fn render_help(&self) { let mut stdout = stdout(); + let text = r#" + Esc q Quit + ? Help + Tab Table of Contents +PageDown Right Space f l Page Down + PageUp Left b h Page Up + d Half Page Down + u Half Page Up + Down j Line Down + Up k Line Up + n Next Chapter + p Previous Chapter + Home g Chapter Start + End G Chapter End + "#; queue!( stdout, terminal::Clear(terminal::ClearType::All), - cursor::MoveTo(0, 0), - Print("TODO: help text") ) .unwrap(); + + for (i, line) in text.lines().enumerate() { + queue!( + stdout, + cursor::MoveTo(0, i as u16), + Print(line) + ).unwrap(); + } stdout.flush().unwrap(); } fn render_read(&self) { @@ -411,7 +482,7 @@ fn main() -> crossterm::Result<()> { std::process::exit(1); }); - let mut bk = Bk::new(&path, chapter, pos, 2).unwrap_or_else(|e| { + let mut bk = Bk::new(&path, chapter, pos, 3).unwrap_or_else(|e| { println!("error reading epub: {}", e); std::process::exit(1); }); @@ -421,12 +492,12 @@ fn main() -> crossterm::Result<()> { stdout, terminal::EnterAlternateScreen, cursor::Hide, - EnableMouseCapture + event::EnableMouseCapture )?; terminal::enable_raw_mode()?; bk.render(); - while bk.run(read()?) { + while bk.run(event::read()?) { bk.render(); } @@ -435,11 +506,12 @@ fn main() -> crossterm::Result<()> { format!("{}\n{}\n{}", path, bk.chapter_idx, bk.pos), ) .unwrap(); + queue!( stdout, terminal::LeaveAlternateScreen, cursor::Show, - DisableMouseCapture + event::DisableMouseCapture )?; stdout.flush()?; terminal::disable_raw_mode() |