Compare commits

..

No commits in common. "723bd32335299c56e382d1aade749b1597e1c7fc" and "7efd98f1f3548090a6a7e523d2f470694049ff46" have entirely different histories.

15 changed files with 105 additions and 169 deletions

View file

@ -1,7 +1,7 @@
use poise;
use serenity::all::{Embed, User};
use crate::types::{Context, ContextExt, Error};
use crate::types::{Error, Context};
#[poise::command(
slash_command,
@ -15,6 +15,6 @@ pub async fn gif(ctx: Context<'_>,
) -> Result<(), Error> {
// let embed;
// send_with_embed(ctx, "hug", &title, &desc).await?;
ctx.reply_ephemeral("Done!").await?;
ctx.reply("Done!").await?;
Ok(())
}

View file

@ -1,7 +1,8 @@
use poise;
use serenity::all::User;
use super::interaction::send_with_embed;
use crate::types::{Context, Error};
use crate::types::{Context, ContextExt, Error};
#[poise::command(
slash_command,
@ -15,5 +16,6 @@ pub async fn headpat(ctx: Context<'_>,
let title = "HEADPATS!";
let desc = format!("{} headpats {}", ctx.author(), user);
send_with_embed(ctx, "headpat", &title, &desc).await?;
ctx.reply_ephemeral("Done!").await?;
Ok(())
}
}

View file

@ -2,7 +2,7 @@ use poise;
use serenity::all::User;
use super::interaction::send_with_embed;
use crate::types::{Context, Error};
use crate::types::{Context, ContextExt, Error};
#[poise::command(
slash_command,
@ -16,5 +16,6 @@ pub async fn hug(ctx: Context<'_>,
let title = "HUGS!";
let desc = format!("{} hugs {}", ctx.author(), user);
send_with_embed(ctx, "hug", &title, &desc).await?;
ctx.reply_ephemeral("Done!").await?;
Ok(())
}

View file

@ -1,5 +1,5 @@
use poise::CreateReply;
use serenity::all::{Colour, CreateEmbed};
use anyhow::anyhow;
use serenity::all::{Colour, CreateEmbed, CreateMessage};
use tenorv2::tenor_builder::Tenor;
use crate::{types::Context, utils::gifs::get_random_tenor_gif};
@ -20,7 +20,15 @@ pub(super) async fn send_with_embed(ctx: Context<'_>, query: &str, title: &str,
.description(desc)
.image(url.as_str());
ctx.send(CreateReply::default().embed(embed)).await?;
if ctx.guild_id().is_none() {
return Err(anyhow!("Guild id not available in context"));
}
ctx.channel_id()
.send_message(
ctx.http(),
CreateMessage::new().add_embed(embed)
).await?;
Ok(())
}

View file

