import json
import os
import time
import urllib.request
import urllib.error

# Configuration
BOT_TOKEN = os.environ["DISCORD_BOT_TOKEN"]
GUILD_ID = "692816967895220344"
BASE_URL = "https://discord.com/api/v10"

# Read Discord IDs from table.md
with open("table.md", "r") as f:
    content = f.read()

lines = content.strip().split("\n")

# Find the table header line (starts with |)
header_line = None
header_idx = 0
for i, line in enumerate(lines):
    if line.startswith("| Discord"):
        header_line = line
        header_idx = i
        break

if not header_line:
    print("Could not find table header!")
    exit(1)

headers = [h.strip() for h in header_line.split("|")[1:-1]]

discord_idx = 0  # Discord ID is the first column

discord_ids = []
for line in lines[header_idx + 2:]:  # Skip header and separator
    if not line.startswith("|"):
        continue
    cols = [c.strip() for c in line.split("|")[1:-1]]
    if len(cols) > discord_idx:
        discord_id = cols[discord_idx].strip()
        if discord_id and discord_id.isdigit():
            discord_ids.append(discord_id)

print(f"Found {len(discord_ids)} Discord IDs to verify")

# Verify each ID against the guild
verified = []
missing = []
errors = []

for i, discord_id in enumerate(discord_ids):
    url = f"{BASE_URL}/guilds/{GUILD_ID}/members/{discord_id}"
    req = urllib.request.Request(url)
    req.add_header("Authorization", f"Bot {BOT_TOKEN}")

    try:
        response = urllib.request.urlopen(req)
        data = json.loads(response.read().decode())
        username = data.get("user", {}).get("username", "Unknown")
        verified.append((discord_id, username))
        print(f"[{i+1}/{len(discord_ids)}] ✓ {discord_id} - {username}")
    except urllib.error.HTTPError as e:
        if e.code == 404:
            missing.append(discord_id)
            print(f"[{i+1}/{len(discord_ids)}] ✗ {discord_id} - NOT IN SERVER")
        elif e.code == 429:
            # Rate limited - wait and retry
            retry_after = json.loads(e.read().decode()).get("retry_after", 1)
            print(f"[{i+1}/{len(discord_ids)}] Rate limited, waiting {retry_after}s...")
            time.sleep(retry_after + 0.5)
            # Retry
            try:
                req2 = urllib.request.Request(url)
                req2.add_header("Authorization", f"Bot {BOT_TOKEN}")
                response = urllib.request.urlopen(req2)
                data = json.loads(response.read().decode())
                username = data.get("user", {}).get("username", "Unknown")
                verified.append((discord_id, username))
                print(f"[{i+1}/{len(discord_ids)}] ✓ {discord_id} - {username} (after retry)")
            except urllib.error.HTTPError as e2:
                if e2.code == 404:
                    missing.append(discord_id)
                    print(f"[{i+1}/{len(discord_ids)}] ✗ {discord_id} - NOT IN SERVER (after retry)")
                else:
                    errors.append((discord_id, f"HTTP {e2.code}"))
                    print(f"[{i+1}/{len(discord_ids)}] ? {discord_id} - Error {e2.code}")
        else:
            errors.append((discord_id, f"HTTP {e.code}"))
            print(f"[{i+1}/{len(discord_ids)}] ? {discord_id} - Error {e.code}")

    # Small delay to avoid rate limits
    time.sleep(0.1)

print(f"\n=== SUMMARY ===")
print(f"Verified: {len(verified)}")
print(f"Missing: {len(missing)}")
print(f"Errors: {len(errors)}")

# Save results
with open("discord_verification.json", "w") as f:
    json.dump({
        "verified": verified,
        "missing": missing,
        "errors": errors
    }, f, indent=2)

print("\nResults saved to discord_verification.json")