1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
use crate::config::SharedConfig;
use nu_ansi_term::{Color, Style};
use reedline::{Highlighter, StyledText};
const MATCH_COLOR: Color = Color::Green;
pub struct ReplHighlighter {
external_commands: Vec<String>,
config: SharedConfig,
}
impl ReplHighlighter {
/// Construct the default highlighter with a given set of extern commands/keywords to detect and highlight
pub fn new(config: SharedConfig, external_commands: Vec<String>) -> Self {
Self {
external_commands,
config,
}
}
}
impl Highlighter for ReplHighlighter {
fn highlight(&self, line: &str, _cursor: usize) -> StyledText {
let mut styled_text = StyledText::new();
let color = if self.config.read().light_theme {
Color::Black
} else {
Color::White
};
let match_color = if self.config.read().highlight {
MATCH_COLOR
} else {
color
};
if self
.external_commands
.clone()
.iter()
.any(|x| line.contains(x))
{
let matches: Vec<&str> = self
.external_commands
.iter()
.filter(|c| line.contains(*c))
.map(std::ops::Deref::deref)
.collect();
let longest_match = matches.iter().fold(String::new(), |acc, &item| {
if item.len() > acc.len() {
item.to_string()
} else {
acc
}
});
let buffer_split: Vec<&str> = line.splitn(2, &longest_match).collect();
styled_text.push((Style::new().fg(color), buffer_split[0].to_string()));
styled_text.push((Style::new().fg(match_color), longest_match));
styled_text.push((Style::new().fg(color), buffer_split[1].to_string()));
} else {
styled_text.push((Style::new().fg(color), line.to_string()));
}
styled_text
}
}
|
