TanTan
2a69b23b47
Dev Build / cleanup (push) Canceled after 0s
Dev Build / typecheck (push) Canceled after 0s
Dev Build / build (linux-arm64, ubuntu-24.04-arm, linux-arm64) (push) Canceled after 0s
Dev Build / build (linux-x64, ubuntu-latest, linux-x64) (push) Canceled after 0s
Dev Build / build (win-x64, windows-latest, win32-x64) (push) Canceled after 0s
Dev Build / dispatch-dev-docker (push) Canceled after 0s
67 lines
2.1 KiB
Plaintext
67 lines
2.1 KiB
Plaintext
import { getBunServer } from "./server.js";
|
|
import { WSContext, createWSMessageEvent, defineWebSocketHelper } from "../../helper/websocket/index.js";
|
|
//#region src/adapter/bun/websocket.ts
|
|
/**
|
|
* @internal
|
|
* @deprecated `hono/bun` will be removed in v5. Install `@hono/bun` and import from there instead.
|
|
*/
|
|
const createWSContext = (ws) => {
|
|
return new WSContext({
|
|
send: (source, options) => {
|
|
ws.send(source, options?.compress);
|
|
},
|
|
raw: ws,
|
|
readyState: ws.readyState,
|
|
url: ws.data.url,
|
|
protocol: ws.data.protocol,
|
|
close(code, reason) {
|
|
ws.close(code, reason);
|
|
}
|
|
});
|
|
};
|
|
/**
|
|
* @deprecated `hono/bun` will be removed in v5. Install `@hono/bun` and import from there instead.
|
|
*/
|
|
const upgradeWebSocket = defineWebSocketHelper((c, events) => {
|
|
const server = getBunServer(c);
|
|
if (!server) throw new TypeError("env has to include the 2nd argument of fetch.");
|
|
if (server.upgrade(c.req.raw, { data: {
|
|
events,
|
|
url: new URL(c.req.url),
|
|
protocol: c.req.header("sec-websocket-protocol")?.split(",")[0]?.trim() ?? ""
|
|
} })) return new Response(null);
|
|
});
|
|
/**
|
|
* @deprecated `hono/bun` will be removed in v5. Install `@hono/bun` and import from there instead.
|
|
*/
|
|
const websocket = {
|
|
open(ws) {
|
|
const websocketListeners = ws.data.events;
|
|
if (websocketListeners.onOpen) websocketListeners.onOpen(new Event("open"), createWSContext(ws));
|
|
},
|
|
close(ws, code, reason) {
|
|
const websocketListeners = ws.data.events;
|
|
if (websocketListeners.onClose) websocketListeners.onClose(new CloseEvent("close", {
|
|
code,
|
|
reason
|
|
}), createWSContext(ws));
|
|
},
|
|
message(ws, message) {
|
|
const websocketListeners = ws.data.events;
|
|
if (websocketListeners.onMessage) {
|
|
const normalizedReceiveData = typeof message === "string" ? message : message.buffer;
|
|
websocketListeners.onMessage(createWSMessageEvent(normalizedReceiveData), createWSContext(ws));
|
|
}
|
|
}
|
|
};
|
|
/**
|
|
* @deprecated Import `upgradeWebSocket` and `websocket` directly from `hono/bun` instead.
|
|
* @returns A function to create a Bun WebSocket handler.
|
|
*/
|
|
const createBunWebSocket = () => ({
|
|
upgradeWebSocket,
|
|
websocket
|
|
});
|
|
//#endregion
|
|
export { createBunWebSocket, createWSContext, upgradeWebSocket, websocket };
|