#!/usr/bin/env -S uv run
# auto-upgrade@fastapi-vue-setup - remove this if you modify this file
"""Run Vite development server for Vue app and FastAPI backend with auto-reload."""

import argparse
import asyncio
import os
import subprocess
import sys
from pathlib import Path

import tracerite

# Import util.py from scripts/fastapi-vue (not a package, so we adjust sys.path)
sys.path.insert(0, str(Path(__file__).with_name("fastapi-vue")))
from devutil import (
    ProcessGroup,
    check_ports_free,
    logger,
    ready,
    setup_cli,
    setup_vite,
)

DEFAULT_VITE_PORT = TEMPLATE_VITE_PORT
DEFAULT_DEV_PORT = TEMPLATE_DEV_PORT
HEALTH = TEMPLATE_HEALTH


async def run_devserver(
    listen: str,
    backend: str,
    extra_args: list[str] | None = None,
) -> None:
    """Start Vite and FastAPI dev servers with hot reload."""
    reporoot = Path(__file__).parent.parent
    front = reporoot / "frontend"
    if not (front / "package.json").exists():
        logger.warning("Frontend source not found at %s", front)
        raise SystemExit(1)

    viteurl, npm_install, vite = setup_vite(listen, DEFAULT_VITE_PORT)
    backurl, MODULE_NAME = setup_cli("PROJECT_CLI", backend, DEFAULT_DEV_PORT)

    # Tell everyone via environment (vite proxy and backend devmode use these)
    os.environ["ENVPREFIX_VITE_URL"] = viteurl
    os.environ["ENVPREFIX_BACKEND_URL"] = backurl
    os.environ["ENVPREFIX_DEV"] = "1"

    async with ProcessGroup() as pg:
        pg.create_task(check_ports_free(viteurl, backurl))
        npm_i = await pg.spawn(*npm_install, cwd=front)
        await pg.spawn(*MODULE_NAME, *(extra_args or []), vital=True)
        await pg.wait(npm_i, ready(backurl, path=HEALTH))
        await pg.spawn(*vite, cwd=front, vital=True)


def main() -> None:
    """Parse CLI arguments and run the devserver."""
    tracerite.load()
    parser = argparse.ArgumentParser(
        description="Run Vite and FastAPI development servers",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog=HELP_EPILOG,
    )
    parser.add_argument(
        "-l",
        "--listen",
        metavar="addr",
        help=f"Vite (default: localhost:{DEFAULT_VITE_PORT})",
    )
    parser.add_argument(
        "--backend",
        metavar="addr",
        help=f"FastAPI (default: localhost:{DEFAULT_DEV_PORT})",
    )
    args, extra_args = parser.parse_known_args()
    try:
        asyncio.run(run_devserver(args.listen, args.backend, extra_args))
    except* KeyboardInterrupt:
        pass  # user stopped the devserver: normal exit
    except* (subprocess.SubprocessError, RuntimeError):
        raise SystemExit(1) from None  # logged in devutil already; exit 1


HELP_EPILOG = """
  Other options are forwarded to PROJECT_CLI [args]

  JS_RUNTIME environment variable can be used to select the JS runtime:
  npm, deno, bun, or full path to the runtime executable (node maps to npm).
"""


if __name__ == "__main__":
    main()