import argparse
import asyncio
import os

import uvicorn
from fastapi_vue.hostutil import parse_endpoint
from uvicorn import Config, Server

from .APP_MODULE import APP_VAR

DEFAULT_PORT = 5080


def run_server(endpoints: list[dict], *, proxy="", devmode=False):
    conf: dict[str, object] = {"app": "MODULE_NAME.APP_MODULE:APP_VAR"}
    if proxy:
        conf["proxy_headers"] = True
        conf["forwarded_allow_ips"] = proxy
    if devmode:
        conf["reload"] = True
        conf["reload_dirs"] = ["MODULE_NAME"]
        APP_VAR.debug = True

    if len(endpoints) > 1:
        # Run separate servers for multiple endpoints
        async def serve_all():
            async with asyncio.TaskGroup() as tg:
                for ep in endpoints:
                    tg.create_task(Server(Config(**conf, **ep)).serve())

        asyncio.run(serve_all())
    else:
        uvicorn.run(**conf, **endpoints[0])


def main():
    parser = argparse.ArgumentParser(description="Run the MODULE_NAME server.")
    parser.add_argument(
        "endpoint",
        nargs="?",
        help=(
            f"Endpoint (default: localhost:{DEFAULT_PORT}). "
            "Forms: host:port | :port | [ipv6]:port | ip | host | unix:/path.sock"
        ),
    )
    args = parser.parse_args()
    proxy = os.getenv("FORWARDED_ALLOW_IPS", "127.0.0.1,::1")
    devmode = bool(os.getenv("FASTAPI_VUE_FRONTEND_URL"))
    endpoints = parse_endpoint(args.endpoint, DEFAULT_PORT)
    run_server(endpoints, proxy=proxy, devmode=devmode)


if __name__ == "__main__":
    main()