110 lines
3.2 KiB
Rust
110 lines
3.2 KiB
Rust
use crate::models::{CookieSource, VideoMetadata};
|
|
use serde::Deserialize;
|
|
use std::process::Command;
|
|
|
|
#[derive(Debug, Deserialize)]
|
|
pub struct YtDlpJson {
|
|
pub title: String,
|
|
pub duration: Option<f64>,
|
|
pub fps: Option<f64>,
|
|
pub thumbnail: Option<String>,
|
|
}
|
|
|
|
pub fn parse_ytdlp_json(json_str: &str) -> Result<YtDlpJson, String> {
|
|
serde_json::from_str::<YtDlpJson>(json_str)
|
|
.map_err(|e| format!("Failed to parse yt-dlp JSON: {e}"))
|
|
}
|
|
|
|
pub fn resolve(url: &str, cookie_source: &CookieSource) -> Result<VideoMetadata, String> {
|
|
let cookie_args = cookie_source.to_ytdlp_args();
|
|
|
|
let mut meta_cmd = Command::new("yt-dlp");
|
|
meta_cmd.arg("--dump-json").arg("--no-download");
|
|
for arg in &cookie_args {
|
|
meta_cmd.arg(arg);
|
|
}
|
|
meta_cmd.arg(url);
|
|
|
|
let meta_output = meta_cmd
|
|
.output()
|
|
.map_err(|e| format!("Failed to run yt-dlp: {e}"))?;
|
|
|
|
if !meta_output.status.success() {
|
|
let stderr = String::from_utf8_lossy(&meta_output.stderr);
|
|
return Err(format!("yt-dlp metadata failed: {stderr}"));
|
|
}
|
|
|
|
let json_str = String::from_utf8_lossy(&meta_output.stdout);
|
|
let meta = parse_ytdlp_json(&json_str)?;
|
|
|
|
let mut stream_cmd = Command::new("yt-dlp");
|
|
stream_cmd.arg("-g").arg("--no-download");
|
|
for arg in &cookie_args {
|
|
stream_cmd.arg(arg);
|
|
}
|
|
stream_cmd.arg(url);
|
|
|
|
let stream_output = stream_cmd
|
|
.output()
|
|
.map_err(|e| format!("Failed to get stream URL: {e}"))?;
|
|
|
|
if !stream_output.status.success() {
|
|
let stderr = String::from_utf8_lossy(&stream_output.stderr);
|
|
return Err(format!("yt-dlp stream URL failed: {stderr}"));
|
|
}
|
|
|
|
let stream_url = String::from_utf8_lossy(&stream_output.stdout)
|
|
.lines()
|
|
.next()
|
|
.unwrap_or("")
|
|
.trim()
|
|
.to_string();
|
|
|
|
if stream_url.is_empty() {
|
|
return Err("yt-dlp returned an empty stream URL".to_string());
|
|
}
|
|
|
|
Ok(VideoMetadata {
|
|
url: url.to_string(),
|
|
title: meta.title,
|
|
duration: meta.duration.unwrap_or(0.0),
|
|
fps: meta.fps.unwrap_or(30.0),
|
|
thumbnail_url: meta.thumbnail,
|
|
stream_url,
|
|
})
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test_parse_ytdlp_json_valid() {
|
|
let json = r#"{"title": "Test Video", "duration": 125.5, "fps": 29.97, "thumbnail": "https://i.ytimg.com/vi/abc/hqdefault.jpg"}"#;
|
|
let result = parse_ytdlp_json(json).unwrap();
|
|
assert_eq!(result.title, "Test Video");
|
|
assert!((result.duration.unwrap() - 125.5).abs() < 0.001);
|
|
assert!((result.fps.unwrap() - 29.97).abs() < 0.001);
|
|
assert_eq!(
|
|
result.thumbnail.unwrap(),
|
|
"https://i.ytimg.com/vi/abc/hqdefault.jpg"
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_ytdlp_json_missing_optional_fields() {
|
|
let json = r#"{"title": "Minimal Video"}"#;
|
|
let result = parse_ytdlp_json(json).unwrap();
|
|
assert_eq!(result.title, "Minimal Video");
|
|
assert!(result.duration.is_none());
|
|
assert!(result.fps.is_none());
|
|
assert!(result.thumbnail.is_none());
|
|
}
|
|
|
|
#[test]
|
|
fn test_parse_ytdlp_json_invalid() {
|
|
let result = parse_ytdlp_json("not json");
|
|
assert!(result.is_err());
|
|
}
|
|
}
|