Added mercy counter

cogs/mercy.py

@@ -0,0 +1,118 @@
+import discord
+from discord.ext import commands
+from utils.DatabaseManager_class import DatabaseManager
+from config import AUTHORIZED_CHANNEL_ID
+from datetime import datetime
+
+# Règles de mercy
+MERCY_RULES = {
+    "ancient": {"threshold": 200, "increment": 0.5, "base": 0},
+    "void": {"threshold": 200, "increment": 0.5, "base": 0},
+    "sacred": {"threshold": 12, "increment": 2, "base": 0},
+    "primal_legendary": {"threshold": 75, "increment": 1, "base": 0},
+    "primal_mythical": {"threshold": 200, "increment": 10, "base": 0},
+    "remnant": {"threshold": 24, "increment": 1, "base": 0},
+}
+
+db_manager = DatabaseManager()
+
+class Mercy(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+        # Création de la table si elle n’existe pas
+        db_manager.execute("""
+            CREATE TABLE IF NOT EXISTS mercy_counters (
+                user_id TEXT,
+                shard_type TEXT,
+                pulls INTEGER DEFAULT 0,
+                last_reset TIMESTAMP,
+                PRIMARY KEY(user_id, shard_type)
+            )
+        """)
+
+    def get_mercy_chance(self, shard_type, pulls):
+        rule = MERCY_RULES[shard_type]
+        if pulls <= rule["threshold"]:
+            return rule["base"]
+        return rule["base"] + (pulls - rule["threshold"]) * rule["increment"]
+
+    @commands.command(name="mercy")
+    async def mercy(self, ctx, action: str = None, arg1: str = None, arg2: str = None):
+        """Gestion de la mercy : !mercy add <nb> <type>, !mercy reset <type>, !mercy show"""
+        if ctx.channel.id != AUTHORIZED_CHANNEL_ID:
+            return
+
+        user_id = str(ctx.author.id)
+
+        # ----- SHOW -----
+        if action == "show":
+            rows = db_manager.fetchall("SELECT shard_type, pulls FROM mercy_counters WHERE user_id = ?", (user_id,))
+            if not rows:
+                await ctx.send("❌ You don't have any mercy data yet.")
+                return
+
+            embed = discord.Embed(
+                title=f"🎲 Mercy Status for {ctx.author.display_name}",
+                color=0x00bfff
+            )
+
+            for shard_type, pulls in rows:
+                chance = self.get_mercy_chance(shard_type, pulls)
+                embed.add_field(
+                    name=shard_type.replace("_", " ").title(),
+                    value=f"**{pulls} pulls** → {chance:.1f}% chance",
+                    inline=False
+                )
+
+            await ctx.send(embed=embed)
+
+        # ----- ADD -----
+        elif action == "add" and arg1 and arg2:
+            try:
+                pulls = int(arg1)
+            except ValueError:
+                await ctx.send("❌ Number of pulls must be an integer.")
+                return
+
+            shard_type = arg2.lower()
+            if shard_type not in MERCY_RULES:
+                await ctx.send(f"❌ Invalid shard type. Available: {', '.join(MERCY_RULES.keys())}")
+                return
+
+            # Update DB
+            current = db_manager.fetchone(
+                "SELECT pulls FROM mercy_counters WHERE user_id = ? AND shard_type = ?",
+                (user_id, shard_type)
+            )
+            if current:
+                new_value = current["pulls"] + pulls
+                db_manager.execute(
+                    "UPDATE mercy_counters SET pulls = ?, last_reset = ? WHERE user_id = ? AND shard_type = ?",
+                    (new_value, datetime.utcnow(), user_id, shard_type)
+                )
+            else:
+                db_manager.execute(
+                    "INSERT INTO mercy_counters (user_id, shard_type, pulls, last_reset) VALUES (?, ?, ?, ?)",
+                    (user_id, shard_type, pulls, datetime.utcnow())
+                )
+
+            await ctx.send(f"✅ Added {pulls} pulls to **{shard_type}** mercy.")
+
+        # ----- RESET -----
+        elif action == "reset" and arg1:
+            shard_type = arg1.lower()
+            if shard_type not in MERCY_RULES:
+                await ctx.send(f"❌ Invalid shard type. Available: {', '.join(MERCY_RULES.keys())}")
+                return
+
+            db_manager.execute(
+                "UPDATE mercy_counters SET pulls = 0, last_reset = ? WHERE user_id = ? AND shard_type = ?",
+                (datetime.utcnow(), user_id, shard_type)
+            )
+            await ctx.send(f"🔄 Mercy for **{shard_type}** has been reset.")
+
+        else:
+            await ctx.send("❌ Usage: `!mercy add <nb> <type>`, `!mercy reset <type>`, `!mercy show`")
+
+def setup(bot):
+    bot.add_cog(Mercy(bot))