canonrs-server 0.1.0

CanonRS server-side rendering support
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116

use pulldown_cmark::{Parser, Event, Tag, TagEnd, HeadingLevel};
use canonrs_core::TocItem;

pub struct TocExtractor;

impl TocExtractor {
    pub fn extract_toc(markdown: &str) -> Vec<TocItem> {
        let parser = Parser::new(markdown);
        let mut toc_items = Vec::new();
        let mut in_heading = false;
        let mut current_level = 0u8;
        let mut current_text = String::new();
        let mut heading_counter = 0;

        for event in parser {
            match event {
                Event::Start(Tag::Heading { level, .. }) => {
                    in_heading = true;
                    current_level = match level {
                        HeadingLevel::H1 => 1,
                        HeadingLevel::H2 => 2,
                        HeadingLevel::H3 => 3,
                        HeadingLevel::H4 => 4,
                        HeadingLevel::H5 => 5,
                        HeadingLevel::H6 => 6,
                    };
                    current_text.clear();
                }
                Event::End(TagEnd::Heading { .. }) => {
                    if in_heading {
                        let id = if current_text.is_empty() {
                            format!("heading-{}", heading_counter)
                        } else {
                            slugify(current_text.trim())
                        };
                        toc_items.push(TocItem::new(id, current_text.clone(), current_level));
                        heading_counter += 1;
                        in_heading = false;
                    }
                }
                Event::Text(text) if in_heading => {
                    current_text.push_str(&text);
                }
                _ => {}
            }
        }

        toc_items
    }
}

pub fn slugify(text: &str) -> String {
    text.to_lowercase()
        .chars()
        .map(|c| {
            if c.is_alphanumeric() { c }
            else if c.is_whitespace() || c == '-' { '-' }
            else { '_' }
        })
        .collect::<String>()
        .split('-')
        .filter(|s| !s.is_empty())
        .collect::<Vec<_>>()
        .join("-")
}

impl TocExtractor {
    pub fn extract_toc_with_prefix(markdown: &str, id_prefix: &str) -> Vec<TocItem> {
        let parser = Parser::new(markdown);
        let mut toc_items = Vec::new();
        let mut in_heading = false;
        let mut current_level = 0u8;
        let mut current_text = String::new();
        let mut heading_counter = 0;

        for event in parser {
            match event {
                Event::Start(Tag::Heading { level, .. }) => {
                    in_heading = true;
                    current_level = match level {
                        HeadingLevel::H1 => 1,
                        HeadingLevel::H2 => 2,
                        HeadingLevel::H3 => 3,
                        HeadingLevel::H4 => 4,
                        HeadingLevel::H5 => 5,
                        HeadingLevel::H6 => 6,
                    };
                    current_text.clear();
                }
                Event::End(TagEnd::Heading { .. }) => {
                    if in_heading {
                        let slug = if current_text.is_empty() {
                            format!("heading-{}", heading_counter)
                        } else {
                            slugify(current_text.trim())
                        };
                        let id = if id_prefix.is_empty() {
                            slug
                        } else {
                            format!("{}-{}", id_prefix, slug)
                        };
                        toc_items.push(TocItem::new(id, current_text.clone(), current_level));
                        heading_counter += 1;
                        in_heading = false;
                    }
                }
                Event::Text(text) if in_heading => {
                    current_text.push_str(&text);
                }
                _ => {}
            }
        }

        toc_items
    }
}