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

//! Node transformations.
use std::iter::FromIterator;
use std::collections::{HashSet, HashMap};
use std::rc::Rc;
use std::borrow::Cow;
use serde::{Serialize, Deserialize};
use crate::frontend::data::{
    Atom,
    Text,
    Enclosure,
    EnclosureKind,
    INLINE_MATH_TAG,
    RewriteRule,
};
use crate::frontend::ast::{Node, Tag};


#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
pub enum HeadingKind {
    H1,
    H2,
    H3,
    H4,
    H5,
    H6,
}

impl HeadingKind {
    pub fn from_str(x: &str) -> Option<Self> {
        match x {
            "h1" => Some(HeadingKind::H1),
            "h2" => Some(HeadingKind::H2),
            "h3" => Some(HeadingKind::H3),
            "h4" => Some(HeadingKind::H4),
            "h5" => Some(HeadingKind::H5),
            "h6" => Some(HeadingKind::H6),
            _ => None
        }
    }
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Heading {
    kind: HeadingKind,
    text: String,
}

pub fn query_heading_nodes<'a>(node: &Node<'a>) -> Vec<Heading> {
    pub fn go<'a>(node: &Node<'a>) -> Vec<Tag<'a>> {
        match node {
            Node::Tag(tag) if tag.has_name("h1") => vec![tag.clone()],
            Node::Tag(tag) if tag.has_name("h2") => vec![tag.clone()],
            Node::Tag(tag) if tag.has_name("h3") => vec![tag.clone()],
            Node::Tag(tag) if tag.has_name("h4") => vec![tag.clone()],
            Node::Tag(tag) if tag.has_name("h5") => vec![tag.clone()],
            Node::Tag(tag) if tag.has_name("h6") => vec![tag.clone()],
            Node::Tag(node) => {
                node.children
                    .iter()
                    .flat_map(go)
                    .collect::<Vec<_>>()
            }
            Node::Enclosure(node) => {
                node.data.children
                    .iter()
                    .flat_map(go)
                    .collect::<Vec<_>>()
            }
            Node::Ident(..) => Vec::new(),
            Node::String(..) => Vec::new(),
            Node::InvalidToken(..) => Vec::new(),
        }
    }
    go(node)
        .into_iter()
        .map(|tag| {
            let kind = HeadingKind::from_str(tag.name()).unwrap();
            let text = tag.children
                .iter()
                .map(Node::to_string)
                .collect::<Vec<_>>()
                .join(" ");
            Heading{kind, text}
        })
        .collect::<Vec<_>>()
}