import json
import os
from pathlib import Path
from dotenv import load_dotenv
from supabase import create_client, Client

BASE_DIR = Path(__file__).parent.parent          # backend/app/
DATA_DIR = BASE_DIR / "data"
ENV_PATH = BASE_DIR.parent / ".env"             # backend/.env


def load_json(filename: str):
    with open(DATA_DIR / filename, encoding="utf-8") as f:
        return json.load(f)


def main():
    print("Iniciando proceso de seeding...")
    load_dotenv(ENV_PATH)

    URL = os.environ.get("SUPABASE_URL")
    KEY = os.environ.get("SUPABASE_SERVICE_ROLE_KEY")

    if not URL or not KEY:
        raise ValueError("Falta SUPABASE_URL o SUPABASE_SERVICE_ROLE_KEY en el .env")

    supabase: Client = create_client(URL, KEY)

    rutas_data = load_json("rutas.json")
    colonias_data = load_json("colonias-rutas.json")

    # 1. Poblar UNITS (Dependencia raíz)
    print("1. Poblando tabla 'units'...")
    units_to_insert = []
    truck_ids = set()
    for r in rutas_data:
        tid = r.get("truckId")
        if tid and tid not in truck_ids:
            truck_ids.add(tid)
            units_to_insert.append({
                "id": tid,
                "plate": f"GTO-{tid}", # Simulado
                "status": "active"
            })
    if units_to_insert:
        supabase.table("units").upsert(units_to_insert).execute()

    # 2. Poblar ROUTES (Depende de units)
    print("2. Poblando tabla 'routes'...")
    routes_to_insert = []
    for r in rutas_data:
        routes_to_insert.append({
            "id": r["routeId"],
            "name": f"Ruta {r['routeId'].split('-')[1]}",
            "truck_id": r["truckId"],
            "turno": r.get("turno", "matutino"),
            "status": r.get("status", "pendiente"),
            "current_position_id": 1
        })
    if routes_to_insert:
        supabase.table("routes").upsert(routes_to_insert).execute()

    # 3. Poblar ROUTE_POSITIONS (Depende de routes)
    print("3. Poblando tabla 'route_positions'...")
    positions_to_insert = []
    for r in rutas_data:
        for pos in r.get("positions", []):
            positions_to_insert.append({
                "route_id": r["routeId"],
                "position_id": pos["positionId"],
                "lat": pos["lat"],
                "lng": pos["lng"],
                "speed": pos.get("speed", 0),
                "ts": pos.get("ts")
            })
    if positions_to_insert:
        supabase.table("route_positions").upsert(positions_to_insert).execute()

    # 4. Poblar COLONIAS (Depende de routes)
    print("4. Poblando tabla 'colonias'...")
    if colonias_data:
        colonias_to_insert = []
        for c in colonias_data:
            colonias_to_insert.append({
                "nombre": c["colonia"],
                "route_id": c["routeId"],
                "turno": c["turno"],
                "horario_estimado": c["horario_estimado"]
            })
        supabase.table("colonias").upsert(colonias_to_insert, on_conflict="nombre").execute()

    print("✅ Seed completado con éxito. Base de datos operativa para la app.")

if __name__ == "__main__":
    main()