Add support for open.spotify.com links (#392)
* No longer necessary * Add support for open.spotify.com links * Reuse struct for insert command * Formatting
This commit is contained in:
102
src/spotify_url.rs
Normal file
102
src/spotify_url.rs
Normal file
@@ -0,0 +1,102 @@
|
||||
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);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user