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

Fix

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-08-08 23:49:44 +02:00
parent 0fca1aea78
commit 39a354df78
1 changed files with 32 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
src/services/downloader/mod.rs
View File

@@ -3,6 +3,7 @@ pub mod utils;
pub mod zip; pub mod zip;
use reqwest::Response; use reqwest::Response;
use tokio::task::JoinSet;
use crate::config; use crate::config;
@@ -14,8 +15,6 @@ use super::book_library::types::BookWithRemote;
use super::covert::convert_file; use super::covert::convert_file;
use super::{book_library::get_remote_book, filename_getter::get_filename_by_book}; use super::{book_library::get_remote_book, filename_getter::get_filename_by_book};
use futures::stream::FuturesUnordered;
use futures::StreamExt;
pub async fn download<'a>( pub async fn download<'a>(
book_id: &'a u32, book_id: &'a u32,
@@ -75,27 +74,27 @@ pub async fn download<'a>(
} }
pub async fn download_chain<'a>( pub async fn download_chain<'a>(
book: &'a BookWithRemote, book: BookWithRemote,
file_type: &'a str, file_type: String,
source_config: &'a config::SourceConfig, source_config: config::SourceConfig,
converting: bool converting: bool
) -> Option<DownloadResult> { ) -> Option<DownloadResult> {
let final_need_zip = file_type == "fb2zip"; let final_need_zip = file_type == "fb2zip";
let file_type_ = if converting { let file_type_ = if converting {
&book.file_type book.file_type.clone()
} else { } else {
file_type file_type.clone()
}; };
let (mut response, is_zip) = match download(&book.remote_id, file_type_, source_config).await { let (mut response, is_zip) = match download(&book.remote_id, &file_type_, &source_config).await {
Some(v) => v, Some(v) => v,
None => return None, None => return None,
}; };
if is_zip && book.file_type.to_lowercase() == "html" { if is_zip && book.file_type.to_lowercase() == "html" {
let filename = get_filename_by_book(book, file_type, true, false); let filename = get_filename_by_book(&book, &file_type, true, false);
let filename_ascii = get_filename_by_book(book, file_type, true, true); let filename_ascii = get_filename_by_book(&book, &file_type, true, true);
let data_size: usize = response.headers().get("Content-Length").unwrap().to_str().unwrap().parse().unwrap(); let data_size: usize = response.headers().get("Content-Length").unwrap().to_str().unwrap().parse().unwrap();
return Some( return Some(
@@ -109,8 +108,8 @@ pub async fn download_chain<'a>(
} }
if !is_zip && !final_need_zip && !converting { if !is_zip && !final_need_zip && !converting {
let filename = get_filename_by_book(book, &book.file_type, false, false); let filename = get_filename_by_book(&book, &book.file_type, false, false);
let filename_ascii = get_filename_by_book(book, file_type, false, true); let filename_ascii = get_filename_by_book(&book, &file_type, false, true);
let data_size: usize = response.headers().get("Content-Length").unwrap().to_str().unwrap().parse().unwrap(); let data_size: usize = response.headers().get("Content-Length").unwrap().to_str().unwrap().parse().unwrap();
return Some( return Some(
@@ -153,8 +152,8 @@ pub async fn download_chain<'a>(
if !final_need_zip { if !final_need_zip {
let t = SpooledTempAsyncRead::new(clean_file); let t = SpooledTempAsyncRead::new(clean_file);
let filename = get_filename_by_book(book, file_type, false, false); let filename = get_filename_by_book(&book, &file_type, false, false);
let filename_ascii = get_filename_by_book(book, file_type, false, true); let filename_ascii = get_filename_by_book(&book, &file_type, false, true);
return Some( return Some(
DownloadResult::new( DownloadResult::new(
@@ -166,13 +165,13 @@ pub async fn download_chain<'a>(
); );
}; };
let t_file_type = if file_type == "fb2zip" { "fb2" } else { file_type }; let t_file_type = if file_type == "fb2zip" { "fb2" } else { &file_type };
let filename = get_filename_by_book(book, t_file_type, false, false); let filename = get_filename_by_book(&book, t_file_type, false, false);
match zip(&mut clean_file, filename.as_str()) { match zip(&mut clean_file, filename.as_str()) {
Some((t_file, data_size)) => { Some((t_file, data_size)) => {
let t = SpooledTempAsyncRead::new(t_file); let t = SpooledTempAsyncRead::new(t_file);
let filename = get_filename_by_book(book, file_type, true, false); let filename = get_filename_by_book(&book, &file_type, true, false);
let filename_ascii = get_filename_by_book(book, file_type, true, true); let filename_ascii = get_filename_by_book(&book, &file_type, true, true);
Some( Some(
DownloadResult::new( DownloadResult::new(
@@ -191,29 +190,31 @@ pub async fn start_download_futures(
book: &BookWithRemote, book: &BookWithRemote,
file_type: &str, file_type: &str,
) -> Option<DownloadResult> { ) -> Option<DownloadResult> {
let mut futures = FuturesUnordered::new(); let mut tasks = JoinSet::new();
for source_config in &config::CONFIG.fl_sources { for source_config in &config::CONFIG.fl_sources {
futures.push(download_chain( tasks.spawn(download_chain(
book, book.clone(),
file_type, file_type.to_string(),
source_config, source_config.clone(),
false false
)); ));
if file_type == "epub" || file_type == "fb2" { if file_type == "epub" || file_type == "fb2" {
futures.push(download_chain( tasks.spawn(download_chain(
book, book.clone(),
file_type, file_type.to_string(),
source_config, source_config.clone(),
true true
)) ));
} }
} }
while let Some(result) = futures.next().await { while let Some(task_result) = tasks.join_next().await {
if let Some(v) = result { if let Ok(task_result) = task_result {
return Some(v) if let Some(v) = task_result {
return Some(v);
}
} }
} }