fastapi_app.py

"""Run the PeerJS broker alongside a FastAPI app, sharing its event loop.

Requires fastapi and uvicorn (not project dependencies):
    pip install fastapi uvicorn

    uvicorn examples.fastapi_app:app --port 8888
"""

import asyncio
import logging
from contextlib import asynccontextmanager

from fastapi import FastAPI

from python_peerjs_server.config import Config
from python_peerjs_server.main import peerjs_serve

logging.basicConfig(level=logging.INFO)

peerjs_config = Config(host="127.0.0.1", port=9000, key="peerjs")


@asynccontextmanager
async def lifespan(app: FastAPI):
    task = asyncio.create_task(peerjs_serve(peerjs_config, handle_signals=False))
    yield
    task.cancel()
    await asyncio.gather(task, return_exceptions=True)


app = FastAPI(lifespan=lifespan)


@app.get("/")
def index() -> str:
    return "hello from fastapi"


if __name__ == "__main__":
    raise SystemExit(
        "This example defines an ASGI app and must be run with uvicorn, not plain python:\n\n"
        "    uvicorn examples.fastapi_app:app --port 8888\n"
    )