From 67e4081aa69f489ad88ee064402d68a5969069c2 Mon Sep 17 00:00:00 2001 From: Cara Salter Date: Wed, 12 Jan 2022 15:45:20 -0500 Subject: rroles: Implement Reaction Roles now requires postgresql. --- src/handler.rs | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 src/handler.rs (limited to 'src/handler.rs') diff --git a/src/handler.rs b/src/handler.rs new file mode 100644 index 0000000..933bd72 --- /dev/null +++ b/src/handler.rs @@ -0,0 +1,36 @@ +use poise::serenity_prelude as serenity; + +use crate::{Data, Error}; +use crate::models::ReactionRole; + +pub async fn event_handler(ctx: &serenity::Context, + event: &poise::Event<'_>, + data: &Data, + ) -> Result<(), Error> { + { + let pool = data.pg.lock().unwrap().clone(); + match event { + poise::Event::ReactionAdd { add_reaction } => { + let rrole = sqlx::query_as!(ReactionRole, "SELECT * FROM reaction_roles WHERE message_id=$1 AND reaction=$2", add_reaction.message_id.0.to_string(), add_reaction.emoji.to_string()).fetch_one(&pool).await?; + let member = ctx.http.get_member(rrole.guild_id.parse::()?, add_reaction.user_id.unwrap().0).await?; + let member_roles = member.roles; + let role_id = serenity::RoleId(rrole.role_id.parse::()?); + if member_roles.contains(&role_id) { + ctx.http.remove_member_role(member.guild_id.0, member.user.id.0, role_id.0, Some("Reaction Role Menu")).await?; + } else { + ctx.http.add_member_role(rrole.guild_id.parse::()?, add_reaction.user_id.unwrap().0, rrole.role_id.parse::()?, Some("Reaction Role")).await?; + } + + let dm_chan = add_reaction.user_id.unwrap().create_dm_channel(&ctx.http).await?; + dm_chan.say(ctx, format!("Toggled the role!")).await?; + + add_reaction.delete(&ctx.http).await?; + + }, + _ => (), + } + } + + + Ok(()) +} -- cgit v1.2.3