markdown_it/plugins/
sourcepos.rs

1//! Add source mapping to resulting HTML, looks like this: `<stuff data-sourcepos="1:1-2:3">`.
2//! ```rust
3//! let md = &mut markdown_it::MarkdownIt::new();
4//! markdown_it::plugins::cmark::add(md);
5//! markdown_it::plugins::sourcepos::add(md);
6//!
7//! let html = md.parse("# hello").render();
8//! assert_eq!(html.trim(), r#"<h1 data-sourcepos="1:1-1:7">hello</h1>"#);
9//! ```
10use crate::common::sourcemap::SourceWithLineStarts;
11use crate::parser::block::builtin::BlockParserRule;
12use crate::parser::core::{CoreRule, Root};
13use crate::parser::inline::builtin::InlineParserRule;
14use crate::{MarkdownIt, Node};
15
16pub fn add(md: &mut MarkdownIt) {
17    md.add_rule::<SyntaxPosRule>()
18        .after::<BlockParserRule>()
19        .after::<InlineParserRule>();
20}
21
22#[doc(hidden)]
23pub struct SyntaxPosRule;
24impl CoreRule for SyntaxPosRule {
25    fn run(root: &mut Node, _: &MarkdownIt) {
26        let source = root.cast::<Root>().unwrap().content.as_str();
27        let mapping = SourceWithLineStarts::new(source);
28
29        root.walk_mut(|node, _| {
30            if let Some(map) = node.srcmap {
31                let ((startline, startcol), (endline, endcol)) = map.get_positions(&mapping);
32                node.attrs.push(("data-sourcepos", format!("{}:{}-{}:{}", startline, startcol, endline, endcol)));
33            }
34        });
35    }
36}
37
38
39#[cfg(test)]
40mod tests {
41    #[test]
42    fn header_test() {
43        // same as doctest, keep in sync!
44        // used for code coverage and quicker rust-analyzer hints
45        let md = &mut crate::MarkdownIt::new();
46        crate::plugins::cmark::add(md);
47        crate::plugins::sourcepos::add(md);
48
49        let html = md.parse("# hello").render();
50        assert_eq!(html.trim(), r#"<h1 data-sourcepos="1:1-1:7">hello</h1>"#);
51    }
52}
markdown_it/plugins/sourcepos.rs

markdown_it/plugins/
sourcepos.rs
