103 lines
3.1 KiB
Rust
103 lines
3.1 KiB
Rust
use crate::spotify::UriType;
|
|
|
|
use url::{Host, Url};
|
|
|
|
pub struct SpotifyUrl {
|
|
pub id: String,
|
|
pub uri_type: UriType,
|
|
}
|
|
|
|
impl SpotifyUrl {
|
|
fn new(id: &str, uri_type: UriType) -> SpotifyUrl {
|
|
SpotifyUrl {
|
|
id: id.to_string(),
|
|
uri_type,
|
|
}
|
|
}
|
|
|
|
/// Get media id and type from open.spotify.com url
|
|
///
|
|
/// ```
|
|
/// let result = spotify_url::SpotifyURL::from_url("https://open.spotify.com/track/4uLU6hMCjMI75M1A2tKUQC").unwrap();
|
|
/// assert_eq!(result.id, "4uLU6hMCjMI75M1A2tKUQC");
|
|
/// assert_eq!(result.uri_type, URIType::Track);
|
|
/// ```
|
|
pub fn from_url(s: &str) -> Option<SpotifyUrl> {
|
|
let url = Url::parse(s).ok()?;
|
|
if url.host() != Some(Host::Domain("open.spotify.com")) {
|
|
return None;
|
|
}
|
|
|
|
let mut path_segments = url.path_segments()?;
|
|
|
|
let entity = path_segments.next()?;
|
|
|
|
let uri_type = match entity.to_lowercase().as_str() {
|
|
"album" => Some(UriType::Album),
|
|
"artist" => Some(UriType::Artist),
|
|
"episode" => Some(UriType::Episode),
|
|
"playlist" => Some(UriType::Playlist),
|
|
"show" => Some(UriType::Show),
|
|
"track" => Some(UriType::Track),
|
|
"user" => {
|
|
let _user_id = path_segments.next()?;
|
|
let entity = path_segments.next()?;
|
|
|
|
if entity != "playlist" {
|
|
return None;
|
|
}
|
|
|
|
Some(UriType::Playlist)
|
|
}
|
|
_ => None,
|
|
}?;
|
|
|
|
let id = path_segments.next()?;
|
|
|
|
Some(SpotifyUrl::new(id, uri_type))
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use std::collections::HashMap;
|
|
|
|
use super::SpotifyUrl;
|
|
use crate::spotify::UriType;
|
|
|
|
#[test]
|
|
fn test_urls() {
|
|
let mut test_cases = HashMap::new();
|
|
test_cases.insert(
|
|
"https://open.spotify.com/playlist/1XFxe8bkTryTODn0lk4CNa?si=FfSpZ6KPQdieClZbwHakOQ",
|
|
SpotifyUrl::new("1XFxe8bkTryTODn0lk4CNa", UriType::Playlist),
|
|
);
|
|
test_cases.insert(
|
|
"https://open.spotify.com/track/6fRJg3R90w0juYoCJXxj2d",
|
|
SpotifyUrl::new("6fRJg3R90w0juYoCJXxj2d", UriType::Track),
|
|
);
|
|
test_cases.insert(
|
|
"https://open.spotify.com/user/~villainy~/playlist/0OgoSs65CLDPn6AF6tsZVg",
|
|
SpotifyUrl::new("0OgoSs65CLDPn6AF6tsZVg", UriType::Playlist),
|
|
);
|
|
test_cases.insert(
|
|
"https://open.spotify.com/show/4MZfJbM2MXzZdPbv6gi5lJ",
|
|
SpotifyUrl::new("4MZfJbM2MXzZdPbv6gi5lJ", UriType::Show),
|
|
);
|
|
test_cases.insert(
|
|
"https://open.spotify.com/episode/3QE6rfmjRaeqXSqeWcIWF6",
|
|
SpotifyUrl::new("3QE6rfmjRaeqXSqeWcIWF6", UriType::Episode),
|
|
);
|
|
test_cases.insert(
|
|
"https://open.spotify.com/artist/6LEeAFiJF8OuPx747e1wxR",
|
|
SpotifyUrl::new("6LEeAFiJF8OuPx747e1wxR", UriType::Artist),
|
|
);
|
|
|
|
for case in test_cases {
|
|
let result = SpotifyUrl::from_url(case.0).unwrap();
|
|
assert_eq!(result.id, case.1.id);
|
|
assert_eq!(result.uri_type, case.1.uri_type);
|
|
}
|
|
}
|
|
}
|