AstralRinth/src/auth/templates/mod.rs

use crate::auth::AuthenticationError;
use actix_web::http::StatusCode;
use actix_web::{HttpResponse, ResponseError};
use std::fmt::{Debug, Display, Formatter};

pub struct Success<'a> {
    pub icon: &'a str,
    pub name: &'a str,
}

impl<'a> Success<'a> {
    pub fn render(self) -> HttpResponse {
        let html = include_str!("success.html");

        HttpResponse::Ok()
            .append_header(("Content-Type", "text/html; charset=utf-8"))
            .body(
                html.replace("{{ icon }}", self.icon)
                    .replace("{{ name }}", self.name),
            )
    }
}

#[derive(Debug)]
pub struct ErrorPage {
    pub code: StatusCode,
    pub message: String,
}

impl Display for ErrorPage {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        let html = include_str!("error.html")
            .replace("{{ code }}", &self.code.to_string())
            .replace("{{ message }}", &self.message);
        write!(f, "{}", html)?;

        Ok(())
    }
}

impl ErrorPage {
    pub fn render(&self) -> HttpResponse {
        HttpResponse::Ok()
            .append_header(("Content-Type", "text/html; charset=utf-8"))
            .body(self.to_string())
    }
}

impl actix_web::ResponseError for ErrorPage {
    fn status_code(&self) -> StatusCode {
        self.code
    }

    fn error_response(&self) -> HttpResponse {
        self.render()
    }
}

impl From<AuthenticationError> for ErrorPage {
    fn from(item: AuthenticationError) -> Self {
        ErrorPage {
            code: item.status_code(),
            message: item.to_string(),
        }
    }
}