diff --git a/src/completion/base.rs b/src/completion/base.rs index 909c467b..c80f2e1a 100644 --- a/src/completion/base.rs +++ b/src/completion/base.rs @@ -78,6 +78,8 @@ pub trait Completer: Send { pub struct Suggestion { /// String replacement that will be introduced to the the buffer pub value: String, + /// Optional display value for the replacement + pub display: Option, /// Optional description for the replacement pub description: Option, /// Optional style for the replacement @@ -91,3 +93,10 @@ pub struct Suggestion { /// This helps to avoid that a completer repeats the complete suggestion. pub append_whitespace: bool, } + +impl Suggestion { + /// Returns display if set, otherwise value. + pub fn display_or_value(&self) -> &String { + self.display.as_ref().unwrap_or(&self.value) + } +} diff --git a/src/completion/default.rs b/src/completion/default.rs index 882debb6..63f351c7 100644 --- a/src/completion/default.rs +++ b/src/completion/default.rs @@ -55,17 +55,17 @@ impl Completer for DefaultCompleter { /// assert_eq!( /// completions.complete("bat",3), /// vec![ - /// Suggestion {value: "batcave".into(), description: None, style: None, extra: None, span: Span { start: 0, end: 3 }, append_whitespace: false}, - /// Suggestion {value: "batman".into(), description: None, style: None, extra: None, span: Span { start: 0, end: 3 }, append_whitespace: false}, - /// Suggestion {value: "batmobile".into(), description: None, style: None, extra: None, span: Span { start: 0, end: 3 }, append_whitespace: false}, + /// Suggestion {value: "batcave".into(), display: None, description: None, style: None, extra: None, span: Span { start: 0, end: 3 }, append_whitespace: false}, + /// Suggestion {value: "batman".into(), display: None, description: None, style: None, extra: None, span: Span { start: 0, end: 3 }, append_whitespace: false}, + /// Suggestion {value: "batmobile".into(), display: None, description: None, style: None, extra: None, span: Span { start: 0, end: 3 }, append_whitespace: false}, /// ]); /// /// assert_eq!( /// completions.complete("to the\r\nbat",11), /// vec![ - /// Suggestion {value: "batcave".into(), description: None, style: None, extra: None, span: Span { start: 8, end: 11 }, append_whitespace: false}, - /// Suggestion {value: "batman".into(), description: None, style: None, extra: None, span: Span { start: 8, end: 11 }, append_whitespace: false}, - /// Suggestion {value: "batmobile".into(), description: None, style: None, extra: None, span: Span { start: 8, end: 11 }, append_whitespace: false}, + /// Suggestion {value: "batcave".into(), display: None, description: None, style: None, extra: None, span: Span { start: 8, end: 11 }, append_whitespace: false}, + /// Suggestion {value: "batman".into(), display: None, description: None, style: None, extra: None, span: Span { start: 8, end: 11 }, append_whitespace: false}, + /// Suggestion {value: "batmobile".into(), display: None, description: None, style: None, extra: None, span: Span { start: 8, end: 11 }, append_whitespace: false}, /// ]); /// ``` fn complete(&mut self, line: &str, pos: usize) -> Vec { @@ -105,6 +105,7 @@ impl Completer for DefaultCompleter { Suggestion { value: format!("{span_line}{ext}"), + display: None, description: None, style: None, extra: None, @@ -182,15 +183,15 @@ impl DefaultCompleter { /// completions.insert(vec!["test-hyphen","test_underscore"].iter().map(|s| s.to_string()).collect()); /// assert_eq!( /// completions.complete("te",2), - /// vec![Suggestion {value: "test".into(), description: None, style: None, extra: None, span: Span { start: 0, end: 2 }, append_whitespace: false}]); + /// vec![Suggestion {value: "test".into(), display: None, description: None, style: None, extra: None, span: Span { start: 0, end: 2 }, append_whitespace: false}]); /// /// let mut completions = DefaultCompleter::with_inclusions(&['-', '_']); /// completions.insert(vec!["test-hyphen","test_underscore"].iter().map(|s| s.to_string()).collect()); /// assert_eq!( /// completions.complete("te",2), /// vec![ - /// Suggestion {value: "test-hyphen".into(), description: None, style: None, extra: None, span: Span { start: 0, end: 2 }, append_whitespace: false}, - /// Suggestion {value: "test_underscore".into(), description: None, style: None, extra: None, span: Span { start: 0, end: 2 }, append_whitespace: false}, + /// Suggestion {value: "test-hyphen".into(), display: None, description: None, style: None, extra: None, span: Span { start: 0, end: 2 }, append_whitespace: false}, + /// Suggestion {value: "test_underscore".into(), display: None, description: None, style: None, extra: None, span: Span { start: 0, end: 2 }, append_whitespace: false}, /// ]); /// ``` pub fn with_inclusions(incl: &[char]) -> Self { @@ -379,6 +380,7 @@ mod tests { [ Suggestion { value: "null".into(), + display: None, description: None, style: None, extra: None, @@ -387,6 +389,7 @@ mod tests { }, Suggestion { value: "number".into(), + display: None, description: None, style: None, extra: None, @@ -395,6 +398,7 @@ mod tests { }, Suggestion { value: "nushell".into(), + display: None, description: None, style: None, extra: None, @@ -423,6 +427,7 @@ mod tests { [ Suggestion { value: "test".into(), + display: None, description: None, style: None, extra: None, @@ -431,6 +436,7 @@ mod tests { }, Suggestion { value: "this is the reedline crate".into(), + display: None, description: None, style: None, extra: None, @@ -439,6 +445,7 @@ mod tests { }, Suggestion { value: "this is the reedline crate".into(), + display: None, description: None, style: None, extra: None, diff --git a/src/completion/history.rs b/src/completion/history.rs index fda3384f..c65605c5 100644 --- a/src/completion/history.rs +++ b/src/completion/history.rs @@ -60,6 +60,7 @@ impl<'menu> HistoryCompleter<'menu> { Suggestion { value: value.to_string(), + display: None, description: None, style: None, extra: None, diff --git a/src/menu/columnar_menu.rs b/src/menu/columnar_menu.rs index 6eac27ab..84003f9d 100644 --- a/src/menu/columnar_menu.rs +++ b/src/menu/columnar_menu.rs @@ -302,7 +302,11 @@ impl ColumnarMenu { let match_len = self.working_details.shortest_base_string.len(); // Split string so the match text can be styled - let (match_str, remaining_str) = suggestion.value.split_at(match_len); + let (match_str, remaining_str) = if let Some(display) = suggestion.display.as_ref() { + ("", display.as_str()) + } else { + suggestion.value.split_at(match_len) + }; let suggestion_style_prefix = suggestion .style @@ -397,7 +401,7 @@ impl ColumnarMenu { format!( "{}{:max$}{}{}", marker, - &suggestion.value, + suggestion.display_or_value(), description .chars() .take(empty_space) @@ -414,7 +418,7 @@ impl ColumnarMenu { format!( "{}{}{:>empty$}{}", marker, - &suggestion.value, + suggestion.display_or_value(), "", self.end_of_line(column), empty = empty_space.saturating_sub(marker.len()), @@ -530,15 +534,17 @@ impl Menu for ColumnarMenu { self.working_details.col_width = painter.screen_width() as usize; self.longest_suggestion = self.get_values().iter().fold(0, |prev, suggestion| { - if prev >= suggestion.value.len() { + let suggestion_length = suggestion.display_or_value().len(); + if prev >= suggestion_length { prev } else { - suggestion.value.len() + suggestion_length } }); } else { let max_width = self.get_values().iter().fold(0, |acc, suggestion| { - let str_len = suggestion.value.len() + self.default_details.col_padding; + let str_len = + suggestion.display_or_value().len() + self.default_details.col_padding; if str_len > acc { str_len } else { @@ -654,7 +660,9 @@ impl Menu for ColumnarMenu { // Correcting the enumerate index based on the number of skipped values let index = index + skip_values; let column = index as u16 % self.get_cols(); - let empty_space = self.get_width().saturating_sub(suggestion.value.len()); + let empty_space = self + .get_width() + .saturating_sub(suggestion.display_or_value().len()); self.create_string(suggestion, index, column, empty_space, use_ansi_coloring) }) @@ -745,6 +753,7 @@ mod tests { fn fake_suggestion(name: &str, pos: usize) -> Suggestion { Suggestion { value: name.to_string(), + display: None, description: None, style: None, extra: None, diff --git a/src/menu/ide_menu.rs b/src/menu/ide_menu.rs index 2ce711f6..7602a237 100644 --- a/src/menu/ide_menu.rs +++ b/src/menu/ide_menu.rs @@ -1376,6 +1376,7 @@ mod tests { fn fake_suggestion(name: &str, pos: usize) -> Suggestion { Suggestion { value: name.to_string(), + display: None, description: None, style: None, extra: None, diff --git a/src/menu/menu_functions.rs b/src/menu/menu_functions.rs index cc48d85a..92d26791 100644 --- a/src/menu/menu_functions.rs +++ b/src/menu/menu_functions.rs @@ -584,6 +584,7 @@ mod tests { .into_iter() .map(|s| Suggestion { value: s.into(), + display: None, description: None, style: None, extra: None, @@ -604,6 +605,7 @@ mod tests { .into_iter() .map(|s| Suggestion { value: s.into(), + display: None, description: None, style: None, extra: None, @@ -659,6 +661,7 @@ mod tests { replace_in_buffer( Some(Suggestion { value, + display: None, description: None, style: None, extra: None,