tezx/ws/index.js

TezX is a high-performance, lightweight JavaScript framework designed for speed, scalability, and flexibility. It enables efficient routing, middleware management, and static file serving with minimal configuration. Fully compatible with Node.js, Deno, an

import { Environment } from "../core/environment.js";
import { DenoTransport } from "./deno.js";
import { NodeTransport } from "./node.js";
export function upgradeWebSocket(callback, options = {}) {
    const { onUpgradeError = (error, ctx) => {
        ctx.setStatus = 401;
        return ctx.text(error.message);
    }, protocol, idleTimeout = 30000, perMessageDeflate, maxPayload = 1048576, } = options;
    return async (ctx, next) => {
        const upgrade = ctx.req.headers.get("upgrade")?.toLowerCase();
        const connection = ctx.req.headers.get("connection")?.toLowerCase();
        const key = ctx.req.headers.get("sec-websocket-key");
        if (upgrade !== "websocket" || !connection?.includes("upgrade") || !key) {
            if (next) {
                ctx.body = { error: "401 Bad Request: Invalid WebSocket headers" };
                return next();
            }
            ctx.setStatus = 401;
            return onUpgradeError(new Error("401 Bad Request: Invalid WebSocket headers"), ctx);
        }
        ctx.wsProtocol = ctx.req.urlRef.protocol === "https:" ? "wss" : "ws";
        try {
            const env = Environment.getEnvironment;
            if (!callback) {
                throw new Error("WebSocket callback is missing. Please provide a valid callback function to handle the WebSocket events.");
            }
            let websocketCallback = callback(ctx);
            switch (env) {
                case "deno":
                    return new DenoTransport().upgrade(ctx, websocketCallback, {
                        idleTimeout,
                        protocol,
                    });
                case "bun":
                    return callback;
                case "node":
                    return new NodeTransport().upgrade(ctx, websocketCallback, {
                        maxPayload,
                        perMessageDeflate,
                    });
                default:
                    throw new Error("Unsupported environment for WebSocket upgrade.");
            }
        }
        catch (err) {
            return onUpgradeError(new Error("WebSocket Upgrade Failed"), ctx);
        }
    };
}
export default upgradeWebSocket;