Ben/RubHub - src/controllers/project/ci/view.rs

Ben (Desktop/Arch) CI ed432d4 18 days ago 222 Commits
Raw
text/x-rust • 3.47 KB • 117 lines
use std::sync::Arc;

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

use crate::{
    AccessType, GlobalState, Project, UserModel,
    extractors::PathUserProject,
    models::{CiJob, ContentPage},
    services::{ci, session},
    views::ThemedRender,
};

#[derive(Template)]
#[template(path = "project/ci/view.html")]
struct CiJobTemplate<'a> {
    owner: Arc<User>,
    project: &'a Project,
    access_level: AccessType,
    job: CiJob,
    log_content: String,
    logged_in_user: Option<Arc<User>>,
    sidebar_projects: Vec<Project>,
    content_pages: Vec<ContentPage>,
    active_tab: &'static str,
    selected_branch: String,
}

pub async fn ci_job_get(
    State(state): State<GlobalState>,
    cookies: Cookies,
    PathUserProject(owner, project): PathUserProject,
    Path((_, _, job_id)): Path<(String, String, String)>,
) -> Response<Body> {
    let logged_in_user = session::current_user(&state, &cookies).await.ok();

    let content_pages = state.config.content_pages.clone();
    let sidebar_projects = if let Some(ref user) = logged_in_user {
        user.sidebar_projects(&state).await
    } else {
        vec![]
    };
    let access_level = project
        .access_level(logged_in_user.as_ref().map(|u| u.slug.clone()))
        .await;

    if access_level == AccessType::None {
        return crate::controllers::not_found(logged_in_user, sidebar_projects, content_pages);
    }

    // Load job
    let job = match ci::load_job(&state.config.ci_root, &owner.slug, &project.slug, &job_id).await {
        Ok(job) => job,
        Err(_) => {
            return crate::controllers::not_found(logged_in_user, sidebar_projects, content_pages);
        }
    };

    // Load log content
    let log_content = ci::load_job_log(&state.config.ci_root, &owner.slug, &project.slug, &job_id)
        .await
        .unwrap_or_else(|_| String::from("(no log available)"));

    let template = CiJobTemplate {
        owner,
        project: &project,
        access_level,
        job,
        log_content,
        logged_in_user,
        sidebar_projects,
        content_pages,
        active_tab: "ci",
        selected_branch: project.main_branch.clone(),
    };
    Html(template.render_with_theme()).into_response()
}

/// Return raw log content as plain text
pub async fn ci_log_get(
    State(state): State<GlobalState>,
    cookies: Cookies,
    PathUserProject(owner, project): PathUserProject,
    Path((_, _, job_id)): Path<(String, String, String)>,
) -> Response<Body> {
    let logged_in_user = session::current_user(&state, &cookies).await.ok();
    let content_pages = state.config.content_pages.clone();
    let sidebar_projects = if let Some(ref user) = logged_in_user {
        user.sidebar_projects(&state).await
    } else {
        vec![]
    };
    let access_level = project
        .access_level(logged_in_user.as_ref().map(|u| u.slug.clone()))
        .await;

    if access_level == AccessType::None {
        return crate::controllers::not_found(logged_in_user, sidebar_projects, content_pages);
    }

    // Load log content
    let log_content = ci::load_job_log(&state.config.ci_root, &owner.slug, &project.slug, &job_id)
        .await
        .unwrap_or_else(|_| String::from("(no log available)"));

    Response::builder()
        .header(header::CONTENT_TYPE, "text/plain; charset=utf-8")
        .body(Body::from(log_content))
        .unwrap()
}
Ben / RubHub
