From 783073b4fbc8bb974b04467b821c15e88d3c970e Mon Sep 17 00:00:00 2001 From: René Kijewski Date: Wed, 5 Jan 2022 11:38:36 +0100 Subject: Update for actix-web beta --- askama_actix/src/lib.rs | 36 +++++++++++++++++------------------- 1 file changed, 17 insertions(+), 19 deletions(-) (limited to 'askama_actix/src') diff --git a/askama_actix/src/lib.rs b/askama_actix/src/lib.rs index d9ac62d..71af8cc 100644 --- a/askama_actix/src/lib.rs +++ b/askama_actix/src/lib.rs @@ -1,31 +1,29 @@ #![deny(elided_lifetimes_in_paths)] +use actix_web::body::BoxBody; +use actix_web::http::StatusCode; +use actix_web::HttpResponseBuilder; +use askama::mime::extension_to_mime_type; pub use askama::*; -use bytes::BytesMut; -use actix_web::{error::ErrorInternalServerError, Error, HttpResponse}; +use actix_web::HttpResponse; pub trait TemplateToResponse { - fn to_response(&self) -> std::result::Result; + fn to_response(&self) -> HttpResponse; } impl TemplateToResponse for T { - fn to_response(&self) -> std::result::Result { - let mut buffer = BytesMut::with_capacity(T::SIZE_HINT); - if self.render_into(&mut buffer).is_err() { - return Err(ErrorInternalServerError("Template parsing error")); + fn to_response(&self) -> HttpResponse { + match self.render() { + Ok(buffer) => { + let ctype = extension_to_mime_type(T::EXTENSION.unwrap_or("txt")); + HttpResponseBuilder::new(StatusCode::OK) + .content_type(ctype) + .body(buffer) + } + Err(err) => { + HttpResponse::from_error(Box::new(err) as Box) + } } - - let ctype = askama::mime::extension_to_mime_type(T::EXTENSION.unwrap_or("txt")).to_string(); - Ok(HttpResponse::Ok() - .content_type(ctype.as_str()) - .body(buffer.freeze())) } } - -// Re-exported for use by generated code -#[doc(hidden)] -pub mod futures { - pub use futures_util::future::ready; - pub use futures_util::future::Ready; -} -- cgit