feat: add simple codeblock indentation

1 files changed, 16 insertions, 8 deletions

diff --git a/src/main.rs b/src/main.rs
index 13824df..67cea60 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -10,10 +10,17 @@ type Blocks<'a> = HashMap<&'a str,&'a str>;
 
 fn build(blocks: &Blocks, entry: &str) -> String {
     lazy_static! {
-        static ref RE: Regex = Regex::new(r"<<([^>\s]+)>>").unwrap();
+        static ref RE: Regex = Regex::new(r"([ \t]*)<<([^>\s]+)>>").unwrap();
     }
     if let Some(entry) = blocks.get(entry).clone() {
-        RE.replace_all(entry, |caps: &Captures| blocks.get(&caps[1]).expect("Block not present")).to_string()
+        RE.replace_all(entry, |caps: &Captures|
+            blocks.get(&caps[2])
+                  .expect("Block not present")
+                  .lines()
+                  .map(|l| format!("{}{}", &caps[1], l) )
+                  .collect::<Vec<_>>()
+                  .join("\n")
+        ).to_string()
     } else {
         String::from("")
     }
@@ -32,16 +39,17 @@ fn main() -> Result<()> {
         let mut blocks: Blocks = HashMap::new();
         pandoc.blocks.iter().for_each(|block|
             if let Block::CodeBlock(attr, code) = block {
+                dbg!(block);
                 if attr.0.len() > 0 {
                     blocks.insert(&attr.0, &code);
-                } /*else {
-                    println!("The following code has no ID:");
-                    code.lines().for_each(|l| println!("    {}", l));
-                }*/
+                } else {
+                    // println!("The following code has no ID:");
+                    // code.lines().for_each(|l| println!("    {}", l));
+                }
             }
         );
-        let program = build(&blocks, "main.rs");
-        println!("{}", program);
+        let program = build(dbg!(&blocks), "main.rs");
+        //println!("{}", program);
         pandoc
     }));
     pandoc.execute().unwrap();