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

Fix download from cache server

Browse Source
This commit is contained in:
Bulat Kurbanov
2022-12-15 20:24:35 +01:00
parent c204b69225
commit 9214b29079
2 changed files with 75 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
src/bots/approved_bot/modules/download.rs
View File

@@ -6,7 +6,7 @@ use tokio_util::compat::FuturesAsyncReadCompatExt;
use crate::{
bots::{
approved_bot::services::book_cache::{
clear_book_cache, download_file, get_cached_message,
download_file, get_cached_message,
types::{CachedMessage, DownloadFile},
},
BotHandlerInternal,
@@ -46,8 +46,8 @@ impl CommandParse<Self> for DownloadData {
}
async fn _send_cached(
message: Message,
bot: AutoSend<Bot>,
message: &Message,
bot: &AutoSend<Bot>,
cached_message: CachedMessage,
) -> BotHandlerInternal {
match bot
@@ -69,42 +69,33 @@ async fn send_cached_message(
bot: AutoSend<Bot>,
download_data: DownloadData,
) -> BotHandlerInternal {
let cached_message = get_cached_message(&download_data).await;
match cached_message {
Ok(v) => match _send_cached(message.clone(), bot.clone(), v).await {
match get_cached_message(&download_data).await {
Ok(v) => match _send_cached(&message, &bot, v).await {
Ok(_) => return Ok(()),
Err(err) => log::info!("{:?}", err),
Err(err) => log::warn!("{:?}", err),
},
Err(err) => return Err(err),
};
match clear_book_cache(&download_data).await {
Ok(_) => (),
Err(err) => log::error!("{:?}", err),
};
let cached_message = get_cached_message(&download_data).await;
match cached_message {
Ok(v) => _send_cached(message, bot, v).await,
Err(err) => return Err(err),
match get_cached_message(&download_data).await {
Ok(v) => match _send_cached(&message, &bot, v).await {
Ok(v_2) => Ok(v_2),
Err(err) => Err(err),
},
Err(err) => Err(err),
}
}
async fn send_with_download_from_channel(
message: Message,
bot: AutoSend<Bot>,
download_data: DownloadData,
async fn _send_downloaded_file(
message: &Message,
bot: &AutoSend<Bot>,
downloaded_data: DownloadFile
) -> BotHandlerInternal {
let downloaded_file = match download_file(&download_data).await {
Ok(v) => v,
Err(err) => return Err(err),
};
let DownloadFile {
response,
filename,
caption,
} = downloaded_file;
} = downloaded_data;
let data = response
.bytes_stream()
@@ -125,6 +116,28 @@ async fn send_with_download_from_channel(
}
}
async fn send_with_download_from_channel(
message: Message,
bot: AutoSend<Bot>,
download_data: DownloadData,
) -> BotHandlerInternal {
match download_file(&download_data).await {
Ok(v) => match _send_downloaded_file(&message, &bot, v).await {
Ok(_) => return Ok(()),
Err(err) => log::warn!("{:?}", err),
},
Err(err) => return Err(err),
};
match download_file(&download_data).await {
Ok(v) => match _send_downloaded_file(&message, &bot, v).await {
Ok(v_2) => Ok(v_2),
Err(err) => Err(err),
},
Err(err) => Err(err)
}
}
async fn download_handler(
message: Message,
bot: AutoSend<Bot>,