#!/usr/bin/env python3
"""
MEP Race Test: Multiple providers compete for the same task.
Simulates a global network of sleeping nodes.
"""
import asyncio
import websockets
import json
import requests
import uuid
import time
import urllib.parse
from identity import MEPIdentity

HUB_URL = "http://localhost:8000"
WS_URL = "ws://localhost:8000"

class RacingProvider:
    def __init__(self, name, location):
        self.name = name
        self.location = location
        self.identity = MEPIdentity(f"{name.replace(' ', '_')}_{uuid.uuid4().hex[:6]}.pem")
        self.node_id = self.identity.node_id
        self.balance = 0
        self.won_race = False
        self.response_time = None
        
    async def compete(self, task_id, task_payload, bounty):
        """Connect and try to win the task."""
        print(f"[{self.name} in {self.location}] Connecting to MEP Hub...")
        
        # Register
        requests.post(f"{HUB_URL}/register", json={"pubkey": self.identity.pub_pem})
        
        # Connect via WebSocket
        start_time = time.time()
        ts = str(int(time.time()))
        sig = self.identity.sign(self.node_id, ts)
        sig_safe = urllib.parse.quote(sig)
        async with websockets.connect(f"{WS_URL}/ws/{self.node_id}?timestamp={ts}&signature={sig_safe}") as ws:
            print(f"[{self.name}] Connected. Waiting for task {task_id[:8]}...")
            
            try:
                # Listen for the task
                msg = await asyncio.wait_for(ws.recv(), timeout=5.0)
                data = json.loads(msg)
                
                if data["event"] == "new_task" and data["data"]["id"] == task_id:
                    self.response_time = time.time() - start_time
                    print(f"[{self.name}] 🏁 GOT THE TASK! Response: {self.response_time:.3f}s")
                    
                    # Simulate processing (faster providers win)
                    process_time = 0.1 if "fast" in self.name.lower() else 0.5
                    await asyncio.sleep(process_time)
                    
                    # Submit result
                    result = f"Processed by {self.name} from {self.location}. Task: {task_payload[:30]}..."
                    payload_str = json.dumps({
                        "task_id": task_id,
                        "provider_id": self.node_id,
                        "result_payload": result
                    })
                    headers = self.identity.get_auth_headers(payload_str)
                    headers["Content-Type"] = "application/json"
                    resp = requests.post(f"{HUB_URL}/tasks/complete", data=payload_str, headers=headers)
                    
                    if resp.status_code == 200:
                        self.won_race = True
                        self.balance = resp.json()["new_balance"]
                        print(f"[{self.name}] 🎉 WON THE RACE! Earned {bounty} SECONDS")
                    else:
                        print(f"[{self.name}] ❌ Race lost (someone else finished first)")
                        
            except asyncio.TimeoutError:
                print(f"[{self.name}] Timeout - task already taken")
            except Exception as e:
                print(f"[{self.name}] Error: {e}")

async def run_race():
    print("=" * 60)
    print("MEP GLOBAL RACE TEST: Multiple Nodes Compete for One Task")
    print("=" * 60)
    
    # Register consumer
    consumer = MEPIdentity(f"race_consumer_{uuid.uuid4().hex[:6]}.pem")
    requests.post(f"{HUB_URL}/register", json={"pubkey": consumer.pub_pem})
    
    # Create 4 providers in different "locations"
    providers = [
        RacingProvider("FastProvider-USA", "New York"),
        RacingProvider("SlowProvider-EU", "Berlin"),
        RacingProvider("QuickProvider-Asia", "Singapore"),
        RacingProvider("SteadyProvider-AU", "Sydney")
    ]
    
    # Submit a task
    task_payload = "Analyze the MEP race dynamics and provide insights"
    bounty = 8.5
    
    print(f"\n📤 Submitting task: {task_payload[:50]}...")
    print(f"   Bounty: {bounty} SECONDS")
    
    payload_str = json.dumps({
        "consumer_id": consumer.node_id,
        "payload": task_payload,
        "bounty": bounty
    })
    headers = consumer.get_auth_headers(payload_str)
    headers["Content-Type"] = "application/json"
    resp = requests.post(f"{HUB_URL}/tasks/submit", data=payload_str, headers=headers)
    
    task_data = resp.json()
    task_id = task_data["task_id"]
    print(f"   Task ID: {task_id[:8]}...")
    
    # Start all providers simultaneously
    print("\n🏁 Starting providers...")
    tasks = [provider.compete(task_id, task_payload, bounty) for provider in providers]
    await asyncio.gather(*tasks)
    
    # Results
    print("\n" + "=" * 60)
    print("RACE RESULTS:")
    print("=" * 60)
    
    winner = None
    for provider in providers:
        status = "🏆 WINNER" if provider.won_race else "❌ Lost"
        time_str = f"{provider.response_time:.3f}s" if provider.response_time else "N/A"
        print(f"{status} {provider.name:20} {provider.location:15} Response: {time_str:8} Balance: {provider.balance}")
        
        if provider.won_race:
            winner = provider
    
    if winner:
        print(f"\n🎯 The market chose: {winner.name} from {winner.location}")
        print(f"   Reason: Fastest response time ({winner.response_time:.3f}s)")
        print(f"   Economics: Earned {bounty} SECONDS for being efficient")
    else:
        print("\n⚠️ No winner - task may have failed")
    
    # Check consumer balance
    balance_resp = requests.get(f"{HUB_URL}/balance/{consumer.node_id}")
    consumer_balance = balance_resp.json()["balance_seconds"]
    print(f"\n💰 Consumer spent {bounty} SECONDS, new balance: {consumer_balance}")
    
    print("\n✅ Race test complete. This simulates how MEP creates a")
    print("   global efficiency market: fastest nodes win SECONDS.")

if __name__ == "__main__":
    asyncio.run(run_race())