feat: add youtube shorts support

This commit is contained in:
2025-09-19 22:36:26 +03:00
parent 7a9ef2c48c
commit 00f0a95d22
8 changed files with 233 additions and 86 deletions

View File

@@ -1,15 +1,10 @@
use crate::error::{Error, Result};
use crate::download::{download_instaloader, process_download_result};
use crate::error::Result;
use crate::handlers::SocialHandler;
use crate::utils::{MediaKind, detect_media_kind_async, send_media_from_path};
use futures::{StreamExt, stream};
use regex::Regex;
use std::path::PathBuf;
use std::{process::Stdio, sync::OnceLock};
use std::sync::OnceLock;
use teloxide::{Bot, types::ChatId};
use tempfile::tempdir;
use tokio::fs::read_dir;
use tokio::process::Command;
use tracing::{error, info};
use tracing::info;
static SHORTCODE_RE: OnceLock<Regex> = OnceLock::new();
@@ -48,77 +43,8 @@ impl SocialHandler for InstagramHandler {
async fn handle(&self, bot: &Bot, chat_id: ChatId, shortcode: String) -> Result<()> {
info!(handler = %self.name(), shortcode = %shortcode, "handling instagram code");
let tmp = tempdir().map_err(Error::from)?;
let cwd = tmp.path().to_path_buf();
let target = format!("-{shortcode}");
let status = Command::new("instaloader")
.current_dir(&cwd)
.args([
"--dirname-pattern=.",
"--no-metadata-json",
"--no-compress-json",
"--quiet",
"--",
&target,
])
.stdin(Stdio::null())
.stdout(Stdio::null())
.stderr(Stdio::null())
.status()
.await
.map_err(Error::from)?;
if !status.success() {
return Err(Error::InstaloaderFaileled(status.to_string()));
}
let mut dir = read_dir(&cwd).await?;
let mut paths = Vec::new();
while let Some(entry) = dir.next_entry().await? {
if entry.file_type().await?.is_file() {
paths.push(entry.path());
}
}
let concurrency = 8;
let results = stream::iter(paths)
.map(|path| async move {
let kind = detect_media_kind_async(&path).await;
match kind {
MediaKind::Unknown => None,
k => Some((path, k)),
}
})
.buffer_unordered(concurrency)
.collect::<Vec<Option<(PathBuf, MediaKind)>>>()
.await;
let mut media = results
.into_iter()
.flatten()
.collect::<Vec<(PathBuf, MediaKind)>>();
if media.is_empty() {
error!("no media found in tmp dir after instaloader");
return Err(Error::NoMediaFound);
}
// deterministic ordering
media.sort_by_key(|(p, _)| p.clone());
// prefer video over image
if let Some((path, MediaKind::Video)) = media.iter().find(|(_, k)| *k == MediaKind::Video) {
return send_media_from_path(bot, chat_id, path.clone(), Some(MediaKind::Video)).await;
}
if let Some((path, MediaKind::Image)) = media.iter().find(|(_, k)| *k == MediaKind::Image) {
return send_media_from_path(bot, chat_id, path.clone(), Some(MediaKind::Image)).await;
}
error!("no supported media kind found after scanning");
Err(Error::NoMediaFound)
let dr = download_instaloader(&shortcode).await?;
process_download_result(bot, chat_id, dr).await
}
fn box_clone(&self) -> Box<dyn SocialHandler> {