"""
Simulation: Handoff from Queue to Recovery Enrichment Agent
Includes browser agent fallback and human validation escalation steps.
"""

import json
import random
from datetime import datetime

# Simulate queue item
queue_item = {
    "id": "rec_101",
    "priority": "high",
    "data": {
        "company_name": "Test Company",
        "domain": "testco.com"
    },
    "created_at": str(datetime.now())
}

# Load agent config
with open("docs/configs/enrichment_config.json", "r") as f:
    config = json.load(f)

activity_log = []

def log(event, details):
    activity_log.append({
        "timestamp": str(datetime.now()),
        "event": event,
        "details": details
    })

log("enrichment_started", {"queue_item_id": queue_item["id"]})

# Prioritized source handoff
for source in config["enrichment_sources"]["priority_order"]:
    log("source_attempted", {"source": source})
    if source == "linkedin_browser":
        # Simulate browser agent fallback
        log("browser_agent_triggered", {"source": source, "item": queue_item["id"]})
        # Simulate human validation being required
        need_human = random.choice([True, False])
        if need_human:
            log("human_validation_required", {"trigger": "confidence_score_below_threshold"})
            log("escalation_created", {"queue": config["agent_handoff"]["human_validation"]["escalation_queue"], "item": queue_item["id"]})
            break
    elif random.random() < 0.05:
        # Simulate error/need retry
        log("source_failed", {"source": source, "error": "Rate limit hit"})
        log("rate_limit_hit", {"source": source})
        continue
    else:
        log("source_succeeded", {"source": source})
        break

log("enrichment_completed", {"queue_item_id": queue_item["id"]})

# Write activity log
with open("activity_log.csv", "a") as f:
    for entry in activity_log:
        f.write(f"{entry['timestamp']},{entry['event']},{json.dumps(entry['details'])}\n")

print("Simulation complete. Activity log updated.")