scrollbar for keyword and entry area

10 files changed, 154 insertions, 24 deletions

diff --git a/Cargo.lock b/Cargo.lock
index 70e5ee8..c8ef416 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -83,7 +83,7 @@ dependencies = [
 
 [[package]]
 name = "bibiman"
-version = "0.2.1"
+version = "0.2.2"
 dependencies = [
  "arboard",
  "biblatex",
diff --git a/Cargo.toml b/Cargo.toml
index 26debff..01ac69f 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "bibiman"
-version = "0.2.1"
+version = "0.2.2"
 authors = ["lukeflo <lukeflo@some.email.not>"]
 license = "GPL-3.0-or-later"
 edition = "2021"
diff --git a/src/errorsetup.rs b/src/errorsetup.rs
new file mode 100644
index 0000000..eaaba0f
--- /dev/null
+++ b/src/errorsetup.rs
@@ -0,0 +1,51 @@
+// bibiman - a TUI for managing BibLaTeX databases
+// Copyright (C) 2024  lukeflo
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+/////
+
+use color_eyre::config::HookBuilder;
+use color_eyre::eyre::Result;
+use crossterm::cursor;
+use crossterm::event::DisableMouseCapture;
+use crossterm::terminal::LeaveAlternateScreen;
+use std::io::stdout;
+
+// Define error hooks to restore the terminal after panic
+pub fn init_error_hooks() -> Result<()> {
+    let (panic, error) = HookBuilder::default().into_hooks();
+    let panic = panic.into_panic_hook();
+    let error = error.into_eyre_hook();
+    color_eyre::eyre::set_hook(Box::new(move |e| {
+        let _ = crossterm::execute!(
+            stdout(),
+            DisableMouseCapture,
+            LeaveAlternateScreen,
+            cursor::Show
+        );
+        let _ = crossterm::terminal::disable_raw_mode();
+        error(e)
+    }))?;
+    std::panic::set_hook(Box::new(move |info| {
+        let _ = crossterm::execute!(
+            stdout(),
+            DisableMouseCapture,
+            LeaveAlternateScreen,
+            cursor::Show
+        );
+        let _ = crossterm::terminal::disable_raw_mode();
+        panic(info)
+    }));
+    Ok(())
+}
diff --git a/src/frontend/app.rs b/src/frontend/app.rs
index 0751da1..26b9a13 100644
--- a/src/frontend/app.rs
+++ b/src/frontend/app.rs
@@ -33,6 +33,7 @@ pub enum CurrentArea {
     TagArea,
     SearchArea,
     HelpArea,
+    InfoArea,
 }
 
 // Check which area was active when popup set active
@@ -137,11 +138,20 @@ impl App {
     // Toggle moveable list between entries and tags
     pub fn toggle_area(&mut self) {
         if let CurrentArea::EntryArea = self.current_area {
+            self.entry_table.entry_scroll_state = self.entry_table.entry_scroll_state.position(0);
             self.current_area = CurrentArea::TagArea;
-            self.tag_list.tag_list_state.select(Some(0))
+            self.tag_list.tag_list_state.select(Some(0));
+            self.tag_list.tag_scroll_state = self
+                .tag_list
+                .tag_scroll_state
+                .position(self.tag_list.tag_list_state.selected().unwrap());
         } else if let CurrentArea::TagArea = self.current_area {
             self.current_area = CurrentArea::EntryArea;
-            self.tag_list.tag_list_state.select(None)
+            self.tag_list.tag_list_state.select(None);
+            self.entry_table.entry_scroll_state = self
+                .entry_table
+                .entry_scroll_state
+                .position(self.entry_table.entry_table_state.selected().unwrap());
         }
     }
 
diff --git a/src/frontend/handler.rs b/src/frontend/handler.rs
index 9a27b3a..ac5b94f 100644
--- a/src/frontend/handler.rs
+++ b/src/frontend/handler.rs
@@ -148,6 +148,7 @@ pub fn handle_key_events(key_event: KeyEvent, app: &mut App, tui: &mut Tui) -> R
             }
             _ => {}
         },
+        CurrentArea::InfoArea => {}
     }
     Ok(())
 }
diff --git a/src/frontend/keywords.rs b/src/frontend/keywords.rs
index 4beec14..6aa4068 100644
--- a/src/frontend/keywords.rs
+++ b/src/frontend/keywords.rs
@@ -18,12 +18,13 @@
 use super::app::{App, FormerArea};
 use super::entries::EntryTable;
 use crate::backend::search::BibiSearch;
