c910d7087d
* Fix(ytdl): Return all results when querying a URL * Publicly export `songbird::input::metadata::ytdl::Output` Closes: https://github.com/serenity-rs/songbird/issues/277 * Make `YoutubeDl::query` public * Abstract getting streams from YoutubeDl into functions `get_stream` and `get_streams` * fmt, remove get_streams * Fix doc comment * fixup doc comments, export `metadata::ytdl::Output` as `YoutubeDlOutput` * fmt * export metadata * fixup! fixup doc comments, export `metadata::ytdl::Output` as `YoutubeDlOutput`
78 lines
2.4 KiB
Rust
78 lines
2.4 KiB
Rust
//! `YoutubeDl` track metadata.
|
|
|
|
use super::AuxMetadata;
|
|
use crate::constants::SAMPLE_RATE_RAW;
|
|
use serde::{Deserialize, Serialize};
|
|
use std::{collections::HashMap, time::Duration};
|
|
|
|
/// Information returned by yt-dlp about a URL.
|
|
///
|
|
/// Returned by [`crate::input::YoutubeDl::query`].
|
|
#[derive(Deserialize, Serialize, Debug)]
|
|
pub struct Output {
|
|
/// The main artist.
|
|
pub artist: Option<String>,
|
|
/// The album name.
|
|
pub album: Option<String>,
|
|
/// The channel name.
|
|
pub channel: Option<String>,
|
|
/// The duration of the stream in seconds.
|
|
pub duration: Option<f64>,
|
|
/// The size of the stream.
|
|
pub filesize: Option<u64>,
|
|
/// Required HTTP headers to fetch the track stream.
|
|
pub http_headers: Option<HashMap<String, String>>,
|
|
/// Release date of this track.
|
|
pub release_date: Option<String>,
|
|
/// The thumbnail URL for this track.
|
|
pub thumbnail: Option<String>,
|
|
/// The title of this track.
|
|
pub title: Option<String>,
|
|
/// The track name.
|
|
pub track: Option<String>,
|
|
/// The date this track was uploaded on.
|
|
pub upload_date: Option<String>,
|
|
/// The name of the uploader.
|
|
pub uploader: Option<String>,
|
|
/// The stream URL.
|
|
pub url: String,
|
|
/// The URL of the public-facing webpage for this track.
|
|
pub webpage_url: Option<String>,
|
|
/// The stream protocol.
|
|
pub protocol: Option<String>,
|
|
}
|
|
|
|
impl Output {
|
|
/// Requests auxiliary metadata which can be accessed without parsing the file.
|
|
pub fn as_aux_metadata(&self) -> AuxMetadata {
|
|
let album = self.album.clone();
|
|
let track = self.track.clone();
|
|
let true_artist = self.artist.as_ref();
|
|
let artist = true_artist.or(self.uploader.as_ref()).cloned();
|
|
let r_date = self.release_date.as_ref();
|
|
let date = r_date.or(self.upload_date.as_ref()).cloned();
|
|
let channel = self.channel.clone();
|
|
let duration = self.duration.map(Duration::from_secs_f64);
|
|
let source_url = self.webpage_url.clone();
|
|
let title = self.title.clone();
|
|
let thumbnail = self.thumbnail.clone();
|
|
|
|
AuxMetadata {
|
|
track,
|
|
artist,
|
|
album,
|
|
date,
|
|
|
|
channels: Some(2),
|
|
channel,
|
|
duration,
|
|
sample_rate: Some(SAMPLE_RATE_RAW as u32),
|
|
source_url,
|
|
title,
|
|
thumbnail,
|
|
|
|
..AuxMetadata::default()
|
|
}
|
|
}
|
|
}
|