Function serveRuntimeNodeHttp

projects/runtime-node/src/server.ts:136–202

(options: RuntimeNodeHttpServerOptions = {})

}

export async function serveRuntimeNodeHttp(options: RuntimeNodeHttpServerOptions = {}): Promise<RuntimeNodeHttpServer> {
    const host = options.host ?? "127.0.0.1"
    const port = options.port ?? 0
    const path = options.path ?? "/v1/runtime"
    const runtime =
        options.runtime ??
        createRuntimeNodeServer({
            serverName: "runtime-node",
            ...options.runtimeOptions,
        })
    const httpServer = http.createServer((_request, response) => {
        response.writeHead(404)
        response.end("not found")
    })
    const socketServer = new WebSocketServer({ noServer: true })
    const httpSockets = new Set<Socket>()

    httpServer.on("connection", (socket) => {
        httpSockets.add(socket)
        socket.once("close", () => httpSockets.delete(socket))
    })

    httpServer.on("upgrade", (request, socket, head) => {
        const url = new URL(request.url ?? "/", `http://${host}`)
        if (url.pathname !== path) {
            socket.destroy()
            return
        }

        if (!isAuthorized(request, { host, token: options.token, allowUnauthenticatedLoopback: options.allowUnauthenticatedLoopback ?? true })) {
            socket.write("HTTP/1.1 401 Unauthorized\r\nConnection: close\r\n\r\n")
            socket.destroy()
            return
        }

        socketServer.handleUpgrade(request, socket, head, (webSocket) => {
            attachSocket(runtime, webSocket, request, options)
        })
    })

    await new Promise<void>((resolve, reject) => {
        httpServer.once("error", reject)
        httpServer.listen(port, host, () => {
            httpServer.off("error", reject)
            resolve()
        })
    })

    const address = httpServer.address()
    const actualPort = typeof address === "object" && address ? address.port : port
    return {
        runtime,
        httpServer,
        url: `ws://${host}:${actualPort}${path}`,
        async close() {
            for (const client of socketServer.clients) client.terminate()
            for (const socket of httpSockets) socket.destroy()
            await new Promise<void>((resolve) => socketServer.close(() => resolve()))