Additional Theme

expikr · expikr · commit c7d58b356821 · 2023-04-09T18:23:01.000+08:00
diff --git a/src/config.rs b/src/config.rs
@@ -496,6 +496,8 @@ pub struct HtmlConfig {
     pub google_analytics: Option<String>,
     /// Additional CSS stylesheets to include in the rendered page's `<head>`.
     pub additional_css: Vec<PathBuf>,
+    /// Additional theme.
+    pub additional_theme: Vec<PathBuf>,
     /// Additional JS scripts to include at the bottom of the rendered page's
     /// `<body>`.
     pub additional_js: Vec<PathBuf>,
@@ -553,6 +555,7 @@ impl Default for HtmlConfig {
             copy_fonts: true,
             google_analytics: None,
             additional_css: Vec::new(),
+            additional_theme: Vec::new(),
             additional_js: Vec::new(),
             fold: Fold::default(),
             playground: Playground::default(),
@@ -742,6 +745,7 @@ mod tests {
         curly-quotes = true
         google-analytics = "123456"
         additional-css = ["./foo/bar/baz.css"]
+        additional-theme = ["foobar"]
         git-repository-url = "https://foo.com/"
         git-repository-icon = "fa-code-fork"
 
@@ -788,6 +792,7 @@ mod tests {
             curly_quotes: true,
             google_analytics: Some(String::from("123456")),
             additional_css: vec![PathBuf::from("./foo/bar/baz.css")],
+            additional_theme: vec![PathBuf::from("foobar")],
             theme: Some(PathBuf::from("./themedir")),
             default_theme: Some(String::from("rust")),
             playground: playground_should_be,
@@ -968,6 +973,7 @@ mod tests {
         curly-quotes = true
         google-analytics = "123456"
         additional-css = ["custom.css", "custom2.css"]
+        additional-theme = ["foobar"]
         additional-js = ["custom.js"]
         "#;
 
@@ -993,6 +999,7 @@ mod tests {
             curly_quotes: true,
             google_analytics: Some(String::from("123456")),
             additional_css: vec![PathBuf::from("custom.css"), PathBuf::from("custom2.css")],
+            additional_theme: vec![PathBuf::from("foobar")],
             additional_js: vec![PathBuf::from("custom.js")],
             ..Default::default()
         };
diff --git a/src/renderer/html_handlebars/hbs_renderer.rs b/src/renderer/html_handlebars/hbs_renderer.rs
@@ -698,6 +698,18 @@ fn make_data(
         data.insert("additional_css".to_owned(), json!(css));
     }
 
+    // Add check to see if there are additional themes
+    if !html_config.additional_theme.is_empty() {
+        let mut theme = Vec::new();
+        for name in &html_config.additional_theme {
+            match name.strip_prefix(root) {
+                Ok(p) => theme.push(p.to_str().expect("Could not convert to str")),
+                Err(_) => theme.push(name.to_str().expect("Could not convert to str")),
+            }
+        }
+        data.insert("additional_theme".to_owned(), json!(theme));
+    }
+
     // Add check to see if there is an additional script
     if !html_config.additional_js.is_empty() {
         let mut js = Vec::new();
