reset loop state on leave/stop
This commit is contained in:
@@ -38,6 +38,9 @@ pub(crate) async fn join_channel(
|
|||||||
call.deafen(true).await?;
|
call.deafen(true).await?;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// create guild config
|
||||||
|
state.guild_settings.entry(guild_id).or_default();
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,26 @@
|
|||||||
use crate::state::State;
|
use crate::state::{State, StateRef};
|
||||||
use std::error::Error;
|
use std::{error::Error, sync::Arc};
|
||||||
use twilight_model::gateway::payload::incoming::InteractionCreate;
|
use twilight_model::{
|
||||||
|
gateway::payload::incoming::InteractionCreate,
|
||||||
|
id::{marker::GuildMarker, Id},
|
||||||
|
};
|
||||||
|
|
||||||
|
pub(crate) async fn leave_channel(
|
||||||
|
guild_id: Id<GuildMarker>,
|
||||||
|
state: Arc<StateRef>,
|
||||||
|
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
||||||
|
// stop playing
|
||||||
|
if let Some(call_lock) = state.songbird.get(guild_id) {
|
||||||
|
let call = call_lock.lock().await;
|
||||||
|
call.queue().stop();
|
||||||
|
}
|
||||||
|
// leave the voice channel
|
||||||
|
state.songbird.leave(guild_id).await?;
|
||||||
|
|
||||||
|
// reset guild settings
|
||||||
|
state.guild_settings.remove(&guild_id);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
pub(crate) async fn leave(
|
pub(crate) async fn leave(
|
||||||
interaction: Box<InteractionCreate>,
|
interaction: Box<InteractionCreate>,
|
||||||
@@ -16,6 +36,8 @@ pub(crate) async fn leave(
|
|||||||
let Some(guild_id) = interaction.guild_id else {
|
let Some(guild_id) = interaction.guild_id else {
|
||||||
return Ok(());
|
return Ok(());
|
||||||
};
|
};
|
||||||
state.songbird.leave(guild_id).await?;
|
|
||||||
|
leave_channel(guild_id, Arc::clone(&state)).await?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
use crate::metadata::MetadataMap;
|
use crate::metadata::MetadataMap;
|
||||||
use crate::state::{Settings, State, StateRef};
|
use crate::state::{State, StateRef};
|
||||||
use async_trait::async_trait;
|
use async_trait::async_trait;
|
||||||
use songbird::{Event, EventContext, EventHandler, TrackEvent};
|
use songbird::{Event, EventContext, EventHandler, TrackEvent};
|
||||||
use std::ops::Sub;
|
use std::ops::Sub;
|
||||||
@@ -29,14 +29,8 @@ pub(crate) async fn loop_queue(
|
|||||||
return Ok(());
|
return Ok(());
|
||||||
};
|
};
|
||||||
|
|
||||||
state
|
|
||||||
.guild_settings
|
|
||||||
.entry(guild_id)
|
|
||||||
.or_insert_with(|| Settings { loop_queue: false });
|
|
||||||
|
|
||||||
state.guild_settings.entry(guild_id).and_modify(|settings| {
|
state.guild_settings.entry(guild_id).and_modify(|settings| {
|
||||||
settings.loop_queue = !settings.loop_queue;
|
settings.loop_queue = !settings.loop_queue;
|
||||||
println!("loop_queue: {}", settings.loop_queue);
|
|
||||||
});
|
});
|
||||||
|
|
||||||
let looping = state
|
let looping = state
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ pub(crate) use join::join;
|
|||||||
|
|
||||||
mod leave;
|
mod leave;
|
||||||
pub(crate) use leave::leave;
|
pub(crate) use leave::leave;
|
||||||
|
pub(crate) use leave::leave_channel;
|
||||||
|
|
||||||
mod pause;
|
mod pause;
|
||||||
pub(crate) use pause::pause;
|
pub(crate) use pause::pause;
|
||||||
|
|||||||
@@ -23,6 +23,10 @@ pub(crate) async fn stop(
|
|||||||
return Ok(());
|
return Ok(());
|
||||||
};
|
};
|
||||||
|
|
||||||
|
state.guild_settings.entry(guild_id).and_modify(|settings| {
|
||||||
|
settings.loop_queue = false;
|
||||||
|
});
|
||||||
|
|
||||||
if let Some(call_lock) = state.songbird.get(guild_id) {
|
if let Some(call_lock) = state.songbird.get(guild_id) {
|
||||||
let call = call_lock.lock().await;
|
let call = call_lock.lock().await;
|
||||||
call.queue().stop();
|
call.queue().stop();
|
||||||
|
|||||||
@@ -1,5 +1,7 @@
|
|||||||
use crate::commands::queue::{build_action_row, build_queue_embeds, TRACKS_PER_PAGE};
|
use crate::commands::queue::{build_action_row, build_queue_embeds, TRACKS_PER_PAGE};
|
||||||
use crate::commands::{delete, join, leave, loop_queue, pause, play, queue, resume, skip, stop};
|
use crate::commands::{
|
||||||
|
delete, join, leave, leave_channel, loop_queue, pause, play, queue, resume, skip, stop,
|
||||||
|
};
|
||||||
use crate::state::State;
|
use crate::state::State;
|
||||||
use futures::Future;
|
use futures::Future;
|
||||||
use std::error::Error;
|
use std::error::Error;
|
||||||
@@ -63,13 +65,7 @@ pub(crate) async fn leave_if_alone(
|
|||||||
|
|
||||||
// count is 1 if the bot is the only one in the channel
|
// count is 1 if the bot is the only one in the channel
|
||||||
if count == 1 {
|
if count == 1 {
|
||||||
// stop playing
|
leave_channel(guild_id, Arc::clone(&state)).await?;
|
||||||
if let Some(call_lock) = state.songbird.get(guild_id) {
|
|
||||||
let call = call_lock.lock().await;
|
|
||||||
call.queue().stop();
|
|
||||||
}
|
|
||||||
// leave the voice channel
|
|
||||||
state.songbird.leave(guild_id).await?;
|
|
||||||
}
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
12
src/state.rs
12
src/state.rs
@@ -13,6 +13,18 @@ pub(crate) struct Settings {
|
|||||||
pub(crate) loop_queue: bool,
|
pub(crate) loop_queue: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl Settings {
|
||||||
|
pub(crate) fn new() -> Self {
|
||||||
|
Self { loop_queue: false }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for Settings {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self::new()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub(crate) struct StateRef {
|
pub(crate) struct StateRef {
|
||||||
pub(crate) http: HttpClient,
|
pub(crate) http: HttpClient,
|
||||||
|
|||||||
Reference in New Issue
Block a user