67 lines
2.2 KiB
TypeScript
67 lines
2.2 KiB
TypeScript
import { createAdaptorServer, type ServerType } from "@hono/node-server"
|
|
import { createNodeWebSocket } from "@hono/node-ws"
|
|
import type { Hono } from "hono"
|
|
import type { Adapter } from "./adapter"
|
|
|
|
export const adapter: Adapter = {
|
|
create(app: Hono) {
|
|
const ws = createNodeWebSocket({ app })
|
|
return {
|
|
upgradeWebSocket: ws.upgradeWebSocket,
|
|
async listen(opts) {
|
|
const start = (port: number) =>
|
|
new Promise<ServerType>((resolve, reject) => {
|
|
const server = createAdaptorServer({ fetch: app.fetch })
|
|
ws.injectWebSocket(server)
|
|
const fail = (err: Error) => {
|
|
cleanup()
|
|
reject(err)
|
|
}
|
|
const ready = () => {
|
|
cleanup()
|
|
resolve(server)
|
|
}
|
|
const cleanup = () => {
|
|
server.off("error", fail)
|
|
server.off("listening", ready)
|
|
}
|
|
server.once("error", fail)
|
|
server.once("listening", ready)
|
|
server.listen(port, opts.hostname)
|
|
})
|
|
|
|
const server = opts.port === 0 ? await start(4096).catch(() => start(0)) : await start(opts.port)
|
|
const addr = server.address()
|
|
if (!addr || typeof addr === "string") {
|
|
throw new Error(`Failed to resolve server address for port ${opts.port}`)
|
|
}
|
|
|
|
let closing: Promise<void> | undefined
|
|
return {
|
|
port: addr.port,
|
|
stop(close?: boolean) {
|
|
closing ??= new Promise((resolve, reject) => {
|
|
server.close((err) => {
|
|
if (err) {
|
|
reject(err)
|
|
return
|
|
}
|
|
resolve()
|
|
})
|
|
if (close) {
|
|
if ("closeAllConnections" in server && typeof server.closeAllConnections === "function") {
|
|
server.closeAllConnections()
|
|
}
|
|
if ("closeIdleConnections" in server && typeof server.closeIdleConnections === "function") {
|
|
server.closeIdleConnections()
|
|
}
|
|
}
|
|
})
|
|
return closing
|
|
},
|
|
}
|
|
},
|
|
}
|
|
},
|
|
}
|