from mem import services, service
import httpx
import asyncio
import time


async def check_service(client: httpx.AsyncClient, s: service):
    try:
        before = time.perf_counter()
        match s.ping_type:
            case 0:
                r = await client.head(
                    url=s.url,
                    follow_redirects=True,
                    timeout=1,
                )
            case 1:
                r = await client.get(
                    url=s.url,
                    follow_redirects=True,
                    timeout=1,
                )
            case _:
                raise httpx.HTTPError("Unknown ping type")

        after = time.perf_counter()
        s.set_error(None)
        s.set_online(r.status_code == 200)
        s.set_status(r.status_code)
        s.set_ping(int((after - before) * 1000))
    except httpx.HTTPError as e:
        s.set_error(str(e))
        s.set_online(False)
        s.set_status(None)
        s.set_ping(None)


def start_async_loop():
    loop = asyncio.new_event_loop()
    asyncio.set_event_loop(loop)
    asyncio.run_coroutine_threadsafe(update_services(loop=loop), loop=loop)
    loop.run_forever()


async def update_services(loop: asyncio.AbstractEventLoop):
    print("Starting service updates...")
    async with (
        httpx.AsyncClient() as public_client,
        httpx.AsyncClient(verify=False) as local_client,
    ):
        while True:
            tasks = [
                check_service(public_client if s.public else local_client, s)
                for s in services
            ]
            await asyncio.gather(*tasks)
            await asyncio.sleep(2)