Introduce the document structure.

src/parser/document.rs

@@ -1,29 +1,42 @@
-//! A single element of text.
-use super::combinator::context_many1;
-use super::error::Res;
-use super::paragraph::paragraph;
-use super::parser_context::ContextElement;
-use super::parser_context::ContextTree;
-use super::token::Paragraph;
-use super::token::Token;
-use super::Context;
-use nom::IResult;
+use super::element::Element;
+use super::source::Source;
 
-type UnboundMatcher<'r, 's, I, O, E> = dyn Fn(Context<'r, 's>, I) -> IResult<I, O, E>;
-
-// TODO: Implement FromStr for Document
-
-pub fn document(input: &str) -> Res<&str, Vec<Paragraph>> {
-    let initial_context: ContextTree<'_, '_> = ContextTree::new();
-    let document_context =
-        initial_context.with_additional_node(ContextElement::DocumentRoot(input));
-    let (remaining, tokens) = context_many1(&document_context, paragraph)(input)?;
-    let paragraphs = tokens
-        .into_iter()
-        .map(|token| match token {
-            Token::TextElement(_) => unreachable!(),
-            Token::Paragraph(paragraph) => paragraph,
-        })
-        .collect();
-    Ok((remaining, paragraphs))
+#[derive(Debug)]
+pub struct Document<'s> {
+    pub source: &'s str,
+    pub zeroth_section: Option<Section<'s>>,
+    pub children: Vec<Heading<'s>>,
+}
+
+#[derive(Debug)]
+pub struct Heading<'s> {
+    pub source: &'s str,
+    pub children: Vec<DocumentElement<'s>>,
+}
+
+#[derive(Debug)]
+pub struct Section<'s> {
+    pub source: &'s str,
+    pub children: Vec<Element<'s>>,
+}
+
+#[derive(Debug)]
+pub enum DocumentElement<'s> {
+    Heading(Heading<'s>),
+    Section(Section<'s>),
+}
+
+impl<'s> Source<'s> for Document<'s> {
+    fn get_source(&'s self) -> &'s str {
+        self.source
+    }
+}
+
+impl<'s> Source<'s> for DocumentElement<'s> {
+    fn get_source(&'s self) -> &'s str {
+        match self {
+            DocumentElement::Heading(obj) => obj.source,
+            DocumentElement::Section(obj) => obj.source,
+        }
+    }
 }