Add spotify url support for music
Signed-off-by: trivernis <trivernis@protonmail.com>pull/2/head
parent
8a94518747
commit
ca1f9e7ecf
GPG Key ID:
DFFFCC2C7A02DB45
@ -0,0 +1,100 @@
|
||||
use crate::utils::error::{BotError, BotResult};
|
||||
use aspotify::{Client, ClientCredentials, PlaylistItemType};
|
||||
|
||||
pub struct SpotifyApi {
|
||||
client: Client,
|
||||
}
|
||||
|
||||
impl SpotifyApi {
|
||||
/// Creates a new spotify api wrapper with the credentials stored
|
||||
/// in the .env files
|
||||
pub fn new() -> Self {
|
||||
let credentials = ClientCredentials {
|
||||
id: dotenv::var("SPOTIFY_CLIENT_ID").expect("Missing Spotify Credentials"),
|
||||
secret: dotenv::var("SPOTIFY_CLIENT_SECRET").expect("Missing Spotify Credentials"),
|
||||
};
|
||||
let client = Client::new(credentials);
|
||||
|
||||
Self { client }
|
||||
}
|
||||
|
||||
/// Returns the song names for a playlist
|
||||
pub async fn get_songs_in_playlist(&self, url: &str) -> BotResult<Vec<String>> {
|
||||
let id = self.get_id_for_url(url)?;
|
||||
let playlist = self.client.playlists().get_playlist(&*id, None).await?.data;
|
||||
let song_names = playlist
|
||||
.tracks
|
||||
.items
|
||||
.into_iter()
|
||||
.filter_map(|item| item.item)
|
||||
.map(|t| match t {
|
||||
PlaylistItemType::Track(t) => format!(
|
||||
"{} - {}",
|
||||
t.artists
|
||||
.into_iter()
|
||||
.map(|a| a.name)
|
||||
.collect::<Vec<String>>()
|
||||
.join(","),
|
||||
t.name
|
||||
),
|
||||
PlaylistItemType::Episode(e) => e.name,
|
||||
})
|
||||
.collect();
|
||||
|
||||
Ok(song_names)
|
||||
}
|
||||
|
||||
/// Returns all song names for a given album
|
||||
pub async fn get_songs_in_album(&self, url: &str) -> BotResult<Vec<String>> {
|
||||
let id = self.get_id_for_url(url)?;
|
||||
let album = self.client.albums().get_album(&*id, None).await?.data;
|
||||
let song_names = album
|
||||
.tracks
|
||||
.items
|
||||
.into_iter()
|
||||
.map(|item| {
|
||||
format!(
|
||||
"{} - {}",
|
||||
item.artists
|
||||
.into_iter()
|
||||
.map(|a| a.name)
|
||||
.collect::<Vec<String>>()
|
||||
.join(","),
|
||||
item.name
|
||||
)
|
||||
})
|
||||
.collect();
|
||||
|
||||
Ok(song_names)
|
||||
}
|
||||
|
||||
/// Returns the name for a spotify song url
|
||||
pub async fn get_song_name(&self, url: &str) -> BotResult<String> {
|
||||
let id = self.get_id_for_url(url)?;
|
||||
let track = self.client.tracks().get_track(&*id, None).await?.data;
|
||||
|
||||
Ok(format!(
|
||||
"{} - {}",
|
||||
track
|
||||
.artists
|
||||
.into_iter()
|
||||
.map(|a| a.name)
|
||||
.collect::<Vec<String>>()
|
||||
.join(","),
|
||||
track.name
|
||||
))
|
||||
}
|
||||
|
||||
/// Returns the id for a given spotify URL
|
||||
fn get_id_for_url(&self, url: &str) -> BotResult<String> {
|
||||
url.split('/')
|
||||
.last()
|
||||
.ok_or(BotError::from("Invalid Spotify URL"))
|
||||
.and_then(|s| {
|
||||
s.split('?')
|
||||
.next()
|
||||
.ok_or(BotError::from("Invalid Spotify URL"))
|
||||
})
|
||||
.map(|s| s.to_string())
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue