colorizer 2.1.0

Colorizer is a tool for highlighting words in your terminal
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

use ansi_term::Color;
use regex::Regex;

pub struct TextProcessor {
    patterns: Vec<(Color, Regex)>,
}

impl TextProcessor {
    pub fn new(patterns: Vec<(Color, Regex)>) -> Self {
        Self { patterns }
    }

    pub fn process_line(&self, mut s: String) -> String {
        for (color, regex) in &self.patterns {
            s = regex
                .replace_all(
                    &s,
                    format!("{}$0{}", color.prefix(), color.suffix()).as_str(),
                )
                .to_string();
        }

        s
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn happy_path() {
        // Arrange
        let patterns = vec![(Color::Red, Regex::new("foo").unwrap())];
        let processor = TextProcessor::new(patterns);
        let input = "foo bar baz".to_string();

        // Act
        let actual = processor.process_line(input);

        // Assert
        let expected = "\x1b[31mfoo\x1b[0m bar baz";
        assert_eq!(actual, expected);
    }
}