Ben/RubHub - src/controllers/project/new.rs

Ben (Desktop/Arch) Cleaned up sidebar 35d8d90 1 month ago 144 Commits
Raw
text/x-rust • 3.98 KB • 134 lines
use askama::Template;
use axum::{
    Form,
    body::Body,
    extract::State,
    http::Response,
    response::{Html, IntoResponse, Redirect},
};
use serde::Deserialize;
use tower_cookies::Cookies;

use crate::{
    AccessType, GlobalState, Project, User,
    models::ContentPage,
    services::{repository::create_bare_repo, session, validation::validate_project_name},
    views::ThemedRender,
};

#[derive(Debug, Deserialize)]
pub struct NewProjectForm {
    pub name: String,
    pub public_access: Option<String>,
}

#[derive(Template)]
#[template(path = "project_new.html")]
struct NewProjectTemplate<'a> {
    message: Option<&'a str>,
    logged_in_user: Option<&'a User>,
    sidebar_projects: Vec<Project>,
    content_pages: Vec<ContentPage>,
}

pub async fn project_new_get(
    State(state): State<GlobalState>,
    cookies: Cookies,
) -> Result<Html<String>, Redirect> {
    let logged_in_user = match session::current_user(&state, &cookies).await {
        Ok(user) => user,
        Err(_) => return Err(Redirect::to("/login")),
    };
    Ok(render_new_project_page(&state, Some(&logged_in_user), None).await)
}

async fn render_new_project_page(
    state: &GlobalState,
    logged_in_user: Option<&User>,
    message: Option<&str>,
) -> Html<String> {
    let sidebar_projects = if let Some(user) = logged_in_user {
        user.sidebar_projects(state).await
    } else {
        vec![]
    };

    let template = NewProjectTemplate {
        message,
        logged_in_user,
        sidebar_projects,
        content_pages: state.config.content_pages.clone(),
    };
    Html(template.render_with_theme())
}

pub async fn project_new_post(
    State(state): State<GlobalState>,
    cookies: Cookies,
    Form(form): Form<NewProjectForm>,
) -> Response<Body> {
    let selected_public_access = form
        .public_access
        .as_deref()
        .and_then(|value| AccessType::parse_public_access(value).ok())
        .unwrap_or(AccessType::Read);

    let current_user = match session::current_user(&state, &cookies).await {
        Ok(user) => user,
        Err(_) => return Redirect::to("/login").into_response(),
    };

    let name = form.name.trim();
    if name.is_empty() {
        return render_new_project_page(&state, Some(&current_user), Some("Name is required."))
            .await
            .into_response();
    }

    if let Err(msg) = validate_project_name(name) {
        return render_new_project_page(&state, Some(&current_user), Some(msg))
            .await
            .into_response();
    }

    let user_slug = current_user.slug.clone();

    match Project::new(&current_user, name, selected_public_access) {
        Ok(project) => {
            if Project::load(&state, &project.owner, &project.slug)
                .await
                .is_ok()
            {
                return render_new_project_page(
                    &state,
                    Some(&current_user),
                    Some("Project already exists"),
                )
                .await
                .into_response();
            };
            match project.save(&state).await {
                Ok(_) => {
                    match create_bare_repo(&state, user_slug.clone(), project.slug.clone()).await {
                        Ok(_) => Redirect::to(&project.uri()).into_response(),
                        Err(_) => render_new_project_page(
                            &state,
                            Some(&current_user),
                            Some("Could not create project."),
                        )
                        .await
                        .into_response(),
                    }
                }
                Err(msg) => {
                    render_new_project_page(&state, Some(&current_user), Some(&msg.to_string()))
                        .await
                        .into_response()
                }
            }
        }
        Err(msg) => render_new_project_page(&state, Some(&current_user), Some(&msg.to_string()))
            .await
            .into_response(),
    }
}
Ben / RubHub
