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<_>>()
}