maybe fix bot leaving too early, and playing multiple sounds at once

Patrick Cleavelin 2023-02-13 21:00:18 -06:00
parent cd4de895eb
commit 2dfe7d779a
2 changed files with 19 additions and 16 deletions

View File

@ -9,7 +9,7 @@ edition = "2021"
serde = "1.0.152" serde = "1.0.152"
serde_json = "1.0.93" serde_json = "1.0.93"
serenity = { version = "0.11.5", default-features = false, features = ["client", "gateway", "rustls_backend", "model", "cache", "voice"] } serenity = { version = "0.11.5", default-features = false, features = ["client", "gateway", "rustls_backend", "model", "cache", "voice"] }
songbird = "0.3.0" songbird = { version = "0.3.0", features = [ "builtin-queue" ] }
tokio = { version = "1.25.0", features = ["rt-multi-thread", "macros", "signal"] } tokio = { version = "1.25.0", features = ["rt-multi-thread", "macros", "signal"] }
tracing = "0.1.37" tracing = "0.1.37"
tracing-subscriber = "0.3.16" tracing-subscriber = "0.3.16"

View File

@ -28,18 +28,21 @@ impl songbird::EventHandler for TrackEventHandler {
&'a self, &'a self,
ctx: &'b songbird::EventContext<'c>, ctx: &'b songbird::EventContext<'c>,
) -> Option<songbird::Event> { ) -> Option<songbird::Event> {
if let songbird::EventContext::Track(track) = ctx { if let songbird::EventContext::Track(tracks) = ctx {
if let Some(context) = track.get(0) { for (track_state, _track_handle) in tracks.iter() {
if context.0.playing == songbird::tracks::PlayMode::End // If any track is still playing, don't leave the channel yet
|| context.0.playing == songbird::tracks::PlayMode::Stop // there are more sounds to be played (I think)
{ if track_state.playing == songbird::tracks::PlayMode::Play {
return None;
}
}
}
let manager = songbird::get(&self.ctx).await.expect("should get manager"); let manager = songbird::get(&self.ctx).await.expect("should get manager");
if let Err(err) = manager.leave(self.guild_id).await { if let Err(err) = manager.leave(self.guild_id).await {
error!("Failed to leave voice channel: {err:?}"); error!("Failed to leave voice channel: {err:?}");
} }
}
}
}
None None
} }
} }
@ -89,6 +92,10 @@ impl EventHandler for Handler {
async fn voice_state_update(&self, ctx: Context, old: Option<VoiceState>, new: VoiceState) { async fn voice_state_update(&self, ctx: Context, old: Option<VoiceState>, new: VoiceState) {
if old.is_none() { if old.is_none() {
if let (Some(member), Some(channel_id)) = (new.member, new.channel_id) { if let (Some(member), Some(channel_id)) = (new.member, new.channel_id) {
if member.user.name == "MemeJoin" {
return;
}
info!( info!(
"{}#{} joined voice channel {:?} in {:?}", "{}#{} joined voice channel {:?} in {:?}",
member.user.name, member.user.name,
@ -100,10 +107,6 @@ impl EventHandler for Handler {
.unwrap_or("no_guild_name".to_string()) .unwrap_or("no_guild_name".to_string())
); );
if member.user.name == "MemeJoin" {
return;
}
let settings = { let settings = {
let data_read = ctx.data.read().await; let data_read = ctx.data.read().await;
@ -151,8 +154,8 @@ impl EventHandler for Handler {
} }
}; };
let track_handle = handler.play_source(source); let track_handler = handler.enqueue_source(source);
if let Err(err) = track_handle.add_event( if let Err(err) = track_handler.add_event(
songbird::Event::Track(songbird::TrackEvent::End), songbird::Event::Track(songbird::TrackEvent::End),
TrackEventHandler { TrackEventHandler {
ctx, ctx,