import asyncio
import random
from datetime import datetime, timedelta, timezone

from app.api.dependencies import prisma

async def main():
    await prisma.connect()
    
    ports_data = [
        {"name": "Haldia", "chartedDepth": 10.0, "permissibleDraft": 7.5, "brackishDensity": 1.010, "lat": 22.02, "lon": 88.06, "typicalTidalRange": 1.5, "maxVesselClass": "Supramax"},
        {"name": "Paradip", "chartedDepth": 17.0, "permissibleDraft": 14.5, "brackishDensity": 1.025, "lat": 20.26, "lon": 86.67, "typicalTidalRange": 2.0, "maxVesselClass": "Capesize"},
        {"name": "Dhamra", "chartedDepth": 18.0, "permissibleDraft": 16.0, "brackishDensity": 1.025, "lat": 20.82, "lon": 86.97, "typicalTidalRange": 2.2, "maxVesselClass": "Capesize"},
        {"name": "Mumbai", "chartedDepth": 14.0, "permissibleDraft": 11.0, "brackishDensity": 1.025, "lat": 18.94, "lon": 72.83, "typicalTidalRange": 4.5, "maxVesselClass": "Panamax"},
        {"name": "Kandla", "chartedDepth": 13.0, "permissibleDraft": 12.0, "brackishDensity": 1.025, "lat": 23.00, "lon": 70.21, "typicalTidalRange": 6.0, "maxVesselClass": "Panamax"},
        {"name": "Mundra", "chartedDepth": 16.0, "permissibleDraft": 14.0, "brackishDensity": 1.025, "lat": 22.73, "lon": 69.70, "typicalTidalRange": 4.0, "maxVesselClass": "Capesize"},
        # Additional missing ports
        {"name": "Vizag", "chartedDepth": 18.0, "permissibleDraft": 16.5, "brackishDensity": 1.025, "lat": 17.68, "lon": 83.21, "typicalTidalRange": 1.8, "maxVesselClass": "Capesize"},
        {"name": "Chennai", "chartedDepth": 16.5, "permissibleDraft": 14.5, "brackishDensity": 1.025, "lat": 13.08, "lon": 80.29, "typicalTidalRange": 1.2, "maxVesselClass": "Capesize"},
        {"name": "Kolkata", "chartedDepth": 8.0, "permissibleDraft": 6.5, "brackishDensity": 1.005, "lat": 22.57, "lon": 88.36, "typicalTidalRange": 1.5, "maxVesselClass": "Handysize"},
        {"name": "Cochin", "chartedDepth": 13.5, "permissibleDraft": 12.5, "brackishDensity": 1.020, "lat": 9.93, "lon": 76.26, "typicalTidalRange": 1.0, "maxVesselClass": "Panamax"},
        {"name": "Mangalore", "chartedDepth": 14.0, "permissibleDraft": 12.0, "brackishDensity": 1.025, "lat": 12.87, "lon": 74.84, "typicalTidalRange": 1.5, "maxVesselClass": "Panamax"},
        {"name": "Tuticorin", "chartedDepth": 14.2, "permissibleDraft": 12.8, "brackishDensity": 1.025, "lat": 8.76, "lon": 78.13, "typicalTidalRange": 1.1, "maxVesselClass": "Panamax"},
        {"name": "Ennore", "chartedDepth": 16.0, "permissibleDraft": 14.5, "brackishDensity": 1.025, "lat": 13.25, "lon": 80.33, "typicalTidalRange": 1.2, "maxVesselClass": "Capesize"},
        {"name": "Gangavaram", "chartedDepth": 21.0, "permissibleDraft": 19.5, "brackishDensity": 1.025, "lat": 17.62, "lon": 83.24, "typicalTidalRange": 1.8, "maxVesselClass": "Capesize"},
        {"name": "Krishnapatnam", "chartedDepth": 18.5, "permissibleDraft": 17.0, "brackishDensity": 1.025, "lat": 14.24, "lon": 80.12, "typicalTidalRange": 1.5, "maxVesselClass": "Capesize"},
    ]
    
    for port in ports_data:
        await prisma.port.upsert(
            where={"name": port["name"]},
            data={
                "create": port,
                "update": port,
            }
        )

    vessels_data = [
        {"name": "Capesize", "capacity": 150000, "laden_draft": 17.5, "ballast_draft": 9.0, "daily_cost": 25000, "block_coeff": 0.85, "speed_knots": 14.5},
        {"name": "Panamax", "capacity": 75000, "laden_draft": 13.5, "ballast_draft": 7.0, "daily_cost": 15000, "block_coeff": 0.82, "speed_knots": 14.0},
        {"name": "Supramax", "capacity": 55000, "laden_draft": 11.5, "ballast_draft": 6.0, "daily_cost": 12000, "block_coeff": 0.80, "speed_knots": 14.0},
        {"name": "Handymax", "capacity": 50000, "laden_draft": 11.5, "ballast_draft": 5.5, "daily_cost": 11000, "block_coeff": 0.79, "speed_knots": 14.0},
        {"name": "Handysize", "capacity": 35000, "laden_draft": 10.0, "ballast_draft": 5.0, "daily_cost": 9000, "block_coeff": 0.75, "speed_knots": 13.5},
    ]

    for vessel in vessels_data:
        await prisma.vessel.upsert(
            where={"name": vessel["name"]},
            data={
                "create": vessel,
                "update": vessel,
            }
        )

    await prisma.requisition.delete_many()
    
    req_data = []
    dest_ports = [p["name"] for p in ports_data]
    commodities = ["Thermal Coal", "Coking Coal", "Metallurgical Coal", "Iron Ore", "Bauxite", "Grain"]
    statuses = ["Infeasible", "Pending Evaluation", "Converted", "Feasible", "Allocated"]
    origins = [
        "Port Hedland, Australia",
        "Newcastle, Australia",
        "Richards Bay, South Africa",
        "Dampier, Australia",
        "Tubarão, Brazil",
        "Samarinda, Indonesia",
        "Baltimore, USA",
        "Gladstone, Australia"
    ]
    
    for i in range(500):
        req_data.append({
            "volume_mt": float(random.randint(30000, 200000)),
            "commodity": random.choice(commodities),
            "origin": random.choice(origins),
            "destPortName": random.choice(dest_ports),
            "status": random.choice(statuses),
            "createdAt": datetime.now(timezone.utc) - timedelta(days=random.randint(0, 730))
        })
        
    await prisma.requisition.create_many(data=req_data)

    print("Seed complete.")
    await prisma.disconnect()

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