diff options
| -rw-r--r-- | src/main.rs | 121 |
1 files changed, 70 insertions, 51 deletions
diff --git a/src/main.rs b/src/main.rs index 8a31bc7..4c66133 100644 --- a/src/main.rs +++ b/src/main.rs @@ -4,7 +4,10 @@ use std::io::{stdout, Error, Read, Write}; use crossterm::{ cursor, - event::{read, Event, KeyCode, KeyEvent}, + event::{ + read, DisableMouseCapture, EnableMouseCapture, Event, KeyCode, + MouseEvent, + }, queue, style::{Attribute, Print}, terminal, @@ -174,57 +177,65 @@ impl Bk { self.chapter = wrap(chapter, self.cols - (self.pad * 2)); self.chapter_idx = idx; } - fn run(&mut self, code: KeyCode) -> bool { - match code { - KeyCode::Char('q') => return true, - KeyCode::Char('p') => { - if self.chapter_idx > 0 { - self.get_chapter(self.chapter_idx - 1); - 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) { + if self.pos > 0 { + self.pos = self.pos.saturating_sub(n); + } else if self.chapter_idx > 0 { + self.get_chapter(self.chapter_idx - 1); + self.pos = (self.chapter.len() / self.rows) * self.rows; + } + } + fn run(&mut self, event: Event) -> bool { + match event { + Event::Key(e) => match e.code { + KeyCode::Char('q') => return true, + 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::Char('n') => { + if self.chapter_idx < self.toc.len() - 1 { + self.get_chapter(self.chapter_idx + 1); + self.pos = 0; + } } - } - KeyCode::Up | KeyCode::Char('k') => { - if self.pos > 0 { - self.pos -= 1; - } else if self.chapter_idx > 0 { - self.get_chapter(self.chapter_idx - 1); - self.pos = (self.chapter.len() / self.rows) * self.rows; + KeyCode::Up | KeyCode::Char('k') => { + self.scroll_up(1); } - } - KeyCode::Left | KeyCode::PageUp | KeyCode::Char('h') => { - if self.pos > 0 { - self.pos = self.pos.saturating_sub(self.rows); - } else if self.chapter_idx > 0 { - self.get_chapter(self.chapter_idx - 1); - self.pos = (self.chapter.len() / self.rows) * self.rows; + KeyCode::Left | KeyCode::PageUp | KeyCode::Char('h') => { + self.scroll_up(self.rows); } - } - KeyCode::Down | KeyCode::Char('j') => { - if self.pos < self.chapter.len() - 1 { - self.pos += 1; - } else if self.chapter_idx < self.toc.len() - 1 { - self.get_chapter(self.chapter_idx + 1); - self.pos = 0; + KeyCode::Down | KeyCode::Char('j') => { + self.scroll_down(1); } - } - KeyCode::Right - | KeyCode::PageDown - | KeyCode::Char('l') - | KeyCode::Char(' ') => { - if self.pos + self.rows < self.chapter.len() { - self.pos += self.rows; - } else if self.chapter_idx < self.toc.len() - 1 { - self.get_chapter(self.chapter_idx + 1); - self.pos = 0; + KeyCode::Right + | KeyCode::PageDown + | KeyCode::Char('l') + | KeyCode::Char(' ') => { + self.scroll_down(self.rows); } + _ => (), + }, + Event::Mouse(e) => match e { + MouseEvent::ScrollDown(_, _, _) => self.scroll_down(3), + MouseEvent::ScrollUp(_, _, _) => self.scroll_up(3), + _ => (), + }, + Event::Resize(cols, rows) => { + self.cols = cols; + self.rows = rows as usize; + self.get_chapter(self.chapter_idx); } - _ => (), } false } @@ -293,15 +304,18 @@ fn main() -> crossterm::Result<()> { }); let mut stdout = stdout(); - queue!(stdout, terminal::EnterAlternateScreen, cursor::Hide)?; + queue!( + stdout, + terminal::EnterAlternateScreen, + cursor::Hide, + EnableMouseCapture + )?; terminal::enable_raw_mode()?; loop { bk.render(); - if let Event::Key(KeyEvent { code, .. }) = read()? { - if bk.run(code) { - break; - } + if bk.run(read()?) { + break; } } @@ -310,7 +324,12 @@ fn main() -> crossterm::Result<()> { format!("{}\n{}\n{}", path, bk.chapter_idx, bk.pos), ) .unwrap(); - queue!(stdout, terminal::LeaveAlternateScreen, cursor::Show)?; + queue!( + stdout, + terminal::LeaveAlternateScreen, + cursor::Show, + DisableMouseCapture + )?; stdout.flush()?; terminal::disable_raw_mode() } |