Include subsearch string when handling messages with uploads

Author: eip618

cogs/JollinessEnforcement.py

@@ -0,0 +1,95 @@
+from __future__ import annotations
+
+import discord
+from discord.ext import commands
+
+
+class JollinessEnforcement(commands.Cog):
+    """Enforces appropriate jolliness."""
+
+    TARGET_CHANNEL_IDS: set[int] = {
+        485138525885431808,
+        1292689984372867173,
+        314856589716750346,
+    }
+
+    ALLOWED_STRINGS: tuple[str, ...] = (
+        "lol",
+        "rofl",
+        "lmao",
+        "kek",
+    )
+
+    NOTICE_TEXT = (
+        "All messages must contain appropriate "
+        "jolliness or they will be refused."
+    )
+
+    def __init__(self, bot: commands.Bot) -> None:
+        self.bot = bot
+        self.enabled: bool = False
+
+    @commands.hybrid_command(name="jolliness", description="Enable or disable jolliness enforcement.")
+    @commands.has_permissions(manage_messages=True)
+    async def jolliness(self, ctx: commands.Context, state: str) -> None:
+        """Enable or disable the enforcement."""
+        state = state.lower()
+
+        if state in ("on", "enable", "enabled", "true"):
+            self.enabled = True
+            await ctx.reply("Jolliness enforcement enabled.")
+        elif state in ("off", "disable", "disabled", "false"):
+            self.enabled = False
+            await ctx.reply("Jolliness enforcement disabled.")
+        else:
+            await ctx.reply("Use 'on' or 'off'.")
+
+    @jolliness.error
+    async def jolliness_error(self, ctx: commands.Context, error: commands.CommandError) -> None:
+        if isinstance(error, commands.MissingPermissions):
+            await ctx.reply("You do not have permission to use this command.")
+        else:
+            raise error
+
+    @commands.Cog.listener()
+    async def on_message(self, message: discord.Message) -> None:
+        if not self.enabled:
+            return
+
+        if message.guild is None:
+            return
+
+        if self.bot.user is not None and message.author.id == self.bot.user.id:
+            return
+
+        channel = message.channel
+        if isinstance(channel, discord.Thread):
+            parent_id = channel.parent_id
+        else:
+            parent_id = channel.id
+
+        if parent_id not in self.TARGET_CHANNEL_IDS:
+            return
+
+        content = message.content.lower()
+
+        if any(term in content for term in self.ALLOWED_STRINGS):
+            return
+
+        try:
+            await message.delete()
+        except (discord.Forbidden, discord.HTTPException):
+            return
+
+        try:
+            await message.channel.send(
+                f"{message.author.mention} {self.NOTICE_TEXT}",
+                delete_after=3,
+                allowed_mentions=discord.AllowedMentions(users=True, roles=False, everyone=False),
+            )
+        except (discord.Forbidden, discord.HTTPException):
+            pass
+
+
+async def setup(bot: commands.Bot) -> None:
+    await bot.add_cog(JollinessEnforcement(bot))

kurisu.py

@@ -66,7 +66,8 @@
     'cogs.server_logs',
     'cogs.soap',
     'cogs.titletxtparse',
-    'cogs.slursearch'
+    'cogs.slursearch',
+    'cogs.JollinessEnforcement'
 )
 
 DEBUG = False