-use ratatui::widgets::ListState;
+use ratatui::widgets::{ListState, ScrollbarState};
 
 #[derive(Debug)]
 pub struct TagList {
     pub tag_list_items: Vec<TagListItem>,
     pub tag_list_state: ListState,
+    pub tag_scroll_state: ScrollbarState,
 }
 
 // Structure of the list items.
@@ -43,14 +44,16 @@ impl TagListItem {
 
 impl FromIterator<String> for TagList {
     fn from_iter<I: IntoIterator<Item = String>>(iter: I) -> Self {
-        let tag_list_items = iter
+        let tag_list_items: Vec<TagListItem> = iter
             .into_iter()
             .map(|info| TagListItem::new(&info))
             .collect();
         let tag_list_state = ListState::default(); // for preselection: .with_selected(Some(0));
+        let tag_scroll_state = ScrollbarState::new(tag_list_items.len());
         Self {
             tag_list_items,
             tag_list_state,
+            tag_scroll_state,
         }
     }
 }
@@ -61,18 +64,31 @@ impl App {
     // Movement
     pub fn select_next_tag(&mut self) {
         self.tag_list.tag_list_state.select_next();
+        self.tag_list.tag_scroll_state = self
+            .tag_list
+            .tag_scroll_state
+            .position(self.tag_list.tag_list_state.selected().unwrap());
     }
 
     pub fn select_previous_tag(&mut self) {
         self.tag_list.tag_list_state.select_previous();
+        self.tag_list.tag_scroll_state = self
+            .tag_list
+            .tag_scroll_state
+            .position(self.tag_list.tag_list_state.selected().unwrap());
     }
 
     pub fn select_first_tag(&mut self) {
         self.tag_list.tag_list_state.select_first();
+        self.tag_list.tag_scroll_state = self.tag_list.tag_scroll_state.position(0);
     }
 
     pub fn select_last_tag(&mut self) {
         self.tag_list.tag_list_state.select_last();
+        self.tag_list.tag_scroll_state = self
+            .tag_list
+            .tag_scroll_state
+            .position(self.tag_list.tag_list_items.len());
     }
 
     pub fn get_selected_tag(&self) -> &str {
diff --git a/src/frontend/tui.rs b/src/frontend/tui.rs
index a4560c1..698407d 100644
--- a/src/frontend/tui.rs
+++ b/src/frontend/tui.rs
@@ -179,7 +179,7 @@ impl Tui {
         // if self.paste {
         //     crossterm::execute!(stdout(), EnableBracketedPaste)?;
         // }
-        Self::init_error_hooks()?;
+        // Self::init_error_hooks()?;
         self.start();
         Ok(())
     }
diff --git a/src/frontend/ui.rs b/src/frontend/ui.rs
index 32a8296..3b79f7a 100644
--- a/src/frontend/ui.rs
+++ b/src/frontend/ui.rs
@@ -47,6 +47,9 @@ const SELECTED_STYLE: Style = Style::new()
 const TEXT_FG_COLOR: Color = Color::Indexed(252);
 const TEXT_UNSELECTED_FG_COLOR: Color = Color::Indexed(245);
 
+const SCROLLBAR_UPPER_CORNER: Option<&str> = Some("┓");
+const SCROLLBAR_LOWER_CORNER: Option<&str> = Some("┛");
+
 pub const fn alternate_colors(i: usize) -> Color {
     if i % 2 == 0 {
         NORMAL_ROW_BG
@@ -82,9 +85,7 @@ impl Widget for &mut App {
         self.render_footer(footer_area, buf);
         // Render list area where entry gets selected
         self.render_entrytable(list_area, buf);
-        self.render_entry_table_scrollbar(list_area, buf);
         // Render infos related to selected entry
-        // TODO: only placeholder at the moment, has to be impl.
         self.render_taglist(tag_area, buf);
         self.render_selected_item(info_area, buf);
     }
@@ -229,20 +230,24 @@ impl App {
             buf,
             &mut self.entry_table.entry_table_state,
         );
-    }
 
-    pub fn render_entry_table_scrollbar(&mut self, area: Rect, buf: &mut Buffer) {
-        StatefulWidget::render(
-            Scrollbar::default()
-                .orientation(ScrollbarOrientation::VerticalRight)
-                .track_symbol(None)
-                .thumb_style(Style::new().fg(Color::DarkGray))
-                .begin_symbol(None)
-                .end_symbol(None),
-            area,
-            buf,
-            &mut self.entry_table.entry_scroll_state,
-        );
+        // Scrollbar for entry table
+        let scrollbar = Scrollbar::default()
+            .orientation(ScrollbarOrientation::VerticalRight)
+            .track_symbol(None)
+            .begin_symbol(SCROLLBAR_UPPER_CORNER)
+            .end_symbol(SCROLLBAR_LOWER_CORNER)
+            .thumb_style(Style::new().fg(Color::DarkGray));
+
+        if let CurrentArea::EntryArea = self.current_area {
+            // render the scrollbar
+            StatefulWidget::render(
+                scrollbar,
+                area,
+                buf,
+                &mut self.entry_table.entry_scroll_state,
+            );
+        }
     }
 
     pub fn render_selected_item(&mut self, area: Rect, buf: &mut Buffer) {
@@ -358,6 +363,25 @@ impl App {
             .wrap(Wrap { trim: false })
             .scroll((scroll_height, 0))
             .render(area, buf);
+
+        let scrollbar = Scrollbar::default()
+            .orientation(ScrollbarOrientation::VerticalRight)
+            .track_symbol(None)
+            .begin_symbol(SCROLLBAR_UPPER_CORNER)
+            .end_symbol(SCROLLBAR_LOWER_CORNER)
+            .thumb_style(Style::new().fg(Color::DarkGray));
+
+        if box_height > area.height.into() {
+            if let CurrentArea::EntryArea = self.current_area {
+                // render the scrollbar
+                StatefulWidget::render(
+                    scrollbar,
+                    area,
+                    buf,
+                    &mut self.entry_table.entry_scroll_state,
+                );
+            }
+        }
     }
 
     pub fn render_taglist(&mut self, area: Rect, buf: &mut Buffer) {
@@ -404,8 +428,32 @@ impl App {
             // .highlight_symbol("> ")
             .highlight_spacing(HighlightSpacing::Always);
 
+        // Save list length for calculating scrollbar need
+        // Add 2 to compmensate lines of the block border
+        let list_length = list.len() + 2;
+
         // We need to disambiguate this trait method as both `Widget` and `StatefulWidget` share the
         // same method name `render`.
         StatefulWidget::render(list, area, buf, &mut self.tag_list.tag_list_state);
+
+        // Scrollbar for keyword list
+        let scrollbar = Scrollbar::default()
+            .orientation(ScrollbarOrientation::VerticalRight)
+            .track_symbol(None)
+            .begin_symbol(SCROLLBAR_UPPER_CORNER)
+            .end_symbol(SCROLLBAR_LOWER_CORNER)
+            .thumb_style(Style::new().fg(Color::DarkGray));
+
+        if list_length > area.height.into() {
+            if let CurrentArea::TagArea = self.current_area {
+                // render the scrollbar
+                StatefulWidget::render(
+                    scrollbar,
+                    area,
+                    buf,
+                    &mut self.entry_table.entry_scroll_state,
+                );
+            }
+        }
     }
 }
diff --git a/src/main.rs b/src/main.rs
index 74e1252..6ef7ee1 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -15,13 +15,14 @@
 // along with this program.  If not, see <https://www.gnu.org/licenses/>.
 /////
 
-use core::panic;
-
 use backend::cliargs::{self, CLIArgs};
 use color_eyre::eyre::Result;
+use core::panic;
+use errorsetup::init_error_hooks;
 use frontend::app::App;
 
 pub mod backend;
+pub mod errorsetup;
 pub mod frontend;
 
 #[tokio::main]
@@ -45,6 +46,8 @@ async fn main() -> Result<()> {
         panic!("No \'.bib\' file passed, aborting")
     }
 
+    init_error_hooks()?;
+
     // Create an application.
     let mut app = App::new(parsed_args)?;
 
diff --git a/test.bib b/test.bib
index b902220..50ea05f 100644
--- a/test.bib
+++ b/test.bib
@@ -117,6 +117,7 @@
     langid        = {english},
     langidopts    = {variant=american},
     annotation    = {A plain \texttt{book} entry},
+    keywords = {chemistry},
 }
 
 @book{averroes/bland,