search for nested includes

src/lib.rs

@@ -1,12 +1,17 @@
-use std::path::PathBuf;
+use std::{
+    fs::{self, read, File},
+    io::BufReader,
+    path::{Path, PathBuf},
+};
 
 use once_cell::sync::Lazy;
 use regex::Regex;
 use roxy_core::roxy::Parse;
 
 const DEFAULT_CONTEXT: Lazy<tera::Context> = Lazy::new(|| tera::Context::default());
-const EXPANSION_RE: Lazy<Regex> =
-    Lazy::new(|| Regex::new("\\{% extends \"?(.+?)\"? %\\}").expect("couldn't load regex"));
+const EXPANSION_RE: Lazy<Regex> = Lazy::new(|| {
+    Regex::new("\\{% (extends|include) \"?(.+?)\"? %\\}").expect("couldn't load regex")
+});
 
 #[derive(Debug)]
 pub struct TeraParserOptions {
@@ -41,18 +46,24 @@ impl<'a> TeraParser<'a> {
         self.context = Some(context);
     }
 
-    fn load_template(&mut self, path: &str, src: &[u8]) -> Result<(), tera::Error> {
+    fn load_template<P: AsRef<Path>>(&mut self, path: &P, src: &[u8]) -> Result<(), tera::Error> {
+        let path = path.as_ref().canonicalize()?;
         let str = String::from_utf8_lossy(src).to_string();
-        if let Some(captures) = EXPANSION_RE.captures(&str.as_str()) {
-            if let Some(layout_path) = captures.get(1) {
-                let layout_path = layout_path.as_str();
-                let path = PathBuf::from(path).parent().map(|p| p.join(layout_path)).unwrap();
+        for (_, [_, layout_path]) in EXPANSION_RE
+            .captures_iter(&str.as_str())
+            .map(|c| c.extract())
+        {
+            let path = path
+                .canonicalize()?
+                .parent()
+                .map(|p| p.join(layout_path))
+                .unwrap();
 
-                self.tera.add_template_file(
-                    &path.canonicalize().unwrap(),
-                    Some(layout_path)
-                )?;
-            }
+            let next_template = fs::read(&path)?;
+            self.load_template(&path, &next_template)?;
+
+            self.tera
+                .add_template_file(&path, Some(layout_path))?;
         }
 
         Ok(())
@@ -60,7 +71,12 @@ impl<'a> TeraParser<'a> {
 }
 
 impl<'a> Parse for TeraParser<'a> {
-    fn parse(&mut self, path: &str, src: &[u8], dst: &mut Vec<u8>) -> Result<(), roxy_core::error::Error> {
+    fn parse(
+        &mut self,
+        path: &str,
+        src: &[u8],
+        dst: &mut Vec<u8>,
+    ) -> Result<(), roxy_core::error::Error> {
         // TODO: This error is a hack
         let err = |e: tera::Error| {
             println!("{e:?}");
@@ -68,7 +84,7 @@ impl<'a> Parse for TeraParser<'a> {
         };
 
         if self.options.auto_load_layouts {
-            self.load_template(path, src).map_err(err)?;
+            self.load_template(&path, src).map_err(err)?;
         }
 
         let template = String::from_utf8_lossy(src).to_string();
@@ -87,6 +103,19 @@ impl<'a> Parse for TeraParser<'a> {
 
 #[cfg(test)]
 mod tests {
-    #[test]
-    fn it_works() {}
+    use std::fs;
+
+    use roxy_core::roxy::Parse;
+
+    use crate::{TeraParser, TeraParserOptions};
+
+    //#[test]
+    //fn capture_test() {
+    //    let mut tera = tera::Tera::default();
+    //    let mut parser = TeraParser::new(&mut tera, TeraParserOptions::default());
+    //    let data = fs::read("./tests/index.md").unwrap();
+    //    let mut dst = Vec::new();
+    //    parser.parse("./tests/index.md", &data, &mut dst);
+    //    println!("{dst:?}");
+    //}
 }

tests/base.html

@@ -0,0 +1 @@
+:3

tests/index.md

@@ -0,0 +1,2 @@
+{% extends "middle.html" %}
+

tests/middle.html

@@ -0,0 +1,2 @@
+{% extends "base.html" %}
+