Discord-Raid-bot/cogs/mercy.py

import discord
from discord.ext import commands
from config import AUTHORIZED_CHANNEL_ID
from utils.MercyManager_class import MercyManager

VALID_SHARDS = ["ancient", "void", "sacred", "primal", "remnant"]

class Mercy(commands.Cog):
    def __init__(self, bot):
        self.bot = bot
        self.mercy_manager = MercyManager()

    def calc_chance_and_guarantee(self, shard_type, pulls):
        """Calcule la chance actuelle et le pull garanti"""
        rules = {
            "ancient": {"start": 200, "increment": 5, "base": 0.5},
            "void": {"start": 200, "increment": 5, "base": 0.5},
            "sacred": {"start": 12, "increment": 2, "base": 6},
            "primal_legendary": {"start": 75, "increment": 1, "base": 1},
            "primal_mythical": {"start": 200, "increment": 10, "base": 0.5},
            "remnant": {"start": 24, "increment": 1, "base": 0},
        }
        rule = rules[shard_type]
        chance = rule["base"] if pulls < rule["start"] else rule["base"] + (pulls - rule["start"]) * rule["increment"]
        guaranteed_at = None
        if chance >= 100:
            guaranteed_at = rule["start"] + (100 - rule["base"]) / rule["increment"]
        return chance, guaranteed_at

    @commands.command(name="mercy")
    async def mercy(self, ctx, action: str = None, arg1: str = None, arg2: str = None):
        if ctx.channel.id != AUTHORIZED_CHANNEL_ID:
            return

        user_id = str(ctx.author.id)

        if action == "show":
            pulls_dict = self.mercy_manager.get_all_pulls(user_id)
            if not pulls_dict:
                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 pulls_dict.items():
                if shard_type == "primal":
                    for sub_type in ["primal_legendary", "primal_mythical"]:
                        chance, guaranteed_at = self.calc_chance_and_guarantee(sub_type, pulls)
                        guaranteed_text = f" (Guaranteed at {int(guaranteed_at)} pulls)" if guaranteed_at else ""
                        embed.add_field(
                            name=sub_type.replace("_", " ").title(),
                            value=f"Pulled: **{pulls} shards** → {chance:.1f}% chance{guaranteed_text}",
                            inline=False
                        )
                else:
                    chance, guaranteed_at = self.calc_chance_and_guarantee(shard_type, pulls)
                    guaranteed_text = f" (Guaranteed at {int(guaranteed_at)} pulls)" if guaranteed_at else ""
                    embed.add_field(
                        name=shard_type.replace("_", " ").title(),
                        value=f"Pulled: **{pulls} shards** → {chance:.1f}% chance{guaranteed_text}",
                        inline=False
                    )

            await ctx.send(embed=embed)

        elif action == "add" and arg1 and arg2:
            try:
                pulls_to_add = int(arg1)
            except ValueError:
                await ctx.send("❌ Number of pulls must be an integer.")
                return

            shard_type = arg2.lower()
            if shard_type not in VALID_SHARDS:
                await ctx.send(f"❌ Invalid shard type. Available: {', '.join(VALID_SHARDS)}")
                return

            new_pulls = self.mercy_manager.add_pulls(user_id, shard_type, pulls_to_add)
            await ctx.send(f"✅ Added {pulls_to_add} pulls to **{shard_type}** mercy. Total: {new_pulls}")

        elif action == "reset" and arg1:
            shard_type = arg1.lower()
            if shard_type not in VALID_SHARDS:
                await ctx.send(f"❌ Invalid shard type. Available: {', '.join(VALID_SHARDS)}")
                return

            self.mercy_manager.reset_pulls(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`")

async def setup(bot):
    await bot.add_cog(Mercy(bot))