treegrep 2.0.0

regex pattern matcher that displays results in a tree structure with an interface to jump to matched text
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

// SPDX-License-Identifier: MIT

use crate::config::Config;
use crossterm::style::{
    Attribute, Color, SetAttribute, SetForegroundColor, StyledContent, Stylize, style,
};
use std::fmt::{self, Display};

pub const RESET: SetAttribute = SetAttribute(Attribute::Reset);
const BOLD: SetAttribute = SetAttribute(Attribute::Bold);
pub const DIM: SetAttribute = SetAttribute(Attribute::Dim);

const RED_FG: SetForegroundColor = SetForegroundColor(Color::Red);
pub const LONG_BRANCH_FILE_SEPARATOR: &str = ", ";

pub const FILE_COLOR_DEFAULT: Color = Color::Cyan;
pub const DIR_COLOR_DEFAULT: Color = Color::Blue;
pub const LINE_NUMBER_COLOR_DEFAULT: Color = Color::Yellow;
pub const MATCHED_COLORS_DEFAULT: [Color; 3] = [Color::Green, Color::Magenta, Color::Red];
pub const SELECTED_BG_DEFAULT: Color = Color::DarkGrey;
pub const FILTER_HIGHLIGHT_DEFAULT: Color = Color::Black;

pub struct DisplayRepeater<T>(T, usize);
impl<T: Display> Display for DisplayRepeater<T> {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        for _ in 0..self.1 {
            self.0.fmt(f)?;
        }
        Ok(())
    }
}
pub fn repeat<T>(item: T, times: usize) -> DisplayRepeater<T> {
    DisplayRepeater(item, times)
}

pub fn error_prefix(bold: bool, colors: bool) -> String {
    let e = "error:";
    match (bold, colors) {
        (true, true) => format!("{BOLD}{RED_FG}{e}{RESET}"),
        (true, false) => format!("{BOLD}{e}{RESET}"),
        (false, true) => format!("{RED_FG}{e}{RESET}"),
        _ => e.into(),
    }
}

pub fn style_with<D: Display>(orig: D, color: Color, config: &Config) -> StyledContent<D> {
    let mut styled = style(orig);
    if config.with_colors {
        styled = styled.with(color);
    }
    if config.with_bold {
        styled = styled.bold();
    }
    styled
}

pub fn match_substring<'a>(
    orig: &'a str,
    regexp_id: usize,
    config: &Config,
) -> StyledContent<&'a str> {
    style_with(
        orig,
        config.colors.matches[regexp_id % config.colors.matches.len()],
        config,
    )
}

pub fn style_with_on<D: Display>(
    orig: D,
    fg: Color,
    bg: Color,
    config: &Config,
) -> StyledContent<D> {
    let mut s = style_with(orig, fg, config);
    if config.with_colors {
        s = s.on(bg);
    }
    s
}

pub fn style_on<D: Display>(orig: D, bg: Color, config: &Config) -> StyledContent<D> {
    let mut s = style(orig);
    if config.with_colors {
        s = s.on(bg);
    }
    s
}