From c2bf6c58e5d651a8191825c0c9bf35237cbb502c Mon Sep 17 00:00:00 2001
From: rsteube <rsteube@users.noreply.github.com>
Date: Fri, 29 Dec 2023 20:41:10 +0100
Subject: [PATCH] suggestion: added style

supports reverse highlighting using nu.config

```nushell
style: {
    text: white
    selected_text: {
        attr: r
    }
    description_text: white_dimmed
}
```
---
 src/completion/base.rs     |  4 ++++
 src/completion/default.rs  | 22 +++++++++++++---------
 src/completion/history.rs  |  1 +
 src/engine.rs              |  2 +-
 src/menu/columnar_menu.rs  | 21 +++++++++++++++++----
 src/menu/menu_functions.rs |  2 ++
 6 files changed, 38 insertions(+), 14 deletions(-)

diff --git a/src/completion/base.rs b/src/completion/base.rs
index 3de63c8d..46937892 100644
--- a/src/completion/base.rs
+++ b/src/completion/base.rs
@@ -1,3 +1,5 @@
+use nu_ansi_term::Style;
+
 /// A span of source code, with positions in bytes
 #[derive(Debug, Default, Clone, Copy, PartialEq, Eq, Ord, PartialOrd, Hash)]
 pub struct Span {
@@ -60,6 +62,8 @@ pub struct Suggestion {
     pub value: String,
     /// Optional description for the replacement
     pub description: Option<String>,
+    /// Optional style for the replacement
+    pub style: Option<Style>,
     /// Optional vector of strings in the suggestion. These can be used to
     /// represent examples coming from a suggestion
     pub extra: Option<Vec<String>>,
diff --git a/src/completion/default.rs b/src/completion/default.rs
index 11062d7b..693885b9 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, extra: None, span: Span { start: 0, end: 3 }, append_whitespace: false},
-    ///         Suggestion {value: "batman".into(), description: None, extra: None, span: Span { start: 0, end: 3 }, append_whitespace: false},
-    ///         Suggestion {value: "batmobile".into(), description: None, extra: None, span: Span { start: 0, end: 3 }, append_whitespace: false},
+    ///         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},
     ///     ]);
     ///
     /// assert_eq!(
     ///     completions.complete("to the bat",10),
     ///     vec![
-    ///         Suggestion {value: "batcave".into(), description: None, extra: None, span: Span { start: 7, end: 10 }, append_whitespace: false},
-    ///         Suggestion {value: "batman".into(), description: None, extra: None, span: Span { start: 7, end: 10 }, append_whitespace: false},
-    ///         Suggestion {value: "batmobile".into(), description: None, extra: None, span: Span { start: 7, end: 10 }, append_whitespace: false},
+    ///         Suggestion {value: "batcave".into(), description: None, style: None, extra: None, span: Span { start: 7, end: 10 }, append_whitespace: false},
+    ///         Suggestion {value: "batman".into(), description: None, style: None, extra: None, span: Span { start: 7, end: 10 }, append_whitespace: false},
+    ///         Suggestion {value: "batmobile".into(), description: None, style: None, extra: None, span: Span { start: 7, end: 10 }, append_whitespace: false},
     ///     ]);
     /// ```
     fn complete(&mut self, line: &str, pos: usize) -> Vec<Suggestion> {
@@ -99,6 +99,7 @@ impl Completer for DefaultCompleter {
                                     Suggestion {
                                         value: format!("{span_line}{ext}"),
                                         description: None,
+                                        style: None,
                                         extra: None,
                                         span,
                                         append_whitespace: false,
@@ -173,15 +174,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, extra: None, span: Span { start: 0, end: 2 }, append_whitespace: false}]);
+    ///     vec![Suggestion {value: "test".into(), 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, extra: None, span: Span { start: 0, end: 2 }, append_whitespace: false},
-    ///         Suggestion {value: "test_underscore".into(), description: None, extra: None, span: Span { start: 0, end: 2 }, append_whitespace: false},
+    ///         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},
     ///     ]);
     /// ```
     pub fn with_inclusions(incl: &[char]) -> Self {
@@ -371,6 +372,7 @@ mod tests {
                 Suggestion {
                     value: "ｎｕｌｌ".into(),
                     description: None,
+                    style: None,
                     extra: None,
                     span: Span { start: 0, end: 3 },
                     append_whitespace: false,
@@ -378,6 +380,7 @@ mod tests {
                 Suggestion {
                     value: "ｎｕｍｂｅｒ".into(),
                     description: None,
+                    style: None,
                     extra: None,
                     span: Span { start: 0, end: 3 },
                     append_whitespace: false,
@@ -385,6 +388,7 @@ mod tests {
                 Suggestion {
                     value: "ｎｕｓｈｅｌｌ".into(),
                     description: None,
+                    style: None,
                     extra: None,
                     span: Span { start: 0, end: 3 },
                     append_whitespace: false,
diff --git a/src/completion/history.rs b/src/completion/history.rs
index ea6aedb6..82c6a89b 100644
--- a/src/completion/history.rs
+++ b/src/completion/history.rs
@@ -59,6 +59,7 @@ impl<'menu> HistoryCompleter<'menu> {
         Suggestion {
             value: value.to_string(),
             description: None,
+            style: None,
             extra: None,
             span,
             append_whitespace: false,
diff --git a/src/engine.rs b/src/engine.rs
index 43555eec..30bb43bc 100644
--- a/src/engine.rs
+++ b/src/engine.rs
@@ -1538,7 +1538,7 @@ impl Reedline {
                         self.get_history_session_id(),
                     )))
                     .unwrap_or_else(|_| Vec::new())
-                    .get(0)
+                    .first()
                     .and_then(|history| history.command_line.split_whitespace().next_back())
                     .map(|token| (parsed.remainder.len(), indicator.len(), token.to_string())),
             });
diff --git a/src/menu/columnar_menu.rs b/src/menu/columnar_menu.rs
index ec391777..22e7e2f6 100644
--- a/src/menu/columnar_menu.rs
+++ b/src/menu/columnar_menu.rs
@@ -344,9 +344,16 @@ impl ColumnarMenu {
                     let left_text_size = self.longest_suggestion + self.default_details.col_padding;
                     let right_text_size = self.get_width().saturating_sub(left_text_size);
                     format!(
-                        "{}{:max$}{}{}{}",
+                        "{}{}{:max$}{}{}{}{}{}",
+                        suggestion
+                            .style
+                            .unwrap_or(self.color.text_style)
+                            .reverse()
+                            .prefix(),
                         self.color.selected_text_style.prefix(),
                         &suggestion.value,
+                        self.color.description_style.reverse().prefix(),
+                        self.color.selected_text_style.prefix(),
                         description
                             .chars()
                             .take(right_text_size)
@@ -358,7 +365,12 @@ impl ColumnarMenu {
                     )
                 } else {
                     format!(
-                        "{}{}{}{:>empty$}{}",
+                        "{}{}{}{}{:>empty$}{}",
+                        suggestion
+                            .style
+                            .unwrap_or(self.color.text_style)
+                            .reverse()
+                            .prefix(),
                         self.color.selected_text_style.prefix(),
                         &suggestion.value,
                         RESET,
@@ -372,7 +384,7 @@ impl ColumnarMenu {
                 let right_text_size = self.get_width().saturating_sub(left_text_size);
                 format!(
                     "{}{:max$}{}{}{}{}{}",
-                    self.color.text_style.prefix(),
+                    suggestion.style.unwrap_or(self.color.text_style).prefix(),
                     &suggestion.value,
                     RESET,
                     self.color.description_style.prefix(),
@@ -388,7 +400,7 @@ impl ColumnarMenu {
             } else {
                 format!(
                     "{}{}{}{}{:>empty$}{}{}",
-                    self.color.text_style.prefix(),
+                    suggestion.style.unwrap_or(self.color.text_style).prefix(),
                     &suggestion.value,
                     RESET,
                     self.color.description_style.prefix(),
@@ -806,6 +818,7 @@ mod tests {
         Suggestion {
             value: name.to_string(),
             description: None,
+            style: None,
             extra: None,
             span: Span { start: 0, end: pos },
             append_whitespace: false,
diff --git a/src/menu/menu_functions.rs b/src/menu/menu_functions.rs
index 8d15d696..b381fa36 100644
--- a/src/menu/menu_functions.rs
+++ b/src/menu/menu_functions.rs
@@ -488,6 +488,7 @@ mod tests {
             .map(|s| Suggestion {
                 value: s.into(),
                 description: None,
+                style: None,
                 extra: None,
                 span: Span::new(0, s.len()),
                 append_whitespace: false,
@@ -507,6 +508,7 @@ mod tests {
             .map(|s| Suggestion {
                 value: s.into(),
                 description: None,
+                style: None,
                 extra: None,
                 span: Span::new(0, s.len()),
                 append_whitespace: false,