49 lines
1.2 KiB
Rust
49 lines
1.2 KiB
Rust
|
use anyhow::{anyhow, Result};
|
||
|
|
use log::{error, info};
|
||
|
|
use std::ffi::OsStr;
|
||
|
|
use tokio::process;
|
||
|
|
|
||
|
|
pub struct YtDlpWrapper {
|
||
|
|
use_aria2c: bool,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl YtDlpWrapper {
|
||
|
|
pub async fn new() -> Option<Self> {
|
||
|
|
if !command_exists("yt-dlp").await {
|
||
|
|
return None;
|
||
|
|
}
|
||
|
|
Some(Self {
|
||
|
|
use_aria2c: command_exists("aria2c").await,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn spawn_yt_dlp_task(&self, url: &str) -> Result<()> {
|
||
|
|
info!("Spawning yt-dlp task for URL: {}", url);
|
||
|
|
|
||
|
|
let mut builder = process::Command::new("yt-dlp");
|
||
|
|
if self.use_aria2c {
|
||
|
|
builder.arg("--downloader=aria2c");
|
||
|
|
}
|
||
|
|
let mut yt_dlp_cmd = builder.arg(url).spawn()?;
|
||
|
|
|
||
|
|
let status = yt_dlp_cmd.wait().await?;
|
||
|
|
|
||
|
|
if status.success() {
|
||
|
|
info!("yt-dlp task completed successfully");
|
||
|
|
Ok(())
|
||
|
|
} else {
|
||
|
|
error!("yt-dlp task failed with status: {}", status);
|
||
|
|
Err(anyhow!("yt-dlp task failed"))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
async fn command_exists<S: AsRef<OsStr>>(name: S) -> bool {
|
||
|
|
process::Command::new(name)
|
||
|
|
.arg("--version")
|
||
|
|
.output()
|
||
|
|
.await
|
||
|
|
.map(|output| output.status.success())
|
||
|
|
.unwrap_or(false)
|
||
|
|
}
|