use super::{Component, EventState, MovableComponent}; use crate::components::command::CommandInfo; use crate::config::KeyConfig; use crate::event::Key; use anyhow::Result; use tui::{ backend::Backend, layout::Rect, style::{Color, Style}, widgets::{Block, Borders, Clear, List, ListItem, ListState}, Frame, }; const RESERVED_WORDS_IN_WHERE_CLAUSE: &[&str] = &["IN", "AND", "OR", "NOT", "NULL", "IS"]; const ALL_RESERVED_WORDS: &[&str] = &[ "IN", "AND", "OR", "NOT", "NULL", "IS", "SELECT", "UPDATE", "DELETE", "FROM", "LIMIT", "WHERE", ]; pub struct CompletionComponent { key_config: KeyConfig, state: ListState, word: String, candidates: Vec, } impl CompletionComponent { pub fn new(key_config: KeyConfig, word: impl Into, all: bool) -> Self { Self { key_config, state: ListState::default(), word: word.into(), candidates: if all { ALL_RESERVED_WORDS.iter().map(|w| w.to_string()).collect() } else { RESERVED_WORDS_IN_WHERE_CLAUSE .iter() .map(|w| w.to_string()) .collect() }, } } pub fn update(&mut self, word: impl Into) { self.word = word.into(); self.state.select(None); self.state.select(Some(0)) } fn next(&mut self) { let i = match self.state.selected() { Some(i) => { if i >= self.filterd_candidates().count() - 1 { 0 } else { i + 1 } } None => 0, }; self.state.select(Some(i)); } fn previous(&mut self) { let i = match self.state.selected() { Some(i) => { if i == 0 { self.filterd_candidates().count() - 1 } else { i - 1 } } None => 0, }; self.state.select(Some(i)); } fn filterd_candidates(&self) -> impl Iterator { self.candidates.iter().filter(move |c| { (c.starts_with(self.word.to_lowercase().as_str()) || c.starts_with(self.word.to_uppercase().as_str())) && !self.word.is_empty() }) } pub fn selected_candidate(&self) -> Option { self.filterd_candidates() .collect::>() .get(self.state.selected()?) .map(|c| c.to_string()) } pub fn word(&self) -> String { self.word.to_string() } } impl MovableComponent for CompletionComponent { fn draw( &mut self, f: &mut Frame, area: Rect, _focused: bool, x: u16, y: u16, ) -> Result<()> { if !self.word.is_empty() { let width = 30; let candidates = self .filterd_candidates() .map(|c| ListItem::new(c.to_string())) .collect::>(); if candidates.clone().is_empty() { return Ok(()); } let candidate_list = List::new(candidates.clone()) .block(Block::default().borders(Borders::ALL)) .highlight_style(Style::default().bg(Color::Blue)) .style(Style::default()); let area = Rect::new( area.x + x, area.y + y + 2, width .min(f.size().width) .min(f.size().right().saturating_sub(area.x + x)), (candidates.len().min(5) as u16 + 2) .min(f.size().bottom().saturating_sub(area.y + y + 2)), ); f.render_widget(Clear, area); f.render_stateful_widget(candidate_list, area, &mut self.state); } Ok(()) } } impl Component for CompletionComponent { fn commands(&self, _out: &mut Vec) {} fn event(&mut self, key: Key) -> Result { if key == self.key_config.move_down { self.next(); return Ok(EventState::Consumed); } else if key == self.key_config.move_up { self.previous(); return Ok(EventState::Consumed); } Ok(EventState::NotConsumed) } } #[cfg(test)] mod test { use super::{CompletionComponent, KeyConfig}; #[test] fn test_filterd_candidates_lowercase() { assert_eq!( CompletionComponent::new(KeyConfig::default(), "an", false) .filterd_candidates() .collect::>(), vec![&"AND".to_string()] ); } #[test] fn test_filterd_candidates_uppercase() { assert_eq!( CompletionComponent::new(KeyConfig::default(), "AN", false) .filterd_candidates() .collect::>(), vec![&"AND".to_string()] ); } #[test] fn test_filterd_candidates_multiple_candidates() { assert_eq!( CompletionComponent::new(KeyConfig::default(), "n", false) .filterd_candidates() .collect::>(), vec![&"NOT".to_string(), &"NULL".to_string()] ); assert_eq!( CompletionComponent::new(KeyConfig::default(), "N", false) .filterd_candidates() .collect::>(), vec![&"NOT".to_string(), &"NULL".to_string()] ); } }