diff options
Diffstat (limited to 'src/handler.rs')
| -rw-r--r-- | src/handler.rs | 139 |
1 files changed, 137 insertions, 2 deletions
diff --git a/src/handler.rs b/src/handler.rs index a52b31a..b83d789 100644 --- a/src/handler.rs +++ b/src/handler.rs @@ -1,4 +1,6 @@ -use poise::serenity_prelude as serenity; +use poise::serenity_prelude::{self as serenity, Interaction, MessageComponentInteraction, MessageId, ChannelId}; +use regex::Regex; +use serde_json::json; use crate::models::*; use crate::{Data, Error}; @@ -78,17 +80,150 @@ pub async fn event_handler( if new_message.author.id.0 != current_user.id.0 { // This message does *not* belong to the bot - println!("Message from not-bot"); let filters = sqlx::query_as::<_, MessageFilter>( "SELECT * FROM message_filter WHERE guild_id=$1") .bind(new_message.guild_id.unwrap().0.to_string()) .fetch_all(&pool) .await?; + + let content = new_message.content.clone(); + + for f in filters { + let r = Regex::new(&f.pattern)?; + + if r.is_match(&content) { + match f.action { + FilterAction::Review => { + if let Ok(c) = sqlx::query_as!(ConfigChannel, "SELECT * FROM channels WHERE purpose='review' AND guild_id=$1", new_message.guild_id.unwrap().0.to_string()) + .fetch_one(&pool) + .await { + let chan = serenity::ChannelId(c.channel_id.parse::<u64>()?); + + chan.send_message(&ctx, |m| { + m.content(format!("Message was flagged for review (using rule {})", f.id)); + m.components(|c| { + c.create_action_row(|r| { + r.create_button(|b| { + b.style(serenity::ButtonStyle::Primary); + b.label("Approve"); + b.custom_id("approve"); + b + }); + r.create_button(|b| { + b.style(serenity::ButtonStyle::Danger); + b.label("Deny"); + b.custom_id("deny"); + b + }); + r + }); + c + }); + m.embed(|e| { + e.author(|a| { + a.name(new_message.clone().author.name); + a.icon_url(new_message.clone().author.face()) + }); + e.description(content.clone()); + e.footer(|f| { + f.text(format!("{}:{}", new_message.channel_id.0, new_message.id.0.to_string())) + }); + e.field("Link", new_message.link().clone(), true) + }) + }).await?; + } else { + let guild = new_message.guild(&ctx).unwrap(); + + let owner = guild.owner_id; + let dm_chan = owner.create_dm_channel(&ctx.http).await?; + dm_chan.say(&ctx.http, format!("Hi! I tried to flag a message for review in {}, but you haven't set a review channel yet! Set one now by running `filter channel` in the server.", guild.name.clone())).await?; + } + }, + FilterAction::Delete => { + new_message.delete(&ctx.http).await?; + } + } + } + } } }, + poise::Event::InteractionCreate { interaction } => { + match interaction { + Interaction::MessageComponent(mci) => { + let data = &mci.data; + + let mci2 = mci.clone(); + + match data.custom_id.as_str() { + "approve" => { + + let msg_report = mci2.message.clone(); + let footer = msg_report.embeds[0].clone().footer.unwrap().text; + let mut footer = footer.split(':'); + let orig_channel_id = footer.next().unwrap(); + let orig_msg_id = footer.next().unwrap(); + let orig_msg_id = MessageId(orig_msg_id.parse::<u64>()?); + let orig_channel_id = ChannelId(orig_channel_id.parse::<u64>()?); + + let orig_msg = ctx.http.get_message(orig_channel_id.0, orig_msg_id.0).await?; + + + disable_buttons(&ctx, mci2).await?; + }, + "deny" => { + let msg_report = mci2.message.clone(); + let footer = msg_report.embeds[0].clone().footer.unwrap().text; + let mut footer = footer.split(':'); + let orig_channel_id = footer.next().unwrap(); + let orig_msg_id = footer.next().unwrap(); + let orig_msg_id = MessageId(orig_msg_id.parse::<u64>()?); + let orig_channel_id = ChannelId(orig_channel_id.parse::<u64>()?); + + let orig_msg = ctx.http.get_message(orig_channel_id.0, orig_msg_id.0).await?; + + orig_msg.delete(&ctx).await?; + + disable_buttons(&ctx, mci2).await?; + }, + _ => (), + }; + }, + _ => () + }; + }, _ => (), } } Ok(()) } + +async fn disable_buttons(ctx: &serenity::Context, mut mci: MessageComponentInteraction) -> Result<(), Error> { + mci.message.edit(&ctx, |m| { + m.components(|c| { + c.create_action_row(|r| { + r.create_button(|b| { + b.style(serenity::ButtonStyle::Primary); + b.label("Approve"); + b.custom_id("approve"); + b.disabled(true); + b + }); + r.create_button(|b| { + b.style(serenity::ButtonStyle::Danger); + b.label("Deny"); + b.custom_id("deny"); + b.disabled(true); + b + }); + r + }); + c + }); + m + }).await?; + + ctx.http.create_interaction_response(mci.id.0, &mci.token, &json!({ "type": 7})).await?; + + Ok(()) +} |