thegeneralist01/infofeeder-bot
1
Fork 0
Code Issues Pull requests Activity

Accept documents in media message ingestion

Browse source
This commit is contained in:
TheGeneralist 2026-02-16 15:00:10 +01:00
parent 31f6beb161
commit 16e938e756
Signed by: thegeneralist01
SSH key fingerprint: SHA256:pp9qddbCNmVNoSjevdvQvM5z0DHN7LTa8qBMbcMq/R4
1 changed files with 11 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

37
src/main.rs
View file

@ -587,34 +587,19 @@ async fn handle_media_message(
if let Some(document) = msg.document() { if let Some(document) = msg.document() {
let mime = document.mime_type.as_ref().map(|m| m.essence_str()); let mime = document.mime_type.as_ref().map(|m| m.essence_str());
let is_media = if let Some(mime) = mime { fs::create_dir_all(&media_dir)
mime.starts_with("image/") || mime.starts_with("video/") .with_context(|| format!("create media dir {}", media_dir.display()))?;
let ext = mime.and_then(extension_from_mime);
let filename = if let Some(name) = document.file_name.as_deref() {
sanitize_filename_with_default(name, ext)
} else { } else {
document format!("file-{}.{}", Uuid::new_v4(), ext.unwrap_or("bin"))
.file_name
.as_deref()
.and_then(|name| Path::new(name).extension().and_then(|ext| ext.to_str()))
.map(|ext| matches!(
ext.to_ascii_lowercase().as_str(),
"png" | "jpg" | "jpeg" | "gif" | "webp" | "bmp" | "mp4" | "mov" | "mkv"
))
.unwrap_or(false)
}; };
if is_media { let dest_path = media_dir.join(&filename);
fs::create_dir_all(&media_dir) download_telegram_file(bot, &document.file.id, &dest_path).await?;
.with_context(|| format!("create media dir {}", media_dir.display()))?; let entry_text = build_media_entry_text(&filename, caption.as_deref());
let ext = mime.and_then(extension_from_mime); handle_single_item(bot.clone(), chat_id, state.clone(), &entry_text, Some(msg.id)).await?;
let filename = if let Some(name) = document.file_name.as_deref() { return Ok(true);
sanitize_filename_with_default(name, ext)
} else {
format!("file-{}.{}", Uuid::new_v4(), ext.unwrap_or("bin"))
};
let dest_path = media_dir.join(&filename);
download_telegram_file(bot, &document.file.id, &dest_path).await?;
let entry_text = build_media_entry_text(&filename, caption.as_deref());
handle_single_item(bot.clone(), chat_id, state.clone(), &entry_text, Some(msg.id)).await?;
return Ok(true);
}
} }
if let Some(video) = msg.video() { if let Some(video) = msg.video() {