blankie
/
autoytarchivers
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Compare commits

base: blankie:445cac2d8fd4f7e2f535c5446ea859750dbebffd
Branches Tags
blankie:master
...
compare: blankie:d7a0a6a3e596e4c80550cd79ac0b15889169b0da
Branches Tags
blankie:master

3 Commits
445cac2d8f ... d7a0a6a3e5

Author SHA1 Message Date
blank X d7a0a6a3e5
Add uploading messages 2021-06-21 20:01:54 +07:00
blank X e5c9ff3aed
(Try to) Fix calculating file size 2021-06-21 19:55:33 +07:00
blank X d109e3f034
Add a micro-optmisation 2021-06-21 19:51:40 +07:00
2 changed files with 96 additions and 7 deletions
Show Stats Expand all files Collapse all files

8
src/main.rs
View File

@ -163,11 +163,6 @@ async fn async_main() {
match utils::get_videos(&rclient, i).await {
Ok(videos) => {
for j in videos {
{
if tmp_handled.lock().unwrap().contains(&j) {
continue;
}
}
if nodl {
let mut seen_videos = seen_videos.write().unwrap();
if !seen_videos.contains(&j) {
@ -175,6 +170,9 @@ async fn async_main() {
}
} else {
{
if tmp_handled.lock().unwrap().contains(&j) {
continue;
}
let seen_videos = seen_videos.read().unwrap();
if seen_videos.contains(&j) {
continue;

95
src/workers.rs
View File

@ -612,14 +612,61 @@ pub async fn upload_worker(
let total_size = file.seek(SeekFrom::End(0)).await.unwrap();
file.seek(SeekFrom::Start(0)).await.unwrap();
let parts = (total_size as f64 / (2000.0 * 1024.0 * 1024.0)).ceil() as usize;
let current_position = 0;
for i in 0..parts {
let filename = if parts == 1 {
video_filename.clone()
} else {
format!("{}.part{:02}", &video_filename, i)
};
let mut size = total_size - current_position;
let message = tclient
.send_message(
&chat,
InputMessage::text(&format!("Uploading {}...", &video_filename)),
)
.await;
if let Err(ref err) = message {
eprintln!("Failed to send uploading message: {:?}", err);
let text = format!("{:#?}", err);
let size = text.len();
let mut stream = Cursor::new(text.into_bytes());
match tclient
.upload_stream(&mut stream, size, "failed-send-uploading.log".to_string())
.await
{
Ok(uploaded) => {
let message = InputMessage::text("Failed to send uploading message")
.mime_type("text/plain")
.file(uploaded);
if let Err(err) = tclient.send_message(&chat, message).await {
eprintln!(
"Failed to send message about failing to send uploading message: {:?}",
err
);
if let Err(err) = tclient.send_message(&chat, InputMessage::text("Failed to send message about failing to send uploading message, see logs")).await {
eprintln!("Failed to send message about failing to send message about failing to send uploading message: {:?}", err);
}
}
}
Err(err) => {
eprintln!(
"Failed to upload logs about failing to send uploading message: {:?}",
err
);
if let Err(err) = tclient
.send_message(
&chat,
InputMessage::text(
"Failed to upload logs about failing to send uploading message, see logs",
),
)
.await
{
eprintln!("Failed to send message about failing to upload logs about failing to send uploading message: {:?}", err);
}
}
};
}
let mut size = file.stream_position().await.unwrap();
if size > 2000 * 1024 * 1024 {
size = 2000 * 1024 * 1024;
}
@ -664,6 +711,50 @@ pub async fn upload_worker(
}
}
};
if let Ok(mut message) = message {
if let Err(err) = message.delete().await {
eprintln!("Failed to delete uploading message: {:?}", err);
let text = format!("{:#?}", err);
let size = text.len();
let mut stream = Cursor::new(text.into_bytes());
match tclient
.upload_stream(&mut stream, size, "failed-delete-uploading.log".to_string())
.await
{
Ok(uploaded) => {
let message = InputMessage::text("Failed to delete uploading message")
.mime_type("text/plain")
.file(uploaded);
if let Err(err) = tclient.send_message(&chat, message).await {
eprintln!(
"Failed to send message about failing to delete uploading message: {:?}",
err
);
if let Err(err) = tclient.send_message(&chat, InputMessage::text("Failed to send message about failing to delete uploading message, see logs")).await {
eprintln!("Failed to send message about failing to send message about failing to delete uploading message: {:?}", err);
}
}
}
Err(err) => {
eprintln!(
"Failed to upload logs about failing to delete uploading message: {:?}",
err
);
if let Err(err) = tclient
.send_message(
&chat,
InputMessage::text(
"Failed to upload logs about failing to delete uploading message, see logs",
),
)
.await
{
eprintln!("Failed to send message about failing to upload logs about failing to delete uploading message: {:?}", err);
}
}
};
}
}
}
if success {
if let Err(err) = remove_file(&video_filename) {