metadata: Acquire the template spec during doc load if possible

Signed-off-by: Daniel Silverstone <dsilvers@digital-scurf.org>

1 files changed, 18 insertions, 5 deletions

diff --git a/src/metadata.rs b/src/metadata.rs
index 69544c4..b947854 100644
--- a/src/metadata.rs
+++ b/src/metadata.rs
@@ -1,5 +1,5 @@
-use crate::Bindings;
-use crate::Result;
+use crate::{resource, Result};
+use crate::{Bindings, TemplateSpec};
 
 use std::ops::Deref;
 use std::path::{Path, PathBuf};
@@ -15,6 +15,7 @@ pub struct Metadata {
     bindings: Bindings,
     functions_filenames: Vec<PathBuf>,
     template: Option<String>,
+    spec: Option<TemplateSpec>,
     bibliographies: Vec<PathBuf>,
     /// Extra class names which should be considered 'correct' for this document
     classes: Vec<String>,
@@ -30,7 +31,12 @@ impl Metadata {
         let date = get_date(&doc.meta);
         let bindings_filenames = get_bindings_filenames(basedir.as_ref(), &doc.meta);
         let functions_filenames = get_functions_filenames(basedir.as_ref(), &doc.meta);
-        let template = get_template_name(&doc.meta)?;
+        let (template, spec) = if let Some((template, spec)) = get_template_spec(&doc.meta)? {
+            resource::set_template(&template);
+            (Some(template), Some(spec))
+        } else {
+            (None, None)
+        };
         let mut bindings = Bindings::new();
 
         let bibliographies = get_bibliographies(basedir.as_ref(), &doc.meta);
@@ -44,6 +50,7 @@ impl Metadata {
             bindings,
             functions_filenames,
             template,
+            spec,
             bibliographies,
             classes,
         })
@@ -132,9 +139,15 @@ where
     get_paths(basedir, map, "functions")
 }
 
-fn get_template_name(map: &Mapp) -> Result<Option<String>> {
+fn get_template_spec(map: &Mapp) -> Result<Option<(String, TemplateSpec)>> {
     match get_string(map, "template") {
-        Some(s) => Ok(Some(s)),
+        Some(s) => {
+            let mut spec_path = PathBuf::from(&s);
+            spec_path.push("template");
+            spec_path.push("template.yaml");
+            let spec = TemplateSpec::from_file(&spec_path)?;
+            Ok(Some((s, spec)))
+        }
         None => Ok(None),
     }
 }