Feeding the templates into the renderer integration.

src/command/build/render.rs

@@ -1,19 +1,51 @@
+use std::ffi::OsStr;
 use std::path::PathBuf;
 
+use include_dir::include_dir;
+use include_dir::Dir;
+
 use crate::blog_post::convert_blog_post_to_render_context;
 use crate::blog_post::BlogPost;
 use crate::error::CustomError;
 use crate::render::DusterRenderer;
 use crate::render::RendererIntegration;
 
+static MAIN_TEMPLATES: Dir = include_dir!("$CARGO_MANIFEST_DIR/default_environment/templates/html");
+
 pub(crate) struct SiteRenderer {
-    pub(crate) output_directory: PathBuf,
-    pub(crate) blog_posts: Vec<BlogPost>,
+    output_directory: PathBuf,
+    blog_posts: Vec<BlogPost>,
 }
 
 impl SiteRenderer {
+    pub(crate) fn new<P: Into<PathBuf>>(
+        output_directory: P,
+        blog_posts: Vec<BlogPost>,
+    ) -> SiteRenderer {
+        SiteRenderer {
+            output_directory: output_directory.into(),
+            blog_posts,
+        }
+    }
     pub(crate) async fn render_blog_posts(&self) -> Result<(), CustomError> {
-        let mut renderer_integration = DusterRenderer {};
+        let mut renderer_integration = DusterRenderer::new();
+
+        let (main_template, other_templates): (Vec<_>, Vec<_>) = MAIN_TEMPLATES
+            .files()
+            .filter(|f| f.path().extension() == Some(OsStr::new("dust")))
+            .partition(|f| f.path().file_stem() == Some(OsStr::new("main")));
+        if main_template.len() != 1 {
+            return Err("Expect exactly 1 main.dust template file.".into());
+        }
+
+        for entry in main_template {
+            load_template_from_include_dir(&mut renderer_integration, entry)?;
+        }
+
+        for entry in other_templates {
+            load_template_from_include_dir(&mut renderer_integration, entry)?;
+        }
+
         for blog_post in &self.blog_posts {
             let render_context = convert_blog_post_to_render_context(blog_post);
             renderer_integration.render(render_context)?;
@@ -22,3 +54,18 @@ impl SiteRenderer {
         Ok(())
     }
 }
+
+fn load_template_from_include_dir<RI: RendererIntegration>(
+    renderer_integration: &mut RI,
+    entry: &include_dir::File<'_>,
+) -> Result<(), CustomError> {
+    let path = entry.path();
+    let name = path
+        .file_stem()
+        .ok_or("All templates should have a stem.")?
+        .to_str()
+        .ok_or("All template filenames should be valid utf-8.")?;
+    let contents = std::str::from_utf8(entry.contents())?;
+    renderer_integration.load_template(name, contents)?;
+    Ok(())
+}

src/command/build/runner.rs

@@ -9,10 +9,7 @@ use crate::error::CustomError;
 pub(crate) async fn build_site(args: BuildArgs) -> Result<(), CustomError> {
     let config = Config::load_from_file(args.config).await?;
     let blog_posts = load_blog_posts(&config).await?;
-    let renderer = SiteRenderer {
-        output_directory: get_output_directory(&config).await?,
-        blog_posts,
-    };
+    let renderer = SiteRenderer::new(get_output_directory(&config).await?, blog_posts);
     renderer.render_blog_posts().await?;
 
     Ok(())