@ -1,7 +1,6 @@
use std::vec;
use songbird::input::YoutubeDl;
use songbird::TrackEvent;
use crate::types::{Context, ContextExt, Error};
use crate::commands::voice::voice_utils::autocomplete_channel;
@ -12,8 +11,7 @@ use super::connect;
#[poise::command(
slash_command,
description_localized("en-US", "Plays music from supported URL"),
category = "Voice",
guild_only
category = "Voice"
)]
pub async fn play(ctx: Context<'_>,
#[autocomplete = "autocomplete_channel"]
@ -22,20 +20,34 @@ pub async fn play(ctx: Context<'_>,
#[description = "Source URL: "]
url: String,
) -> Result<(), Error> {
let events = vec![TrackEvent::End, TrackEvent::Error];
let (manager, guild_id) = match connect(&ctx, channel, events).await {
Ok(result) => result,
Err(e) => {
if ctx.guild().is_none() {
ctx.reply_ephemeral("Can't use this outside of guild").await?;
return Ok(());
}
let manager = songbird::get(ctx.serenity_context())
.await
.expect("Songbird Voice client placed in at initialisation.")
.clone();
let Some(guild_id) = ctx.guild_id() else {
ctx.reply_ephemeral("Guild id not found").await?;
return Ok(())
};
let http_client = &ctx.data().http_client;
if manager.get(guild_id).is_none() {
if let Err(e) = connect(&ctx, guild_id, channel).await {
ctx.reply_ephemeral(&e.to_string()).await?;
println!("SONGBIRD MANAGER ERROR: {}", e.to_string());
return Ok(())
}
};
}
if let Some(handler_lock) = manager.get(guild_id) {
let mut handler = handler_lock.lock().await;
let http_client = &ctx.data().http_client;
let src = YoutubeDl::new(http_client.clone(), url);
handler.enqueue_input(src.into()).await;
}

View file

@ -2,7 +2,6 @@ use std::vec;
use poise::CreateReply;
use songbird::input::{File, Input};
use songbird::TrackEvent;
use crate::utils::debug::send_error;
use crate::types::{Context, ContextExt, Error};
@ -19,8 +18,7 @@ use super::voice_utils::{connect, autocomplete_channel};
#[poise::command(
slash_command,
description_localized("en-US", "Disconnect from voice channel"),
category = "Voice",
guild_only
category = "Voice"
)]
pub async fn disconnect(
ctx: Context<'_>
@ -70,8 +68,7 @@ async fn autocomplete_song(
#[poise::command(
slash_command,
description_localized("en-US", "Play song from server storage"),
category = "Voice",
guild_only
category = "Voice"
)]
pub async fn play_local(ctx: Context<'_>,
#[autocomplete = "autocomplete_channel"]
@ -81,16 +78,32 @@ pub async fn play_local(ctx: Context<'_>,
#[description = "Filename of local song: "]
file_name: String
) -> Result<(), Error> {
let events = vec![TrackEvent::End, TrackEvent::Error];
let (manager, guild_id) = match connect(&ctx, channel, events).await {
Ok(result) => result,
Err(e) => {
ctx.reply_ephemeral(&e.to_string()).await?;
println!("SONGBIRD MANAGER ERROR: {}", e.to_string());
return Ok(())
}
if ctx.guild().is_none() {
ctx.reply_ephemeral("Can't use this outside of guild").await?;
return Ok(());
}
let manager = songbird::get(ctx.serenity_context())
.await
.expect("Songbird Voice client placed in at initialisation.")
.clone();
let Some(guild_id) = ctx.guild_id() else {
ctx.reply_ephemeral("Guild id not found").await?;
return Ok(())
};
if manager.get(guild_id).is_none() {
match connect(&ctx, guild_id, channel).await {
Ok(_) => (),
Err(e) => {
ctx.reply_ephemeral(&e.to_string()).await?;
return Ok(())
}
}
}
if let Some(handler_lock) = manager.get(guild_id) {
let mut handler = handler_lock.lock().await;
@ -111,8 +124,7 @@ pub async fn play_local(ctx: Context<'_>,
#[poise::command(
slash_command,
description_localized("en-US", "Display currently playing info"),
category = "Voice",
guild_only
category = "Voice"
)]
pub async fn playing(ctx: Context<'_>) -> Result<(), Error> {

View file

@ -5,7 +5,6 @@ use reqwest::Client;
use songbird::input::Input;
use songbird::input::HttpRequest;
use songbird::TrackEvent;
use super::super::connect;
use super::link_or_string;
@ -22,8 +21,7 @@ use crate::commands::voice::voice_utils::autocomplete_channel;
#[poise::command(
slash_command,
description_localized("en-US", "Plays music from URL source"),
subcommands("search", "play"),
guild_only
subcommands("search", "play")
)]
pub async fn radio(_ctx: Context<'_>) -> Result<(), Error> {
Ok(())
@ -43,6 +41,12 @@ pub async fn play(ctx: Context<'_>,
#[description = "Radio station: "]
name: String,
) -> Result<(), Error> {
if ctx.guild().is_none() {
ctx.reply_ephemeral("Can't use this outside of guild").await?;
return Ok(());
}
let api = &ctx.data().radio_browser;
let stations_result = match link_or_string(&name) {
@ -84,15 +88,23 @@ pub async fn play(ctx: Context<'_>,
return Ok(())
};
let events = vec![TrackEvent::End, TrackEvent::Error];
let (manager, guild_id) = match connect(&ctx, channel, events).await {
Ok(result) => result,
Err(e) => {
let manager = songbird::get(ctx.serenity_context())
.await
.expect("Songbird Voice client placed in at initialisation.")
.clone();
let Some(guild_id) = ctx.guild_id() else {
ctx.reply_ephemeral("Guild id not found").await?;
return Ok(())
};
if manager.get(guild_id).is_none() {
if let Err(e) = connect(&ctx, guild_id, channel).await {
ctx.reply_ephemeral(&e.to_string()).await?;
println!("SONGBIRD MANAGER ERROR: {}", e.to_string());
return Ok(())
}
};
}
if let Some(handler_lock) = manager.get(guild_id) {
let mut handler = handler_lock.lock().await;

View file

@ -35,6 +35,7 @@ pub struct GeneralInfo {
pub duration: Option<String>
}
#[async_trait]
impl GenerateEmbed for GeneralInfo {
async fn generate_embed(&self) -> CreateEmbed {

View file

@ -1,11 +1,8 @@
use std::sync::Arc;
use serenity::all::{ChannelId, GuildId};
use serenity::async_trait;
use serenity::model::voice;
use songbird::events::{Event, EventContext, EventHandler as VoiceEventHandler};
use songbird::{Songbird, TrackEvent};
use songbird::TrackEvent;
use crate::{types::{Context, Error}, utils::utilities::get_channel_by_name};
@ -52,15 +49,7 @@ impl VoiceEventHandler for TrackErrorNotifier {
}
}
pub async fn connect(ctx: &Context<'_>, channel: Option<String>, events: Vec<TrackEvent>) -> Result<(Arc<Songbird>, GuildId), String> {
if ctx.guild().is_none() {
return Err("dadsa".to_string())
}
let Some(guild_id) = ctx.guild_id() else {
return Err("dadsa".to_string())
};
pub async fn connect(ctx: &Context<'_>, guild_id: GuildId, channel: Option<String>) -> Result<(), Error> {
let voice_channel = get_voice_channel(&ctx, channel).await?;
let manager = songbird::get(ctx.serenity_context())
@ -70,12 +59,10 @@ pub async fn connect(ctx: &Context<'_>, channel: Option<String>, events: Vec<Tra
if let Ok(handler_lock) = manager.join(guild_id, voice_channel).await {
let mut handler = handler_lock.lock().await;
for event in events {
handler.add_global_event(event.into(), TrackErrorNotifier);
}
handler.add_global_event(TrackEvent::Error.into(), TrackErrorNotifier);
}
Ok((manager, guild_id))
Ok(())
}
pub async fn autocomplete_channel(

View file

@ -6,6 +6,7 @@ use songbird::input::YoutubeDl;
use crate::commands::util::connect;
use crate::util::poise_context_extension::ContextExt;
use crate::types::{Context, Error};
use crate::commands::voice::util::autocomplete_channel;
// TODO: search, queue
#[poise::command(
@ -13,9 +14,9 @@ use crate::types::{Context, Error};
description_localized("en-US", "Plays music from YouTube URL")
)]
pub async fn play_yt(ctx: Context<'_>,
#[autocomplete = "autocomplete_channel"]
#[description = "Voice channel name: "]
#[channel_types("Voice")]
channel: Option<GuildChannel>,
channel: Option<String>,
#[description = "Source URL: "]
url: String,
) -> Result<(), Error> {

View file

@ -1,2 +0,0 @@
pub mod message_handler;
pub mod voice_state_handler;

View file

@ -1,75 +0,0 @@
use std::time::Duration;
use serenity::all::{Context, CreateMessage, GuildChannel, VoiceState};
use tokio::time::sleep;
use crate::utils::{debug::send_error, voice_state_to_guild_channel};
async fn get_channel_info(ctx: &Context, voice_state: &VoiceState) -> Option<GuildChannel> {
let voice_channel = voice_state_to_guild_channel(&ctx, &voice_state).await?;
let Ok(members) = voice_channel.members(&ctx.cache) else {
return None
};
let mut is_connected = false;
let mut users_connected: usize = 0;
for member in &members {
if member.user.id == ctx.cache.current_user().id {
is_connected = true
}
if ! member.user.bot {
users_connected += 1;
}
}
// Check if there is no real user in the voice channel
if ! is_connected || users_connected > 0 {
return None;
}
Some(voice_channel)
}
pub async fn handle_voice_update(ctx: Context, voice_state_old: VoiceState, voice_state: VoiceState) -> Option<()> {
// The user did not disconnect so we don't need to handle if the bot was left alone
// Logic is as follows:
// User connected -> voice_state_old is None (we handle this before this function)
// User moved to different VC or disconnected -> we only check old channel for number of users anyway
if voice_state.channel_id.is_some() {
return None
}
let mut voice_channel = get_channel_info(&ctx, &voice_state_old).await?;
let manager = songbird::get(&ctx)
.await
.expect("Songbird Voice client placed in at initialisation.")
.clone();
// bot is not playing any music so we don't need to handle this
if manager.get(voice_channel.guild_id).is_none() {
return None;
}
// There is a problem with this implementation
// if bot is left alone and users joins and disconnects while this counts down
// this is not a big problem since we want to disconnect anyway
sleep(Duration::from_secs(5)).await;
voice_channel = get_channel_info(&ctx, &voice_state_old).await?;
match manager.remove(voice_channel.guild_id).await {
Ok(()) => {
let _ = voice_channel.send_message(ctx.http,
CreateMessage::new()
.content("Disconnected to save bandwidth")
).await;
}
Err(e) => {
let _ = send_error(ctx.http, e.to_string()).await;
}
}
None
}

View file

@ -6,7 +6,6 @@ use std::time::Duration;
use std::error;
use std::env;
use serenity::all::VoiceState;
use serenity::async_trait;
use serenity::futures::lock::Mutex;
use serenity::prelude::GatewayIntents;
@ -20,14 +19,12 @@ use reqwest::Client as HttpClient;
use dotenv::dotenv;
use songbird::SerenityInit;
use tokio_cron_scheduler::{JobScheduler, Job};
use radiobrowser::RadioBrowserAPI;
mod handlers;
use handlers::message_handler::handle;
use handlers::voice_state_handler::handle_voice_update;
mod message_handler;
use message_handler::handle;
mod commands;
mod utils;
@ -38,7 +35,6 @@ use other::notice;
mod types;
use types::{Data, Error};
struct Handler;
async fn on_error(error: poise::FrameworkError<'_, Data, Error>) {
@ -57,15 +53,6 @@ async fn on_error(error: poise::FrameworkError<'_, Data, Error>) {
#[async_trait]
impl EventHandler for Handler {
async fn voice_state_update(&self, ctx: Context, old: Option<VoiceState>, new: VoiceState) {
// User was not connected to any channel and connected to VC
let Some(old_channel) = old else {
return
};
handle_voice_update(ctx, old_channel, new).await;
}
async fn message(&self, ctx: Context, msg: Message) {
handle(ctx, msg).await;
}
@ -126,11 +113,11 @@ async fn main() -> anyhow::Result<()> {
..Default::default()
};
let framework = poise::Framework::builder()
.setup(move |ctx, _ready, framework| {
Box::pin(async move {
#[cfg(feature="GUILD_COMMAND")] {
use poise::samples::register_in_guild;
use serenity::all::GuildId;
@ -138,7 +125,7 @@ async fn main() -> anyhow::Result<()> {
let debug_guild_id = env::var("DEBUG_GUILD_ID")
.context("DEBUG_GUILD_ID not found in env")?
.parse::<u64>().unwrap();
register_in_guild(ctx, &framework.options().commands, GuildId::new(debug_guild_id)).await?;
}
@ -147,7 +134,7 @@ async fn main() -> anyhow::Result<()> {
register_globally(ctx, &framework.options().commands).await?;
}
Ok(Data {
http_client: HttpClient::new(),
radio_browser: RadioBrowserAPI::new().await?,

View file

@ -1,6 +1,6 @@
use std::{fs, hash::{DefaultHasher, Hash, Hasher}, io, path::Path, sync::Arc, vec};
use serenity::{all::{ChannelId, ChannelType, Context, CreateMessage, GuildChannel, GuildId, GuildRef, Message, VoiceState}, http::Http};
use serenity::{all::{ChannelId, ChannelType, CreateMessage, GuildId, GuildRef, Message}, http::Http};
pub async fn get_system_channel(guild_id: GuildId, http: &Http) -> anyhow::Result<ChannelId> {
@ -15,16 +15,6 @@ pub async fn replace_msg(http: Arc<Http>, msg: Message, content: String) -> Resu
return ChannelId::new(msg.channel_id.get()).send_message(http.clone(), CreateMessage::new().content(content)).await;
}
pub async fn voice_state_to_guild_channel(ctx: &Context, voice_state: &VoiceState) -> Option<GuildChannel> {
let Ok(channel) = voice_state.channel_id?
.to_channel(&ctx.http)
.await else {
return None
};
Some(channel.guild()?)
}
pub fn get_vc_names(guild: GuildRef) -> Vec<String> {
let mut result: Vec<String> = [].to_vec();