import sqlite3
from datetime import datetime
from config import DATABASE_PATH

# Règles de mercy pour stockage
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},
}

class MercyManager:
    def __init__(self, db_path=DATABASE_PATH):
        self.db_path = db_path
        self.init_table()

    def init_table(self):
        conn = sqlite3.connect(self.db_path)
        cursor = conn.cursor()
        cursor.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)
            )
        """)
        conn.commit()
        conn.close()

    def get_pulls(self, user_id, shard_type):
        conn = sqlite3.connect(self.db_path)
        cursor = conn.cursor()
        cursor.execute(
            "SELECT pulls FROM mercy_counters WHERE user_id = ? AND shard_type = ?",
            (user_id, shard_type)
        )
        row = cursor.fetchone()
        conn.close()
        return row[0] if row else 0

    def add_pulls(self, user_id, shard_type, pulls):
        current = self.get_pulls(user_id, shard_type)
        conn = sqlite3.connect(self.db_path)
        cursor = conn.cursor()
        if current:
            cursor.execute(
                "UPDATE mercy_counters SET pulls = ?, last_reset = ? WHERE user_id = ? AND shard_type = ?",
                (current + pulls, datetime.utcnow(), user_id, shard_type)
            )
        else:
            cursor.execute(
                "INSERT INTO mercy_counters (user_id, shard_type, pulls, last_reset) VALUES (?, ?, ?, ?)",
                (user_id, shard_type, pulls, datetime.utcnow())
            )
        conn.commit()
        conn.close()
        return current + pulls

    def reset_pulls(self, user_id, shard_type):
        conn = sqlite3.connect(self.db_path)
        cursor = conn.cursor()
        cursor.execute(
            "UPDATE mercy_counters SET pulls = 0, last_reset = ? WHERE user_id = ? AND shard_type = ?",
            (datetime.utcnow(), user_id, shard_type)
        )
        conn.commit()
        conn.close()

    def get_all_pulls(self, user_id):
        conn = sqlite3.connect(self.db_path)
        cursor = conn.cursor()
        cursor.execute(
            "SELECT shard_type, pulls FROM mercy_counters WHERE user_id = ?",
            (user_id,)
        )
        rows = cursor.fetchall()
        conn.close()
        return {shard_type: pulls for shard_type, pulls in rows}

    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"]

    def get_guaranteed_pull(self, shard_type):
        """Retourne le pull où la loot est garanti à 100%"""
        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},
        }
        if shard_type not in rules:
            return None
        rule = rules[shard_type]
        return int(rule["start"] + (100 - rule["base"]) / rule["increment"])