dmc-transform 0.3.6

Transform pipeline and built-in transformers for the dmc compiler
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

//! Heading id assignment. See `transformers/assign-heading-ids.md` for
//! full docs.

use crate::pipeline::Transformer;
use dmc_diagnostic::Code;
use dmc_diagnostic::metadata::SourceMeta;
use dmc_parser::Slugger;
use dmc_parser::ast::*;

#[derive(Default, Debug)]
pub struct AssignHeadingIds;

impl AssignHeadingIds {
  pub fn new() -> Self {
    Self
  }
}

impl Transformer for AssignHeadingIds {
  fn name(&self) -> &str {
    "assign-heading-ids"
  }

  fn transform(
    &self,
    doc: &mut Document,
    _meta: &SourceMeta,
    _diag_engine: &mut duck_diagnostic::DiagnosticEngine<Code>,
  ) {
    let mut slugger = Slugger::new();
    walk(&mut doc.children, &mut slugger);
  }
}

fn walk(nodes: &mut [Node], slugger: &mut Slugger) {
  for node in nodes {
    if let Node::Heading(h) = node {
      if h.id.is_none() {
        let text = Heading::plain_text(&h.children);
        h.id = Some(slugger.slug(&text));
      }
      // Headings can't contain other headings; no need to recurse.
      continue;
    }
    if let Some(children) = Node::children_of_mut(node) {
      walk(children, slugger);
    }
  }
}