Fixed extension to ignore common jinja extensions (fixes #457)

rinja-rs · Mar 10, 2021 · 429fd07 · 429fd07
1 parent 2b06170
commit 429fd07
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/askama_shared/src/input.rs b/askama_shared/src/input.rs
@@ -205,7 +205,17 @@ impl<'a> TemplateInput<'a> {
 }
 
 fn extension(path: &Path) -> Option<&str> {
-    path.extension().map(|s| s.to_str().unwrap())
+    let ext = path.extension().map(|s| s.to_str().unwrap())?;
+
+    const JINJA_EXTENSIONS: [&str; 3] = ["j2", "jinja", "jinja2"];
+    if JINJA_EXTENSIONS.contains(&ext) {
+        Path::new(path.file_stem().unwrap())
+            .extension()
+            .map(|s| s.to_str().unwrap())
+            .or(Some(ext))
+    } else {
+        Some(ext)
+    }
 }
 
 pub enum Source {