Ben/RubHub - src/controllers/not

Ben (Desktop/Arch) CI 63f0089 17 days ago 229 Commits
Raw
text/x-rust • 1.27 KB • 52 lines
use std::sync::Arc;

use askama::Template;
use axum::{
    body::Body,
    extract::State,
    http::{Response, StatusCode},
    response::{Html, IntoResponse},
};
use tower_cookies::Cookies;

use crate::{
    GlobalState, Project, User, UserModel, models::ContentPage, services::session,
    views::ThemedRender,
};

#[derive(Template)]
#[template(path = "404.html")]
struct NotFoundTemplate {
    logged_in_user: Option<Arc<User>>,
    sidebar_projects: Vec<Project>,
    content_pages: Vec<ContentPage>,
}

pub fn not_found(
    logged_in_user: Option<Arc<User>>,
    sidebar_projects: Vec<Project>,
    content_pages: Vec<ContentPage>,
) -> Response<Body> {
    let template = NotFoundTemplate {
        logged_in_user,
        sidebar_projects,
        content_pages,
    };
    (StatusCode::NOT_FOUND, Html(template.render_with_theme())).into_response()
}

pub async fn not_found_get(State(state): State<GlobalState>, cookies: Cookies) -> Response<Body> {
    let logged_in_user = session::current_user(&state, &cookies).await.ok();

    let sidebar_projects = if let Some(ref user) = logged_in_user {
        user.sidebar_projects(&state).await
    } else {
        vec![]
    };

    not_found(
        logged_in_user,
        sidebar_projects,
        state.config.content_pages.clone(),
    )
}
Ben / RubHub
