#!/usr/bin/env python3
"""
Assign team-specific roles to all 155 participants.
Respects Discord rate limits with proper backoff and retry logic.
"""

import json
import os
import time
import requests

BOT_TOKEN = os.environ["DISCORD_BOT_TOKEN"]
GUILD_ID = "692816967895220344"

BASE_URL = "https://discord.com/api/v10"
HEADERS = {
    "Authorization": f"Bot {BOT_TOKEN}",
    "Content-Length": "0"
}

TEAM_ROLE_IDS = {
    "Jade Jasmine": "1464314923780931677",
    "Crimson Dahlia": "1464315093402784015",
    "Rose Camellia": "1464315098452726106",
    "Amber Wisteria": "1464315105264275600",
    "Ivory Orchid": "1464315109873684593",
    "Teal Iris": "1464315114378498152",
    "Peach Gardenia": "1464315118904152107",
    "Violet Carnation": "1464315124251754559",
    "Azure Lotus": "1464315128437801177",
    "Coral Sunflower": "1464315132896088168",
    "Indigo Tulip": "1464315138428633241",
    "Scarlet Hydrangea": "1464315142710890520",
    "Mint Narcissus": "1464315149203804405",
    "Sage Marigold": "1464315153599299803",
}


def assign_role_with_retry(user_id: str, role_id: str, max_retries: int = 5) -> bool:
    url = f"{BASE_URL}/guilds/{GUILD_ID}/members/{user_id}/roles/{role_id}"

    for attempt in range(max_retries):
        response = requests.put(url, headers=HEADERS)

        if response.status_code == 204:
            return True
        elif response.status_code == 429:
            # Check headers first, fall back to JSON body
            retry_after = response.headers.get("Retry-After")
            if retry_after is None:
                retry_after = response.headers.get("X-RateLimit-Reset-After")
            if retry_after is None:
                try:
                    retry_after = response.json().get("retry_after", 1)
                except:
                    retry_after = 1
            retry_after = float(retry_after)
            print(f"  Rate limited! Waiting {retry_after:.2f}s before retry...")
            time.sleep(retry_after)
        else:
            print(f"  Error {response.status_code}: {response.text}")
            backoff_time = (2 ** attempt) * 0.5
            print(f"  Retrying in {backoff_time:.2f}s...")
            time.sleep(backoff_time)

    return False


def main():
    with open("team_assignments.json", "r") as f:
        teams = json.load(f)

    print(f"Assigning team roles to {len(teams)} teams...")
    print("-" * 50)

    total_success = 0
    total_fail = 0

    for team in teams:
        team_name = team["name"]
        role_id = TEAM_ROLE_IDS[team_name]
        all_members = team["leaders"] + team["participants"]

        print(f"\n[{team_name}] Assigning role to {len(all_members)} members...")

        for user_id in all_members:
            print(f"  {user_id}...", end=" ")

            if assign_role_with_retry(user_id, role_id):
                print("✓")
                total_success += 1
            else:
                print("✗ FAILED")
                total_fail += 1

            # Small delay between requests to be nice to the API
            time.sleep(0.1)

    print("-" * 50)
    print(f"Complete! Success: {total_success}, Failed: {total_fail}")


if __name__ == "__main__":
    main()