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.
69 lines
1.9 KiB
Rust
69 lines
1.9 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_queue_for_guild;
|
|
use crate::messages::music::queue::create_queue_menu;
|
|
use crate::providers::music::queue::Song;
|
|
|
|
#[command]
|
|
#[only_in(guilds)]
|
|
#[description("Shows the song queue")]
|
|
#[usage("(<query...>)")]
|
|
#[aliases("q")]
|
|
#[bucket("general")]
|
|
async fn queue(ctx: &Context, msg: &Message, mut args: Args) -> CommandResult {
|
|
let guild = msg.guild(&ctx.cache).await.unwrap();
|
|
log::trace!("Displaying queue for guild {}", guild.id);
|
|
|
|
let query = args
|
|
.iter::<String>()
|
|
.map(|s| s.unwrap().to_lowercase())
|
|
.collect::<Vec<String>>();
|
|
|
|
let queue = forward_error!(
|
|
ctx,
|
|
msg.channel_id,
|
|
get_queue_for_guild(ctx, &guild.id).await
|
|
);
|
|
let queue_lock = queue.lock().await;
|
|
let songs: Vec<(usize, Song)> = queue_lock
|
|
.entries()
|
|
.into_iter()
|
|
.enumerate()
|
|
.filter(|(i, s)| {
|
|
if query.is_empty() {
|
|
return true;
|
|
}
|
|
for kw in &query {
|
|
if s.title().to_lowercase().contains(kw)
|
|
|| s.author().to_lowercase().contains(kw)
|
|
|| &i.to_string() == kw
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
false
|
|
})
|
|
.map(|(i, s)| (i, s.clone()))
|
|
.collect();
|
|
log::trace!("Songs are {:?}", songs);
|
|
|
|
if songs.len() == 0 {
|
|
msg.channel_id
|
|
.send_message(ctx, |m| {
|
|
m.embed(|e| e.title("Queue").description("*The queue is empty*"))
|
|
})
|
|
.await?;
|
|
|
|
return Ok(());
|
|
}
|
|
create_queue_menu(ctx, msg.channel_id, songs).await?;
|
|
|
|
handle_autodelete(ctx, msg).await?;
|
|
|
|
Ok(())
|
|
}
|