Function highlight_code_with_line_numbers

src/apps/cli/src/ui/syntax_highlight.rs:133–179

Highlight code with line numbers prepended. Returns lines in the format: `{line_number} | {highlighted_code}`

(
    content: &str,
    file_ext: &str,
    hl_theme: HighlightTheme,
    line_num_style: Style,
    separator_style: Style,
)

///
/// Returns lines in the format: `{line_number} | {highlighted_code}`
pub fn highlight_code_with_line_numbers<'a>(
    content: &str,
    file_ext: &str,
    hl_theme: HighlightTheme,
    line_num_style: Style,
    separator_style: Style,
) -> Vec<Line<'a>> {
    let syntax = SYNTAX_SET
        .find_syntax_by_extension(file_ext)
        .unwrap_or_else(|| SYNTAX_SET.find_syntax_plain_text());

    let theme = &THEME_SET.themes[hl_theme.syntect_theme_name()];
    let mut highlighter = HighlightLines::new(syntax, theme);
    let mut lines = Vec::new();

    let total_lines = content.lines().count();
    let num_width = total_lines.to_string().len().max(3);

    for (i, line_str) in LinesWithEndings::from(content).enumerate() {
        let line_num = i + 1;
        let mut spans = vec![
            Span::styled(
                format!("{:>width$}", line_num, width = num_width),
                line_num_style,
            ),
            Span::styled(" \u{2502} ", separator_style), // │
        ];

        match highlighter.highlight_line(line_str, &SYNTAX_SET) {
            Ok(ranges) => {
                for (style, text) in ranges {
                    spans.push(Span::styled(
                        text.trim_end_matches('\n').to_string(),
                        syntect_to_ratatui_style(&style),
                    ));
                }
            }
            Err(_) => {
                spans.push(Span::raw(line_str.trim_end_matches('\n').to_string()));
            }
        }

        lines.push(Line::from(spans));
    }

    lines
}

/// Extract file extension from a file path.
/// Returns "txt" if no extension is found.