aboutsummaryrefslogtreecommitdiffstats
path: root/askama_actix/src/lib.rs
blob: d9ac62db95046231ded4fbc0283e4834b9f1a51b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#![deny(elided_lifetimes_in_paths)]

pub use askama::*;
use bytes::BytesMut;

use actix_web::{error::ErrorInternalServerError, Error, HttpResponse};

pub trait TemplateToResponse {
    fn to_response(&self) -> std::result::Result<HttpResponse, Error>;
}

impl<T: askama::Template> TemplateToResponse for T {
    fn to_response(&self) -> std::result::Result<HttpResponse, Error> {
        let mut buffer = BytesMut::with_capacity(T::SIZE_HINT);
        if self.render_into(&mut buffer).is_err() {
            return Err(ErrorInternalServerError("Template parsing error"));
        }

        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;
}