ratatui-markdown 0.3.6

Markdown rendering, syntax highlighting, collapsible trees, and rich scroll widgets for ratatui
Documentation 
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

use std::sync::Arc;

use ratatui::{
    style::Style,
    text::{Line, Span},
};

use super::CodeHighlighter;
use crate::constants::list_prefix::{HLINE, ROUNDED_BL, ROUNDED_TL, VLINE};

pub struct HighlightHooks {
    highlighter: Arc<dyn CodeHighlighter>,
    max_width: usize,
    border_color: ratatui::style::Color,
}

impl HighlightHooks {
    pub fn new(highlighter: Arc<dyn CodeHighlighter>, max_width: usize) -> Self {
        Self {
            highlighter,
            max_width,
            border_color: ratatui::style::Color::DarkGray,
        }
    }

    pub fn with_border_color(mut self, color: ratatui::style::Color) -> Self {
        self.border_color = color;
        self
    }
}

#[cfg(feature = "markdown")]
impl crate::markdown::RenderHooks for HighlightHooks {
    fn render_code_block(&self, lang: &str, content: &str) -> Option<Vec<Line<'static>>> {
        let segments = self.highlighter.highlight(lang, content);
        if segments.is_empty() {
            return None;
        }

        let border_style = Style::default().fg(self.border_color);
        let display_lang = if lang.is_empty() { "code" } else { lang };

        let mut lines = Vec::new();

        lines.push(Line::from(Span::styled(
            format!("{ROUNDED_TL}{HLINE} {display_lang}"),
            border_style,
        )));

        let prefix = format!("{VLINE} ");
        let content_width = self.max_width.saturating_sub(2);

        let code_lines = super::segment::segments_to_lines(
            content,
            &segments,
            &prefix,
            border_style,
            content_width,
        );
        lines.extend(code_lines);

        lines.push(Line::from(Span::styled(
            format!("{ROUNDED_BL}{HLINE}"),
            border_style,
        )));

        Some(lines)
    }
}