tetratto/crates/app/src/routes/api/v1/uploads.rs

use std::fs::exists;
use axum::{body::Body, extract::Path, response::IntoResponse, Extension, Json};
use crate::cookie::CookieJar;
use pathbufd::PathBufD;
use crate::{get_user_from_token, routes::api::v1::UpdateUploadAlt, State};
use super::auth::images::read_image;
use tetratto_core::model::{carp::CarpGraph, oauth, uploads::MediaType, ApiReturn, Error};

pub async fn get_request(
    Path(id): Path<usize>,
    Extension(data): Extension<State>,
) -> impl IntoResponse {
    let data = &(data.read().await).0;

    let upload = match data.get_upload_by_id(id).await {
        Ok(u) => u,
        Err(_) => {
            return Err((
                [("Content-Type", "image/svg+xml")],
                Body::from(read_image(PathBufD::current().extend(&[
                    data.0.0.dirs.media.as_str(),
                    "images",
                    "default-banner.svg",
                ]))),
            ));
        }
    };

    let path = upload.path(&data.0.0);

    if !exists(&path).unwrap() {
        return Err((
            [("Content-Type", "image/svg+xml")],
            Body::from(read_image(PathBufD::current().extend(&[
                data.0.0.dirs.media.as_str(),
                "images",
                "default-banner.svg",
            ]))),
        ));
    }

    let bytes = read_image(path);

    if upload.what == MediaType::Carpgraph {
        // conver to svg and return
        return Ok((
            [("Content-Type", "image/svg+xml".to_string())],
            Body::from(CarpGraph::from_bytes(bytes).to_svg()),
        ));
    }

    Ok(([("Content-Type", upload.what.mime())], Body::from(bytes)))
}

pub async fn get_json_request(
    Path(id): Path<usize>,
    Extension(data): Extension<State>,
) -> impl IntoResponse {
    let data = &(data.read().await).0;

    let upload = match data.get_upload_by_id(id).await {
        Ok(u) => u,
        Err(e) => return Json(e.into()),
    };

    Json(ApiReturn {
        ok: true,
        message: "Success".to_string(),
        payload: Some(upload),
    })
}

pub async fn delete_request(
    jar: CookieJar,
    Extension(data): Extension<State>,
    Path(id): Path<usize>,
) -> impl IntoResponse {
    let data = &(data.read().await).0;
    let user = match get_user_from_token!(jar, data, oauth::AppScope::UserManageUploads) {
        Some(ua) => ua,
        None => return Json(Error::NotAllowed.into()),
    };

    match data.delete_upload_checked(id, &user).await {
        Ok(_) => Json(ApiReturn {
            ok: true,
            message: "Upload deleted".to_string(),
            payload: (),
        }),
        Err(e) => Json(e.into()),
    }
}

pub async fn update_alt_request(
    jar: CookieJar,
    Extension(data): Extension<State>,
    Path(id): Path<usize>,
    Json(props): Json<UpdateUploadAlt>,
) -> impl IntoResponse {
    let data = &(data.read().await).0;
    let user = match get_user_from_token!(jar, data, oauth::AppScope::UserManageUploads) {
        Some(ua) => ua,
        None => return Json(Error::NotAllowed.into()),
    };

    match data.update_upload_alt(id, &user, &props.alt).await {
        Ok(_) => Json(ApiReturn {
            ok: true,
            message: "Upload updated".to_string(),
            payload: (),
        }),
        Err(e) => Json(e.into()),
    }
}