summaryrefslogtreecommitdiff
path: root/src/handlers
diff options
context:
space:
mode:
authorseth <[email protected]>2024-08-09 23:35:41 -0400
committerGitHub <[email protected]>2024-08-09 23:35:41 -0400
commitb643a6a235b0c1c9902b97421f24eff2b0d0a5ac (patch)
tree350794c0e9330fb77367838313bc6bb97278a0aa /src/handlers
parent372780546b508684839916e5ad54c9e90456a94f (diff)
tree-wide: end of summer cleanup (#214)
* api: refactor & rename module to http * client: split from main.rs * tree-wide: use eyre::Report as error * nix: alejandra -> nixfmt * nix: start using treefmt-nix * nix: simplify flake * nix: refactor derivation & docker image * nix: remove overlay * ci: update & cleanup workflows * commands: assign all commands automatically * commands/copypasta: remove * http/teawie: update response struct for upstream rust rewrite * handlers: rename modules to events; flatten * crates: rename self to teawie-bot * nix: fenix -> rust-overlay i want a specific rust version grrrrrrr * ci: pin rust to 1.79 this is what our nix dev shell uses and what we can compile on. it seems the time crate doesn't like v1.80 of the compiler :( * ci: always run release gates * nix: fix static toolchain * nix: rust-overlay -> nixpkgs * ci: adopt actions-rust-lang actions * nix: use docker arch names for containers * crates/time: 0.3.30 -> 0.3.36 fixes building on rust 1.80.0
Diffstat (limited to 'src/handlers')
-rw-r--r--src/handlers/error.rs51
-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
-rw-r--r--src/handlers/mod.rs2
7 files changed, 0 insertions, 409 deletions
diff --git a/src/handlers/error.rs b/src/handlers/error.rs
deleted file mode 100644
index e706fec..0000000
--- a/src/handlers/error.rs
+++ /dev/null
@@ -1,51 +0,0 @@
-use crate::{consts::Colors, Data, Error};
-
-use log::error;
-use poise::serenity_prelude::{CreateEmbed, Timestamp};
-use poise::{CreateReply, FrameworkError};
-
-pub async fn handle(error: poise::FrameworkError<'_, Data, Error>) {
- match error {
- FrameworkError::Setup {
- error, framework, ..
- } => {
- error!("Error setting up client! Bailing out");
- framework.shard_manager().shutdown_all().await;
-
- panic!("{error}")
- }
-
- FrameworkError::Command { error, ctx, .. } => {
- error!("Error in command {}:\n{error}", ctx.command().name);
-
- let embed = CreateEmbed::default()
- .title("Something went wrong!")
- .description("oopsie")
- .timestamp(Timestamp::now())
- .color(Colors::Orange);
-
- let reply = CreateReply::default().embed(embed);
-
- ctx.send(reply).await.ok();
- }
-
- FrameworkError::EventHandler {
- error,
- ctx: _,
- event,
- framework: _,
- ..
- } => {
- error!(
- "Error while handling event {}:\n{error}",
- event.snake_case_name()
- );
- }
-
- error => {
- if let Err(e) = poise::builtins::on_error(error).await {
- error!("Unhandled error occurred:\n{e:#?}");
- }
- }
- }
-}
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(())
-}
diff --git a/src/handlers/mod.rs b/src/handlers/mod.rs
deleted file mode 100644
index 1610d23..0000000
--- a/src/handlers/mod.rs
+++ /dev/null
@@ -1,2 +0,0 @@
-pub mod error;
-pub mod event;