Implement get_additional_properties for Document.

This commit is contained in:
Tom Alexander 2023-10-11 16:14:03 -04:00
parent 42dcd41e48
commit d965dd6fd1
Signed by: talexander
GPG Key ID: D3A179C9A53C0EDE
1 changed files with 24 additions and 4 deletions

View File

@ -93,8 +93,7 @@ impl<'s> Heading<'s> {
}
pub fn get_additional_properties(&self) -> impl Iterator<Item = &NodeProperty<'s>> {
let foo = self
.children
self.children
.iter()
.take(1)
.filter_map(|c| match c {
@ -107,7 +106,28 @@ impl<'s> Heading<'s> {
Element::PropertyDrawer(property_drawer) => Some(property_drawer),
_ => None,
})
.flat_map(|property_drawer| property_drawer.children.iter());
foo
.flat_map(|property_drawer| property_drawer.children.iter())
}
}
impl<'s> Document<'s> {
pub fn get_additional_properties(&self) -> impl Iterator<Item = &NodeProperty<'s>> {
let zeroth_section_children = self
.zeroth_section
.iter()
.flat_map(|zeroth_section| zeroth_section.children.iter());
let property_drawer = zeroth_section_children
.take_while(|element| match element {
Element::Comment(_) => true,
Element::PropertyDrawer(_) => true,
_ => false,
})
.find_map(|element| match element {
Element::PropertyDrawer(property_drawer) => Some(property_drawer),
_ => None,
});
property_drawer
.into_iter()
.flat_map(|property_drawer| property_drawer.children.iter())
}
}