Merge pull request 'paginate /queue' (#3) from dev into main
All checks were successful
tests / fmt (push) Successful in 1m30s
tests / build (push) Successful in 1m36s
tests / test (push) Successful in 1m41s
tests / clippy (push) Successful in 1m39s
tests / pre-commit (push) Successful in 1m37s
deploy / release-image (push) Successful in 1h1m56s
All checks were successful
tests / fmt (push) Successful in 1m30s
tests / build (push) Successful in 1m36s
tests / test (push) Successful in 1m41s
tests / clippy (push) Successful in 1m39s
tests / pre-commit (push) Successful in 1m37s
deploy / release-image (push) Successful in 1h1m56s
Reviewed-on: #3
This commit was merged in pull request #3.
This commit is contained in:
1032
Cargo.lock
generated
1032
Cargo.lock
generated
File diff suppressed because it is too large
Load Diff
@@ -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
2
src/colors.rs
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
pub(crate) const BLURPLE: u32 = 0x58_65_F2;
|
||||||
|
pub(crate) const YELLOW: u32 = 0xFE_E7_5C;
|
||||||
@@ -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 {
|
||||||
|
|||||||
@@ -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!(
|
||||||
|
|||||||
@@ -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!(
|
||||||
|
|||||||
@@ -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;
|
||||||
|
|||||||
@@ -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!(
|
||||||
|
|||||||
@@ -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)
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -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(())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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!(
|
||||||
|
|||||||
@@ -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!(
|
||||||
|
|||||||
182
src/handler.rs
182
src/handler.rs
@@ -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,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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?;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user