Add partial support for multiple template dirs

2 files changed, 35 insertions, 1 deletions

diff --git a/askama_shared/src/lib.rs b/askama_shared/src/lib.rs
index 9427dff..64d8d25 100644
--- a/askama_shared/src/lib.rs
+++ b/askama_shared/src/lib.rs
@@ -1,9 +1,11 @@
 #![cfg_attr(feature = "cargo-clippy", allow(unused_parens))]
 
-#[cfg(feature = "serde-json")]
 extern crate serde;
+#[macro_use]
+extern crate serde_derive;
 #[cfg(feature = "serde-json")]
 extern crate serde_json;
+extern crate toml;
 
 pub use error::{Error, Result};
 pub use escaping::MarkupDisplay;
diff --git a/askama_shared/src/path.rs b/askama_shared/src/path.rs
index 8b92250..37fb261 100644
--- a/askama_shared/src/path.rs
+++ b/askama_shared/src/path.rs
@@ -3,6 +3,8 @@ use std::fs::File;
 use std::io::Read;
 use std::path::{Path, PathBuf};
 
+use toml;
+
 pub fn get_template_source(tpl_path: &Path) -> String {
     let mut path = template_dir();
     path.push(tpl_path);
@@ -52,6 +54,36 @@ pub fn template_dir() -> PathBuf {
     path
 }
 
+#[derive(Deserialize)]
+struct Config {
+    dirs: Option<Vec<String>>,
+}
+
+pub fn template_dirs() -> Vec<PathBuf> {
+    let root = PathBuf::from(env::var("CARGO_MANIFEST_DIR").unwrap());
+    let askama = root.join("askama.toml");
+    let default = vec![root.join("templates")];
+    if askama.exists() {
+        let mut contents = String::new();
+        File::open(askama).unwrap().read_to_string(&mut contents).unwrap();
+        let config: Config = toml::from_str(&contents).unwrap();
+        if let Some(dirs) = config.dirs {
+            let paths: Vec<PathBuf> = dirs.into_iter().map(|directory| {
+                root.join(directory)
+            }).collect();
+            if paths.len() > 0 {
+                paths
+            } else {
+                default
+            }
+        } else {
+            default
+        }
+    } else {
+        vec![root.join("templates")]
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use super::Path;