Function first_message_handler

experiments/authentication/app.py:111–120

Handler that sends credentials in the first WebSocket message.

(websocket)

async def first_message_handler(websocket):
    """Handler that sends credentials in the first WebSocket message."""
    token = await websocket.recv()
    user = get_user(token)
    if user is None:
        await websocket.close(CloseCode.INTERNAL_ERROR, "authentication failed")
        return

    websocket.username = user
    await handler(websocket)


async def query_param_auth(connection, request):