import network
import time
import json
import urequests



print("\n" + "*" * 40)
print("       The Bull Runner Tracker")
print("*" * 40 + "\n")

# Define the API endpoint URLs for all stops
url_stop_1 = 'https://passiogo.com/mapGetData.php?eta=3&deviceId=22898923&stopIds=70778&routeId=40482&position=7&userId=2343&routeIds=40482,41819'
url_stop_2 = 'https://passiogo.com/mapGetData.php?eta=3&deviceId=22898923&stopIds=70779&routeId=40482&position=8&userId=2343&routeIds=40482,41819'
url_stop_3 = 'https://passiogo.com/mapGetData.php?eta=3&deviceId=22898923&stopIds=70780&routeId=40482&position=9&userId=2343&routeIds=40482'
url_stop_4 = 'https://passiogo.com/mapGetData.php?eta=3&deviceId=22898923&stopIds=70781&routeId=40482&position=10&userId=2343&routeIds=40482'
url_stop_5 = 'https://passiogo.com/mapGetData.php?eta=3&deviceId=22898923&stopIds=70783&routeId=40482&position=11&userId=2343&routeIds=40482'
url_stop_6 = 'https://passiogo.com/mapGetData.php?eta=3&deviceId=22898923&stopIds=70749&routeId=40482&position=12&userId=2343&routeIds=40482'
url_stop_7 = 'https://passiogo.com/mapGetData.php?eta=3&deviceId=22898923&stopIds=70777&routeId=40482&position=6&userId=2343&routeIds=40482'
url_stop_8 = 'https://passiogo.com/mapGetData.php?eta=3&deviceId=22898923&stopIds=70776&routeId=40482&position=5&userId=2343&routeIds=40482'
url_stop_9 = 'https://passiogo.com/mapGetData.php?eta=3&deviceId=22898923&stopIds=70787&routeId=40482&position=4&userId=2343&routeIds=40482'
url_stop_10 = 'https://passiogo.com/mapGetData.php?eta=3&deviceId=22898923&stopIds=70786&routeId=40482&position=3&userId=2343&routeIds=40482'

# Define a dictionary to map stop IDs to their names
stop_names = {
    "70778": "00807 - N 46th St at USF Golf Center",
    "70779": "00815 - Skipper Rd at 43rd St",
    "70780": "00821 - N 42nd St at Hellenic Dr",
    "70781": "00827 - N 42nd St at Cambridge Woods Dr",
    "70783": "00829 - N 42nd St at Rocky Circle",
    "70749": "00411 - Palm Dr at Laurel Dr",
    "70777": "00803 - N 46th St at Shadow Moss Ln",
    "70776": "00801 - N 46th St at The Claw",
    "70787": "00970 - Parking Lot 56",
    "70786": "00966 - The Hub",
}

# Function to connect to WiFi
# Function to connect to WiFi
def connect_to_wifi():
    print("\n" + "*" * 40)
    print("          Connecting to WiFi")
    print("*" * 40 + "\n")

    print("Grabbing the Bull by the Horns... Connecting", end="")
    sta_if = network.WLAN(network.STA_IF)
    sta_if.active(True)
    sta_if.connect('Wokwi-GUEST', '')
    while not sta_if.isconnected():
        print(".", end="")
        time.sleep(0.1)
    print(" Bull Tamed! Connected!\n")



# Function to make an HTTP GET request and return the response as text
def make_http_request(url):
    response = urequests.get(url)
    return response.text

# Function to parse and display ETA data for each stop
def display_eta_for_stop(response, stop_id):
    try:
        api_data = json.loads(response)
        eta_data = api_data.get("ETAs", {})
        stop_eta_list = eta_data.get(stop_id, [])
        stop_name = stop_names.get(stop_id, "Unknown Stop")

        if stop_eta_list:
            print(f"Stop Name: {stop_name}")
            for eta_entry in stop_eta_list:
                bus_name = eta_entry.get("busName")
                eta_time = eta_entry.get("eta")
                print(f"Bus Name: {bus_name}, ETA: {eta_time}")

                # Record wait time in historical data
                current_time = time.time()
                historical_wait_times[stop_id].append(eta_time - current_time)

        # Display average wait time
        display_average_wait_time(stop_id)

    except json.JSONDecodeError as e:
        print(f"Failed to parse JSON for Stop ID {stop_id}: {e}")
    except Exception as e:
        print(f"An error occurred: {e}")
        # Move the 'else' block to the same indentation level as the 'try' block
    else:
        print(f"Stop Name: {stop_name}")
        print("No vehicles")


# Function to calculate and display average wait time for each stop
def display_average_wait_time(stop_id):
    wait_times = historical_wait_times.get(stop_id, [])
    if wait_times:
        average_wait_time = sum(wait_times) / len(wait_times)
        print(f"Average wait time for {stop_names[stop_id]}: {average_wait_time:.2f} minutes")
    else:
        print(f"No historical data available for {stop_names[stop_id]}")

# Connect to WiFi
connect_to_wifi()

# Make HTTP GET requests for all stops
response_stop_1 = make_http_request(url_stop_1)
response_stop_2 = make_http_request(url_stop_2)
response_stop_3 = make_http_request(url_stop_3)
response_stop_4 = make_http_request(url_stop_4)
response_stop_5 = make_http_request(url_stop_5)
response_stop_6 = make_http_request(url_stop_6)
response_stop_7 = make_http_request(url_stop_7)
response_stop_8 = make_http_request(url_stop_8)
response_stop_9 = make_http_request(url_stop_9)
response_stop_10 = make_http_request(url_stop_10)

# Dictionary to store historical wait times for each stop
historical_wait_times = {
    "70778": [],
    "70779": [],
    "70780": [],
    "70781": [],
    "70783": [],
    "70749": [],
    "70777": [],
    "70776": [],
    "70787": [],
    "70786": [],
}

# Function to parse and display ETA data for each stop
def display_eta_for_stop(response, stop_id):
    try:
        api_data = json.loads(response)
        eta_data = api_data.get("ETAs", {})
        stop_eta_list = eta_data.get(stop_id, [])
        stop_name = stop_names.get(stop_id, "Unknown Stop")

        if stop_eta_list:
            print(f"Stop Name: {stop_name}")
            for eta_entry in stop_eta_list:
                bus_name = eta_entry.get("busName")
                eta_time = eta_entry.get("eta")
                print(f"Bus Name: {bus_name}, ETA: {eta_time}")

                # Record wait time in historical data
                current_time = time.time()
                historical_wait_times[stop_id].append(eta_time - current_time)

            # Display average wait time
            display_average_wait_time(stop_id)

        else:
            print(f"Stop Name: {stop_name}")
            print("No vehicles")
    except json.JSONDecodeError as e:
        print(f"Failed to parse JSON for Stop ID {stop_id}: {e}")
    except Exception as e:
        print(f"An error occurred: {e}")

# Display ETA for all stops with their names
display_eta_for_stop(response_stop_1, "70778")
display_eta_for_stop(response_stop_2, "70779")
display_eta_for_stop(response_stop_3, "70780")
display_eta_for_stop(response_stop_4, "70781")
display_eta_for_stop(response_stop_5, "70783")
display_eta_for_stop(response_stop_6, "70749")
display_eta_for_stop(response_stop_7, "70777")
display_eta_for_stop(response_stop_8, "70776")
display_eta_for_stop(response_stop_9, "70787")
display_eta_for_stop(response_stop_10, "70786")