Add Template method to return template extension

Thanks to Ryan McGrath for the suggestion and initial implementation.

3 files changed, 8 insertions, 0 deletions

diff --git a/askama/src/lib.rs b/askama/src/lib.rs
index 77e49f2..c7cd182 100644
--- a/askama/src/lib.rs
+++ b/askama/src/lib.rs
@@ -337,6 +337,8 @@ pub trait Template {
         self.render_into(&mut buf)?;
         Ok(buf)
     }
+    /// Helper method to inspect the template's extension
+    fn extension(&self) -> Option<&str>;
 }
 
 pub use askama_derive::*;
diff --git a/askama_derive/src/generator.rs b/askama_derive/src/generator.rs
index 5af38c0..db9ac17 100644
--- a/askama_derive/src/generator.rs
+++ b/askama_derive/src/generator.rs
@@ -117,6 +117,11 @@ impl<'a> Generator<'a> {
         self.flush_ws(WS(false, false));
         self.writeln("Ok(())");
         self.writeln("}");
+
+        self.writeln("fn extension(&self) -> Option<&str> {");
+        self.writeln(&format!("{:?}", self.input.path.extension().map(|s| s.to_str().unwrap())));
+        self.writeln("}");
+
         self.writeln("}");
     }
 
diff --git a/testing/tests/simple.rs b/testing/tests/simple.rs
index 95c14b1..4af2cea 100644
--- a/testing/tests/simple.rs
+++ b/testing/tests/simple.rs
@@ -27,6 +27,7 @@ fn test_variables() {
          Iñtërnâtiônàlizætiøn is important\n\
          in vars too: Iñtërnâtiônàlizætiøn"
     );
+    assert_eq!(s.extension(), Some("html"));
 }
 
 #[derive(Template)]