thegeneralist01/social-media-video-downloader-bot
1
Fork 0
mirror of https://github.com/thegeneralist01/social-media-video-downloader-bot synced 2026-01-09 14:50:30 +01:00
Code Issues Activity

keep videos that start with !

Browse source
This commit is contained in:
TheGeneralist 2025-04-04 18:44:12 +02:00
parent b8214e5a12
commit 62a5166757
No known key found for this signature in database
GPG key ID: C391D4D52D630F45
1 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/main.rs
View file

@ -31,11 +31,17 @@ async fn main() -> Result<()> {
if let Some(text) = msg.text() {
if text.to_lowercase() == "/start" {
bot.send_message(msg.chat.id, "Send a social media video link to download it!\nCurrently, the following social medias are supported: YouTube, X (twitter).").await.ok();
} else if URL_WHITELIST.iter().any(|url| text.starts_with(url)) {
} else if (text.starts_with("!") && URL_WHITELIST.iter().any(|url| (text[1..]).starts_with(url))) || URL_WHITELIST.iter().any(|url| text.starts_with(url)) {
log::info!("Video downloaded");
let prepping_msg = bot.send_message(msg.chat.id, "Prepping the video...").await;
let user_id = msg.from.as_ref().map(|user| user.id).unwrap();
let sent = download(text, user_id).await.unwrap();
let delete_msg = !text.starts_with("!");
let url = if let Some(stripped) = text.strip_prefix("!") {
stripped
} else {
text
};
let sent = download(url, user_id).await.unwrap();
match sent {
true => {
@ -52,6 +58,9 @@ async fn main() -> Result<()> {
};
if let Ok(prepping_msg) = prepping_msg {
bot.delete_message(msg.chat.id, prepping_msg.id).await.ok();
if delete_msg {
bot.delete_message(msg.chat.id, msg.id).await.ok();
}
}
}
};