196 lines
6.9 KiB
Rust
196 lines
6.9 KiB
Rust
|
use indextree::NodeId;
|
||
|
use std::ops::RangeInclusive;
|
||
|
|
||
|
use crate::elements::{Element, Table, TableRow, Title};
|
||
|
use crate::Org;
|
||
|
|
||
|
/// Validation Error
|
||
|
#[derive(Debug)]
|
||
|
pub enum ValidationError {
|
||
|
/// Expected at least one child
|
||
|
ExpectedChildren {
|
||
|
at: NodeId,
|
||
|
},
|
||
|
/// Expected no children
|
||
|
UnexpectedChildren {
|
||
|
at: NodeId,
|
||
|
},
|
||
|
UnexpectedElement {
|
||
|
expected: &'static str,
|
||
|
at: NodeId,
|
||
|
},
|
||
|
/// Expect a detached element
|
||
|
ExpectedDetached {
|
||
|
at: NodeId,
|
||
|
},
|
||
|
HeadlineLevelMismatch {
|
||
|
range: RangeInclusive<usize>,
|
||
|
at: NodeId,
|
||
|
},
|
||
|
}
|
||
|
|
||
|
impl ValidationError {
|
||
|
pub fn element<'a, 'b>(&self, org: &'a Org<'b>) -> &'a Element<'b> {
|
||
|
match &self {
|
||
|
ValidationError::ExpectedChildren { at }
|
||
|
| ValidationError::UnexpectedChildren { at }
|
||
|
| ValidationError::UnexpectedElement { at, .. }
|
||
|
| ValidationError::ExpectedDetached { at }
|
||
|
| ValidationError::HeadlineLevelMismatch { at, .. } => org.arena[*at].get(),
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub type ValidationResult<T> = Result<T, ValidationError>;
|
||
|
|
||
|
impl Org<'_> {
|
||
|
/// Validate an `Org` struct.
|
||
|
pub fn validate(&self) -> Vec<ValidationError> {
|
||
|
let mut errors = Vec::new();
|
||
|
|
||
|
macro_rules! expect {
|
||
|
($node:ident, $expect:expr, $($pattern:pat)|+) => {
|
||
|
match self.arena[$node].get() {
|
||
|
$($pattern)|+ => (),
|
||
|
_ => errors.push(ValidationError::UnexpectedElement {
|
||
|
expected: $expect,
|
||
|
at: $node
|
||
|
}),
|
||
|
}
|
||
|
};
|
||
|
}
|
||
|
|
||
|
for node_id in self.root.descendants(&self.arena) {
|
||
|
let node = &self.arena[node_id];
|
||
|
match node.get() {
|
||
|
Element::Document => {
|
||
|
let mut children = node_id.children(&self.arena);
|
||
|
if let Some(node) = children.next() {
|
||
|
expect!(
|
||
|
node,
|
||
|
"Headline,Section",
|
||
|
Element::Headline { .. } | Element::Section
|
||
|
);
|
||
|
}
|
||
|
for node in children {
|
||
|
expect!(
|
||
|
node,
|
||
|
"Headline",
|
||
|
Element::Headline { .. }
|
||
|
);
|
||
|
}
|
||
|
}
|
||
|
Element::Headline { .. } => {
|
||
|
if node.first_child().is_some() {
|
||
|
let mut children = node_id.children(&self.arena);
|
||
|
if let Some(node) = children.next() {
|
||
|
expect!(node, "Title", Element::Title(_));
|
||
|
}
|
||
|
if let Some(node) = children.next() {
|
||
|
expect!(
|
||
|
node,
|
||
|
"Headline,Section",
|
||
|
Element::Headline { .. } | Element::Section
|
||
|
);
|
||
|
}
|
||
|
for node in children {
|
||
|
expect!(
|
||
|
node,
|
||
|
"Headline",
|
||
|
Element::Headline { .. }
|
||
|
);
|
||
|
}
|
||
|
} else {
|
||
|
errors.push(ValidationError::ExpectedChildren { at: node_id });
|
||
|
}
|
||
|
}
|
||
|
Element::Title(Title { raw, .. }) => {
|
||
|
if !raw.is_empty() && node.first_child().is_none() {
|
||
|
errors.push(ValidationError::ExpectedChildren { at: node_id });
|
||
|
}
|
||
|
}
|
||
|
Element::CommentBlock(_)
|
||
|
| Element::ExampleBlock(_)
|
||
|
| Element::ExportBlock(_)
|
||
|
| Element::SourceBlock(_)
|
||
|
| Element::BabelCall(_)
|
||
|
| Element::InlineSrc(_)
|
||
|
| Element::Code { .. }
|
||
|
| Element::FnRef(_)
|
||
|
| Element::InlineCall(_)
|
||
|
| Element::Link(_)
|
||
|
| Element::Macros(_)
|
||
|
| Element::RadioTarget
|
||
|
| Element::Snippet(_)
|
||
|
| Element::Target(_)
|
||
|
| Element::Text { .. }
|
||
|
| Element::Timestamp(_)
|
||
|
| Element::Verbatim { .. }
|
||
|
| Element::FnDef(_)
|
||
|
| Element::Clock(_)
|
||
|
| Element::Comment { .. }
|
||
|
| Element::FixedWidth { .. }
|
||
|
| Element::Keyword(_)
|
||
|
| Element::Rule
|
||
|
| Element::Cookie(_)
|
||
|
| Element::Table(Table::TableEl { .. })
|
||
|
| Element::TableRow(TableRow::Rule) => {
|
||
|
if node.first_child().is_some() {
|
||
|
errors.push(ValidationError::UnexpectedChildren { at: node_id });
|
||
|
}
|
||
|
}
|
||
|
Element::List(_) => {
|
||
|
if node.first_child().is_some() {
|
||
|
for node in node_id.children(&self.arena) {
|
||
|
expect!(node, "ListItem", Element::ListItem(_));
|
||
|
}
|
||
|
} else {
|
||
|
errors.push(ValidationError::ExpectedChildren { at: node_id });
|
||
|
}
|
||
|
}
|
||
|
Element::SpecialBlock(_)
|
||
|
| Element::QuoteBlock(_)
|
||
|
| Element::CenterBlock(_)
|
||
|
| Element::VerseBlock(_)
|
||
|
| Element::Paragraph
|
||
|
| Element::Section
|
||
|
| Element::Table(Table::Org { .. })
|
||
|
| Element::TableRow(TableRow::Standard)
|
||
|
| Element::Bold
|
||
|
| Element::Italic
|
||
|
| Element::Underline
|
||
|
| Element::Strike
|
||
|
| Element::DynBlock(_)
|
||
|
| Element::ListItem(_) => {
|
||
|
if node.first_child().is_none() {
|
||
|
errors.push(ValidationError::ExpectedChildren { at: node_id });
|
||
|
}
|
||
|
}
|
||
|
// TableCell is a container but it might
|
||
|
// not contains anything, e.g. `||||||`
|
||
|
Element::Drawer(_) | Element::TableCell => (),
|
||
|
}
|
||
|
}
|
||
|
errors
|
||
|
}
|
||
|
|
||
|
#[deprecated(since = "0.3.1", note = "rename to validate")]
|
||
|
/// Validate an `Org` struct.
|
||
|
pub fn check(&self) -> Vec<ValidationError> {
|
||
|
self.validate()
|
||
|
}
|
||
|
|
||
|
pub(crate) fn debug_validate(&self) {
|
||
|
if cfg!(debug_assertions) {
|
||
|
let errors = self.validate();
|
||
|
if !errors.is_empty() {
|
||
|
eprintln!("Validation failed. {} error(s) found:", errors.len());
|
||
|
for err in &errors {
|
||
|
eprintln!("{:?} at {:?}", err, err.element(self));
|
||
|
}
|
||
|
panic!();
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|