flibusta-apps/fb2converter_server
1
0
Fork 0
mirror of https://github.com/flibusta-apps/fb2converter_server.git synced 2025-12-06 06:55:36 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix
Some checks failed
Build docker image / Build-Docker-Image (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Bulat Kurbanov
2024-08-22 22:00:24 +02:00
parent 84c7fbf1c0
commit a9da07f216
1 changed files with 9 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/main.rs
View File

@@ -64,24 +64,22 @@ async fn convert_file(Path(file_format): Path<String>, body: Body) -> impl IntoR
while let Some(chunk) = data_stream.next().await { while let Some(chunk) = data_stream.next().await {
let data = match chunk { let data = match chunk {
Ok(v) => v, Ok(v) => v,
Err(err) => { Err(_err) => {
log::error!("{:?}", err); return StatusCode::BAD_REQUEST.into_response();
return StatusCode::INTERNAL_SERVER_ERROR.into_response();
} }
}; };
match tempfile_rw.write(data.as_ref()).await { match tempfile_rw.write(data.as_ref()).await {
Ok(_) => (), Ok(_) => (),
Err(err) => { Err(_err) => {
log::error!("{:?}", err); return StatusCode::NO_CONTENT.into_response();
return StatusCode::INTERNAL_SERVER_ERROR.into_response();
} }
} }
} }
let _ = tempfile_rw.flush().await; let _ = tempfile_rw.flush().await;
let allowed_formats = vec!["epub".to_string(), "mobi".to_string()]; let allowed_formats = ["epub".to_string(), "mobi".to_string()];
if !allowed_formats.contains(&file_format.clone().to_lowercase()) { if !allowed_formats.contains(&file_format.clone().to_lowercase()) {
return StatusCode::BAD_REQUEST.into_response(); return StatusCode::BAD_REQUEST.into_response();
} }
@@ -96,9 +94,8 @@ async fn convert_file(Path(file_format): Path<String>, body: Body) -> impl IntoR
.await .await
{ {
Ok(v) => v, Ok(v) => v,
Err(err) => { Err(_err) => {
log::error!("{:?}", err); return StatusCode::NO_CONTENT.into_response();
return StatusCode::INTERNAL_SERVER_ERROR.into_response();
} }
}; };
@@ -109,9 +106,8 @@ async fn convert_file(Path(file_format): Path<String>, body: Body) -> impl IntoR
let mut result_file = match File::open(format!("/tmp/{prefix}.{file_format}")).await { let mut result_file = match File::open(format!("/tmp/{prefix}.{file_format}")).await {
Ok(v) => v, Ok(v) => v,
Err(err) => { Err(_err) => {
log::error!("{:?}", err); return StatusCode::NO_CONTENT.into_response();
return StatusCode::INTERNAL_SERVER_ERROR.into_response();
} }
}; };