a06673b
use axum::body::Bytes;
use axum::extract::{Path, Query, State};
use axum::http::header;
use axum::http::{HeaderValue, StatusCode};
use axum::response::{IntoResponse, Response};
use jwt::extractor::AuthUser;
use uuid::Uuid;
use crate::error::RepoResult;
use crate::features::RepoService;
use crate::features::archive::ArchiveFormatQuery;
#[utoipa::path(
get,
path = "/repo/{channel_id}/{repo_id}/archive/{archive_format}/{ref_name}",
tag = "repo",
params(ArchiveFormatQuery),
responses((status = 200, description = "Download an archive")),
security(("bearer_auth" = []))
)]
pub(crate) async fn handler(
State(svc): State<RepoService>,
AuthUser { claims }: AuthUser,
Path((_, repo_id, archive_format, ref_name)): Path<(Uuid, Uuid, String, String)>,
Query(q): Query<ArchiveFormatQuery>,
) -> RepoResult<Response> {
let bytes = svc
.get_archive(claims.sub, repo_id, &archive_format, &ref_name, q)
.await?;
let mime = match archive_format.as_str() {
"zipball" => "application/zip",
"tarball" | "bundle" => "application/x-tar",
_ => "application/octet-stream",
};
let resp = (
StatusCode::OK,
[
(header::CONTENT_TYPE, HeaderValue::from_static(mime)),
(
header::CONTENT_DISPOSITION,
HeaderValue::from_str(&format!(
"attachment; filename=\"{ref_name}.{archive_format}\""
))
.unwrap_or(HeaderValue::from_static("attachment")),
),
],
Bytes::from(bytes),
)
.into_response();
Ok(resp)
}