From a33e225e71ceeb31944d13f04291bcb1d69f488f Mon Sep 17 00:00:00 2001
From: Marcus Ball <marcus@marcusball.me>
Date: Wed, 28 Dec 2016 22:29:14 -0500
Subject: [PATCH] Properly resolve nested template names on Windows.

---
 contrib/src/templates/mod.rs | 77 +++++++++++++++++++++++++++++++++---
 1 file changed, 71 insertions(+), 6 deletions(-)

diff --git a/contrib/src/templates/mod.rs b/contrib/src/templates/mod.rs
index 28539371..82214a81 100644
--- a/contrib/src/templates/mod.rs
+++ b/contrib/src/templates/mod.rs
@@ -185,6 +185,23 @@ fn remove_extension<P: AsRef<Path>>(path: P) -> PathBuf {
     }
 }
 
+/// Splits a path into a relative path from TEMPLATE_DIR, a name that
+/// may be used to identify the template, and the template's data type.
+fn split_path(path: &Path) -> (PathBuf, String, Option<String>) {
+    let rel_path = path.strip_prefix(&*TEMPLATE_DIR).unwrap().to_path_buf();
+    let path_no_ext = remove_extension(&rel_path);
+    let data_type = path_no_ext.extension();
+    let mut name = remove_extension(&path_no_ext).to_string_lossy().into_owned();
+
+    // Ensure template name consistency on Windows systems
+    if cfg!(windows) {
+        name = name.replace("\\", "/");
+    }
+
+    (rel_path, name, data_type.map(|d| d.to_string_lossy().into_owned()))
+}
+
+
 /// Returns a HashMap of `TemplateInfo`'s for all of the templates in
 /// `TEMPLATE_DIR`. Templates are all files that match one of the extensions for
 /// engine's in `engine_set`.
@@ -200,15 +217,12 @@ fn discover_templates() -> HashMap<String, TemplateInfo> {
         let mut glob_path: PathBuf = [&*TEMPLATE_DIR, "**", "*"].iter().collect();
         glob_path.set_extension(ext);
         for path in glob(glob_path.to_str().unwrap()).unwrap().filter_map(Result::ok) {
-            let rel_path = path.strip_prefix(&*TEMPLATE_DIR).unwrap().to_path_buf();
-            let path_no_ext = remove_extension(&rel_path);
-            let data_type = path_no_ext.extension();
-            let name = remove_extension(&path_no_ext);
-            templates.insert(name.to_string_lossy().into_owned(), TemplateInfo {
+            let (rel_path, name, data_type) = split_path(&path);
+            templates.insert(name, TemplateInfo {
                 full_path: path.to_path_buf(),
                 path: rel_path,
                 extension: path.extension().unwrap().to_string_lossy().into_owned(),
-                data_type: data_type.map(|d| d.to_string_lossy().into_owned())
+                data_type: data_type,
             });
         }
     }
@@ -216,3 +230,54 @@ fn discover_templates() -> HashMap<String, TemplateInfo> {
     templates
 }
 
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    /// Combines a `relative_path` and the `TEMPLATE_DIR` path into a full path.
+    fn template_path(relative_path: &str) -> PathBuf {
+        let mut path = PathBuf::from(&*TEMPLATE_DIR);
+        path.push(relative_path);
+        path
+    }
+
+    /// Returns the template system name, given a relative path to a file.
+    fn relative_path_to_name(relative_path: &str) -> String {
+        let path = template_path(relative_path);
+        let (_, name, _) = split_path(&path);
+        name
+    }
+
+    #[test]
+    fn template_path_index_html() {
+        let path = template_path("index.html.hbs");
+        let (rel_path, name, data_type) = split_path(&path);
+
+        assert_eq!(rel_path.to_string_lossy(), "index.html.hbs");
+        assert_eq!(name, "index");
+        assert_eq!(data_type, Some("html".to_owned()));
+    }
+
+    #[test]
+    fn template_path_subdir_index_html() {
+        let path = template_path("subdir/index.html.hbs");
+        let (rel_path, name, data_type) = split_path(&path);
+
+        assert_eq!(rel_path.to_string_lossy(), "subdir/index.html.hbs");
+        assert_eq!(name, "subdir/index");
+        assert_eq!(data_type, Some("html".to_owned()));
+    }
+
+    #[test]
+    fn template_path_doc_examples() {
+        assert_eq!(relative_path_to_name("index.html.hbs"), "index");
+        assert_eq!(relative_path_to_name("index.tera"), "index");
+        assert_eq!(relative_path_to_name("index.hbs"), "index");
+        assert_eq!(relative_path_to_name("dir/index.hbs"), "dir/index");
+        assert_eq!(relative_path_to_name("dir/index.html.tera"), "dir/index");
+        assert_eq!(relative_path_to_name("index.template.html.hbs"),
+                   "index.template");
+        assert_eq!(relative_path_to_name("subdir/index.template.html.hbs"),
+                   "subdir/index.template");
+    }
+}