Add TemplateResponder trait for integration with actix-web

1 files changed, 15 insertions, 0 deletions

diff --git a/askama/src/lib.rs b/askama/src/lib.rs
index a0d641b..1d66feb 100644
--- a/askama/src/lib.rs
+++ b/askama/src/lib.rs
@@ -492,6 +492,21 @@ pub mod actix_web {
         let ctype = get_mime_type(ext).to_string();
         Ok(HttpResponse::Ok().content_type(ctype.as_str()).body(rsp))
     }
+
+    pub trait TemplateResponder {
+        fn responder(&self) -> Result<HttpResponse, Error>;
+    }
+
+    impl<T: super::Template> TemplateResponder for T {
+        fn responder(&self) -> Result<HttpResponse, Error> {
+            let rsp = self
+                .render()
+                .map_err(|_| ErrorInternalServerError("Template render error"))?;
+            let ext = T::extension().unwrap_or("");
+            let ctype = get_mime_type(ext).to_string();
+            Ok(HttpResponse::Ok().content_type(ctype.as_str()).body(rsp))
+        }
+    }
 }
 
 #[cfg(feature = "with-gotham")]