reaction_roles.rs

use std::collections::HashMap;

use serenity::all::{Context as SerenityContext, MessageId, Reaction, ReactionType, RoleId};
use tracing::{debug, error};

use crate::{
    ids::{
        AI_ROLE_ID, ARCHIVE_ROLE_ID, DEVOPS_ROLE_ID, MOBILE_ROLE_ID, RESEARCH_ROLE_ID,
        ROLES_MESSAGE_ID, SYSTEMS_ROLE_ID, WEB_ROLE_ID,
    },
    Data,
};

pub fn populate_data_with_reaction_roles(data: &mut Data) {
    let roles = [
        (
            ReactionType::Unicode("📁".to_string()),
            RoleId::new(ARCHIVE_ROLE_ID),
        ),
        (
            ReactionType::Unicode("📱".to_string()),
            RoleId::new(MOBILE_ROLE_ID),
        ),
        (
            ReactionType::Unicode("⚙️".to_string()),
            RoleId::new(SYSTEMS_ROLE_ID),
        ),
        (
            ReactionType::Unicode("🤖".to_string()),
            RoleId::new(AI_ROLE_ID),
        ),
        (
            ReactionType::Unicode("📜".to_string()),
            RoleId::new(RESEARCH_ROLE_ID),
        ),
        (
            ReactionType::Unicode("🚀".to_string()),
            RoleId::new(DEVOPS_ROLE_ID),
        ),
        (
            ReactionType::Unicode("🌐".to_string()),
            RoleId::new(WEB_ROLE_ID),
        ),
    ];

    data.reaction_roles
        .extend::<HashMap<ReactionType, RoleId>>(roles.into());
}

pub async fn handle_reaction(
    ctx: &SerenityContext,
    reaction: &Reaction,
    data: &Data,
    is_add: bool,
) {
    if !is_relevant_reaction(reaction.message_id, &reaction.emoji, data) {
        return;
    }

    debug!("Handling {:?} from {:?}.", reaction.emoji, reaction.user_id);

    // TODO Log these errors
    let Some(guild_id) = reaction.guild_id else {
        return;
    };
    let Some(user_id) = reaction.user_id else {
        return;
    };
    let Ok(member) = guild_id.member(ctx, user_id).await else {
        return;
    };
    let Some(role_id) = data.reaction_roles.get(&reaction.emoji) else {
        return;
    };

    let result = if is_add {
        member.add_role(&ctx.http, *role_id).await
    } else {
        member.remove_role(&ctx.http, *role_id).await
    };

    if let Err(e) = result {
        error!(
            "Could not handle {:?} from {:?}. Error: {}",
            reaction.emoji, reaction.user_id, e
        );
    }
}

fn is_relevant_reaction(message_id: MessageId, emoji: &ReactionType, data: &Data) -> bool {
    message_id == MessageId::new(ROLES_MESSAGE_ID) && data.reaction_roles.contains_key(emoji)
}