flibusta-apps/books_downloader
1
0
Fork 0
mirror of https://github.com/flibusta-apps/books_downloader.git synced 2026-03-03 23:20:50 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add x-filename-b64

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-01-19 16:36:46 +01:00
parent 192fac1d78
commit f5fce67a33
6 changed files with 57 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/views.rs
View File

@@ -30,17 +30,21 @@ pub async fn download(
},
};
let (data, filename) = match download_result {
let data = match download_result {
Some(v) => v,
None => return Err((StatusCode::NO_CONTENT, "Can't download!".to_string())),
};
let filename = data.filename.clone();
let filename_ascii = data.filename_ascii.clone();
let reader = data.get_async_read();
let stream = ReaderStream::new(reader);
let body = StreamBody::new(stream);
let headers = AppendHeaders([
(header::CONTENT_DISPOSITION, format!("attachment; filename={filename}"))
(header::CONTENT_DISPOSITION, format!("attachment; filename={filename_ascii}")),
(header::HeaderName::from_static("x-filename-b64"), base64::encode(filename))
]);
Ok((headers, body))
@@ -62,7 +66,7 @@ pub async fn get_filename(
}
let filename = match get_book(book_id).await {
Ok(book) => get_filename_by_book(&book, file_type.as_str(), false),
Ok(book) => get_filename_by_book(&book, file_type.as_str(), false, false),
Err(_) => return (StatusCode::BAD_REQUEST, "Book not found!".to_string()),
};