aichat 0.30.0

All-in-one LLM CLI Tool
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

use super::REPL_COMMANDS;

use crate::{config::GlobalConfig, utils::NO_COLOR};

use nu_ansi_term::{Color, Style};
use reedline::{Highlighter, StyledText};

const DEFAULT_COLOR: Color = Color::Default;
const MATCH_COLOR: Color = Color::Green;

pub struct ReplHighlighter;

impl ReplHighlighter {
    pub fn new(_config: &GlobalConfig) -> Self {
        Self
    }
}

impl Highlighter for ReplHighlighter {
    fn highlight(&self, line: &str, _cursor: usize) -> StyledText {
        let mut styled_text = StyledText::new();

        if *NO_COLOR {
            styled_text.push((Style::default(), line.to_string()));
        } else if REPL_COMMANDS.iter().any(|cmd| line.contains(cmd.name)) {
            let matches: Vec<&str> = REPL_COMMANDS
                .iter()
                .filter(|cmd| line.contains(cmd.name))
                .map(|cmd| cmd.name)
                .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(DEFAULT_COLOR), buffer_split[0].to_string()));
            styled_text.push((Style::new().fg(MATCH_COLOR), longest_match));
            styled_text.push((Style::new().fg(DEFAULT_COLOR), buffer_split[1].to_string()));
        } else {
            styled_text.push((Style::new().fg(DEFAULT_COLOR), line.to_string()));
        }

        styled_text
    }
}