Compare commits
22 Commits
1ca7d5204d
...
cleanup
| Author | SHA1 | Date | |
|---|---|---|---|
|
4a69e0f578
|
|||
|
cad6cf22fa
|
|||
|
9f0e934f92
|
|||
|
92be6c26a6
|
|||
|
c5d9c49452
|
|||
|
dbaf1a1374
|
|||
|
c0d9f6cad5
|
|||
|
1aa0efc4a6
|
|||
|
ab7f56fe05
|
|||
| feccbebd4d | |||
| d6a54dfc06 | |||
|
fa17ed75f6
|
|||
| b19f6e552c | |||
|
0bee437e53
|
|||
|
3781785a50
|
|||
| cabae9ea19 | |||
|
54c36688b7
|
|||
|
55fe212424
|
|||
| 2c91c744dc | |||
|
a0ac87959b
|
|||
| b4a1efb22a | |||
|
e831ede37c
|
@@ -2,7 +2,9 @@ name: tests
|
|||||||
|
|
||||||
on:
|
on:
|
||||||
push:
|
push:
|
||||||
pull_request:
|
|
||||||
|
paths-ignore:
|
||||||
|
- 'README.md'
|
||||||
|
|
||||||
env:
|
env:
|
||||||
CARGO_TERM_COLOR: always
|
CARGO_TERM_COLOR: always
|
||||||
@@ -36,6 +38,11 @@ jobs:
|
|||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
- run: sudo apt-get update
|
- run: sudo apt-get update
|
||||||
- run: sudo apt-get install -y cmake
|
- run: sudo apt-get install -y cmake
|
||||||
|
- uses: actions/setup-python@v5
|
||||||
|
with:
|
||||||
|
python-version: '3.12'
|
||||||
|
- name: Install yt-dlp
|
||||||
|
run: pip install yt-dlp
|
||||||
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
||||||
- name: Run tests
|
- name: Run tests
|
||||||
run: cargo test --verbose --all-features
|
run: cargo test --verbose --all-features
|
||||||
|
|||||||
@@ -5,6 +5,9 @@ on:
|
|||||||
branches:
|
branches:
|
||||||
- main
|
- main
|
||||||
|
|
||||||
|
paths-ignore:
|
||||||
|
- 'README.md'
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
release-image:
|
release-image:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
@@ -26,6 +29,5 @@ jobs:
|
|||||||
with:
|
with:
|
||||||
platforms: |
|
platforms: |
|
||||||
linux/amd64
|
linux/amd64
|
||||||
linux/arm64
|
|
||||||
push: true
|
push: true
|
||||||
tags: jheuel/ohrwurm:latest
|
tags: jheuel/ohrwurm:latest
|
||||||
|
|||||||
5
.github/workflows/ci.yml
vendored
5
.github/workflows/ci.yml
vendored
@@ -37,6 +37,11 @@ jobs:
|
|||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
- run: sudo apt-get update
|
- run: sudo apt-get update
|
||||||
- run: sudo apt-get install -y cmake
|
- run: sudo apt-get install -y cmake
|
||||||
|
- uses: actions/setup-python@v5
|
||||||
|
with:
|
||||||
|
python-version: '3.12'
|
||||||
|
- name: Install yt-dlp
|
||||||
|
run: pip install yt-dlp
|
||||||
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
||||||
- name: Run tests
|
- name: Run tests
|
||||||
run: cargo test --verbose --all-features
|
run: cargo test --verbose --all-features
|
||||||
|
|||||||
1035
Cargo.lock
generated
1035
Cargo.lock
generated
File diff suppressed because it is too large
Load Diff
@@ -21,5 +21,10 @@ 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.1"
|
||||||
|
anyhow = "1.0.86"
|
||||||
|
dashmap = "5.5.3"
|
||||||
|
async-trait = "0.1.80"
|
||||||
|
uuid = "1.8.0"
|
||||||
|
|||||||
@@ -8,7 +8,7 @@
|
|||||||
Ohrwurm is a user-friendly bot designed to play music in Discord voice chats. Once added to your server, you can request a song using the command `/play {query}`, where query can be a URL to a YouTube video or playlist, or a simple search term. The bot will fetch the song and start playing it. You can pause the music with `/pause`, resume playback with `/resume`, and stop and clear the queue with `/stop`. If the bot is alone in a voice chat, it will automatically leave, but you can also manually make it leave with the `/leave` command.
|
Ohrwurm is a user-friendly bot designed to play music in Discord voice chats. Once added to your server, you can request a song using the command `/play {query}`, where query can be a URL to a YouTube video or playlist, or a simple search term. The bot will fetch the song and start playing it. You can pause the music with `/pause`, resume playback with `/resume`, and stop and clear the queue with `/stop`. If the bot is alone in a voice chat, it will automatically leave, but you can also manually make it leave with the `/leave` command.
|
||||||
|
|
||||||
# Deployment
|
# Deployment
|
||||||
To deploy Ohrwurm with Docker, you can use the [provided Docker](https://hub.docker.com/r/jheuel/ohrwurm) image:
|
To deploy Ohrwurm with Docker, you can use the [provided Docker image](https://hub.docker.com/r/jheuel/ohrwurm):
|
||||||
```bash
|
```bash
|
||||||
docker run -d \
|
docker run -d \
|
||||||
--name ohrwurm \
|
--name ohrwurm \
|
||||||
|
|||||||
20
src/colors.rs
Normal file
20
src/colors.rs
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
#[allow(dead_code)]
|
||||||
|
pub(crate) const BLURPLE: u32 = 0x58_65_F2;
|
||||||
|
#[allow(dead_code)]
|
||||||
|
pub(crate) const YELLOW: u32 = 0xFE_E7_5C;
|
||||||
|
#[allow(dead_code)]
|
||||||
|
pub(crate) const WHITE: u32 = 0xFF_FF_FF;
|
||||||
|
#[allow(dead_code)]
|
||||||
|
pub(crate) const GREYPLE: u32 = 0x99_AA_B5;
|
||||||
|
#[allow(dead_code)]
|
||||||
|
pub(crate) const BLACK: u32 = 0x23_27_2A;
|
||||||
|
#[allow(dead_code)]
|
||||||
|
pub(crate) const DARK_BUT_NOT_BLACK: u32 = 0x2C_2F_33;
|
||||||
|
#[allow(dead_code)]
|
||||||
|
pub(crate) const NOT_QUITE_BLACK: u32 = 0x23_27_2A;
|
||||||
|
#[allow(dead_code)]
|
||||||
|
pub(crate) const GREEN: u32 = 0x57_F2_87;
|
||||||
|
#[allow(dead_code)]
|
||||||
|
pub(crate) const FUCHSIA: u32 = 0xEB_45_9E;
|
||||||
|
#[allow(dead_code)]
|
||||||
|
pub(crate) const RED: u32 = 0xED_42_45;
|
||||||
@@ -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},
|
||||||
@@ -38,11 +38,14 @@ 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(())
|
||||||
}
|
}
|
||||||
|
|
||||||
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,59 @@
|
|||||||
use crate::state::State;
|
use crate::state::{State, StateRef};
|
||||||
use std::error::Error;
|
use anyhow::Context;
|
||||||
use twilight_model::application::interaction::Interaction;
|
use std::{error::Error, sync::Arc};
|
||||||
|
use twilight_model::{
|
||||||
|
gateway::payload::incoming::InteractionCreate,
|
||||||
|
id::{marker::GuildMarker, Id},
|
||||||
|
};
|
||||||
|
|
||||||
|
pub(crate) async fn leave_if_alone(
|
||||||
|
guild_id: Id<GuildMarker>,
|
||||||
|
state: State,
|
||||||
|
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
||||||
|
let user = state
|
||||||
|
.cache
|
||||||
|
.current_user()
|
||||||
|
.context("Cannot get current user")?;
|
||||||
|
let user_voice_state = state
|
||||||
|
.cache
|
||||||
|
.voice_state(user.id, guild_id)
|
||||||
|
.context("Cannot get voice state")?;
|
||||||
|
let channel = state
|
||||||
|
.cache
|
||||||
|
.channel(user_voice_state.channel_id())
|
||||||
|
.context("Cannot get channel")?;
|
||||||
|
let channel_voice_states = state
|
||||||
|
.cache
|
||||||
|
.voice_channel_states(channel.id)
|
||||||
|
.context("Cannot get voice channel")?;
|
||||||
|
let count = channel_voice_states.count();
|
||||||
|
|
||||||
|
// count is 1 if the bot is the only one in the channel
|
||||||
|
if count == 1 {
|
||||||
|
leave_channel(guild_id, Arc::clone(&state)).await?;
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
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: 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!(
|
||||||
@@ -16,6 +66,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(())
|
||||||
}
|
}
|
||||||
|
|||||||
123
src/commands/loop_queue.rs
Normal file
123
src/commands/loop_queue.rs
Normal file
@@ -0,0 +1,123 @@
|
|||||||
|
use crate::metadata::MetadataMap;
|
||||||
|
use crate::state::{State, StateRef};
|
||||||
|
use async_trait::async_trait;
|
||||||
|
use songbird::{Event, EventContext, EventHandler, TrackEvent};
|
||||||
|
use std::ops::Sub;
|
||||||
|
use std::time::Duration;
|
||||||
|
use std::{error::Error, sync::Arc};
|
||||||
|
use twilight_model::{
|
||||||
|
gateway::payload::incoming::InteractionCreate,
|
||||||
|
http::interaction::{InteractionResponse, InteractionResponseType},
|
||||||
|
id::{marker::GuildMarker, Id},
|
||||||
|
};
|
||||||
|
use twilight_util::builder::InteractionResponseDataBuilder;
|
||||||
|
|
||||||
|
pub(crate) async fn loop_queue(
|
||||||
|
interaction: Box<InteractionCreate>,
|
||||||
|
state: State,
|
||||||
|
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
||||||
|
tracing::debug!(
|
||||||
|
"loop command in guild {:?} in channel {:?} by {:?}",
|
||||||
|
interaction.guild_id,
|
||||||
|
interaction.channel,
|
||||||
|
interaction.author(),
|
||||||
|
);
|
||||||
|
|
||||||
|
let guild_id: Id<GuildMarker> = if let Some(guild_id) = interaction.guild_id {
|
||||||
|
guild_id
|
||||||
|
} else {
|
||||||
|
return Ok(());
|
||||||
|
};
|
||||||
|
|
||||||
|
state.guild_settings.entry(guild_id).and_modify(|settings| {
|
||||||
|
settings.loop_queue = !settings.loop_queue;
|
||||||
|
});
|
||||||
|
|
||||||
|
let looping = state
|
||||||
|
.guild_settings
|
||||||
|
.get(&guild_id)
|
||||||
|
.expect("Cannot get loop state")
|
||||||
|
.loop_queue;
|
||||||
|
|
||||||
|
if let Some(call_lock) = state.songbird.get(guild_id) {
|
||||||
|
let mut call = call_lock.lock().await;
|
||||||
|
call.add_global_event(
|
||||||
|
Event::Track(TrackEvent::End),
|
||||||
|
TrackEndNotifier {
|
||||||
|
guild_id,
|
||||||
|
state: Arc::clone(&state),
|
||||||
|
},
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let message = if looping {
|
||||||
|
"I'm now looping the current queue!".to_string()
|
||||||
|
} else {
|
||||||
|
"I'm not looping anymore!".to_string()
|
||||||
|
};
|
||||||
|
|
||||||
|
let interaction_response_data = InteractionResponseDataBuilder::new()
|
||||||
|
.content(message)
|
||||||
|
.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?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
struct TrackEndNotifier {
|
||||||
|
guild_id: Id<GuildMarker>,
|
||||||
|
state: Arc<StateRef>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[async_trait]
|
||||||
|
impl EventHandler for TrackEndNotifier {
|
||||||
|
async fn act(&self, ctx: &EventContext<'_>) -> Option<Event> {
|
||||||
|
if !self
|
||||||
|
.state
|
||||||
|
.guild_settings
|
||||||
|
.get(&self.guild_id)
|
||||||
|
.unwrap()
|
||||||
|
.loop_queue
|
||||||
|
{
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
let EventContext::Track(track_list) = ctx else {
|
||||||
|
return None;
|
||||||
|
};
|
||||||
|
let (_, track_handle) = track_list.first()?;
|
||||||
|
if let Some(call_lock) = self.state.songbird.get(self.guild_id) {
|
||||||
|
let mut call = call_lock.lock().await;
|
||||||
|
|
||||||
|
// get metadata from finished track
|
||||||
|
let old_typemap_lock = track_handle.typemap().read().await;
|
||||||
|
let old_metadata = old_typemap_lock.get::<MetadataMap>().unwrap();
|
||||||
|
|
||||||
|
// enqueue track
|
||||||
|
let handle = call.enqueue_with_preload(
|
||||||
|
old_metadata.src.clone().into(),
|
||||||
|
old_metadata.duration.map(|duration| -> Duration {
|
||||||
|
if duration.as_secs() > 5 {
|
||||||
|
duration.sub(Duration::from_secs(5))
|
||||||
|
} else {
|
||||||
|
duration
|
||||||
|
}
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
|
||||||
|
// insert metadata into new track
|
||||||
|
let mut new_typemap = handle.typemap().write().await;
|
||||||
|
new_typemap.insert::<MetadataMap>(old_metadata.clone());
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -3,14 +3,21 @@ pub(crate) use join::join;
|
|||||||
|
|
||||||
mod leave;
|
mod leave;
|
||||||
pub(crate) use leave::leave;
|
pub(crate) use leave::leave;
|
||||||
|
pub(crate) use leave::leave_if_alone;
|
||||||
|
|
||||||
mod pause;
|
mod pause;
|
||||||
pub(crate) use pause::pause;
|
pub(crate) use pause::pause;
|
||||||
|
|
||||||
|
mod skip;
|
||||||
|
pub(crate) use skip::skip;
|
||||||
|
|
||||||
|
mod loop_queue;
|
||||||
|
pub(crate) use loop_queue::loop_queue;
|
||||||
|
|
||||||
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;
|
||||||
@@ -29,12 +36,14 @@ pub(crate) fn get_chat_commands() -> Vec<twilight_model::application::command::C
|
|||||||
vec![
|
vec![
|
||||||
CommandBuilder::new("join", "Join the channel", CommandType::ChatInput).build(),
|
CommandBuilder::new("join", "Join the channel", CommandType::ChatInput).build(),
|
||||||
CommandBuilder::new("leave", "Leave the channel", CommandType::ChatInput).build(),
|
CommandBuilder::new("leave", "Leave the channel", CommandType::ChatInput).build(),
|
||||||
|
CommandBuilder::new("loop", "Loop queue", CommandType::ChatInput).build(),
|
||||||
|
CommandBuilder::new("skip", "Skip track", CommandType::ChatInput).build(),
|
||||||
|
CommandBuilder::new("queue", "Print track queue", CommandType::ChatInput).build(),
|
||||||
|
CommandBuilder::new("stop", "Stop playing", CommandType::ChatInput).build(),
|
||||||
CommandBuilder::new("pause", "Pause playing", CommandType::ChatInput).build(),
|
CommandBuilder::new("pause", "Pause playing", CommandType::ChatInput).build(),
|
||||||
|
CommandBuilder::new("resume", "Resume playing", CommandType::ChatInput).build(),
|
||||||
CommandBuilder::new("play", "Add a song to the queue", CommandType::ChatInput)
|
CommandBuilder::new("play", "Add a song to the queue", CommandType::ChatInput)
|
||||||
.option(StringBuilder::new("query", "URL of a song").required(true))
|
.option(StringBuilder::new("query", "URL of a song").required(true))
|
||||||
.build(),
|
.build(),
|
||||||
CommandBuilder::new("queue", "Print track queue", CommandType::ChatInput).build(),
|
|
||||||
CommandBuilder::new("resume", "Resume playing", CommandType::ChatInput).build(),
|
|
||||||
CommandBuilder::new("stop", "Stop playing", CommandType::ChatInput).build(),
|
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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,19 +1,73 @@
|
|||||||
|
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 songbird::tracks::Track;
|
||||||
use std::io::{BufRead, BufReader};
|
use std::io::{BufRead, BufReader};
|
||||||
use std::{error::Error, ops::Sub, time::Duration};
|
use std::ops::Sub;
|
||||||
|
use std::{error::Error, 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::embed::{
|
||||||
|
EmbedAuthor, EmbedField, EmbedFooter, EmbedThumbnail,
|
||||||
|
};
|
||||||
|
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::InteractionResponseDataBuilder;
|
use twilight_util::builder::InteractionResponseDataBuilder;
|
||||||
use url::Url;
|
use url::Url;
|
||||||
|
|
||||||
|
#[derive(Debug, Serialize, Deserialize)]
|
||||||
|
struct YouTubeTrack {
|
||||||
|
url: Option<String>,
|
||||||
|
original_url: Option<String>,
|
||||||
|
title: String,
|
||||||
|
channel: String,
|
||||||
|
playlist: Option<String>,
|
||||||
|
playlist_id: Option<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();
|
||||||
|
|
||||||
|
if tracks.is_empty() {
|
||||||
|
if let Ok(stderr) = String::from_utf8(output.stderr) {
|
||||||
|
if stderr.contains("This video is only available to Music Premium members") {
|
||||||
|
return Err("This video is only available to Music Premium members".into());
|
||||||
|
}
|
||||||
|
if stderr.contains("YouTube said: The playlist does not exist.") {
|
||||||
|
return Err("YouTube said: The playlist does not exist.".into());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return Err("No tracks found".into());
|
||||||
|
}
|
||||||
|
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>> {
|
||||||
@@ -23,11 +77,17 @@ pub(crate) async fn play(
|
|||||||
interaction.author(),
|
interaction.author(),
|
||||||
);
|
);
|
||||||
|
|
||||||
|
let content = format!("Adding track(s) to the queue: {}", query);
|
||||||
|
let embeds = vec![EmbedBuilder::new()
|
||||||
|
.description(content)
|
||||||
|
.color(colors::YELLOW)
|
||||||
|
.build()];
|
||||||
let interaction_response_data = InteractionResponseDataBuilder::new()
|
let interaction_response_data = InteractionResponseDataBuilder::new()
|
||||||
.content(format!("Adding track(s) to queue: {}", query))
|
.flags(MessageFlags::LOADING)
|
||||||
|
.embeds(embeds)
|
||||||
.build();
|
.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
|
||||||
@@ -54,29 +114,90 @@ pub(crate) async fn play(
|
|||||||
|
|
||||||
debug!("query: {:?}", query);
|
debug!("query: {:?}", query);
|
||||||
|
|
||||||
// handle playlist links
|
let tracks = match get_tracks(query).await {
|
||||||
let urls = if query.contains("list=") {
|
Err(e) => {
|
||||||
get_playlist_urls(query).await?
|
let content = format!("{}", e);
|
||||||
} else {
|
let embeds = vec![EmbedBuilder::new()
|
||||||
vec![query]
|
.description(content)
|
||||||
|
.color(colors::RED)
|
||||||
|
.build()];
|
||||||
|
state
|
||||||
|
.http
|
||||||
|
.interaction(interaction.application_id)
|
||||||
|
.update_response(&interaction.token)
|
||||||
|
.embeds(Some(&embeds))?
|
||||||
|
.await?;
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
Ok(tracks) => tracks,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
if tracks.len() > 1 {
|
||||||
|
let first_track = tracks.first().unwrap();
|
||||||
|
let content = format!(
|
||||||
|
"Adding playlist [{}]({})",
|
||||||
|
first_track
|
||||||
|
.playlist
|
||||||
|
.clone()
|
||||||
|
.unwrap_or("Unknown".to_string()),
|
||||||
|
build_playlist_url(
|
||||||
|
&first_track
|
||||||
|
.playlist_id
|
||||||
|
.clone()
|
||||||
|
.unwrap_or("Unknown".to_string())
|
||||||
|
)
|
||||||
|
);
|
||||||
|
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;
|
||||||
call.queue().resume()?;
|
call.queue().resume()?;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
struct TrackType {
|
||||||
|
url: String,
|
||||||
|
title: Option<String>,
|
||||||
|
duration_string: String,
|
||||||
|
channel: String,
|
||||||
|
thumbnail: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
let mut tracks_added = vec![];
|
let mut tracks_added = vec![];
|
||||||
for url in urls {
|
for yttrack in &tracks {
|
||||||
let mut src = YoutubeDl::new(reqwest::Client::new(), url.to_string());
|
tracing::debug!("track: {:?}", yttrack);
|
||||||
|
let url = yttrack
|
||||||
|
.original_url
|
||||||
|
.clone()
|
||||||
|
.or(yttrack.url.clone())
|
||||||
|
.ok_or("Could not find url")?;
|
||||||
|
let mut src = YoutubeDl::new(reqwest::Client::new(), url.clone());
|
||||||
|
let src_copy = src.clone();
|
||||||
|
let track: Track = src_copy.into();
|
||||||
|
|
||||||
if let Ok(metadata) = src.aux_metadata().await {
|
if let Ok(metadata) = src.aux_metadata().await {
|
||||||
debug!("metadata: {:?}", metadata);
|
debug!("metadata: {:?}", metadata);
|
||||||
tracks_added.push(metadata.title.clone());
|
tracks_added.push(TrackType {
|
||||||
|
url: url.clone(),
|
||||||
|
title: metadata.title.clone(),
|
||||||
|
duration_string: yttrack.duration_string.clone(),
|
||||||
|
channel: yttrack.channel.clone(),
|
||||||
|
thumbnail: metadata.thumbnail.clone(),
|
||||||
|
});
|
||||||
|
|
||||||
if let Some(call_lock) = state.songbird.get(guild_id) {
|
if let Some(call_lock) = state.songbird.get(guild_id) {
|
||||||
let mut call = call_lock.lock().await;
|
let mut call = call_lock.lock().await;
|
||||||
let handle = call.enqueue_with_preload(
|
let handle = call.enqueue_with_preload(
|
||||||
src.into(),
|
track,
|
||||||
metadata.duration.map(|duration| -> Duration {
|
metadata.duration.map(|duration| -> Duration {
|
||||||
if duration.as_secs() > 5 {
|
if duration.as_secs() > 5 {
|
||||||
duration.sub(Duration::from_secs(5))
|
duration.sub(Duration::from_secs(5))
|
||||||
@@ -89,60 +210,150 @@ pub(crate) async fn play(
|
|||||||
x.insert::<MetadataMap>(Metadata {
|
x.insert::<MetadataMap>(Metadata {
|
||||||
title: metadata.title,
|
title: metadata.title,
|
||||||
duration: metadata.duration,
|
duration: metadata.duration,
|
||||||
|
url,
|
||||||
|
src,
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
let mut content = String::new();
|
let mut content = String::new();
|
||||||
let num_tracks_added = tracks_added.len();
|
let num_tracks_added = tracks_added.len();
|
||||||
match num_tracks_added {
|
let embeds = match num_tracks_added {
|
||||||
0 => {}
|
0 => {
|
||||||
|
vec![]
|
||||||
|
}
|
||||||
1 => {
|
1 => {
|
||||||
content = format!(
|
let track = tracks_added.first().unwrap();
|
||||||
"Added \"{}\" to queue",
|
|
||||||
tracks_added.first().unwrap().clone().unwrap()
|
let host = Url::parse(&track.url)?;
|
||||||
);
|
let host = host
|
||||||
|
.host_str()
|
||||||
|
.unwrap_or_default()
|
||||||
|
.trim_start_matches("www.");
|
||||||
|
let mut embed = EmbedBuilder::new()
|
||||||
|
.author(EmbedAuthor {
|
||||||
|
name: "🔊 Added to queue".to_string(),
|
||||||
|
icon_url: None,
|
||||||
|
proxy_icon_url: None,
|
||||||
|
url: None,
|
||||||
|
})
|
||||||
|
.title(track.title.clone().unwrap_or("Unknown".to_string()))
|
||||||
|
.url(track.url.clone())
|
||||||
|
.color(colors::BLURPLE)
|
||||||
|
.footer(EmbedFooter {
|
||||||
|
text: format!("Streaming from {}", host),
|
||||||
|
icon_url: Some(format!(
|
||||||
|
"https://www.google.com/s2/favicons?domain={}",
|
||||||
|
host
|
||||||
|
)),
|
||||||
|
proxy_icon_url: None,
|
||||||
|
})
|
||||||
|
.field(EmbedField {
|
||||||
|
inline: true,
|
||||||
|
name: "Duration".to_string(),
|
||||||
|
value: track.duration_string.clone(),
|
||||||
|
})
|
||||||
|
.field(EmbedField {
|
||||||
|
inline: true,
|
||||||
|
name: "Channel".to_string(),
|
||||||
|
value: track.channel.clone(),
|
||||||
|
})
|
||||||
|
.build();
|
||||||
|
|
||||||
|
if let Some(thumbnail) = &track.thumbnail {
|
||||||
|
embed.thumbnail = Some(EmbedThumbnail {
|
||||||
|
height: None,
|
||||||
|
proxy_url: None,
|
||||||
|
url: thumbnail.to_string(),
|
||||||
|
width: None,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
vec![embed]
|
||||||
}
|
}
|
||||||
_ => {
|
_ => {
|
||||||
content = format!("Added {} tracks to 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!(
|
||||||
content.push_str(&format!(" \"{}\"\n", track.unwrap()));
|
"Adding playlist: [{}]({})\n",
|
||||||
}
|
&first_track
|
||||||
|
.playlist
|
||||||
|
.clone()
|
||||||
|
.unwrap_or("Unknown".to_string()),
|
||||||
|
build_playlist_url(
|
||||||
|
&first_track
|
||||||
|
.playlist_id
|
||||||
|
.clone()
|
||||||
|
.unwrap_or("Unknown".to_string())
|
||||||
|
)
|
||||||
|
));
|
||||||
|
content.push_str(&format!(
|
||||||
|
"Added {} tracks to the queue:\n",
|
||||||
|
num_tracks_added
|
||||||
|
));
|
||||||
|
let embed = EmbedBuilder::new()
|
||||||
|
.description(content)
|
||||||
|
.color(colors::BLURPLE)
|
||||||
|
.build();
|
||||||
|
vec![embed]
|
||||||
}
|
}
|
||||||
}
|
};
|
||||||
|
|
||||||
state
|
state
|
||||||
.http
|
.http
|
||||||
.interaction(interaction.application_id)
|
.interaction(interaction.application_id)
|
||||||
.update_response(&interaction.token)
|
.update_response(&interaction.token)
|
||||||
.content(Some(&content))
|
.embeds(Some(&embeds))?
|
||||||
.unwrap()
|
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn get_playlist_urls(
|
#[cfg(test)]
|
||||||
url: String,
|
mod tests {
|
||||||
) -> Result<Vec<String>, Box<dyn Error + Send + Sync + 'static>> {
|
use super::*;
|
||||||
let output = Command::new("yt-dlp")
|
|
||||||
.args(vec![&url, "--flat-playlist", "-j"])
|
|
||||||
.output()
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
let reader = BufReader::new(output.stdout.as_slice());
|
#[tokio::test]
|
||||||
let urls = reader
|
async fn test_get_tracks() {
|
||||||
.lines()
|
let urls = [
|
||||||
.map_while(Result::ok)
|
"https://www.youtube.com/playlist?list=PLFxxhcEeloYa1OlnWD6UgxlVQKJH5i_0p",
|
||||||
.map(|line| {
|
"https://music.youtube.com/watch?v=RO75ZzqUOJw",
|
||||||
let entry: Value = serde_json::from_str(&line).unwrap();
|
"https://www.youtube.com/watch?v=qVHyl0P_P-M",
|
||||||
entry
|
"https://www.youtube.com/watch?v=34CZjsEI1yU",
|
||||||
.get("webpage_url")
|
];
|
||||||
.unwrap()
|
for url in urls.iter() {
|
||||||
.as_str()
|
println!("url: {:?}", url);
|
||||||
|
let tracks = get_tracks(url.to_string()).await.unwrap();
|
||||||
|
assert!(!tracks.is_empty());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_premium_tracks() {
|
||||||
|
let urls = ["https://www.youtube.com/watch?v=QgMZRmxQ0Dc"];
|
||||||
|
for url in urls.iter() {
|
||||||
|
println!("url: {:?}", url);
|
||||||
|
let tracks = get_tracks(url.to_string()).await;
|
||||||
|
assert!(tracks.is_err());
|
||||||
|
assert!(tracks
|
||||||
|
.err()
|
||||||
.unwrap()
|
.unwrap()
|
||||||
.to_string()
|
.to_string()
|
||||||
})
|
.contains("This video is only available to Music Premium members"));
|
||||||
.collect();
|
}
|
||||||
|
}
|
||||||
|
|
||||||
Ok(urls)
|
#[tokio::test]
|
||||||
|
async fn test_playlist_does_not_exist_tracks() {
|
||||||
|
let urls = ["https://www.youtube.com/playlist?list=PLox0oG0uy8Lc1IaIfGyrvtuRItuEyJiyG"];
|
||||||
|
for url in urls.iter() {
|
||||||
|
println!("url: {:?}", url);
|
||||||
|
let tracks = get_tracks(url.to_string()).await;
|
||||||
|
assert!(tracks.is_err());
|
||||||
|
assert!(tracks
|
||||||
|
.err()
|
||||||
|
.unwrap()
|
||||||
|
.to_string()
|
||||||
|
.contains("YouTube said: The playlist does not exist."));
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,15 +1,113 @@
|
|||||||
|
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::{
|
use twilight_util::builder::embed::EmbedBuilder;
|
||||||
application::interaction::Interaction, channel::message::MessageFlags,
|
|
||||||
http::interaction::InteractionResponse,
|
|
||||||
};
|
|
||||||
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 n_pages = (queue.len() + TRACKS_PER_PAGE - 1) / TRACKS_PER_PAGE;
|
||||||
|
if n_pages > 1 {
|
||||||
|
message.push_str(&format!("page {}/{}", 1 + page, n_pages));
|
||||||
|
}
|
||||||
|
vec![EmbedBuilder::new()
|
||||||
|
.description(&message)
|
||||||
|
.color(colors::BLURPLE)
|
||||||
|
.build()]
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) fn build_action_row(page: usize, n_pages: usize) -> Vec<Component> {
|
||||||
|
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)),
|
||||||
|
style: ButtonStyle::Primary,
|
||||||
|
label: Some("Refresh".to_string()),
|
||||||
|
emoji: Some(ReactionType::Unicode {
|
||||||
|
name: "🔄".to_string(),
|
||||||
|
}),
|
||||||
|
url: None,
|
||||||
|
disabled: false,
|
||||||
|
}),
|
||||||
|
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 >= n_pages - 1,
|
||||||
|
}),
|
||||||
|
],
|
||||||
|
})]
|
||||||
|
}
|
||||||
|
|
||||||
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!(
|
||||||
@@ -21,57 +119,43 @@ 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()),
|
|
||||||
)
|
|
||||||
.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_str("`\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
let interaction_response_data = InteractionResponseDataBuilder::new()
|
|
||||||
.content(&message)
|
|
||||||
.flags(MessageFlags::EPHEMERAL)
|
|
||||||
.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 n_pages = (queue.len() + TRACKS_PER_PAGE - 1) / TRACKS_PER_PAGE;
|
||||||
|
let action_row = build_action_row(0, n_pages);
|
||||||
|
|
||||||
|
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!(
|
||||||
|
|||||||
45
src/commands/skip.rs
Normal file
45
src/commands/skip.rs
Normal file
@@ -0,0 +1,45 @@
|
|||||||
|
use crate::state::State;
|
||||||
|
use std::error::Error;
|
||||||
|
use twilight_model::{
|
||||||
|
gateway::payload::incoming::InteractionCreate,
|
||||||
|
http::interaction::{InteractionResponse, InteractionResponseType},
|
||||||
|
};
|
||||||
|
use twilight_util::builder::InteractionResponseDataBuilder;
|
||||||
|
|
||||||
|
pub(crate) async fn skip(
|
||||||
|
interaction: Box<InteractionCreate>,
|
||||||
|
state: State,
|
||||||
|
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
||||||
|
tracing::debug!(
|
||||||
|
"skip command in guild {:?} in channel {:?} by {:?}",
|
||||||
|
interaction.guild_id,
|
||||||
|
interaction.channel,
|
||||||
|
interaction.author(),
|
||||||
|
);
|
||||||
|
|
||||||
|
let Some(guild_id) = interaction.guild_id else {
|
||||||
|
return Ok(());
|
||||||
|
};
|
||||||
|
|
||||||
|
if let Some(call_lock) = state.songbird.get(guild_id) {
|
||||||
|
let call = call_lock.lock().await;
|
||||||
|
call.queue().skip()?;
|
||||||
|
}
|
||||||
|
|
||||||
|
let interaction_response_data = InteractionResponseDataBuilder::new()
|
||||||
|
.content("Skipped a track")
|
||||||
|
.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?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
@@ -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!(
|
||||||
@@ -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();
|
||||||
|
|||||||
244
src/handler.rs
244
src/handler.rs
@@ -1,71 +1,18 @@
|
|||||||
use crate::commands::{delete, join, leave, pause, play, queue, resume, stop};
|
use crate::commands::queue::{build_action_row, build_queue_embeds, TRACKS_PER_PAGE};
|
||||||
|
use crate::commands::{
|
||||||
|
delete, join, leave, leave_if_alone, loop_queue, pause, play, queue, resume, skip, stop,
|
||||||
|
};
|
||||||
|
use crate::interaction_commands::InteractionCommand;
|
||||||
use crate::state::State;
|
use crate::state::State;
|
||||||
use futures::Future;
|
use crate::utils::spawn;
|
||||||
use std::error::Error;
|
use anyhow::Context;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use tracing::debug;
|
|
||||||
use twilight_gateway::Event;
|
use twilight_gateway::Event;
|
||||||
use twilight_model::application::interaction::application_command::CommandOptionValue;
|
use twilight_model::application::interaction::message_component::MessageComponentInteractionData;
|
||||||
use twilight_model::application::interaction::{Interaction, InteractionData};
|
use twilight_model::application::interaction::InteractionData;
|
||||||
use twilight_model::gateway::payload::incoming::VoiceStateUpdate;
|
use twilight_model::gateway::payload::incoming::InteractionCreate;
|
||||||
|
use twilight_model::http::interaction::{InteractionResponse, InteractionResponseType};
|
||||||
#[derive(Debug)]
|
use twilight_util::builder::InteractionResponseDataBuilder;
|
||||||
enum InteractionCommand {
|
|
||||||
Play(Interaction, String),
|
|
||||||
Stop(Interaction),
|
|
||||||
Pause(Interaction),
|
|
||||||
Resume(Interaction),
|
|
||||||
Leave(Interaction),
|
|
||||||
Join(Interaction),
|
|
||||||
Queue(Interaction),
|
|
||||||
NotImplemented,
|
|
||||||
}
|
|
||||||
|
|
||||||
fn spawn(
|
|
||||||
fut: impl Future<Output = Result<(), Box<dyn Error + Send + Sync + 'static>>> + Send + 'static,
|
|
||||||
) {
|
|
||||||
tokio::spawn(async move {
|
|
||||||
if let Err(why) = fut.await {
|
|
||||||
tracing::debug!("handler error: {:?}", why);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) async fn leave_if_alone(
|
|
||||||
update: VoiceStateUpdate,
|
|
||||||
state: State,
|
|
||||||
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
|
||||||
let guild_id = update.guild_id.ok_or("Guild ID not found")?;
|
|
||||||
let user = state
|
|
||||||
.cache
|
|
||||||
.current_user()
|
|
||||||
.ok_or("Cannot get current user")?;
|
|
||||||
let user_voice_state = state
|
|
||||||
.cache
|
|
||||||
.voice_state(user.id, guild_id)
|
|
||||||
.ok_or("Cannot get voice state")?;
|
|
||||||
let channel = state
|
|
||||||
.cache
|
|
||||||
.channel(user_voice_state.channel_id())
|
|
||||||
.ok_or("Cannot get channel")?;
|
|
||||||
let channel_voice_states = state
|
|
||||||
.cache
|
|
||||||
.voice_channel_states(channel.id)
|
|
||||||
.ok_or("Cannot get voice channel")?;
|
|
||||||
let count = channel_voice_states.count();
|
|
||||||
|
|
||||||
// count is 1 if the bot is the only one in the channel
|
|
||||||
if count == 1 {
|
|
||||||
// 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?;
|
|
||||||
}
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) struct Handler {
|
pub(crate) struct Handler {
|
||||||
state: State,
|
state: State,
|
||||||
@@ -75,80 +22,117 @@ 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 {
|
self.handle_messages(&event).await?;
|
||||||
|
self.handle_voice_state_update(&event).await?;
|
||||||
|
self.handle_interaction(&event).await?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle_messages(&self, event: &Event) -> anyhow::Result<()> {
|
||||||
|
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) => {
|
_ => Ok(()),
|
||||||
spawn(leave_if_alone(*update.clone(), Arc::clone(&self.state)))
|
|
||||||
}
|
|
||||||
_ => {}
|
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
let interaction_command = match event {
|
async fn handle_voice_state_update(&self, event: &Event) -> anyhow::Result<()> {
|
||||||
Event::InteractionCreate(interaction) => {
|
match event {
|
||||||
debug!("interaction: {:?}", &interaction);
|
Event::VoiceStateUpdate(update) => {
|
||||||
match &interaction.data {
|
let guild_id = update.guild_id.context("Guild ID not found")?;
|
||||||
Some(InteractionData::ApplicationCommand(command)) => {
|
spawn(leave_if_alone(guild_id, Arc::clone(&self.state)));
|
||||||
debug!("command: {:?}", command);
|
Ok(())
|
||||||
match command.name.as_str() {
|
}
|
||||||
"play" => {
|
_ => Ok(()),
|
||||||
if let Some(query_option) =
|
}
|
||||||
command.options.iter().find(|opt| opt.name == "query")
|
}
|
||||||
{
|
|
||||||
if let CommandOptionValue::String(query) = &query_option.value {
|
async fn handle_interaction(&self, event: &Event) -> anyhow::Result<()> {
|
||||||
InteractionCommand::Play(
|
match event {
|
||||||
interaction.0.clone(),
|
Event::InteractionCreate(interaction) => match &interaction.data {
|
||||||
query.clone(),
|
Some(InteractionData::ApplicationCommand(command)) => {
|
||||||
)
|
self.handle_application_command(command.clone().into(), interaction.clone())
|
||||||
} else {
|
|
||||||
InteractionCommand::NotImplemented
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
InteractionCommand::NotImplemented
|
|
||||||
}
|
|
||||||
}
|
|
||||||
"stop" => InteractionCommand::Stop(interaction.0.clone()),
|
|
||||||
"pause" => InteractionCommand::Pause(interaction.0.clone()),
|
|
||||||
"resume" => InteractionCommand::Resume(interaction.0.clone()),
|
|
||||||
"leave" => InteractionCommand::Leave(interaction.0.clone()),
|
|
||||||
"join" => InteractionCommand::Join(interaction.0.clone()),
|
|
||||||
"queue" => InteractionCommand::Queue(interaction.0.clone()),
|
|
||||||
_ => InteractionCommand::NotImplemented,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
_ => InteractionCommand::NotImplemented,
|
|
||||||
}
|
}
|
||||||
|
Some(InteractionData::MessageComponent(data)) => {
|
||||||
|
self.handle_message_component(data, interaction.clone())
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
_ => Ok(()),
|
||||||
|
},
|
||||||
|
_ => Ok(()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn handle_application_command(
|
||||||
|
&self,
|
||||||
|
command: InteractionCommand,
|
||||||
|
interaction: Box<InteractionCreate>,
|
||||||
|
) -> anyhow::Result<()> {
|
||||||
|
{
|
||||||
|
match command {
|
||||||
|
InteractionCommand::Play(query) => {
|
||||||
|
spawn(play(interaction, Arc::clone(&self.state), query))
|
||||||
|
}
|
||||||
|
InteractionCommand::Stop => spawn(stop(interaction, Arc::clone(&self.state))),
|
||||||
|
InteractionCommand::Pause => spawn(pause(interaction, Arc::clone(&self.state))),
|
||||||
|
InteractionCommand::Skip => spawn(skip(interaction, Arc::clone(&self.state))),
|
||||||
|
InteractionCommand::Loop => spawn(loop_queue(interaction, Arc::clone(&self.state))),
|
||||||
|
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))),
|
||||||
|
_ => {}
|
||||||
}
|
}
|
||||||
_ => InteractionCommand::NotImplemented,
|
Ok(())
|
||||||
};
|
}
|
||||||
debug!("{:?}", interaction_command);
|
}
|
||||||
match interaction_command {
|
|
||||||
InteractionCommand::Play(interaction, query) => {
|
async fn handle_message_component(
|
||||||
spawn(play(interaction, Arc::clone(&self.state), query))
|
&self,
|
||||||
|
data: &MessageComponentInteractionData,
|
||||||
|
interaction: Box<InteractionCreate>,
|
||||||
|
) -> anyhow::Result<()> {
|
||||||
|
if !data.custom_id.starts_with("page:") {
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
|
let page = data
|
||||||
|
.custom_id
|
||||||
|
.trim_start_matches("page:")
|
||||||
|
.parse::<usize>()
|
||||||
|
.unwrap_or(0);
|
||||||
|
|
||||||
|
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();
|
||||||
}
|
}
|
||||||
InteractionCommand::Stop(interaction) => {
|
let n_pages = (queue.len() + TRACKS_PER_PAGE - 1) / TRACKS_PER_PAGE;
|
||||||
spawn(stop(interaction, Arc::clone(&self.state)))
|
let page = page.min(n_pages - 1).max(0);
|
||||||
}
|
let embeds = build_queue_embeds(&queue, page).await;
|
||||||
InteractionCommand::Pause(interaction) => {
|
let action_row = build_action_row(page, n_pages);
|
||||||
spawn(pause(interaction, Arc::clone(&self.state)))
|
|
||||||
}
|
let interaction_response_data = InteractionResponseDataBuilder::new()
|
||||||
InteractionCommand::Resume(interaction) => {
|
.embeds(embeds)
|
||||||
spawn(resume(interaction, Arc::clone(&self.state)))
|
.components(action_row)
|
||||||
}
|
.build();
|
||||||
InteractionCommand::Leave(interaction) => {
|
let response = InteractionResponse {
|
||||||
spawn(leave(interaction, Arc::clone(&self.state)))
|
kind: InteractionResponseType::UpdateMessage,
|
||||||
}
|
data: Some(interaction_response_data),
|
||||||
InteractionCommand::Join(interaction) => {
|
};
|
||||||
spawn(join(interaction, Arc::clone(&self.state)))
|
self.state
|
||||||
}
|
.http
|
||||||
InteractionCommand::Queue(interaction) => {
|
.interaction(interaction.application_id)
|
||||||
spawn(queue(interaction, Arc::clone(&self.state)))
|
.create_response(interaction.id, &interaction.token, &response)
|
||||||
}
|
.await?;
|
||||||
_ => {}
|
Ok(())
|
||||||
};
|
} else {
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
44
src/interaction_commands.rs
Normal file
44
src/interaction_commands.rs
Normal file
@@ -0,0 +1,44 @@
|
|||||||
|
use twilight_model::application::interaction::application_command::{
|
||||||
|
CommandData, CommandOptionValue,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub(crate) enum InteractionCommand {
|
||||||
|
Play(String),
|
||||||
|
Stop,
|
||||||
|
Pause,
|
||||||
|
Skip,
|
||||||
|
Loop,
|
||||||
|
Resume,
|
||||||
|
Leave,
|
||||||
|
Join,
|
||||||
|
Queue,
|
||||||
|
NotImplemented,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<Box<CommandData>> for InteractionCommand {
|
||||||
|
fn from(command: Box<CommandData>) -> InteractionCommand {
|
||||||
|
match command.name.as_str() {
|
||||||
|
"play" => {
|
||||||
|
if let Some(query_option) = command.options.iter().find(|opt| opt.name == "query") {
|
||||||
|
if let CommandOptionValue::String(query) = &query_option.value {
|
||||||
|
InteractionCommand::Play(query.clone())
|
||||||
|
} else {
|
||||||
|
InteractionCommand::NotImplemented
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
InteractionCommand::NotImplemented
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"stop" => InteractionCommand::Stop,
|
||||||
|
"pause" => InteractionCommand::Pause,
|
||||||
|
"skip" => InteractionCommand::Skip,
|
||||||
|
"loop" => InteractionCommand::Loop,
|
||||||
|
"resume" => InteractionCommand::Resume,
|
||||||
|
"leave" => InteractionCommand::Leave,
|
||||||
|
"join" => InteractionCommand::Join,
|
||||||
|
"queue" => InteractionCommand::Queue,
|
||||||
|
_ => InteractionCommand::NotImplemented,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,9 +1,13 @@
|
|||||||
mod handler;
|
mod handler;
|
||||||
use handler::Handler;
|
use handler::Handler;
|
||||||
|
mod colors;
|
||||||
mod commands;
|
mod commands;
|
||||||
|
mod interaction_commands;
|
||||||
mod metadata;
|
mod metadata;
|
||||||
mod signal;
|
mod signal;
|
||||||
mod state;
|
mod state;
|
||||||
|
mod utils;
|
||||||
|
|
||||||
use crate::commands::get_chat_commands;
|
use crate::commands::get_chat_commands;
|
||||||
use dotenv::dotenv;
|
use dotenv::dotenv;
|
||||||
use futures::StreamExt;
|
use futures::StreamExt;
|
||||||
@@ -75,6 +79,7 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
|||||||
cache,
|
cache,
|
||||||
songbird,
|
songbird,
|
||||||
standby: Standby::new(),
|
standby: Standby::new(),
|
||||||
|
guild_settings: Default::default(),
|
||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
};
|
};
|
||||||
@@ -118,7 +123,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?;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,9 +1,12 @@
|
|||||||
use songbird::typemap::TypeMapKey;
|
use songbird::{input::YoutubeDl, typemap::TypeMapKey};
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
pub(crate) struct Metadata {
|
pub(crate) struct Metadata {
|
||||||
pub(crate) title: Option<String>,
|
pub(crate) title: Option<String>,
|
||||||
pub(crate) duration: Option<Duration>,
|
pub(crate) duration: Option<Duration>,
|
||||||
|
pub(crate) url: String,
|
||||||
|
pub(crate) src: YoutubeDl,
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) struct MetadataMap;
|
pub(crate) struct MetadataMap;
|
||||||
|
|||||||
20
src/state.rs
20
src/state.rs
@@ -1,15 +1,35 @@
|
|||||||
|
use dashmap::DashMap;
|
||||||
use songbird::Songbird;
|
use songbird::Songbird;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use twilight_cache_inmemory::InMemoryCache;
|
use twilight_cache_inmemory::InMemoryCache;
|
||||||
use twilight_http::Client as HttpClient;
|
use twilight_http::Client as HttpClient;
|
||||||
|
use twilight_model::id::{marker::GuildMarker, Id};
|
||||||
use twilight_standby::Standby;
|
use twilight_standby::Standby;
|
||||||
|
|
||||||
pub(crate) type State = Arc<StateRef>;
|
pub(crate) type State = Arc<StateRef>;
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub(crate) struct Settings {
|
||||||
|
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,
|
||||||
pub(crate) cache: InMemoryCache,
|
pub(crate) cache: InMemoryCache,
|
||||||
pub(crate) songbird: Songbird,
|
pub(crate) songbird: Songbird,
|
||||||
pub(crate) standby: Standby,
|
pub(crate) standby: Standby,
|
||||||
|
pub(crate) guild_settings: DashMap<Id<GuildMarker>, Settings>,
|
||||||
}
|
}
|
||||||
|
|||||||
12
src/utils.rs
Normal file
12
src/utils.rs
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
use futures::Future;
|
||||||
|
use std::error::Error;
|
||||||
|
|
||||||
|
pub(crate) fn spawn(
|
||||||
|
fut: impl Future<Output = Result<(), Box<dyn Error + Send + Sync + 'static>>> + Send + 'static,
|
||||||
|
) {
|
||||||
|
tokio::spawn(async move {
|
||||||
|
if let Err(why) = fut.await {
|
||||||
|
tracing::debug!("handler error: {:?}", why);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user