tezx/ws/index.js

TezX is a modern, ultra-lightweight, and high-performance JavaScript framework built specifically for Bun. It provides a minimal yet powerful API, seamless environment management, and a high-concurrency HTTP engine for building fast, scalable web applicat

export function upgradeWebSocket(callback, options = {}) {
    const { onUpgradeError = (error, ctx) => {
        return ctx.status(401).text(error.message);
    }, } = options;
    return async (ctx, next) => {
        const upgrade = ctx.req.header("upgrade")?.toLowerCase();
        const connection = ctx.req.header("connection")?.toLowerCase();
        const key = ctx.req.header("sec-websocket-key");
        if (upgrade !== "websocket" || !connection?.includes("upgrade") || !key) {
            if (next) {
                ctx.body = { error: "401 Bad Request: Invalid WebSocket headers" };
                return next();
            }
            ctx.status(401);
            return onUpgradeError(new Error("401 Bad Request: Invalid WebSocket headers"), ctx);
        }
        ctx.wsProtocol = ctx.url?.startsWith("https") ? "wss" : "ws";
        if (!callback) {
            throw new Error("WebSocket callback is missing. Please provide a valid callback function to handle the WebSocket events.");
        }
        let server = ctx.server;
        if (!server?.upgrade) {
            return onUpgradeError(new Error("Bun server instance missing for WebSocket"), ctx);
        }
        const success = server.upgrade(ctx.rawRequest, {
            data: callback(ctx),
        });
        if (success)
            return undefined;
        return next();
    };
}
export const wsHandlers = (options) => {
    return {
        open(ws) {
            return ws.data?.open?.(ws);
        },
        message(ws, msg) {
            return ws.data?.message?.(ws, msg);
        },
        close(ws, code, reason) {
            return ws.data?.close?.(ws, { code, reason });
        },
        ping(ws, data) {
            return ws.data?.ping?.(ws, data);
        },
        pong(ws, data) {
            return ws.data?.pong?.(ws, data);
        },
        drain(ws) {
            return ws.data?.drain?.(ws);
        },
        ...options,
    };
};
export default upgradeWebSocket;