Merge pull request #308 from RemainingDelta/220-Feature

220-Feature add counting game feature with sequential validation and set-count command

Author: RemainingDelta

database/mongo.py

@@ -179,6 +179,36 @@ async def set_setting(key: str, value: str):
     await db.settings.update_one({"_id": key}, {"$set": {"value": value}}, upsert=True)
 
 
+# --- COUNTING HELPERS ---
+
+
+async def get_counting_state() -> dict:
+    if db is None:
+        return {"current_count": 0, "last_user_id": None}
+    doc = await db.counting.find_one({"_id": "state"})
+    if not doc:
+        return {"current_count": 0, "last_user_id": None}
+    return {
+        "current_count": doc.get("current_count", 0),
+        "last_user_id": doc.get("last_user_id"),
+    }
+
+
+async def update_counting_state(current_count: int, last_user_id: int | None):
+    if db is None:
+        return
+    await db.counting.update_one(
+        {"_id": "state"},
+        {
+            "$set": {
+                "current_count": current_count,
+                "last_user_id": last_user_id,
+            }
+        },
+        upsert=True,
+    )
+
+
 # --- LEADERBOARD HELPERS ---
 
 

features/config.py

@@ -61,6 +61,7 @@
 
     ECONOMY_COMMANDS_CHANNEL_ID = 1352841132735860746
     BOT_COMMANDS_CHANNEL_ID = 546405233736155151
+    COUNTING_CHANNEL_ID = 909853678863790092
 
     EMOJIS_CURRENCY = {
         "coins": "<:bs_coin:1454305320015888516>",
@@ -239,6 +240,7 @@
 
     ECONOMY_COMMANDS_CHANNEL_ID = 1484768262649675776
     BOT_COMMANDS_CHANNEL_ID = 1484768347756167228
+    COUNTING_CHANNEL_ID = 1513450829707874354
 
     EMOJIS_CURRENCY = {
         "coins": "<:bs_coin:1454290949780934678>",

features/counting.py

@@ -0,0 +1,89 @@
+import discord
+from discord import app_commands
+from discord.ext import commands
+
+from database.mongo import get_counting_state, update_counting_state
+from features.config import (
+    ADMIN_ROLE_ID,
+    COUNTING_CHANNEL_ID,
+    FOUNDER_ROLE_ID,
+    MODERATOR_ROLE_ID,
+)
+
+COUNTING_MOD_ROLES = {MODERATOR_ROLE_ID, ADMIN_ROLE_ID, FOUNDER_ROLE_ID}
+
+
+class Counting(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @commands.Cog.listener()
+    async def on_message(self, message: discord.Message):
+        if message.author.bot:
+            return
+        if not message.guild:
+            return
+        if message.channel.id != COUNTING_CHANNEL_ID:
+            return
+
+        state = await get_counting_state()
+        current_count = state["current_count"]
+        last_user_id = state["last_user_id"]
+        expected = current_count + 1
+
+        if message.author.id == last_user_id:
+            await message.delete()
+            await message.channel.send(
+                f"{message.author.mention} You can't count twice in a row!",
+                delete_after=5,
+            )
+            return
+
+        try:
+            number = int(message.content.strip())
+        except ValueError:
+            await message.delete()
+            await message.channel.send(
+                f"{message.author.mention} That's not a number! Next number is **{expected}**.",
+                delete_after=5,
+            )
+            return
+
+        if number != expected:
+            await message.delete()
+            await message.channel.send(
+                f"{message.author.mention} Wrong number! Next number is **{expected}**.",
+                delete_after=5,
+            )
+            return
+
+        await update_counting_state(number, message.author.id)
+
+    @app_commands.command(
+        name="set-count", description="Set the current count (staff only)."
+    )
+    @app_commands.describe(count="The number to set the count to")
+    async def set_count(self, interaction: discord.Interaction, count: int):
+        if not isinstance(interaction.user, discord.Member) or not any(
+            role.id in COUNTING_MOD_ROLES for role in interaction.user.roles
+        ):
+            await interaction.response.send_message(
+                "❌ You don't have permission to use this command.", ephemeral=True
+            )
+            return
+
+        if interaction.channel_id != COUNTING_CHANNEL_ID:
+            await interaction.response.send_message(
+                f"❌ This command can only be used in <#{COUNTING_CHANNEL_ID}>.",
+                ephemeral=True,
+            )
+            return
+
+        await update_counting_state(count, None)
+        await interaction.response.send_message(
+            f"✅ Count set to **{count}**.", ephemeral=True
+        )
+
+
+async def setup(bot):
+    await bot.add_cog(Counting(bot))

main.py

@@ -73,6 +73,9 @@ async def on_ready():
         await bot.load_extension("features.sticky")
         print("✅ Loaded Feature: Sticky Messages")
 
+        await bot.load_extension("features.counting")
+        print("✅ Loaded Feature: Counting")
+
     except Exception as e:
         print(f"❌ Error loading features: {e}")