You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
2b-rs/src/commands/music/play_next.rs

68 lines
2.3 KiB
Rust

use serenity::client::Context;
use serenity::framework::standard::macros::command;
use serenity::framework::standard::{Args, CommandResult};
use serenity::model::channel::Message;
use crate::commands::common::handle_autodelete;
use crate::commands::music::{
get_channel_for_author, get_queue_for_guild, get_songs_for_query, get_voice_manager,
join_channel, play_next_in_queue, DJ_CHECK,
};
use crate::messages::music::now_playing::create_now_playing_msg;
use crate::providers::music::lavalink::Lavalink;
#[command]
#[only_in(guilds)]
#[description("Puts a song as the next to play in the queue")]
#[usage("(<spotify_ur>|<youtube_url>|<query>|pl:<saved_playlist>)")]
#[min_args(1)]
#[aliases("pn", "play-next", "playnext")]
#[bucket("music_api")]
#[checks(DJ)]
async fn play_next(ctx: &Context, msg: &Message, args: Args) -> CommandResult {
let query = args.message();
let guild = msg.guild(&ctx.cache).await.unwrap();
log::debug!("Playing song as next song for guild {}", guild.id);
let manager = get_voice_manager(ctx).await;
let handler = manager.get(guild.id);
if handler.is_none() {
log::debug!("Not in a voice channel. Joining authors channel");
msg.guild(&ctx.cache).await.unwrap();
let channel_id = get_channel_for_author(&msg.author.id, &guild)?;
join_channel(ctx, channel_id, guild.id).await;
}
let mut songs = get_songs_for_query(&ctx, msg, query).await?;
let queue = get_queue_for_guild(ctx, &guild.id).await?;
let (play_first, create_now_playing) = {
let mut queue_lock = queue.lock().await;
songs.reverse();
log::debug!("Enqueueing songs as next songs in the queue");
for song in songs {
queue_lock.add_next(song);
}
(
queue_lock.current().is_none(),
queue_lock.now_playing_msg.is_none(),
)
};
if play_first {
let data = ctx.data.read().await;
let player = data.get::<Lavalink>().unwrap();
while !play_next_in_queue(&ctx.http, &msg.channel_id, &guild.id, &queue, &player).await {}
}
if create_now_playing {
let handle = create_now_playing_msg(ctx, queue.clone(), msg.channel_id).await?;
let mut queue_lock = queue.lock().await;
queue_lock.now_playing_msg = Some(handle);
}
handle_autodelete(ctx, msg).await?;
Ok(())
}