Compare commits
No commits in common. "6d0302d3bf8e060a5d50afe3b9cef2486d7b8b4c" and "0101fc0c6b87581d14ff1fb0b8bcd3c301ee5bb9" have entirely different histories.
6d0302d3bf
...
0101fc0c6b
6 changed files with 32 additions and 96 deletions
|
@ -414,8 +414,7 @@ impl MusicListMapper {
|
||||||
Ok(MusicEntityType::Playlist)
|
Ok(MusicEntityType::Playlist)
|
||||||
}
|
}
|
||||||
PageType::Channel => {
|
PageType::Channel => {
|
||||||
// There may be broken YT channels from the artist search. They can be skipped.
|
Err(format!("channel items unsupported. id: {}", id))
|
||||||
Ok(MusicEntityType::Artist)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -139,7 +139,7 @@ impl PageType {
|
||||||
pub(crate) fn to_url_target(self, id: String) -> UrlTarget {
|
pub(crate) fn to_url_target(self, id: String) -> UrlTarget {
|
||||||
match self {
|
match self {
|
||||||
PageType::Artist => UrlTarget::Channel { id },
|
PageType::Artist => UrlTarget::Channel { id },
|
||||||
PageType::Album => UrlTarget::Album { id },
|
PageType::Album => UrlTarget::Playlist { id },
|
||||||
PageType::Channel => UrlTarget::Channel { id },
|
PageType::Channel => UrlTarget::Channel { id },
|
||||||
PageType::Playlist => UrlTarget::Playlist { id },
|
PageType::Playlist => UrlTarget::Playlist { id },
|
||||||
}
|
}
|
||||||
|
|
|
@ -20,7 +20,7 @@ struct QResolveUrl<'a> {
|
||||||
}
|
}
|
||||||
|
|
||||||
impl RustyPipeQuery {
|
impl RustyPipeQuery {
|
||||||
pub async fn resolve_url(self, url: &str, resolve_albums: bool) -> Result<UrlTarget, Error> {
|
pub async fn resolve_url(self, url: &str) -> Result<UrlTarget, Error> {
|
||||||
let (url, params) = util::url_to_params(url)?;
|
let (url, params) = util::url_to_params(url)?;
|
||||||
|
|
||||||
let mut is_shortlink = url.domain().and_then(|d| match d {
|
let mut is_shortlink = url.domain().and_then(|d| match d {
|
||||||
|
@ -61,17 +61,8 @@ impl RustyPipeQuery {
|
||||||
.ok_or(Error::Other(Cow::Borrowed("invalid url: no playlist id")))?
|
.ok_or(Error::Other(Cow::Borrowed("invalid url: no playlist id")))?
|
||||||
.to_string();
|
.to_string();
|
||||||
|
|
||||||
// YouTube Music album has to be resolved by the YTM API
|
|
||||||
if resolve_albums && id.starts_with(util::PLAYLIST_ID_ALBUM_PREFIX) {
|
|
||||||
self._navigation_resolve_url(
|
|
||||||
&format!("/playlist?list={}", id),
|
|
||||||
ClientType::DesktopMusic,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
} else {
|
|
||||||
Ok(UrlTarget::Playlist { id })
|
Ok(UrlTarget::Playlist { id })
|
||||||
}
|
}
|
||||||
}
|
|
||||||
// Channel vanity URL or youtu.be shortlink
|
// Channel vanity URL or youtu.be shortlink
|
||||||
Some(mut id) => {
|
Some(mut id) => {
|
||||||
if id == "c" || id == "user" {
|
if id == "c" || id == "user" {
|
||||||
|
@ -95,8 +86,7 @@ impl RustyPipeQuery {
|
||||||
}
|
}
|
||||||
Some(false) => {
|
Some(false) => {
|
||||||
// Vanity URL (e.g. youtube.com/LinusTechTips) has to be resolved by the Innertube API
|
// Vanity URL (e.g. youtube.com/LinusTechTips) has to be resolved by the Innertube API
|
||||||
self._navigation_resolve_url(url.path(), ClientType::Desktop)
|
self._navigation_resolve_url(url.path()).await
|
||||||
.await
|
|
||||||
}
|
}
|
||||||
None => {
|
None => {
|
||||||
// We dont have the original YT domain, so this can be both
|
// We dont have the original YT domain, so this can be both
|
||||||
|
@ -108,10 +98,7 @@ impl RustyPipeQuery {
|
||||||
.is_match(url.path())
|
.is_match(url.path())
|
||||||
.unwrap_or_default()
|
.unwrap_or_default()
|
||||||
{
|
{
|
||||||
match self
|
match self._navigation_resolve_url(url.path()).await {
|
||||||
._navigation_resolve_url(url.path(), ClientType::Desktop)
|
|
||||||
.await
|
|
||||||
{
|
|
||||||
Ok(target) => Ok(target),
|
Ok(target) => Ok(target),
|
||||||
Err(Error::Extraction(ExtractionError::ContentUnavailable(e))) => {
|
Err(Error::Extraction(ExtractionError::ContentUnavailable(e))) => {
|
||||||
match util::VIDEO_ID_REGEX.is_match(id).unwrap_or_default() {
|
match util::VIDEO_ID_REGEX.is_match(id).unwrap_or_default() {
|
||||||
|
@ -144,19 +131,14 @@ impl RustyPipeQuery {
|
||||||
Ok(target)
|
Ok(target)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn resolve_string(
|
pub async fn resolve_string(self, string: &str) -> Result<UrlTarget, Error> {
|
||||||
self,
|
|
||||||
string: &str,
|
|
||||||
resolve_albums: bool,
|
|
||||||
) -> Result<UrlTarget, Error> {
|
|
||||||
// URL with protocol
|
// URL with protocol
|
||||||
if string.starts_with("http://") || string.starts_with("https://") {
|
if string.starts_with("http://") || string.starts_with("https://") {
|
||||||
self.resolve_url(string, resolve_albums).await
|
self.resolve_url(string).await
|
||||||
}
|
}
|
||||||
// URL without protocol
|
// URL without protocol
|
||||||
else if string.contains('/') && string.contains('.') {
|
else if string.contains('/') && string.contains('.') {
|
||||||
self.resolve_url(&format!("https://{}", string), resolve_albums)
|
self.resolve_url(&format!("https://{}", string)).await
|
||||||
.await
|
|
||||||
}
|
}
|
||||||
// ID only
|
// ID only
|
||||||
else if util::VIDEO_ID_REGEX.is_match(string).unwrap_or_default() {
|
else if util::VIDEO_ID_REGEX.is_match(string).unwrap_or_default() {
|
||||||
|
@ -169,54 +151,28 @@ impl RustyPipeQuery {
|
||||||
id: string.to_owned(),
|
id: string.to_owned(),
|
||||||
})
|
})
|
||||||
} else if util::PLAYLIST_ID_REGEX.is_match(string).unwrap_or_default() {
|
} else if util::PLAYLIST_ID_REGEX.is_match(string).unwrap_or_default() {
|
||||||
if resolve_albums && string.starts_with(util::PLAYLIST_ID_ALBUM_PREFIX) {
|
|
||||||
self._navigation_resolve_url(
|
|
||||||
&format!("/playlist?list={}", string),
|
|
||||||
ClientType::DesktopMusic,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
} else {
|
|
||||||
Ok(UrlTarget::Playlist {
|
Ok(UrlTarget::Playlist {
|
||||||
id: string.to_owned(),
|
id: string.to_owned(),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
} else if util::ALBUM_ID_REGEX.is_match(string).unwrap_or_default() {
|
|
||||||
Ok(UrlTarget::Album {
|
|
||||||
id: string.to_owned(),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
// Channel name only
|
// Channel name only
|
||||||
else if util::VANITY_PATH_REGEX.is_match(string).unwrap_or_default() {
|
else if util::VANITY_PATH_REGEX.is_match(string).unwrap_or_default() {
|
||||||
self._navigation_resolve_url(
|
self._navigation_resolve_url(&format!("/{}", string.trim_start_matches('/')))
|
||||||
&format!("/{}", string.trim_start_matches('/')),
|
|
||||||
ClientType::Desktop,
|
|
||||||
)
|
|
||||||
.await
|
.await
|
||||||
} else {
|
} else {
|
||||||
Err(Error::Other("invalid input string".into()))
|
Err(Error::Other("invalid input string".into()))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn _navigation_resolve_url(
|
async fn _navigation_resolve_url(&self, url_path: &str) -> Result<UrlTarget, Error> {
|
||||||
&self,
|
let context = self.get_context(ClientType::Desktop, true, None).await;
|
||||||
url_path: &str,
|
|
||||||
ctype: ClientType,
|
|
||||||
) -> Result<UrlTarget, Error> {
|
|
||||||
let context = self.get_context(ctype, true, None).await;
|
|
||||||
let request_body = QResolveUrl {
|
let request_body = QResolveUrl {
|
||||||
context,
|
context,
|
||||||
url: format!(
|
url: format!("https://www.youtube.com{}", url_path),
|
||||||
"https://{}.youtube.com{}",
|
|
||||||
match ctype {
|
|
||||||
ClientType::DesktopMusic => "music",
|
|
||||||
_ => "www",
|
|
||||||
},
|
|
||||||
url_path
|
|
||||||
),
|
|
||||||
};
|
};
|
||||||
|
|
||||||
self.execute_request::<response::ResolvedUrl, _, _>(
|
self.execute_request::<response::ResolvedUrl, _, _>(
|
||||||
ctype,
|
ClientType::Desktop,
|
||||||
"channel_id",
|
"channel_id",
|
||||||
&request_body.url,
|
&request_body.url,
|
||||||
"navigation/resolve_url",
|
"navigation/resolve_url",
|
||||||
|
@ -233,24 +189,23 @@ impl MapResponse<UrlTarget> for response::ResolvedUrl {
|
||||||
_lang: Language,
|
_lang: Language,
|
||||||
_deobf: Option<&crate::deobfuscate::Deobfuscator>,
|
_deobf: Option<&crate::deobfuscate::Deobfuscator>,
|
||||||
) -> Result<MapResult<UrlTarget>, ExtractionError> {
|
) -> Result<MapResult<UrlTarget>, ExtractionError> {
|
||||||
let browse_endpoint = self
|
|
||||||
.endpoint
|
|
||||||
.browse_endpoint
|
|
||||||
.ok_or(ExtractionError::InvalidData(Cow::Borrowed("No browse ID")))?;
|
|
||||||
|
|
||||||
let page_type = self
|
let page_type = self
|
||||||
.endpoint
|
.endpoint
|
||||||
.command_metadata
|
.command_metadata
|
||||||
.map(|c| c.web_command_metadata.web_page_type)
|
.ok_or(ExtractionError::InvalidData(Cow::Borrowed(
|
||||||
.or_else(|| {
|
"No command metadata",
|
||||||
browse_endpoint
|
)))?
|
||||||
.browse_endpoint_context_supported_configs
|
.web_command_metadata
|
||||||
.map(|c| c.browse_endpoint_context_music_config.page_type)
|
.web_page_type;
|
||||||
})
|
|
||||||
.ok_or(ExtractionError::InvalidData(Cow::Borrowed("No page type")))?;
|
let id = self
|
||||||
|
.endpoint
|
||||||
|
.browse_endpoint
|
||||||
|
.ok_or(ExtractionError::InvalidData(Cow::Borrowed("No browse ID")))?
|
||||||
|
.browse_id;
|
||||||
|
|
||||||
Ok(MapResult {
|
Ok(MapResult {
|
||||||
c: page_type.to_url_target(browse_endpoint.browse_id),
|
c: page_type.to_url_target(id),
|
||||||
warnings: Vec::new(),
|
warnings: Vec::new(),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
|
@ -41,7 +41,6 @@ pub enum UrlTarget {
|
||||||
Video { id: String, start_time: u32 },
|
Video { id: String, start_time: u32 },
|
||||||
Channel { id: String },
|
Channel { id: String },
|
||||||
Playlist { id: String },
|
Playlist { id: String },
|
||||||
Album { id: String },
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ToString for UrlTarget {
|
impl ToString for UrlTarget {
|
||||||
|
@ -67,11 +66,6 @@ impl UrlTarget {
|
||||||
UrlTarget::Playlist { id } => {
|
UrlTarget::Playlist { id } => {
|
||||||
format!("{}/playlist?list={}", yt_host, id)
|
format!("{}/playlist?list={}", yt_host, id)
|
||||||
}
|
}
|
||||||
UrlTarget::Album { id } => {
|
|
||||||
// The official album URLs use the playlist ID
|
|
||||||
// This looks weird, but it works
|
|
||||||
format!("{}/channel/{}", yt_host, id)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -95,12 +89,6 @@ impl UrlTarget {
|
||||||
false => Err(Error::Other("invalid playlist id".into())),
|
false => Err(Error::Other("invalid playlist id".into())),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
UrlTarget::Album { id } => {
|
|
||||||
match util::ALBUM_ID_REGEX.is_match(id).unwrap_or_default() {
|
|
||||||
true => Ok(()),
|
|
||||||
false => Err(Error::Other("invalid album id".into())),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -23,9 +23,7 @@ pub static VIDEO_ID_REGEX: Lazy<Regex> = Lazy::new(|| Regex::new(r"^[A-Za-z0-9_-
|
||||||
pub static CHANNEL_ID_REGEX: Lazy<Regex> =
|
pub static CHANNEL_ID_REGEX: Lazy<Regex> =
|
||||||
Lazy::new(|| Regex::new(r"^UC[A-Za-z0-9_-]{22}$").unwrap());
|
Lazy::new(|| Regex::new(r"^UC[A-Za-z0-9_-]{22}$").unwrap());
|
||||||
pub static PLAYLIST_ID_REGEX: Lazy<Regex> =
|
pub static PLAYLIST_ID_REGEX: Lazy<Regex> =
|
||||||
Lazy::new(|| Regex::new(r"^(?:PL|RD|OLAK)[A-Za-z0-9_-]{30,}$").unwrap());
|
Lazy::new(|| Regex::new(r"^(?:PL|RD)[A-Za-z0-9_-]{30,}$").unwrap());
|
||||||
pub static ALBUM_ID_REGEX: Lazy<Regex> =
|
|
||||||
Lazy::new(|| Regex::new(r"^MPREb_[A-Za-z0-9_-]{11}$").unwrap());
|
|
||||||
pub static VANITY_PATH_REGEX: Lazy<Regex> =
|
pub static VANITY_PATH_REGEX: Lazy<Regex> =
|
||||||
Lazy::new(|| Regex::new(r"^/?(?:(?:c\/|user\/)?[A-z0-9]+)|(?:@[A-z0-9-_.]+)$").unwrap());
|
Lazy::new(|| Regex::new(r"^/?(?:(?:c\/|user\/)?[A-z0-9]+)|(?:@[A-z0-9-_.]+)$").unwrap());
|
||||||
|
|
||||||
|
@ -34,7 +32,6 @@ pub const DOT_SEPARATOR: &str = " • ";
|
||||||
/// YouTube Music name (author of official playlists)
|
/// YouTube Music name (author of official playlists)
|
||||||
pub const YT_MUSIC_NAME: &str = "YouTube Music";
|
pub const YT_MUSIC_NAME: &str = "YouTube Music";
|
||||||
pub const VARIOUS_ARTISTS: &str = "Various Artists";
|
pub const VARIOUS_ARTISTS: &str = "Various Artists";
|
||||||
pub const PLAYLIST_ID_ALBUM_PREFIX: &str = "OLAK";
|
|
||||||
|
|
||||||
const CONTENT_PLAYBACK_NONCE_ALPHABET: &[u8; 64] =
|
const CONTENT_PLAYBACK_NONCE_ALPHABET: &[u8; 64] =
|
||||||
b"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_";
|
b"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_";
|
||||||
|
|
|
@ -1165,11 +1165,10 @@ async fn search_suggestion_empty() {
|
||||||
#[case("https://piped.mha.fi/dQw4w9WgXcQ", UrlTarget::Channel {id: "UCoG6BrhgmivrkcbEHcYtK4Q".to_owned()})]
|
#[case("https://piped.mha.fi/dQw4w9WgXcQ", UrlTarget::Channel {id: "UCoG6BrhgmivrkcbEHcYtK4Q".to_owned()})]
|
||||||
// Both a video ID and a channel name + video time param => returns video
|
// Both a video ID and a channel name + video time param => returns video
|
||||||
#[case("https://piped.mha.fi/dQw4w9WgXcQ?t=0", UrlTarget::Video {id: "dQw4w9WgXcQ".to_owned(), start_time: 0})]
|
#[case("https://piped.mha.fi/dQw4w9WgXcQ?t=0", UrlTarget::Video {id: "dQw4w9WgXcQ".to_owned(), start_time: 0})]
|
||||||
#[case("https://music.youtube.com/playlist?list=OLAK5uy_k0yFrZlFRgCf3rLPza-lkRmCrtLPbK9pE", UrlTarget::Album {id: "MPREb_GyH43gCvdM5".to_owned()})]
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn resolve_url(#[case] url: &str, #[case] expect: UrlTarget) {
|
async fn resolve_url(#[case] url: &str, #[case] expect: UrlTarget) {
|
||||||
let rp = RustyPipe::builder().strict().build();
|
let rp = RustyPipe::builder().strict().build();
|
||||||
let target = rp.query().resolve_url(url, true).await.unwrap();
|
let target = rp.query().resolve_url(url).await.unwrap();
|
||||||
assert_eq!(target, expect);
|
assert_eq!(target, expect);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1183,12 +1182,10 @@ async fn resolve_url(#[case] url: &str, #[case] expect: UrlTarget) {
|
||||||
#[case("dQw4w9WgXcQ", UrlTarget::Video {id: "dQw4w9WgXcQ".to_owned(), start_time: 0})]
|
#[case("dQw4w9WgXcQ", UrlTarget::Video {id: "dQw4w9WgXcQ".to_owned(), start_time: 0})]
|
||||||
#[case("PL4lEESSgxM_5O81EvKCmBIm_JT5Q7JeaI", UrlTarget::Playlist {id: "PL4lEESSgxM_5O81EvKCmBIm_JT5Q7JeaI".to_owned()})]
|
#[case("PL4lEESSgxM_5O81EvKCmBIm_JT5Q7JeaI", UrlTarget::Playlist {id: "PL4lEESSgxM_5O81EvKCmBIm_JT5Q7JeaI".to_owned()})]
|
||||||
#[case("RDCLAK5uy_kFQXdnqMaQCVx2wpUM4ZfbsGCDibZtkJk", UrlTarget::Playlist {id: "RDCLAK5uy_kFQXdnqMaQCVx2wpUM4ZfbsGCDibZtkJk".to_owned()})]
|
#[case("RDCLAK5uy_kFQXdnqMaQCVx2wpUM4ZfbsGCDibZtkJk", UrlTarget::Playlist {id: "RDCLAK5uy_kFQXdnqMaQCVx2wpUM4ZfbsGCDibZtkJk".to_owned()})]
|
||||||
#[case("OLAK5uy_k0yFrZlFRgCf3rLPza-lkRmCrtLPbK9pE", UrlTarget::Album {id: "MPREb_GyH43gCvdM5".to_owned()})]
|
|
||||||
#[case("MPREb_GyH43gCvdM5", UrlTarget::Album {id: "MPREb_GyH43gCvdM5".to_owned()})]
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn resolve_string(#[case] string: &str, #[case] expect: UrlTarget) {
|
async fn resolve_string(#[case] string: &str, #[case] expect: UrlTarget) {
|
||||||
let rp = RustyPipe::builder().strict().build();
|
let rp = RustyPipe::builder().strict().build();
|
||||||
let target = rp.query().resolve_string(string, true).await.unwrap();
|
let target = rp.query().resolve_string(string).await.unwrap();
|
||||||
assert_eq!(target, expect);
|
assert_eq!(target, expect);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1197,7 +1194,7 @@ async fn resolve_channel_not_found() {
|
||||||
let rp = RustyPipe::builder().strict().build();
|
let rp = RustyPipe::builder().strict().build();
|
||||||
let err = rp
|
let err = rp
|
||||||
.query()
|
.query()
|
||||||
.resolve_url("https://www.youtube.com/feeqegnhq3rkwghjq43ruih43io3", true)
|
.resolve_url("https://www.youtube.com/feeqegnhq3rkwghjq43ruih43io3")
|
||||||
.await
|
.await
|
||||||
.unwrap_err();
|
.unwrap_err();
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue