Compare commits
2 Commits
f0b904f17e
...
34c294ef0f
Author | SHA1 | Date |
---|---|---|
Davide Polonio | 34c294ef0f | |
Davide Polonio | 06c0db1574 |
34
src/main.rs
34
src/main.rs
|
@ -1,11 +1,18 @@
|
||||||
use crate::SpotifyURL::Track;
|
use crate::SpotifyURL::Track;
|
||||||
use aspotify::{Client, ClientCredentials};
|
use aspotify::{Client, ClientCredentials};
|
||||||
|
use std::time::Duration;
|
||||||
use teloxide::prelude::*;
|
use teloxide::prelude::*;
|
||||||
|
|
||||||
enum SpotifyURL {
|
enum SpotifyURL {
|
||||||
Track(String),
|
Track(String),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
struct TrackInfo {
|
||||||
|
name: String,
|
||||||
|
artist: Vec<String>,
|
||||||
|
duration: Duration,
|
||||||
|
}
|
||||||
|
|
||||||
fn get_spotify_entry(url: &str) -> Option<SpotifyURL> {
|
fn get_spotify_entry(url: &str) -> Option<SpotifyURL> {
|
||||||
if url.contains("https://open.spotify.com/track/") {
|
if url.contains("https://open.spotify.com/track/") {
|
||||||
let track_id = url.rsplit('/').next().and_then(|x| x.split('?').next());
|
let track_id = url.rsplit('/').next().and_then(|x| x.split('?').next());
|
||||||
|
@ -17,16 +24,12 @@ fn get_spotify_entry(url: &str) -> Option<SpotifyURL> {
|
||||||
return None;
|
return None;
|
||||||
}
|
}
|
||||||
|
|
||||||
struct TrackInfo {
|
|
||||||
name: String,
|
|
||||||
artist: Vec<String>,
|
|
||||||
}
|
|
||||||
|
|
||||||
async fn get_spotify_track(spotify: Box<Client>, id: &String) -> Option<TrackInfo> {
|
async fn get_spotify_track(spotify: Box<Client>, id: &String) -> Option<TrackInfo> {
|
||||||
match spotify.tracks().get_track(id.as_str(), None).await {
|
match spotify.tracks().get_track(id.as_str(), None).await {
|
||||||
Ok(track) => Some(TrackInfo {
|
Ok(track) => Some(TrackInfo {
|
||||||
name: track.data.name,
|
name: track.data.name,
|
||||||
artist: track.data.artists.iter().map(|x| x.name.clone()).collect(),
|
artist: track.data.artists.iter().map(|x| x.name.clone()).collect(),
|
||||||
|
duration: track.data.duration,
|
||||||
}),
|
}),
|
||||||
Err(_e) => None,
|
Err(_e) => None,
|
||||||
}
|
}
|
||||||
|
@ -39,31 +42,34 @@ async fn main() {
|
||||||
|
|
||||||
let bot = Bot::from_env().auto_send();
|
let bot = Bot::from_env().auto_send();
|
||||||
teloxide::repl(bot, |message| async move {
|
teloxide::repl(bot, |message| async move {
|
||||||
|
let text = message.update.text().and_then(get_spotify_entry);
|
||||||
|
match text {
|
||||||
|
Some(spotify) => {
|
||||||
let spotify_creds =
|
let spotify_creds =
|
||||||
ClientCredentials::from_env().expect("CLIENT_ID and CLIENT_SECRET not found.");
|
ClientCredentials::from_env().expect("CLIENT_ID and CLIENT_SECRET not found.");
|
||||||
let spotify_client = Box::new(Client::new(spotify_creds));
|
let spotify_client = Box::new(Client::new(spotify_creds));
|
||||||
|
match spotify {
|
||||||
log::info!("Connected to Spotify");
|
|
||||||
let text = message.update.text().and_then(get_spotify_entry);
|
|
||||||
match text {
|
|
||||||
Some(spotify) => match spotify {
|
|
||||||
Track(id) => {
|
Track(id) => {
|
||||||
|
log::debug!("Parsing spotify song: {}", id);
|
||||||
let track_info = get_spotify_track(spotify_client, &id).await;
|
let track_info = get_spotify_track(spotify_client, &id).await;
|
||||||
match track_info {
|
match track_info {
|
||||||
Some(info) => {
|
Some(info) => {
|
||||||
let reply = format!(
|
let reply = format!(
|
||||||
"Track information:\n\
|
"Track information:\n\
|
||||||
Track name: {}\n\
|
🎵 Track name: {}\n\
|
||||||
Artists: {}",
|
🧑🎤 Artist(s): {}\n\
|
||||||
|
⏳ Duration: {} second(s)",
|
||||||
info.name,
|
info.name,
|
||||||
info.artist.join(", ")
|
info.artist.join(", "),
|
||||||
|
info.duration.as_secs()
|
||||||
);
|
);
|
||||||
Some(message.reply_to(reply).await?)
|
Some(message.reply_to(reply).await?)
|
||||||
}
|
}
|
||||||
None => None,
|
None => None,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
}
|
||||||
|
}
|
||||||
None => None,
|
None => None,
|
||||||
};
|
};
|
||||||
respond(())
|
respond(())
|
||||||
|
|
Loading…
Reference in New Issue