feat! allow multiple markdown files for a subplot

Sponsored-by: author

1 files changed, 51 insertions, 17 deletions

diff --git a/src/doc.rs b/src/doc.rs
index 0b0b9e7..640968c 100644
--- a/src/doc.rs
+++ b/src/doc.rs
@@ -66,7 +66,7 @@ static KNOWN_BLOCK_CLASSES: &[&str] = &["numberLines", "noNumberLines"];
 #[derive(Debug)]
 pub struct Document {
     subplot: PathBuf,
-    md: Markdown,
+    markdowns: Vec<Markdown>,
     meta: Metadata,
     files: EmbeddedFiles,
     style: Style,
@@ -75,14 +75,14 @@ pub struct Document {
 impl Document {
     fn new(
         subplot: PathBuf,
-        md: Markdown,
+        markdowns: Vec<Markdown>,
         meta: Metadata,
         files: EmbeddedFiles,
         style: Style,
     ) -> Document {
         let doc = Document {
             subplot,
-            md,
+            markdowns,
             meta,
             files,
             style,
@@ -95,7 +95,7 @@ impl Document {
         basedir: P,
         subplot: PathBuf,
         yamlmeta: &YamlMetadata,
-        md: Markdown,
+        markdowns: Vec<Markdown>,
         style: Style,
         template: Option<&str>,
     ) -> Result<Document, SubplotError>
@@ -104,12 +104,22 @@ impl Document {
     {
         let meta = Metadata::from_yaml_metadata(basedir, yamlmeta, template)?;
         trace!("metadata from YAML: {:#?}", meta);
-        let files = md.embedded_files();
-        let doc = Document::new(subplot, md, meta, files, style);
+        let files = Self::all_files(&markdowns);
+        let doc = Document::new(subplot, markdowns, meta, files, style);
         trace!("Loaded from JSON OK");
         Ok(doc)
     }
 
+    fn all_files(markdowns: &[Markdown]) -> EmbeddedFiles {
+        let mut files = EmbeddedFiles::default();
+        for md in markdowns {
+            for file in md.embedded_files().files() {
+                files.push(file.clone());
+            }
+        }
+        files
+    }
+
     /// Construct a Document from a named file.
     pub fn from_file(
         basedir: &Path,
@@ -126,11 +136,13 @@ impl Document {
         let meta = load_metadata_from_yaml_file(filename)?;
         trace!("metadata from YAML file: {:#?}", meta);
 
-        let mdfile = meta.markdown();
-        let mdfile = basedir.join(mdfile);
-        let md = Markdown::load_file(mdfile.as_path())?;
+        let mut markdowns = vec![];
+        for filename in meta.markdowns() {
+            let filename = basedir.join(filename);
+            markdowns.push(Markdown::load_file(&filename)?);
+        }
 
-        let doc = Self::from_ast(basedir, filename.into(), &meta, md, style, template)?;
+        let doc = Self::from_ast(basedir, filename.into(), &meta, markdowns, style, template)?;
 
         trace!("Loaded document OK");
         Ok(doc)
@@ -146,7 +158,9 @@ impl Document {
         self.meta.set_date(date.into());
 
         let mut body = self.typeset_meta();
-        body.push_child(Content::Elt(self.md.root_element().clone()));
+        for md in self.markdowns.iter() {
+            body.push_child(Content::Elt(md.root_element().clone()));
+        }
         let page = HtmlPage::new(head, body);
         page.serialize().unwrap() // FIXME
     }
@@ -230,10 +244,14 @@ impl Document {
             names.push(PathBuf::from(x))
         }
 
-        names.push(self.meta().markdown_filename().to_path_buf());
+        for name in self.meta().markdown_filenames() {
+            names.push(name.into());
+        }
 
-        let mut images = self.md.images();
-        names.append(&mut images);
+        for md in self.markdowns.iter() {
+            let mut images = md.images();
+            names.append(&mut images);
+        }
 
         names
     }
@@ -276,7 +294,7 @@ impl Document {
 
     /// Check that all the block classes in the document are known
     fn check_block_classes(&self) -> Result<(), SubplotError> {
-        let classes_in_doc = self.md.block_classes();
+        let classes_in_doc = self.all_block_classes();
 
         // Build the set of known good classes
         let mut known_classes: HashSet<String> = HashSet::new();
@@ -299,6 +317,16 @@ impl Document {
         }
     }
 
+    fn all_block_classes(&self) -> HashSet<String> {
+        let mut set = HashSet::new();
+        for md in self.markdowns.iter() {
+            for class in md.block_classes() {
+                set.insert(class);
+            }
+        }
+        set
+    }
+
     /// Check that all named files (in matched steps) are actually present in the
     /// document.
     pub fn check_named_files_exist(
@@ -402,12 +430,18 @@ impl Document {
 
     /// Typeset a Subplot document.
     pub fn typeset(&mut self, warnings: &mut Warnings) {
-        warnings.push_all(self.md.typeset(self.style.clone(), self.meta.bindings()));
+        for md in self.markdowns.iter_mut() {
+            warnings.push_all(md.typeset(self.style.clone(), self.meta.bindings()));
+        }
     }
 
     /// Return all scenarios in a document.
     pub fn scenarios(&self) -> Result<Vec<Scenario>, SubplotError> {
-        self.md.scenarios()
+        let mut scenarios = vec![];
+        for md in self.markdowns.iter() {
+            scenarios.append(&mut md.scenarios()?);
+        }
+        Ok(scenarios)
     }
 
     /// Return matched scenarios in a document.