From 9a7afe9f8271ccf199d361719da6d085ee4b19aa Mon Sep 17 00:00:00 2001
From: Nadir Fejzic <nadirfejzo@gmail.com>
Date: Thu, 6 Jan 2022 22:41:54 +0100
Subject: [PATCH] feat(frontend): setup parser for enclosed blocks

---
 src/frontend/parser.rs | 31 +++++++++++++++++++++++++++----
 1 file changed, 27 insertions(+), 4 deletions(-)

diff --git a/src/frontend/parser.rs b/src/frontend/parser.rs
index 91bedc28..8e3aaee6 100644
--- a/src/frontend/parser.rs
+++ b/src/frontend/parser.rs
@@ -4,7 +4,11 @@ use pest::{iterators::Pair, iterators::Pairs, Parser, Span};
 use pest_derive::Parser;
 use std::{fs, path::Path};
 
-use crate::{um_elements::HeadingBlock, um_elements::ParagraphBlock, um_error::UmError};
+use crate::{
+    um_elements::HeadingBlock,
+    um_elements::{ParagraphBlock, VerbatimBlock},
+    um_error::UmError,
+};
 
 use super::UnimarkupBlocks;
 
@@ -65,9 +69,17 @@ pub fn parse_unimarkup(um_file: &Path) -> Result<UnimarkupBlocks, UmError> {
 
     if let Some(unimarkup) = rule_pairs.next() {
         for pair in unimarkup.into_inner() {
-            if pair.as_rule() == Rule::atomic_block {
-                let mut atomic_blocks = parse_atomic_block(pair)?;
-                blocks.append(&mut atomic_blocks);
+            match pair.as_rule() {
+                Rule::atomic_block => {
+                    let mut atomic_blocks = parse_atomic_block(pair)?;
+                    blocks.append(&mut atomic_blocks);
+                }
+                Rule::enclosed_block => {
+                    let mut enclosed_blocks = parse_enclosed_block(pair)?;
+
+                    blocks.append(&mut enclosed_blocks);
+                }
+                _ => unreachable!("Unimarkup consists only of atomic and enclosed blocks."),
             }
         }
     }
@@ -84,3 +96,14 @@ fn parse_atomic_block(input: Pair<Rule>) -> Result<UnimarkupBlocks, UmError> {
 
     Ok(vec![])
 }
+
+fn parse_enclosed_block(input: Pair<Rule>) -> Result<UnimarkupBlocks, UmError> {
+    if let Ok(ref mut pairs) = UnimarkupParser::parse(Rule::verbatim, input.as_str()) {
+        return VerbatimBlock::parse(pairs, input.as_span());
+    } else if let Ok(ref mut pairs) = UnimarkupParser::parse(Rule::paragraph, input.as_str()) {
+        // fallback to paragraph for now
+        return ParagraphBlock::parse(pairs, input.as_span());
+    }
+
+    Ok(vec![])
+}