Add simple moderation honeypot channel feature to the 'pgc' extension

Author: Mega-JC

pcbot/exts/pgc_activity.py pcbot/exts/pgc.py

@@ -92,8 +92,15 @@
 
 
 class PGCActivityCog(BaseExtensionCog, name="pgc-activity"):
-    def __init__(self, bot: BotT, theme_color: int | discord.Color = 0) -> None:
+    def __init__(
+        self,
+        bot: BotT,
+        theme_color: int | discord.Color = 0,
+        honeypot_channel_id: int | None = None,
+    ) -> None:
         super().__init__(bot, theme_color)
+        self.honeypot_channel_id = honeypot_channel_id
+        self.honeypot_victims = set()
 
     @commands.Cog.listener()
     async def on_ready(self):
@@ -120,6 +127,7 @@ async def on_message(self, message: discord.Message):
         if not (
             message.guild
             and message.guild.id == PGC_GUILD_ID
+            and isinstance(message.author, discord.Member)
             and (
                 message.type
                 in (
@@ -136,6 +144,17 @@ async def on_message(self, message: discord.Message):
         ):
             return
 
+        if message.channel.id == self.honeypot_channel_id:
+            await message.author.ban(
+                reason="Fell into snake pit", delete_message_days=1
+            )
+
+            if message.author.id not in self.honeypot_victims:
+                self.honeypot_victims.add(message.author.id)
+                await message.author.unban(reason="Second chance")
+            else:
+                self.honeypot_victims.remove(message.author.id)
+
         if message.type == discord.MessageType.premium_guild_tier_1:
             await message.channel.send(
                 "LETS GO 🎉 ! Thanks for boosting us to **LEVEL 1** "
@@ -191,5 +210,5 @@ async def on_member_join(self, member: discord.Member):
 
 
 @snakecore.commands.decorators.with_config_kwargs
-async def setup(bot: BotT):
-    await bot.add_cog(PGCActivityCog(bot))
+async def setup(bot: BotT, honeypot_channel_id: int | None = None) -> None:
+    await bot.add_cog(PGCActivityCog(bot, honeypot_channel_id=honeypot_channel_id))