diff options
Diffstat (limited to '')
-rw-r--r-- | askama_shared/Cargo.toml | 6 | ||||
-rw-r--r-- | askama_shared/src/lib.rs | 4 | ||||
-rw-r--r-- | askama_shared/src/path.rs | 32 |
3 files changed, 39 insertions, 3 deletions
diff --git a/askama_shared/Cargo.toml b/askama_shared/Cargo.toml index 8b18805..cd052cf 100644 --- a/askama_shared/Cargo.toml +++ b/askama_shared/Cargo.toml @@ -10,10 +10,12 @@ workspace = ".." [features] default = [] -serde-json = ["serde", "serde_json"] +serde-json = ["serde_json"] iron = [] rocket = [] [dependencies] -serde = { version = "1.0", optional = true } +serde = "1.0" +serde_derive = "1.0" serde_json = { version = "1.0", optional = true } +toml = "0.4" 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; |