summaryrefslogtreecommitdiff
path: root/src/handlers/event
diff options
context:
space:
mode:
Diffstat (limited to 'src/handlers/event')
-rw-r--r--src/handlers/event/guild.rs37
-rw-r--r--src/handlers/event/message.rs44
-rw-r--r--src/handlers/event/mod.rs51
-rw-r--r--src/handlers/event/pinboard.rs81
-rw-r--r--src/handlers/event/reactboard.rs143
5 files changed, 0 insertions, 356 deletions
diff --git a/src/handlers/event/guild.rs b/src/handlers/event/guild.rs
deleted file mode 100644
index 774179c..0000000
--- a/src/handlers/event/guild.rs
+++ /dev/null
@@ -1,37 +0,0 @@
-use eyre::Result;
-use log::{debug, warn};
-use poise::serenity_prelude::{Guild, UnavailableGuild};
-
-use crate::{storage, Data};
-use storage::settings::Settings;
-
-pub async fn handle_create(guild: &Guild, data: &Data) -> Result<()> {
- if let Some(storage) = &data.storage {
- if storage.guild_settings_exist(&guild.id).await? {
- debug!("Not recreating settings key for {}", guild.id);
- return Ok(());
- }
-
- let settings = Settings {
- guild_id: guild.id,
- ..Default::default()
- };
-
- warn!("Creating new settings key for {}:\n{settings:#?}", guild.id);
- storage.create_guild_settings(settings).await?;
- } else {
- warn!("Can't create guild settings; no storage backend found!");
- }
-
- Ok(())
-}
-
-pub async fn handle_delete(guild: &UnavailableGuild, data: &Data) -> Result<()> {
- if let Some(storage) = &data.storage {
- storage.delete_guild_settings(&guild.id).await?;
- } else {
- warn!("Can't delete guild settings; no storage backend found!");
- }
-
- Ok(())
-}
diff --git a/src/handlers/event/message.rs b/src/handlers/event/message.rs
deleted file mode 100644
index 67dbb21..0000000
--- a/src/handlers/event/message.rs
+++ /dev/null
@@ -1,44 +0,0 @@
-use crate::{consts, Data};
-
-use eyre::{eyre, Result};
-use log::{debug, warn};
-use poise::serenity_prelude::{Context, Message};
-
-pub async fn handle(ctx: &Context, msg: &Message, data: &Data) -> Result<()> {
- if should_echo(ctx, msg, data).await? {
- msg.reply(ctx, &msg.content).await?;
- }
-
- Ok(())
-}
-
-async fn should_echo(ctx: &Context, msg: &Message, data: &Data) -> Result<bool> {
- if (msg.author.bot && msg.webhook_id.is_none()) || msg.is_own(ctx) {
- debug!("Not repeating another bot");
- return Ok(false);
- }
-
- let gid = msg
- .guild_id
- .ok_or_else(|| eyre!("Couldn't get GuildId from {}!", msg.id))?;
-
- if let Some(storage) = &data.storage {
- let settings = storage.get_guild_settings(&gid).await?;
-
- if !settings.optional_commands_enabled {
- debug!("Not echoing in guild {gid}");
- return Ok(false);
- }
- } else {
- warn!("Ignoring restrictions on echoing messages; no storage backend is attached!");
- }
-
- let content = &msg.content;
-
- Ok(content == "🗿"
- || consts::TEAMOJIS.contains(&content.as_str())
- || content.to_ascii_lowercase() == "moyai"
- || content
- .to_ascii_lowercase()
- .contains("twitter's recommendation algorithm"))
-}
diff --git a/src/handlers/event/mod.rs b/src/handlers/event/mod.rs
deleted file mode 100644
index cc7d727..0000000
--- a/src/handlers/event/mod.rs
+++ /dev/null
@@ -1,51 +0,0 @@
-use crate::{consts, Data, Error};
-
-use eyre::Result;
-use log::{debug, info};
-use poise::serenity_prelude::{self as serenity, CreateBotAuthParameters};
-use serenity::FullEvent;
-
-mod guild;
-mod message;
-mod pinboard;
-mod reactboard;
-
-pub async fn handle(ctx: &serenity::Context, event: &FullEvent, data: &Data) -> Result<(), Error> {
- match event {
- FullEvent::Ready { data_about_bot } => {
- info!("Logged in as {}!", data_about_bot.user.name);
-
- if let Ok(invite_link) = CreateBotAuthParameters::new().auto_client_id(ctx).await {
- let link = invite_link
- .scopes(consts::bot_scopes())
- .permissions(*consts::bot_permissions())
- .build();
- info!("Invite me to your server at {link}");
- } else {
- debug!("Not displaying invite_link since we couldn't find our client ID");
- }
- }
-
- FullEvent::Message { new_message } => {
- message::handle(ctx, new_message, data).await?;
- pinboard::handle(ctx, new_message, data).await?;
- }
-
- FullEvent::ReactionAdd { add_reaction } => {
- reactboard::handle(ctx, add_reaction, data).await?;
- }
-
- FullEvent::GuildCreate { guild, is_new: _ } => {
- guild::handle_create(guild, data).await?;
- }
-
- FullEvent::GuildDelete {
- incomplete,
- full: _,
- } => guild::handle_delete(incomplete, data).await?,
-
- _ => {}
- }
-
- Ok(())
-}
diff --git a/src/handlers/event/pinboard.rs b/src/handlers/event/pinboard.rs
deleted file mode 100644
index 5b7d454..0000000
--- a/src/handlers/event/pinboard.rs
+++ /dev/null
@@ -1,81 +0,0 @@
-use crate::{utils, Data};
-
-use eyre::{eyre, Context as _, OptionExt as _, Result};
-use log::{debug, warn};
-use poise::serenity_prelude::{
- ChannelId, Context, CreateAllowedMentions, CreateMessage, Message, MessageType, User,
-};
-
-pub async fn handle(ctx: &Context, message: &Message, data: &Data) -> Result<()> {
- if message.kind != MessageType::PinsAdd {
- return Ok(());
- }
-
- let gid = message.guild_id.unwrap_or_default();
- let Some(storage) = &data.storage else {
- warn!("Can't create PinBoard entry; no storage backend found!");
- return Ok(());
- };
-
- let settings = storage.get_guild_settings(&gid).await?;
-
- if !settings.pinboard_enabled {
- debug!("PinBoard is disabled in {gid}, ignoring");
- return Ok(());
- }
-
- let Some(target) = settings.pinboard_channel else {
- debug!("PinBoard is disabled in {gid}, ignoring");
- return Ok(());
- };
-
- if let Some(sources) = settings.pinboard_watch {
- if !sources.contains(&message.channel_id) {
- debug!(
- "{} not listed in PinBoard settings for {gid}, ignoring",
- message.channel_id
- );
-
- return Ok(());
- }
- }
-
- let reference_id = message
- .clone()
- .message_reference
- .ok_or_eyre("Couldn't get referenced message of pin!")?
- .message_id
- .ok_or_eyre("Couldn't get id of referenced message of pin!")?;
-
- let pins = message
- .channel_id
- .pins(ctx)
- .await
- .wrap_err("Couldn't get a list of pins!?")?;
-
- let pin = pins
- .iter()
- .find(|pin| pin.id == reference_id)
- .ok_or_else(|| eyre!("Couldn't find a pin for message {reference_id}!"))?;
-
- redirect(ctx, pin, &message.author, &target).await?;
- pin.unpin(ctx).await?;
-
- Ok(())
-}
-
-async fn redirect(ctx: &Context, pin: &Message, pinner: &User, target: &ChannelId) -> Result<()> {
- let embed = utils::resolve_message_to_embed(ctx, pin).await;
- let mentions = CreateAllowedMentions::new().empty_roles().empty_users();
- let message = CreateMessage::default()
- .allowed_mentions(mentions)
- .content(format!("📌'd by {pinner} in {}", pin.link()))
- .embed(embed);
-
- target
- .send_message(&ctx.http, message)
- .await
- .wrap_err("Couldn't redirect message")?;
-
- Ok(())
-}
diff --git a/src/handlers/event/reactboard.rs b/src/handlers/event/reactboard.rs
deleted file mode 100644
index 75fc858..0000000
--- a/src/handlers/event/reactboard.rs
+++ /dev/null
@@ -1,143 +0,0 @@
-use crate::{storage, utils, Data};
-use storage::reactboard::ReactBoardEntry;
-
-use eyre::{eyre, Context as _, Result};
-use log::{debug, warn};
-use poise::serenity_prelude::{
- Context, CreateMessage, EditMessage, GuildId, Message, MessageReaction, Reaction,
-};
-
-pub async fn handle(ctx: &Context, reaction: &Reaction, data: &Data) -> Result<()> {
- // TODO @getchoo: don't do anything if this message is old
- let msg = reaction
- .message(&ctx.http)
- .await
- .wrap_err("Couldn't get reaction from message!")?;
-
- let matched = msg
- .clone()
- .reactions
- .into_iter()
- .find(|r| r.reaction_type == reaction.emoji)
- .ok_or_else(|| {
- eyre!(
- "Couldn't find any matching reactions for {} in message {}!",
- reaction.emoji.as_data(),
- msg.id
- )
- })?;
-
- send_to_reactboard(
- ctx,
- &matched,
- &msg,
- &reaction.guild_id.unwrap_or_default(),
- data,
- )
- .await?;
-
- Ok(())
-}
-
-async fn send_to_reactboard(
- ctx: &Context,
- reaction: &MessageReaction,
- msg: &Message,
- guild_id: &GuildId,
- data: &Data,
-) -> Result<()> {
- let Some(storage) = &data.storage else {
- warn!("Can't make ReactBoard entry; no storage backend found!");
- return Ok(());
- };
-
- let settings = storage.get_guild_settings(guild_id).await?;
-
- // make sure everything is in order...
- if !settings.reactboard_enabled {
- debug!("ReactBoard is disabled in {guild_id}, ignoring");
- return Ok(());
- }
-
- let Some(target) = settings.reactboard_channel else {
- debug!("ReactBoard is disabled in {guild_id}, ignoring");
- return Ok(());
- };
-
- if !settings.can_use_reaction(&reaction.reaction_type) {
- debug!("Reaction {} can't be used!", reaction.reaction_type);
- return Ok(());
- }
-
- let count = if msg
- .reaction_users(ctx, reaction.reaction_type.clone(), None, None)
- .await?
- .contains(&msg.author)
- {
- reaction.count - 1
- } else {
- reaction.count
- };
-
- if count < settings.reactboard_requirement.unwrap_or(5) {
- debug!(
- "Ignoring message {} on ReactBoard, not enough reactions",
- msg.id
- );
- return Ok(());
- }
-
- let content = format!("{} **#{}**", reaction.reaction_type, count);
-
- let entry = if storage.reactboard_entry_exists(guild_id, &msg.id).await? {
- // bump reaction count if previous entry exists
- let mut entry = storage.get_reactboard_entry(guild_id, &msg.id).await?;
-
- // bail if we don't need to edit anything
- if entry.reaction_count >= count {
- debug!("Message {} doesn't need updating", msg.id);
- return Ok(());
- }
-
- debug!(
- "Bumping {} reaction count from {} to {}",
- msg.id, entry.reaction_count, count
- );
-
- let edited = EditMessage::new().content(content);
-
- ctx.http
- .get_message(entry.posted_channel_id, entry.posted_message_id)
- .await
- .wrap_err_with(|| {
- format!(
- "Couldn't get previous message from ReactBoardEntry {} in Redis DB!",
- entry.original_message_id
- )
- })?
- .edit(ctx, edited)
- .await?;
-
- // update reaction count in redis
- entry.reaction_count = count;
- entry
- } else {
- // make new message and add entry to redis otherwise
- let embed = utils::resolve_message_to_embed(ctx, msg).await;
- let message = CreateMessage::default().content(content).embed(embed);
-
- let resp = target.send_message(ctx, message).await?;
-
- ReactBoardEntry {
- original_message_id: msg.id,
- reaction_count: count,
- posted_channel_id: resp.channel_id,
- posted_message_id: resp.id,
- }
- };
-
- debug!("Creating new ReactBoard entry:\n{entry:#?}");
- storage.create_reactboard_entry(guild_id, entry).await?;
-
- Ok(())
-}