2b-rs/src/handler.rs

use serenity::async_trait;
use serenity::client::Context;
use serenity::model::channel::Reaction;
use serenity::model::event::ResumedEvent;
use serenity::model::gateway::{Activity, Ready};
use serenity::model::guild::Member;
use serenity::model::id::{ChannelId, GuildId, MessageId};
use serenity::model::voice::VoiceState;
use serenity::prelude::*;

use crate::commands::music::get_music_player_for_guild;
use crate::utils::delete_messages_from_database;
use bot_serenityutils::menu::{
    handle_message_delete, handle_message_delete_bulk, handle_reaction_add, handle_reaction_remove,
    start_update_loop,
};

pub(crate) struct Handler;

#[async_trait]
impl EventHandler for Handler {
    async fn cache_ready(&self, ctx: Context, _: Vec<GuildId>) {
        log::info!("Cache Ready");
        start_update_loop(&ctx).await;
        if let Err(e) = delete_messages_from_database(&ctx).await {
            log::error!("Failed to delete expired messages {:?}", e);
        }
    }

    /// Fired when a message was deleted
    async fn message_delete(
        &self,
        ctx: Context,
        channel_id: ChannelId,
        message_id: MessageId,
        _: Option<GuildId>,
    ) {
        tokio::spawn(async move {
            log::trace!("Handling message delete event");
            if let Err(e) = handle_message_delete(&ctx, channel_id, message_id).await {
                log::error!("Failed to handle event: {:?}", e);
            }
            log::trace!("Message delete event handled");
        });
    }

    /// Fired when multiple messages were deleted
    async fn message_delete_bulk(
        &self,
        ctx: Context,
        channel_id: ChannelId,
        message_ids: Vec<MessageId>,
        _: Option<GuildId>,
    ) {
        tokio::spawn(async move {
            log::trace!("Handling message delete bulk event");
            if let Err(e) = handle_message_delete_bulk(&ctx, channel_id, &message_ids).await {
                log::error!("Failed to handle event: {:?}", e);
            }
            log::debug!("Message delte bulk event handled");
        });
    }

    /// Fired when a reaction was added to a message
    async fn reaction_add(&self, ctx: Context, reaction: Reaction) {
        tokio::spawn(async move {
            log::trace!("Handling reaction add event...");
            if let Err(e) = handle_reaction_add(&ctx, &reaction).await {
                log::error!("Failed to handle event: {:?}", e);
            }
            log::trace!("Reaction add event handled");
        });
    }

    /// Fired when a reaction was added to a message
    async fn reaction_remove(&self, ctx: Context, reaction: Reaction) {
        tokio::spawn(async move {
            log::trace!("Handling reaction remove event");
            if let Err(e) = handle_reaction_remove(&ctx, &reaction).await {
                log::error!("Failed to handle event: {:?}", e);
            }
            log::trace!("Reaction remove event handled");
        });
    }

    async fn ready(&self, ctx: Context, ready: Ready) {
        log::info!("Connected as {}", ready.user.name);
        let prefix = dotenv::var("BOT_PREFIX").unwrap_or("~!".to_string());
        ctx.set_activity(Activity::listening(format!("{}help", prefix).as_str()))
            .await;
    }

    async fn resume(&self, _: Context, _: ResumedEvent) {
        log::info!("Reconnected to gateway")
    }

    async fn voice_state_update(
        &self,
        ctx: Context,
        guild_id: Option<GuildId>,
        old_state: Option<VoiceState>,
        new_state: VoiceState,
    ) {
        let mut member_count = None;

        let guild_id = if let Some(gid) = guild_id {
            gid
        } else {
            return;
        };

        if let Some(old_id) = old_state.and_then(|c| c.channel_id) {
            member_count = get_own_channel_member_count(&ctx, &old_id).await;
        }
        if member_count.is_none() {
            if let Some(new_id) = new_state.channel_id {
                member_count = get_own_channel_member_count(&ctx, &new_id).await;
            }
        }

        if let Some(count) = member_count {
            log::debug!("{} Members in channel", count);
            if let Some(player) = get_music_player_for_guild(&ctx, guild_id).await {
                let mut player = player.lock().await;
                log::debug!("Setting leave flag to {}", count == 0);
                player.set_leave_flag(count == 0);
            }
        }
    }
}

/// Returns the number of members in the channel if it's the bots voice channel
async fn get_own_channel_member_count(ctx: &Context, channel_id: &ChannelId) -> Option<usize> {
    let channel = ctx.http.get_channel(channel_id.0).await.ok()?;
    let guild_channel = channel.guild()?;
    let current_user = ctx.http.get_current_user().await.ok()?;

    let members = guild_channel.members(&ctx).await.ok()?;
    let own_channel = members
        .iter()
        .find(|m| m.user.id == current_user.id)
        .is_some();

    if !own_channel {
        return None;
    }
    let members: Vec<Member> = members.into_iter().filter(|m| !m.user.bot).collect();

    Some(members.len())
}