chore: clippy improvements

This commit is contained in:
Miguel da Mota 2024-03-09 20:09:17 +01:00
parent e1cf394362
commit 9c1f6bee6d
6 changed files with 143 additions and 92 deletions

View file

@ -1,14 +1,15 @@
use crate::music::music_manager;
use serenity::all::{CommandDataOptionValue, CommandInteraction, Context}; use serenity::all::{CommandDataOptionValue, CommandInteraction, Context};
use serenity::builder::{CreateCommand, CreateCommandOption, CreateEmbed}; use serenity::builder::{CreateCommand, CreateCommandOption, CreateEmbed};
use serenity::model::application::CommandOptionType; use serenity::model::application::CommandOptionType;
use crate::music::music_manager;
use crate::util::embed::Embed; use crate::util::embed::Embed;
pub async unsafe fn run(ctx: &Context, command: &CommandInteraction) -> CreateEmbed { pub async unsafe fn run(ctx: &Context, command: &CommandInteraction) -> CreateEmbed {
let username = command.user.name.as_str(); let username = command.user.name.as_str();
let options = &command.data.options; let options = &command.data.options;
let query = options.first().and_then(|option| { let query = options.first().and_then(|option| {
if let CommandDataOptionValue::String(query) = &option.value { if let CommandDataOptionValue::String(query) = &option.value {
Some(query) Some(query)
@ -16,7 +17,7 @@ pub async unsafe fn run(ctx: &Context, command: &CommandInteraction) -> CreateEm
None None
} }
}); });
if query.is_none() { if query.is_none() {
return Embed::create(username, "Error 400", "There is no query provided"); return Embed::create(username, "Error 400", "There is no query provided");
} }
@ -27,8 +28,15 @@ pub async unsafe fn run(ctx: &Context, command: &CommandInteraction) -> CreateEm
return Embed::create(username, "GuildId not found", "Could not find guild id."); return Embed::create(username, "GuildId not found", "Could not find guild id.");
} }
}; };
music_manager::attempt_to_queue_song(&ctx, &guild_id, &command.user.id, &command.user.name, query.unwrap()).await music_manager::attempt_to_queue_song(
ctx,
guild_id,
&command.user.id,
&command.user.name,
query.unwrap(),
)
.await
} }
pub fn register() -> CreateCommand { pub fn register() -> CreateCommand {

View file

@ -1,7 +1,7 @@
use serenity::all::{CommandInteraction, Context};
use serenity::builder::{CreateCommand, CreateEmbed};
use crate::music::music_manager; use crate::music::music_manager;
use crate::util::embed::Embed; use crate::util::embed::Embed;
use serenity::all::{CommandInteraction, Context};
use serenity::builder::{CreateCommand, CreateEmbed};
pub async unsafe fn run(ctx: &Context, command: &CommandInteraction) -> CreateEmbed { pub async unsafe fn run(ctx: &Context, command: &CommandInteraction) -> CreateEmbed {
let username = command.user.name.as_str(); let username = command.user.name.as_str();
@ -13,7 +13,7 @@ pub async unsafe fn run(ctx: &Context, command: &CommandInteraction) -> CreateEm
} }
}; };
music_manager::attempt_to_stop(&ctx, &guild_id, &command.user.id, &command.user.name).await music_manager::attempt_to_stop(ctx, guild_id, &command.user.id, &command.user.name).await
} }
pub fn register() -> CreateCommand { pub fn register() -> CreateCommand {

View file

@ -1,8 +1,8 @@
use std::sync::Arc; use crate::music::{music_manager, music_queue};
use serenity::all::{ChannelId, GuildId, Http}; use serenity::all::{ChannelId, GuildId, Http};
use serenity::async_trait; use serenity::async_trait;
use songbird::{Event, EventContext, EventHandler}; use songbird::{Event, EventContext, EventHandler};
use crate::music::{music_manager, music_queue}; use std::sync::Arc;
pub struct TrackEndNotifier { pub struct TrackEndNotifier {
pub guild_id: GuildId, pub guild_id: GuildId,
@ -14,12 +14,13 @@ pub struct TrackEndNotifier {
#[async_trait] #[async_trait]
impl EventHandler for TrackEndNotifier { impl EventHandler for TrackEndNotifier {
async fn act(&self, ctx: &EventContext<'_>) -> Option<Event> { async fn act(&self, ctx: &EventContext<'_>) -> Option<Event> {
unsafe { // TODO: Does this need to be unsafe? unsafe {
if let EventContext::Track(track_list) = ctx { // TODO: Does this need to be unsafe?
if let EventContext::Track(..) = ctx {
println!("The track ended!"); println!("The track ended!");
let music_queue = music_queue::get_queue(&self.guild_id); let music_queue = music_queue::get_queue(&self.guild_id);
let q = &music_queue.queue; let q = &music_queue.queue;
if q.len() == 0 { if q.is_empty() {
// No more songs in queue, exit the vc // No more songs in queue, exit the vc
let stopped = match music_manager::stop(&self.cmdctx, &self.guild_id).await { let stopped = match music_manager::stop(&self.cmdctx, &self.guild_id).await {
Ok(stopped) => stopped, Ok(stopped) => stopped,
@ -30,7 +31,7 @@ impl EventHandler for TrackEndNotifier {
}; };
if stopped { if stopped {
println!("Stopped playing successfully."); println!("Stopped playing successfully.");
} else { } else {
println!("Failed to stop playing."); println!("Failed to stop playing.");
} }
return None; return None;

View file

@ -1,26 +1,35 @@
use std::sync::{Arc};
use std::time::Duration;
use serenity::all::{Context, CreateEmbed, GuildId, UserId};
use songbird::{Event, TrackEvent};
use songbird::error::JoinError;
use songbird::input::{Compose, YoutubeDl};
use crate::HttpKey;
use crate::music::{music_events, music_queue}; use crate::music::{music_events, music_queue};
use crate::util::embed::Embed; use crate::util::embed::Embed;
use crate::util::user_util; use crate::util::user_util;
use crate::util::user_util::get_vc_id; use crate::util::user_util::get_vc_id;
use crate::HttpKey;
use serenity::all::{Context, CreateEmbed, GuildId, UserId};
use songbird::error::JoinError;
use songbird::input::{Compose, YoutubeDl};
use songbird::{Event, TrackEvent};
use std::sync::Arc;
use std::time::Duration;
pub async unsafe fn attempt_to_queue_song(ctx: &Context, guild_id: &GuildId, user_id: &UserId, username: &str, query: &str) -> CreateEmbed { pub async unsafe fn attempt_to_queue_song(
ctx: &Context,
guild_id: &GuildId,
user_id: &UserId,
username: &str,
query: &str,
) -> CreateEmbed {
if !user_util::is_user_connected_to_vc(ctx, guild_id, user_id).await { if !user_util::is_user_connected_to_vc(ctx, guild_id, user_id).await {
return Embed::create(username, "You are not connected to a VC", "Connect to my VC to control the music."); return Embed::create(
username,
"You are not connected to a VC",
"Connect to my VC to control the music.",
);
} }
let connect_to = match get_vc_id(ctx, &guild_id, &user_id).await { let connect_to = match get_vc_id(ctx, guild_id, user_id).await {
Some(channel_id) => channel_id, Some(channel_id) => channel_id,
None => { None => {
return Embed::create(username, "Error", "Cannot get channel id"); return Embed::create(username, "Error", "Cannot get channel id");
} }
}; };
let manager = &songbird::get(ctx) // TODO match let manager = &songbird::get(ctx) // TODO match
@ -28,18 +37,18 @@ pub async unsafe fn attempt_to_queue_song(ctx: &Context, guild_id: &GuildId, use
.expect("Cannot get Songbird.") .expect("Cannot get Songbird.")
.clone(); .clone();
let self_channel = user_util::get_self_vc_id(ctx, &guild_id).await; let self_channel = user_util::get_self_vc_id(ctx, guild_id).await;
if !user_util::is_self_connected_to_vc(ctx, &guild_id).await { if !user_util::is_self_connected_to_vc(ctx, guild_id).await {
// self is connected to vc, check if user is in same vc // self is connected to vc, check if user is in same vc
if self_channel.is_none() { // TODO This could maybe be removed? if self_channel.is_none() {
// TODO This could maybe be removed?
// Connect to VC // Connect to VC
manager manager
.join(*guild_id, connect_to) .join(*guild_id, connect_to)
.await .await
.expect("Cannot connect>..."); .expect("Cannot connect>...");
} }
} else { } else {
let self_channel = self_channel.expect("Cannot get self channel"); let self_channel = self_channel.expect("Cannot get self channel");
@ -52,7 +61,7 @@ pub async unsafe fn attempt_to_queue_song(ctx: &Context, guild_id: &GuildId, use
); );
} }
} }
// Get query // Get query
let do_search = !query.starts_with("http"); let do_search = !query.starts_with("http");
let http_client = { let http_client = {
@ -61,30 +70,33 @@ pub async unsafe fn attempt_to_queue_song(ctx: &Context, guild_id: &GuildId, use
.cloned() .cloned()
.expect("Guaranteed to exist in the typemap.") .expect("Guaranteed to exist in the typemap.")
}; };
// Create source // Create source
let mut src = if do_search { let mut src = if do_search {
YoutubeDl::new_search(http_client, query.to_string()) YoutubeDl::new_search(http_client, query.to_string())
} else { } else {
YoutubeDl::new(http_client, query.to_string()) YoutubeDl::new(http_client, query.to_string())
}; };
let currently_playing = music_queue::get_now_playing(&guild_id); let currently_playing = music_queue::get_now_playing(guild_id);
music_queue::add_to_queue(&guild_id, src.clone()); music_queue::add_to_queue(guild_id, src.clone());
if currently_playing.is_some() { if currently_playing.is_some() {
// Add to queue // Add to queue
return Embed::create(username, "Added to queue", "The song was added to the queue."); return Embed::create(
username,
"Added to queue",
"The song was added to the queue.",
);
} }
let query = music_queue::get_head(&guild_id).expect("Cannot get head of queue"); let _query = music_queue::get_head(guild_id).expect("Cannot get head of queue");
music_queue::set_now_playing(&guild_id, Some(src.clone())); music_queue::set_now_playing(guild_id, Some(src.clone()));
let handler_lock = match manager.get(*guild_id) { let handler_lock = match manager.get(*guild_id) {
Some(handler) => handler, Some(handler) => handler,
None => { None => {
return Embed::create("", "Error", "Cannot get handler"); return Embed::create("", "Error", "Cannot get handler");
}, }
}; };
// Start playing // Start playing
@ -99,18 +111,32 @@ pub async unsafe fn attempt_to_queue_song(ctx: &Context, guild_id: &GuildId, use
cmdctx: Arc::new(ctx.clone()), cmdctx: Arc::new(ctx.clone()),
}, },
); );
// Get metadata // Get metadata
let metadata = src.aux_metadata().await.expect("Cannot get metadata"); let metadata = src.aux_metadata().await.expect("Cannot get metadata");
let title = metadata.title.unwrap_or("Unknown title".to_string()); let title = metadata.title.unwrap_or("Unknown title".to_string());
let author = metadata.artist.unwrap_or("Unknown artist".to_string()); let author = metadata.artist.unwrap_or("Unknown artist".to_string());
let duration = metadata.duration.unwrap_or(Duration::from_millis(0)); let duration = metadata.duration.unwrap_or(Duration::from_millis(0));
let thumbnail = metadata.thumbnail.unwrap_or("https://http.cat/images/404.jpg".to_string()); let thumbnail = metadata
let link = metadata.source_url.unwrap_or("https://piped.moonleay.net/404".to_string()); .thumbnail
.unwrap_or("https://http.cat/images/404.jpg".to_string());
let link = metadata
.source_url
.unwrap_or("https://piped.moonleay.net/404".to_string());
Embed::create(username, "Added to queue", format!("{} by {} ({}min {}sec) was added to the queue.\n [[Link]({})]", Embed::create(
title, author, duration.as_secs() / 60, duration.as_secs() % 60, link)) username,
.thumbnail(thumbnail) "Added to queue",
format!(
"{} by {} ({}min {}sec) was added to the queue.\n [[Link]({})]",
title,
author,
duration.as_secs() / 60,
duration.as_secs() % 60,
link
),
)
.thumbnail(thumbnail)
} }
pub async unsafe fn play_song(ctx: &Context, guild_id: &GuildId, target: YoutubeDl) { pub async unsafe fn play_song(ctx: &Context, guild_id: &GuildId, target: YoutubeDl) {
@ -118,25 +144,28 @@ pub async unsafe fn play_song(ctx: &Context, guild_id: &GuildId, target: Youtube
.await .await
.expect("Cannot get Songbird.") .expect("Cannot get Songbird.")
.clone(); .clone();
if !user_util::is_self_connected_to_vc(ctx, guild_id).await { if !user_util::is_self_connected_to_vc(ctx, guild_id).await {
println!("Bot is not connected to a VC, cannot play."); println!("Bot is not connected to a VC, cannot play.");
return; return;
} }
music_queue::set_now_playing(&guild_id, Some(target.clone())); music_queue::set_now_playing(guild_id, Some(target.clone()));
let handler_lock = match manager.get(*guild_id) { let handler_lock = match manager.get(*guild_id) {
Some(handler) => handler, Some(handler) => handler,
None => { None => return,
return
},
}; };
let mut handler = handler_lock.lock().await; let mut handler = handler_lock.lock().await;
let _ = handler.play_input(target.into()); // TODO: Add event handlers let _ = handler.play_input(target.into()); // TODO: Add event handlers
} }
pub async unsafe fn attempt_to_stop(ctx: &Context, guild_id: &GuildId, user_id: &UserId, username: &str) -> CreateEmbed { pub async unsafe fn attempt_to_stop(
ctx: &Context,
guild_id: &GuildId,
user_id: &UserId,
username: &str,
) -> CreateEmbed {
if !user_util::is_self_connected_to_vc(ctx, guild_id).await { if !user_util::is_self_connected_to_vc(ctx, guild_id).await {
// Bot is not connectd to vc; no need to dc // Bot is not connectd to vc; no need to dc
return Embed::create( return Embed::create(
@ -145,8 +174,12 @@ pub async unsafe fn attempt_to_stop(ctx: &Context, guild_id: &GuildId, user_id:
"And therefore there is no need to do anything.", "And therefore there is no need to do anything.",
); );
} }
let self_channel = user_util::get_self_vc_id(ctx, &guild_id).await.expect("Cannot get self channel"); let self_channel = user_util::get_self_vc_id(ctx, guild_id)
let connect_to = get_vc_id(ctx, &guild_id, &user_id).await.expect("Cannot get channel id"); .await
.expect("Cannot get self channel");
let connect_to = get_vc_id(ctx, guild_id, user_id)
.await
.expect("Cannot get channel id");
// Check if user is in the same VC as the bot // Check if user is in the same VC as the bot
if self_channel != connect_to { if self_channel != connect_to {
@ -161,20 +194,24 @@ pub async unsafe fn attempt_to_stop(ctx: &Context, guild_id: &GuildId, user_id:
Ok(stopped) => stopped, Ok(stopped) => stopped,
Err(e) => { Err(e) => {
println!("Error while stopping: {:?}", e); println!("Error while stopping: {:?}", e);
return Embed::create(username, "There was an error", "Tell moonleay to check the logs.".to_string()); return Embed::create(
username,
"There was an error",
"Tell moonleay to check the logs.".to_string(),
);
} }
}; };
return if !stopped { if !stopped {
Embed::create(username, "Can't stop, what ain't running.", "I am not connected. I cant stop doing something, when I'm not doing it".to_string())
} else {
music_queue::delete_queue(guild_id); // Clear queue
Embed::create( Embed::create(
username, username,
"I stopped and left", "Can't stop, what ain't running.",
"Just like your girlfriend.", "I am not connected. I cant stop doing something, when I'm not doing it".to_string(),
) )
} else {
music_queue::delete_queue(guild_id); // Clear queue
Embed::create(username, "I stopped and left", "Just like your girlfriend.")
} }
} }
@ -188,10 +225,8 @@ pub async fn stop(ctx: &Context, guild_id: &GuildId) -> Result<bool, JoinError>
let has_handler = manager.get(*guild_id).is_some(); let has_handler = manager.get(*guild_id).is_some();
if has_handler { if has_handler {
if let Err(e) = manager.remove(*guild_id).await { manager.remove(*guild_id).await?;
return Err(e); // Failed to remove handler return Ok(true); // Handler removed
}
return Ok(true) // Handler removed
} }
Ok(false) // No handler, so it's already stopped Ok(false) // No handler, so it's already stopped
} }

View file

@ -1,7 +1,8 @@
use serenity::all::GuildId; use serenity::all::GuildId;
use songbird::input::YoutubeDl; use songbird::input::YoutubeDl;
pub struct MusicQueue { // God this sucks. This needs to be reprogrammed properly. pub struct MusicQueue {
// God this sucks. This needs to be reprogrammed properly.
guild_id: GuildId, guild_id: GuildId,
pub queue: Vec<YoutubeDl>, pub queue: Vec<YoutubeDl>,
pub now_playing: Option<YoutubeDl>, pub now_playing: Option<YoutubeDl>,
@ -20,14 +21,20 @@ pub unsafe fn get_queue(guild_id: &GuildId) -> &MusicQueue {
now_playing: None, now_playing: None,
}; };
MUSIC_QUEUE.push(new_queue); MUSIC_QUEUE.push(new_queue);
MUSIC_QUEUE.iter().find(|q| q.guild_id == *guild_id).expect("Cannot get queue") MUSIC_QUEUE
.iter()
.find(|q| q.guild_id == *guild_id)
.expect("Cannot get queue")
} }
} }
} }
unsafe fn update_queue(guild_id: &GuildId, queue: Vec<YoutubeDl>, now_playing: Option<YoutubeDl>) { unsafe fn update_queue(guild_id: &GuildId, queue: Vec<YoutubeDl>, now_playing: Option<YoutubeDl>) {
let index = MUSIC_QUEUE.iter().position(|q| q.guild_id == *guild_id).expect("Cannot get index"); let index = MUSIC_QUEUE
MUSIC_QUEUE[index] = MusicQueue{ .iter()
.position(|q| q.guild_id == *guild_id)
.expect("Cannot get index");
MUSIC_QUEUE[index] = MusicQueue {
guild_id: *guild_id, guild_id: *guild_id,
queue, queue,
now_playing, now_playing,
@ -35,7 +42,10 @@ unsafe fn update_queue(guild_id: &GuildId, queue: Vec<YoutubeDl>, now_playing: O
} }
pub unsafe fn delete_queue(guild_id: &GuildId) { pub unsafe fn delete_queue(guild_id: &GuildId) {
let index = MUSIC_QUEUE.iter().position(|q| q.guild_id == *guild_id).expect("Cannot get index"); let index = MUSIC_QUEUE
.iter()
.position(|q| q.guild_id == *guild_id)
.expect("Cannot get index");
MUSIC_QUEUE.remove(index); MUSIC_QUEUE.remove(index);
} }
@ -49,7 +59,7 @@ pub unsafe fn add_to_queue(guild_id: &GuildId, input: YoutubeDl) {
pub unsafe fn get_head(guild_id: &GuildId) -> Option<YoutubeDl> { pub unsafe fn get_head(guild_id: &GuildId) -> Option<YoutubeDl> {
let queue = get_queue(guild_id); let queue = get_queue(guild_id);
let mut q = queue.queue.clone(); let mut q = queue.queue.clone();
let result = q.first().map(|s| s.clone()); let result = q.first().cloned();
q.remove(0); q.remove(0);
update_queue(guild_id, q, queue.now_playing.clone()); update_queue(guild_id, q, queue.now_playing.clone());
result result
@ -63,4 +73,4 @@ pub unsafe fn set_now_playing(guild_id: &GuildId, now_playing: Option<YoutubeDl>
pub unsafe fn get_now_playing(guild_id: &GuildId) -> &Option<YoutubeDl> { pub unsafe fn get_now_playing(guild_id: &GuildId) -> &Option<YoutubeDl> {
let queue = get_queue(guild_id); let queue = get_queue(guild_id);
&queue.now_playing &queue.now_playing
} }

View file

@ -1,52 +1,47 @@
use std::collections::HashMap;
use serenity::all::{ChannelId, Context, Guild, GuildId, PartialGuild, UserId, VoiceState}; use serenity::all::{ChannelId, Context, Guild, GuildId, PartialGuild, UserId, VoiceState};
use std::collections::HashMap;
/// Request a guild by id, get it from cache /// Request a guild by id, get it from cache
pub fn request_guild(ctx: &Context, guild_id: &GuildId) -> Guild { pub fn request_guild(ctx: &Context, guild_id: &GuildId) -> Guild {
let guild = match guild_id.to_guild_cached(&ctx.cache) { match guild_id.to_guild_cached(&ctx.cache) {
Some(guild) => guild.clone(), Some(guild) => guild.clone(),
None => { None => {
panic!("Cannot get guild with id {:?}!", guild_id); panic!("Cannot get guild with id {:?}!", guild_id);
} }
}; }
guild
} }
/// Request a guild by id, get it from Discord, not from cache, this is a partial guild /// Request a guild by id, get it from Discord, not from cache, this is a partial guild
pub async fn request_partial_guild(ctx: &Context, guild_id: &GuildId) -> PartialGuild { pub async fn request_partial_guild(ctx: &Context, guild_id: &GuildId) -> PartialGuild {
let guild = match ctx.http.get_guild(*guild_id).await { match ctx.http.get_guild(*guild_id).await {
Ok(guild) => guild, Ok(guild) => guild,
Err(error) => { Err(error) => {
panic!("error whilest getting guild from Discord {}", error); panic!("error whilest getting guild from Discord {}", error);
} }
}; }
guild
} }
/// Get the voice channel id of a user /// Get the voice channel id of a user
pub async fn get_vc_id(ctx: &Context, guild_id: &GuildId, user_id: &UserId) -> Option<ChannelId> { pub async fn get_vc_id(ctx: &Context, guild_id: &GuildId, user_id: &UserId) -> Option<ChannelId> {
let guild = request_guild(&ctx, guild_id); let guild = request_guild(ctx, guild_id);
guild guild
.voice_states .voice_states
.get(user_id) .get(user_id)
.and_then(|voice_state| voice_state.channel_id) .and_then(|voice_state| voice_state.channel_id)
} }
/// Check if the bot is connected to a voice channel /// Check if the bot is connected to a voice channel
pub async fn is_self_connected_to_vc(ctx: &Context, guild_id: &GuildId) -> bool { pub async fn is_self_connected_to_vc(ctx: &Context, guild_id: &GuildId) -> bool {
let channel_id = get_self_vc_id(ctx, guild_id); let channel_id = get_self_vc_id(ctx, guild_id);
!channel_id.await.is_none() channel_id.await.is_some()
} }
/// Check if a user is connected to a voice channel /// Check if a user is connected to a voice channel
pub async fn is_user_connected_to_vc(ctx: &Context, guild_id: &GuildId, user_id: &UserId) -> bool { pub async fn is_user_connected_to_vc(ctx: &Context, guild_id: &GuildId, user_id: &UserId) -> bool {
let channel_id = get_vc_id(ctx, guild_id, user_id).await; let channel_id = get_vc_id(ctx, guild_id, user_id).await;
!channel_id.is_none() channel_id.is_some()
} }
/// Get the voice channel id of the bot /// Get the voice channel id of the bot
@ -57,15 +52,17 @@ pub async fn get_self_vc_id(ctx: &Context, guild_id: &GuildId) -> Option<Channel
} }
/// Get all voice states of a guild /// Get all voice states of a guild
pub async fn get_voice_states(ctx: &Context, guild_id: &GuildId) -> HashMap<UserId, VoiceState>{ pub async fn get_voice_states(ctx: &Context, guild_id: &GuildId) -> HashMap<UserId, VoiceState> {
let guild = request_guild(ctx, guild_id); let guild = request_guild(ctx, guild_id);
let voice_states = guild.voice_states.clone(); guild.voice_states.clone()
voice_states
} }
/// Get the amount of members in a voice channel /// Get the amount of members in a voice channel
pub async fn get_amount_of_members_in_vc(ctx: &Context, guild_id: &GuildId, channel_id: &ChannelId,) -> usize { pub async fn get_amount_of_members_in_vc(
ctx: &Context,
guild_id: &GuildId,
channel_id: &ChannelId,
) -> usize {
let voice_states = get_voice_states(ctx, guild_id).await; let voice_states = get_voice_states(ctx, guild_id).await;
let amount = voice_states let amount = voice_states
.iter() .iter()