markdown-ppp 2.9.2

Feature-rich Markdown Parsing and Pretty-Printing library
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
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
117
118
119
120
121
122
123
124
125
126

use crate::ast::*;
use std::collections::HashMap;

struct FootnoteIndex {
    indices: HashMap<String, usize>,
    counter: usize,
}

impl FootnoteIndex {
    fn new() -> Self {
        FootnoteIndex {
            indices: HashMap::new(),
            counter: 1,
        }
    }

    fn add(&mut self, label: String) {
        if let std::collections::hash_map::Entry::Vacant(e) = self.indices.entry(label) {
            e.insert(self.counter);
            self.counter += 1;
        }
    }
}

/// Build footnote indices ordered by first reference appearance in inline content,
/// and collect link definitions.
///
/// Unlike `ast::index::get_footnote_indices` which orders by definition position,
/// this orders by the first `FootnoteReference` encountered in inline traversal.
pub(crate) fn get_indices(
    ast: &Document,
) -> (HashMap<String, usize>, HashMap<Vec<Inline>, LinkDefinition>) {
    let mut footnotes = FootnoteIndex::new();

    for block in &ast.blocks {
        collect_footnote_refs(&mut footnotes, block);
    }

    let link_definitions = crate::ast::index::get_link_definitions(ast);

    (footnotes.indices, link_definitions)
}

fn collect_footnote_refs(footnotes: &mut FootnoteIndex, block: &Block) {
    match block {
        Block::Paragraph(v) => {
            for inline in v {
                collect_inline_refs(footnotes, inline);
            }
        }
        Block::Heading(v) => {
            for inline in &v.content {
                collect_inline_refs(footnotes, inline);
            }
        }
        Block::BlockQuote(v) => {
            for block in v {
                collect_footnote_refs(footnotes, block);
            }
        }
        Block::List(v) => {
            for item in &v.items {
                for block in &item.blocks {
                    collect_footnote_refs(footnotes, block);
                }
            }
        }
        Block::Definition(v) => {
            for inline in &v.label {
                collect_inline_refs(footnotes, inline);
            }
        }
        Block::Table(v) => {
            for row in &v.rows {
                for cell in row {
                    for inline in cell {
                        collect_inline_refs(footnotes, inline);
                    }
                }
            }
        }
        Block::FootnoteDefinition(v) => {
            for block in &v.blocks {
                collect_footnote_refs(footnotes, block);
            }
        }
        Block::GitHubAlert(alert) => {
            for block in &alert.blocks {
                collect_footnote_refs(footnotes, block);
            }
        }
        Block::ThematicBreak | Block::CodeBlock(_) | Block::HtmlBlock(_) | Block::Empty => (),
    }
}

fn collect_inline_refs(footnotes: &mut FootnoteIndex, inline: &Inline) {
    match inline {
        Inline::FootnoteReference(label) => {
            footnotes.add(label.clone());
        }
        Inline::Emphasis(children)
        | Inline::Strong(children)
        | Inline::Strikethrough(children) => {
            for child in children {
                collect_inline_refs(footnotes, child);
            }
        }
        Inline::Link(Link { children, .. }) => {
            for child in children {
                collect_inline_refs(footnotes, child);
            }
        }
        Inline::LinkReference(LinkReference { text, .. }) => {
            for child in text {
                collect_inline_refs(footnotes, child);
            }
        }
        Inline::Text(_)
        | Inline::LineBreak
        | Inline::Code(_)
        | Inline::Html(_)
        | Inline::Image(_)
        | Inline::Autolink(_)
        | Inline::Empty => {}
    }
}