Skip to content

Commit

Permalink
Fix movement keys in inverted mode (#1035)
Browse files Browse the repository at this point in the history
* Extract scrolling code into functions

* Fix movement keys when inverted

* remove extra len param

---------

Co-authored-by: Conrad Ludgate <conrad.ludgate@truelayer.com>
  • Loading branch information
majutsushi and conradludgate committed Jun 8, 2023
1 parent 49f0def commit a224a8e
Showing 1 changed file with 47 additions and 35 deletions.
82 changes: 47 additions & 35 deletions atuin/src/command/client/search/interactive.rs
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,7 @@ struct State {
results_state: ListState,
switched_search_mode: bool,
search_mode: SearchMode,
results_len: usize,

search: SearchState,
engine: Box<dyn SearchEngine>,
Expand All @@ -58,27 +59,26 @@ impl State {
async fn query_results(&mut self, db: &mut dyn Database) -> Result<Vec<History>> {
let results = self.engine.query(&self.search, db).await?;
self.results_state.select(0);
self.results_len = results.len();
Ok(results)
}

fn handle_input(&mut self, settings: &Settings, input: &Event, len: usize) -> Option<usize> {
fn handle_input(&mut self, settings: &Settings, input: &Event) -> Option<usize> {
match input {
Event::Key(k) => self.handle_key_input(settings, k, len),
Event::Mouse(m) => self.handle_mouse_input(*m, len),
Event::Key(k) => self.handle_key_input(settings, k),
Event::Mouse(m) => self.handle_mouse_input(*m),
Event::Paste(d) => self.handle_paste_input(d),
_ => None,
}
}

fn handle_mouse_input(&mut self, input: MouseEvent, len: usize) -> Option<usize> {
fn handle_mouse_input(&mut self, input: MouseEvent) -> Option<usize> {
match input.kind {
event::MouseEventKind::ScrollDown => {
let i = self.results_state.selected().saturating_sub(1);
self.results_state.select(i);
self.scroll_down(1);
}
event::MouseEventKind::ScrollUp => {
let i = self.results_state.selected() + 1;
self.results_state.select(i.min(len - 1));
self.scroll_up(1);
}
_ => {}
}
Expand All @@ -94,12 +94,7 @@ impl State {

#[allow(clippy::too_many_lines)]
#[allow(clippy::cognitive_complexity)]
fn handle_key_input(
&mut self,
settings: &Settings,
input: &KeyEvent,
len: usize,
) -> Option<usize> {
fn handle_key_input(&mut self, settings: &Settings, input: &KeyEvent) -> Option<usize> {
if input.kind == event::KeyEventKind::Release {
return None;
}
Expand Down Expand Up @@ -211,46 +206,62 @@ impl State {
})
}
KeyCode::Down if !settings.invert => {
let i = self.results_state.selected().saturating_sub(1);
self.results_state.select(i);
self.scroll_down(1);
}
KeyCode::Up if settings.invert => {
let i = self.results_state.selected().saturating_sub(1);
self.results_state.select(i);
self.scroll_down(1);
}
KeyCode::Char('n' | 'j') if ctrl && !settings.invert => {
self.scroll_down(1);
}
KeyCode::Char('n' | 'j') if ctrl => {
let i = self.results_state.selected().saturating_sub(1);
self.results_state.select(i);
KeyCode::Char('n' | 'j') if ctrl && settings.invert => {
self.scroll_up(1);
}
KeyCode::Up if !settings.invert => {
let i = self.results_state.selected() + 1;
self.results_state.select(i.min(len - 1));
self.scroll_up(1);
}
KeyCode::Down if settings.invert => {
let i = self.results_state.selected() + 1;
self.results_state.select(i.min(len - 1));
self.scroll_up(1);
}
KeyCode::Char('p' | 'k') if ctrl && !settings.invert => {
self.scroll_up(1);
}
KeyCode::Char('p' | 'k') if ctrl => {
let i = self.results_state.selected() + 1;
self.results_state.select(i.min(len - 1));
KeyCode::Char('p' | 'k') if ctrl && settings.invert => {
self.scroll_down(1);
}
KeyCode::Char(c) => self.search.input.insert(c),
KeyCode::PageDown => {
KeyCode::PageDown if !settings.invert => {
let scroll_len = self.results_state.max_entries() - settings.scroll_context_lines;
let i = self.results_state.selected().saturating_sub(scroll_len);
self.results_state.select(i);
self.scroll_down(scroll_len);
}
KeyCode::PageUp => {
KeyCode::PageDown if settings.invert => {
let scroll_len = self.results_state.max_entries() - settings.scroll_context_lines;
let i = self.results_state.selected() + scroll_len;
self.results_state.select(i.min(len - 1));
self.scroll_up(scroll_len);
}
KeyCode::PageUp if !settings.invert => {
let scroll_len = self.results_state.max_entries() - settings.scroll_context_lines;
self.scroll_up(scroll_len);
}
KeyCode::PageUp if settings.invert => {
let scroll_len = self.results_state.max_entries() - settings.scroll_context_lines;
self.scroll_down(scroll_len);
}
_ => {}
};

None
}

fn scroll_down(&mut self, scroll_len: usize) {
let i = self.results_state.selected().saturating_sub(scroll_len);
self.results_state.select(i);
}

fn scroll_up(&mut self, scroll_len: usize) {
let i = self.results_state.selected() + scroll_len;
self.results_state.select(i.min(self.results_len - 1));
}

#[allow(clippy::cast_possible_truncation)]
#[allow(clippy::bool_to_int_with_if)]
fn draw<T: Backend>(&mut self, f: &mut Frame<'_, T>, results: &[History], settings: &Settings) {
Expand Down Expand Up @@ -579,6 +590,7 @@ pub async fn history(
},
},
engine: engines::engine(settings.search_mode),
results_len: 0,
};

let mut results = app.query_results(&mut db).await?;
Expand All @@ -596,7 +608,7 @@ pub async fn history(
event_ready = event_ready => {
if event_ready?? {
loop {
if let Some(i) = app.handle_input(settings, &event::read()?, results.len()) {
if let Some(i) = app.handle_input(settings, &event::read()?) {
break 'render i;
}
if !event::poll(Duration::ZERO)? {
Expand Down

1 comment on commit a224a8e

@vercel
Copy link

@vercel vercel bot commented on a224a8e Jun 8, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Successfully deployed to the following URLs:

atuin – ./

atuin-git-main-ellieh.vercel.app
atuin-ellieh.vercel.app
atuin.sh

Please sign in to comment.