import requests
from utils import nextMonday
from collections import defaultdict

API_KEY = "AIzaSyBoBHzeQFgR7O-NlNsuHXQcC1B7ccEHpl8"
url = "https://routes.googleapis.com/directions/v2:computeRoutes"


def travel_time(origin_lat: float, origin_lon: float, dest_lat: float, dest_lon: float):
    monday9am = nextMonday()

    header = {
        "X-Goog-Api-Key": API_KEY,
        "Content-Type": "application/json",
        "X-Goog-FieldMask": "routes.distanceMeters,routes.duration,routes.staticDuration,routes.legs.steps.distanceMeters,routes.legs.steps.staticDuration,routes.legs.steps.travelMode",
    }

    body = {
        "origin": {
            "location": {"latLng": {"latitude": origin_lat, "longitude": origin_lon}}
        },
        "destination": {
            "location": {"latLng": {"latitude": dest_lat, "longitude": dest_lon}}
        },
        "travelMode": "TRANSIT",
        # "2023-10-15T15:01:23.045123456Z"
        "departureTime": monday9am.strftime("%Y-%m-%dT%H:%M:%S.%fZ"),
        "computeAlternativeRoutes": False,
        #   "routeModifiers": {
        #     "avoidTolls": false,
        #     "avoidHighways": false,
        #     "avoidFerries": false
        #   },
        "languageCode": "en-US",
        "units": "METRIC",
    }

    r = requests.post(url, json=body, headers=header)
    if r.status_code == 200:
        return r.json()

    raise Exception(r.json())


def extract_time(d):
    r = d["routes"][0]
    print(r.keys())
    distance = r["distanceMeters"]
    duration = r["duration"]
    duration_static = r["staticDuration"]

    steps = r["legs"][0]["steps"]
    # print(steps)
    duration_per_transit = defaultdict(lambda: 0)
    distance_per_transit = defaultdict(lambda: 0)

    for step in steps:
        duration_per_transit[step["travelMode"]] += int(
            step["staticDuration"].strip("s")
        )
        distance_per_transit[step["travelMode"]] += step.get("distanceMeters", 0)

    print(
        f"dis {distance}, dur {duration}, duration per transit {dict(duration_per_transit)}, distance per transit {dict(distance_per_transit)}"
    )


if __name__ == "__main__":
    import json

    with open("code/json/routing_routeapi.json", "r") as f:
        d = json.load(f)

    extract_time(d)


# if __name__ == "__main__":
#    origin = 51.5635664310333, -0.1107173751570373 # home
#    dest = 51.50475678313417, 0.04915321000190009 # london city airport
#    d = travel_time(origin[0], origin[1], dest[0], dest[1])
#    import json
#    with open('code/json/routing_routeapi.json', 'w') as f:
#       json.dump(d, f)