paginate /queue #3

Merged
jheuel merged 1 commits from dev into main 2024-06-13 00:54:17 +02:00
14 changed files with 1031 additions and 535 deletions
Showing only changes of commit a0ac87959b - Show all commits

1032
Cargo.lock generated

File diff suppressed because it is too large Load Diff

View File

@@ -21,5 +21,7 @@ twilight-standby = "0.15"
twilight-cache-inmemory = "0.15" twilight-cache-inmemory = "0.15"
twilight-util = { version = "0.15", features=["builder"] } twilight-util = { version = "0.15", features=["builder"] }
dotenv = "0.15.0" dotenv = "0.15.0"
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0" serde_json = "1.0"
url = "2.5.0" url = "2.5.0"
anyhow = "1.0.86"

2
src/colors.rs Normal file
View File

@@ -0,0 +1,2 @@
pub(crate) const BLURPLE: u32 = 0x58_65_F2;
pub(crate) const YELLOW: u32 = 0xFE_E7_5C;

View File

@@ -9,14 +9,14 @@ pub(crate) async fn delete(
state: State, state: State,
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> { ) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
let admin = env::var("ADMIN")?.parse::<u64>()?; let admin = env::var("ADMIN")?.parse::<u64>()?;
if msg.author.id != Id::from(NonZeroU64::new(admin).unwrap()) { if msg.author.id != Id::from(NonZeroU64::new(admin).expect("Could not get author id")) {
return Ok(()); return Ok(());
} }
let n = msg let n = msg
.content .content
.split(' ') .split(' ')
.last() .last()
.unwrap() .unwrap_or("1")
.parse::<u16>() .parse::<u16>()
.unwrap_or(1); .unwrap_or(1);
if n > 100 { if n > 100 {

View File

@@ -2,8 +2,8 @@ use crate::state::State;
use std::error::Error; use std::error::Error;
use tracing::debug; use tracing::debug;
use twilight_model::{ use twilight_model::{
application::interaction::Interaction,
channel::message::MessageFlags, channel::message::MessageFlags,
gateway::payload::incoming::InteractionCreate,
http::interaction::{InteractionResponse, InteractionResponseType}, http::interaction::{InteractionResponse, InteractionResponseType},
id::{ id::{
marker::{GuildMarker, UserMarker}, marker::{GuildMarker, UserMarker},
@@ -42,7 +42,7 @@ pub(crate) async fn join_channel(
} }
pub(crate) async fn join( pub(crate) async fn join(
interaction: Interaction, interaction: Box<InteractionCreate>,
state: State, state: State,
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> { ) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
debug!( debug!(

View File

@@ -1,9 +1,9 @@
use crate::state::State; use crate::state::State;
use std::error::Error; use std::error::Error;
use twilight_model::application::interaction::Interaction; use twilight_model::gateway::payload::incoming::InteractionCreate;
pub(crate) async fn leave( pub(crate) async fn leave(
interaction: Interaction, interaction: Box<InteractionCreate>,
state: State, state: State,
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> { ) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
tracing::debug!( tracing::debug!(

View File

@@ -10,7 +10,7 @@ pub(crate) use pause::pause;
mod play; mod play;
pub(crate) use play::play; pub(crate) use play::play;
mod queue; pub(crate) mod queue;
pub(crate) use queue::queue; pub(crate) use queue::queue;
mod resume; mod resume;

View File

@@ -1,14 +1,14 @@
use crate::state::State; use crate::state::State;
use std::error::Error; use std::error::Error;
use twilight_model::{ use twilight_model::{
application::interaction::Interaction,
channel::message::MessageFlags, channel::message::MessageFlags,
gateway::payload::incoming::InteractionCreate,
http::interaction::{InteractionResponse, InteractionResponseType}, http::interaction::{InteractionResponse, InteractionResponseType},
}; };
use twilight_util::builder::InteractionResponseDataBuilder; use twilight_util::builder::InteractionResponseDataBuilder;
pub(crate) async fn pause( pub(crate) async fn pause(
interaction: Interaction, interaction: Box<InteractionCreate>,
state: State, state: State,
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> { ) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
tracing::debug!( tracing::debug!(

View File

@@ -1,20 +1,55 @@
use crate::colors;
use crate::commands::join::join_channel; use crate::commands::join::join_channel;
use crate::metadata::{Metadata, MetadataMap}; use crate::metadata::{Metadata, MetadataMap};
use crate::state::State; use crate::state::State;
use serde_json::Value;
use serde::{Deserialize, Serialize};
use songbird::input::{Compose, YoutubeDl}; use songbird::input::{Compose, YoutubeDl};
use std::io::{BufRead, BufReader}; use std::io::{BufRead, BufReader};
use std::{error::Error, ops::Sub, time::Duration}; use std::{error::Error, ops::Sub, time::Duration};
use tokio::process::Command; use tokio::process::Command;
use tracing::debug; use tracing::debug;
use twilight_model::application::interaction::Interaction; use twilight_model::channel::message::MessageFlags;
use twilight_model::gateway::payload::incoming::InteractionCreate;
use twilight_model::http::interaction::{InteractionResponse, InteractionResponseType}; use twilight_model::http::interaction::{InteractionResponse, InteractionResponseType};
use twilight_util::builder::embed::EmbedBuilder; use twilight_util::builder::embed::EmbedBuilder;
use twilight_util::builder::InteractionResponseDataBuilder; use twilight_util::builder::InteractionResponseDataBuilder;
use url::Url; use url::Url;
#[derive(Debug, Serialize, Deserialize)]
struct YouTubeTrack {
url: String,
title: String,
channel: String,
playlist: String,
playlist_id: String,
duration_string: String,
}
fn build_playlist_url(playlist_id: &str) -> String {
format!("https://www.youtube.com/playlist?list={}", playlist_id)
}
async fn get_tracks(
url: String,
) -> Result<Vec<YouTubeTrack>, Box<dyn Error + Send + Sync + 'static>> {
let output = Command::new("yt-dlp")
.args(vec![&url, "--flat-playlist", "-j"])
.output()
.await?;
let reader = BufReader::new(output.stdout.as_slice());
let tracks: Vec<YouTubeTrack> = reader
.lines()
.map_while(Result::ok)
.flat_map(|line| serde_json::from_str(&line))
.collect();
tracing::debug!("tracks: {:?}", tracks);
Ok(tracks)
}
pub(crate) async fn play( pub(crate) async fn play(
interaction: Interaction, interaction: Box<InteractionCreate>,
state: State, state: State,
query: String, query: String,
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> { ) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
@@ -25,14 +60,16 @@ pub(crate) async fn play(
); );
let content = format!("Adding track(s) to the queue: {}", query); let content = format!("Adding track(s) to the queue: {}", query);
let yellow = 0xFE_E7_5C;
let embeds = vec![EmbedBuilder::new() let embeds = vec![EmbedBuilder::new()
.description(content) .description(content)
.color(yellow) .color(colors::YELLOW)
.build()]; .build()];
let interaction_response_data = InteractionResponseDataBuilder::new().embeds(embeds).build(); let interaction_response_data = InteractionResponseDataBuilder::new()
.flags(MessageFlags::LOADING)
.embeds(embeds)
.build();
let response = InteractionResponse { let response = InteractionResponse {
kind: InteractionResponseType::ChannelMessageWithSource, kind: InteractionResponseType::DeferredChannelMessageWithSource,
data: Some(interaction_response_data), data: Some(interaction_response_data),
}; };
state state
@@ -59,7 +96,26 @@ pub(crate) async fn play(
debug!("query: {:?}", query); debug!("query: {:?}", query);
let urls = get_playlist_urls(query).await?; let tracks = get_tracks(query).await?;
if tracks.len() > 1 {
let first_track = tracks.first().unwrap();
let content = format!(
"Adding playlist [{}]({})",
first_track.playlist,
build_playlist_url(&first_track.playlist_id)
);
let embeds = vec![EmbedBuilder::new()
.description(content)
.color(colors::BLURPLE)
.build()];
state
.http
.interaction(interaction.application_id)
.update_response(&interaction.token)
.embeds(Some(&embeds))?
.await?;
}
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;
@@ -67,7 +123,9 @@ pub(crate) async fn play(
} }
let mut tracks_added = vec![]; let mut tracks_added = vec![];
for url in urls { for track in &tracks {
tracing::debug!("track: {:?}", track);
let url = track.url.clone();
let mut src = YoutubeDl::new(reqwest::Client::new(), url.to_string()); let mut src = YoutubeDl::new(reqwest::Client::new(), url.to_string());
if let Ok(metadata) = src.aux_metadata().await { if let Ok(metadata) = src.aux_metadata().await {
debug!("metadata: {:?}", metadata); debug!("metadata: {:?}", metadata);
@@ -99,59 +157,38 @@ pub(crate) async fn play(
match num_tracks_added { match num_tracks_added {
0 => {} 0 => {}
1 => { 1 => {
let track = tracks_added.first().unwrap().clone(); let (title, url) = if let Some(track) = tracks_added.first() {
let title = track.1.unwrap(); let track = track.clone();
let url = track.0; (track.1.unwrap_or("Unknown".to_string()), track.0)
} else {
("Unknown".to_string(), "".to_string())
};
content = format!("Added [{}]({}) to the queue", title, url); content = format!("Added [{}]({}) to the queue", title, url);
} }
_ => { _ => {
content = format!("Added {} tracks to the queue:\n", num_tracks_added); let first_track = tracks.first().unwrap();
for track in tracks_added.into_iter().take(num_tracks_added.min(3)) { content.push_str(&format!(
let title = track.1.unwrap(); "Adding playlist: [{}]({})\n",
let url = track.0; &first_track.playlist,
content.push_str(&format!(" \"[{}]({})\"\n", title, url)); build_playlist_url(&first_track.playlist_id)
} ));
content.push_str(&format!(
"Added {} tracks to the queue:\n",
num_tracks_added
));
} }
} }
let blurple = 0x58_65_F2;
let embeds = vec![EmbedBuilder::new() let embeds = vec![EmbedBuilder::new()
.description(content) .description(content)
.color(blurple) .color(colors::BLURPLE)
.build()]; .build()];
state state
.http .http
.interaction(interaction.application_id) .interaction(interaction.application_id)
.update_response(&interaction.token) .update_response(&interaction.token)
.embeds(Some(&embeds)) .embeds(Some(&embeds))?
.unwrap()
.await?; .await?;
Ok(()) Ok(())
} }
async fn get_playlist_urls(
url: String,
) -> Result<Vec<String>, Box<dyn Error + Send + Sync + 'static>> {
let output = Command::new("yt-dlp")
.args(vec![&url, "--flat-playlist", "-j"])
.output()
.await?;
let reader = BufReader::new(output.stdout.as_slice());
let urls = reader
.lines()
.map_while(Result::ok)
.map(|line| {
let entry: Value = serde_json::from_str(&line).unwrap();
entry
.get("webpage_url")
.unwrap()
.as_str()
.unwrap()
.to_string()
})
.collect();
Ok(urls)
}

View File

@@ -1,16 +1,106 @@
use songbird::tracks::TrackHandle;
use twilight_model::channel::message::component::{ActionRow, Button, ButtonStyle};
use twilight_model::channel::message::{Component, Embed, MessageFlags, ReactionType};
use twilight_model::gateway::payload::incoming::InteractionCreate;
use twilight_model::http::interaction::InteractionResponse;
use twilight_model::http::interaction::InteractionResponseType; use twilight_model::http::interaction::InteractionResponseType;
use twilight_model::{
application::interaction::Interaction, channel::message::MessageFlags,
http::interaction::InteractionResponse,
};
use twilight_util::builder::embed::EmbedBuilder; use twilight_util::builder::embed::EmbedBuilder;
use twilight_util::builder::InteractionResponseDataBuilder; use twilight_util::builder::InteractionResponseDataBuilder;
use crate::colors;
use crate::{metadata::MetadataMap, state::State}; use crate::{metadata::MetadataMap, state::State};
use std::error::Error; use std::error::Error;
pub(crate) const TRACKS_PER_PAGE: usize = 5;
fn format_duration(duration: std::time::Duration) -> String {
let res = duration.as_secs();
let hours = res / (60 * 60);
let res = res - hours * 60 * 60;
let minutes = res / 60;
let res = res - minutes * 60;
let seconds = res;
let mut s = String::new();
if hours > 0 {
s.push_str(format!("{:02}:", hours).as_str());
}
s.push_str(format!("{:02}:{:02}", minutes, seconds).as_str());
s
}
pub(crate) async fn build_queue_embeds(queue: &[TrackHandle], page: usize) -> Vec<Embed> {
let mut message = String::new();
if queue.is_empty() {
message.push_str("There are no tracks in the queue.\n");
}
for track in queue
.iter()
.skip(TRACKS_PER_PAGE * page)
.take(TRACKS_PER_PAGE)
{
let map = track.typemap().read().await;
let metadata = map
.get::<MetadataMap>()
.expect("Could not get metadata map");
message.push_str(
format!(
"* [{}]({})",
metadata.title.clone().unwrap_or("Unknown".to_string()),
metadata.url,
)
.as_str(),
);
if let Some(duration) = metadata.duration {
message.push_str(" (");
message.push_str(&format_duration(duration));
message.push(')');
}
message.push('\n');
}
message.push('\n');
let max_pages = queue.len() / TRACKS_PER_PAGE;
if max_pages > 0 {
message.push_str(&format!("page {}/{}", 1 + page, 1 + max_pages));
}
vec![EmbedBuilder::new()
.description(&message)
.color(colors::BLURPLE)
.build()]
}
pub(crate) fn build_action_row(page: usize, max_pages: usize) -> Vec<Component> {
if max_pages == 0 {
return Vec::new();
}
vec![Component::ActionRow(ActionRow {
components: vec![
Component::Button(Button {
custom_id: Some(format!("page:{}", page as i32 - 1)),
style: ButtonStyle::Primary,
label: Some("Previous page".to_string()),
emoji: Some(ReactionType::Unicode {
name: "⬅️".to_string(),
}),
url: None,
disabled: page == 0,
}),
Component::Button(Button {
custom_id: Some(format!("page:{}", page + 1)),
style: ButtonStyle::Primary,
label: Some("Next page".to_string()),
emoji: Some(ReactionType::Unicode {
name: "➡️".to_string(),
}),
url: None,
disabled: page >= max_pages,
}),
],
})]
}
pub(crate) async fn queue( pub(crate) async fn queue(
interaction: Interaction, interaction: Box<InteractionCreate>,
state: State, state: State,
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> { ) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
tracing::debug!( tracing::debug!(
@@ -22,59 +112,42 @@ pub(crate) async fn queue(
let Some(guild_id) = interaction.guild_id else { let Some(guild_id) = interaction.guild_id else {
return Ok(()); return Ok(());
}; };
let content = "Fetching queue".to_string();
let embeds = vec![EmbedBuilder::new()
.description(content)
.color(colors::YELLOW)
.build()];
let interaction_response_data = InteractionResponseDataBuilder::new()
.embeds(embeds)
.flags(MessageFlags::LOADING)
.build();
let response = InteractionResponse {
kind: InteractionResponseType::DeferredChannelMessageWithSource,
data: Some(interaction_response_data),
};
state
.http
.interaction(interaction.application_id)
.create_response(interaction.id, &interaction.token, &response)
.await?;
let mut queue = Vec::new();
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;
let queue = call.queue().current_queue(); queue = call.queue().current_queue();
let mut message = String::new();
if queue.is_empty() {
message.push_str("There are no tracks in the queue.\n");
} else {
message.push_str("Next songs are:\n");
}
for track in queue.iter().take(5) {
let map = track.typemap().read().await;
let metadata = map.get::<MetadataMap>().unwrap();
message.push_str(
format!(
"* [{}]({})",
metadata.title.clone().unwrap_or("Unknown".to_string()),
metadata.url,
)
.as_str(),
);
if let Some(duration) = metadata.duration {
let res = duration.as_secs();
let hours = res / (60 * 60);
let res = res - hours * 60 * 60;
let minutes = res / 60;
let res = res - minutes * 60;
let seconds = res;
message.push_str(" (");
if hours > 0 {
message.push_str(format!("{:02}:", hours).as_str());
}
message.push_str(format!("{:02}:{:02}", minutes, seconds).as_str());
message.push(')');
}
message.push('\n');
}
let embeds = vec![EmbedBuilder::new().description(&message).build()];
let interaction_response_data = InteractionResponseDataBuilder::new()
.flags(MessageFlags::EPHEMERAL)
.embeds(embeds)
.build();
let response = InteractionResponse {
kind: InteractionResponseType::ChannelMessageWithSource,
data: Some(interaction_response_data),
};
state
.http
.interaction(interaction.application_id)
.create_response(interaction.id, &interaction.token, &response)
.await?;
} }
let embeds = build_queue_embeds(&queue, 0).await;
let action_row = build_action_row(0, queue.len() / TRACKS_PER_PAGE);
state
.http
.interaction(interaction.application_id)
.update_response(&interaction.token)
.embeds(Some(&embeds))?
.components(Some(&action_row))?
.await?;
Ok(()) Ok(())
} }

View File

@@ -1,14 +1,14 @@
use crate::state::State; use crate::state::State;
use std::error::Error; use std::error::Error;
use twilight_model::{ use twilight_model::{
application::interaction::Interaction,
channel::message::MessageFlags, channel::message::MessageFlags,
gateway::payload::incoming::InteractionCreate,
http::interaction::{InteractionResponse, InteractionResponseType}, http::interaction::{InteractionResponse, InteractionResponseType},
}; };
use twilight_util::builder::InteractionResponseDataBuilder; use twilight_util::builder::InteractionResponseDataBuilder;
pub(crate) async fn resume( pub(crate) async fn resume(
interaction: Interaction, interaction: Box<InteractionCreate>,
state: State, state: State,
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> { ) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
tracing::debug!( tracing::debug!(

View File

@@ -1,6 +1,6 @@
use twilight_model::{ use twilight_model::{
application::interaction::Interaction,
channel::message::MessageFlags, channel::message::MessageFlags,
gateway::payload::incoming::InteractionCreate,
http::interaction::{InteractionResponse, InteractionResponseType}, http::interaction::{InteractionResponse, InteractionResponseType},
}; };
use twilight_util::builder::InteractionResponseDataBuilder; use twilight_util::builder::InteractionResponseDataBuilder;
@@ -9,7 +9,7 @@ use crate::state::State;
use std::error::Error; use std::error::Error;
pub(crate) async fn stop( pub(crate) async fn stop(
interaction: Interaction, interaction: Box<InteractionCreate>,
state: State, state: State,
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> { ) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
tracing::debug!( tracing::debug!(

View File

@@ -1,3 +1,4 @@
use crate::commands::queue::{build_action_row, build_queue_embeds, TRACKS_PER_PAGE};
use crate::commands::{delete, join, leave, pause, play, queue, resume, stop}; use crate::commands::{delete, join, leave, pause, play, queue, resume, stop};
use crate::state::State; use crate::state::State;
use futures::Future; use futures::Future;
@@ -5,19 +6,23 @@ use std::error::Error;
use std::sync::Arc; use std::sync::Arc;
use tracing::debug; use tracing::debug;
use twilight_gateway::Event; use twilight_gateway::Event;
use twilight_model::application::interaction::application_command::CommandOptionValue; use twilight_model::application::interaction::application_command::{
use twilight_model::application::interaction::{Interaction, InteractionData}; CommandData, CommandOptionValue,
};
use twilight_model::application::interaction::InteractionData;
use twilight_model::gateway::payload::incoming::VoiceStateUpdate; use twilight_model::gateway::payload::incoming::VoiceStateUpdate;
use twilight_model::http::interaction::{InteractionResponse, InteractionResponseType};
use twilight_util::builder::InteractionResponseDataBuilder;
#[derive(Debug)] #[derive(Debug)]
enum InteractionCommand { enum InteractionCommand {
Play(Interaction, String), Play(String),
Stop(Interaction), Stop,
Pause(Interaction), Pause,
Resume(Interaction), Resume,
Leave(Interaction), Leave,
Join(Interaction), Join,
Queue(Interaction), Queue,
NotImplemented, NotImplemented,
} }
@@ -75,80 +80,121 @@ impl Handler {
pub(crate) fn new(state: State) -> Self { pub(crate) fn new(state: State) -> Self {
Self { state } Self { state }
} }
pub(crate) async fn act(&self, event: Event) { pub(crate) async fn act(&self, event: Event) -> anyhow::Result<()> {
match &event { match event {
Event::MessageCreate(message) if message.content.starts_with('!') => { Event::MessageCreate(message) if message.content.starts_with('!') => {
if message.content.contains("!delete") { if message.content.contains("!delete") {
spawn(delete(message.0.clone(), Arc::clone(&self.state))); spawn(delete(message.0.clone(), Arc::clone(&self.state)));
} }
Ok(())
} }
Event::VoiceStateUpdate(update) => { Event::VoiceStateUpdate(update) => {
spawn(leave_if_alone(*update.clone(), Arc::clone(&self.state))) spawn(leave_if_alone(*update.clone(), Arc::clone(&self.state)));
Ok(())
} }
_ => {}
}
let interaction_command = match event {
Event::InteractionCreate(interaction) => { Event::InteractionCreate(interaction) => {
debug!("interaction: {:?}", &interaction); tracing::info!("interaction: {:?}", &interaction);
match &interaction.data { match &interaction.data {
Some(InteractionData::ApplicationCommand(command)) => { Some(InteractionData::ApplicationCommand(command)) => {
debug!("command: {:?}", command); let interaction_command = parse_interaction_command(command);
match command.name.as_str() { debug!("{:?}", interaction_command);
"play" => { match interaction_command {
if let Some(query_option) = InteractionCommand::Play(query) => {
command.options.iter().find(|opt| opt.name == "query") spawn(play(interaction, Arc::clone(&self.state), query))
{
if let CommandOptionValue::String(query) = &query_option.value {
InteractionCommand::Play(
interaction.0.clone(),
query.clone(),
)
} else {
InteractionCommand::NotImplemented
}
} else {
InteractionCommand::NotImplemented
}
} }
"stop" => InteractionCommand::Stop(interaction.0.clone()), InteractionCommand::Stop => {
"pause" => InteractionCommand::Pause(interaction.0.clone()), spawn(stop(interaction, Arc::clone(&self.state)))
"resume" => InteractionCommand::Resume(interaction.0.clone()), }
"leave" => InteractionCommand::Leave(interaction.0.clone()), InteractionCommand::Pause => {
"join" => InteractionCommand::Join(interaction.0.clone()), spawn(pause(interaction, Arc::clone(&self.state)))
"queue" => InteractionCommand::Queue(interaction.0.clone()), }
_ => InteractionCommand::NotImplemented, InteractionCommand::Resume => {
spawn(resume(interaction, Arc::clone(&self.state)))
}
InteractionCommand::Leave => {
spawn(leave(interaction, Arc::clone(&self.state)))
}
InteractionCommand::Join => {
spawn(join(interaction, Arc::clone(&self.state)))
}
InteractionCommand::Queue => {
spawn(queue(interaction, Arc::clone(&self.state)))
}
_ => {}
}
Ok(())
}
Some(InteractionData::MessageComponent(data)) => {
tracing::info!("message component: {:?}", data);
if !data.custom_id.starts_with("page:") {
return Ok(());
}
let page = data
.custom_id
.trim_start_matches("page:")
.parse::<usize>()
.unwrap_or(0);
tracing::info!("page: {:?}", page);
if let Some(guild_id) = interaction.guild_id {
let mut queue = Vec::new();
if let Some(call_lock) = self.state.songbird.get(guild_id) {
let call = call_lock.lock().await;
queue = call.queue().current_queue();
}
let embeds = build_queue_embeds(&queue, page).await;
let action_row = build_action_row(page, queue.len() / TRACKS_PER_PAGE);
let interaction_response_data = InteractionResponseDataBuilder::new()
.embeds(embeds)
.components(action_row)
.build();
let response = InteractionResponse {
kind: InteractionResponseType::UpdateMessage,
data: Some(interaction_response_data),
};
self.state
.http
.interaction(interaction.application_id)
.create_response(interaction.id, &interaction.token, &response)
.await?;
Ok(())
} else {
Ok(())
} }
} }
_ => InteractionCommand::NotImplemented, _ => Ok(()),
} }
} }
_ => InteractionCommand::NotImplemented, event => {
}; tracing::info!("unhandled event: {:?}", event);
debug!("{:?}", interaction_command); Ok(())
match interaction_command {
InteractionCommand::Play(interaction, query) => {
spawn(play(interaction, Arc::clone(&self.state), query))
} }
InteractionCommand::Stop(interaction) => { }
spawn(stop(interaction, Arc::clone(&self.state))) }
} }
InteractionCommand::Pause(interaction) => {
spawn(pause(interaction, Arc::clone(&self.state))) fn parse_interaction_command(command: &CommandData) -> InteractionCommand {
} debug!("command: {:?}", command);
InteractionCommand::Resume(interaction) => { match command.name.as_str() {
spawn(resume(interaction, Arc::clone(&self.state))) "play" => {
} if let Some(query_option) = command.options.iter().find(|opt| opt.name == "query") {
InteractionCommand::Leave(interaction) => { if let CommandOptionValue::String(query) = &query_option.value {
spawn(leave(interaction, Arc::clone(&self.state))) InteractionCommand::Play(query.clone())
} } else {
InteractionCommand::Join(interaction) => { InteractionCommand::NotImplemented
spawn(join(interaction, Arc::clone(&self.state))) }
} } else {
InteractionCommand::Queue(interaction) => { InteractionCommand::NotImplemented
spawn(queue(interaction, Arc::clone(&self.state))) }
} }
_ => {} "stop" => InteractionCommand::Stop,
}; "pause" => InteractionCommand::Pause,
"resume" => InteractionCommand::Resume,
"leave" => InteractionCommand::Leave,
"join" => InteractionCommand::Join,
"queue" => InteractionCommand::Queue,
_ => InteractionCommand::NotImplemented,
} }
} }

View File

@@ -1,9 +1,11 @@
mod handler; mod handler;
use handler::Handler; use handler::Handler;
mod colors;
mod commands; mod commands;
mod metadata; mod metadata;
mod signal; mod signal;
mod state; mod state;
use crate::commands::get_chat_commands; use crate::commands::get_chat_commands;
use dotenv::dotenv; use dotenv::dotenv;
use futures::StreamExt; use futures::StreamExt;
@@ -118,7 +120,7 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
state.standby.process(&event); state.standby.process(&event);
state.songbird.process(&event).await; state.songbird.process(&event).await;
handler.act(event).await; handler.act(event).await?;
} }
} }
} }