Add MCDonalds Cog

cogs/mcdonalds.py

@@ -0,0 +1,63 @@
+import asyncio
+import typing
+
+import discord
+from discord.ext import commands
+
+
+class McDonaldsCog(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+        self.targets: dict[discord.Member, float] = {}
+        self.lock = asyncio.Lock()
+        self.cooldown: dict[discord.Member, float] = {}
+
+    @commands.Cog.listener()
+    async def on_message(self, message: discord.Message):
+        if not message.channel.permissions_for(message.guild.me or self.bot.user).manage_messages:
+            return
+
+        async with self.lock:
+            if message.author in self.targets:
+                if message.content.upper() != "MCDONALDS!":
+                    await message.delete()
+                    if (message.created_at.timestamp() - self.targets[message.author]) > 10:
+                        await message.channel.send(
+                            f"{message.author.mention} Please say `MCDONALDS!` to end commercial.",
+                            delete_after=30
+                        )
+                        self.targets[message.author] = message.created_at.timestamp()
+                else:
+                    await message.reply(
+                        "Thank you. You may now resume your activity.",
+                        delete_after=60
+                    )
+
+                    self.cooldown[message.author] = message.created_at.timestamp()
+
+    @commands.user_command(name="Commercial Break")
+    @commands.cooldown(1, 30, commands.BucketType.user)
+    async def commercial_break(self, ctx: discord.ApplicationContext, member: discord.Member):
+        await ctx.defer()
+        if member in self.targets:
+            await ctx.respond(f"{member.mention} is already in a commercial break.")
+            return
+        elif member in self.cooldown and self.cooldown[member] + 300 > discord.utils.utcnow().timestamp():
+            await ctx.respond(
+                f"{member.mention} is not due another ad break yet. Their next commercial break will start "
+                f"<t:{int(self.cooldown[member] + 300)}:R> at the earliest."
+            )
+            return
+
+        self.targets[member] = discord.utils.utcnow().timestamp()
+        await ctx.send(
+            f"{member.mention} Commercial break! Please say `MCDONALDS!` to end commercial.\n"
+            f"*This commercial break is sponsored by {ctx.user.mention}.*",
+            delete_after=300
+        )
+        await ctx.respond("Commercial break started.", ephemeral=True)
+        await ctx.delete(delay=120)
+
+
+def setup(bot):
+    bot.add_cog(McDonaldsCog(bot))