rjweb-server
Version:
Easy and Robust Way to create a Web Server with Many Easy-to-use Features in NodeJS
64 lines (63 loc) • 2.05 kB
JavaScript
import handleEvent from "../handleEvent";
function handleWSOpen(ws, ctg) {
const { custom, ctx } = ws.getUserData();
ctg.logger.debug("WebSocket connection established");
ctx.response.content = [Buffer.allocUnsafe(0)];
ctx.continueSend = true;
ctx.executeSelf = () => true;
ctx.executeCode = true;
ctx.handleError = (err) => {
if (!err)
return;
ctx.error = err;
ctx.execute.event = "wsConnectError";
};
{
(async () => {
const ctr = new ctg.classContexts.wsConnect(ctg.controller, ctx, ws);
ctr["@"] = custom;
if (ctg.middlewares.length > 0 && !ctx.error) {
for (let middlewareIndex = 0; middlewareIndex < ctg.middlewares.length; middlewareIndex++) {
const middleware = ctg.middlewares[middlewareIndex];
if (!middleware.data.wsConnectEvent)
continue;
try {
await Promise.resolve(middleware.data.wsConnectEvent(middleware.localContext, () => ctx.executeCode = false, ctr, ctx, ctg));
if (ctx.error)
throw ctx.error;
} catch (err) {
ctx.handleError(err);
break;
}
}
}
await handleEvent("wsConnect", ctr, ctx, ctg);
const runPageLogic = (eventOnly) => new Promise(async (resolve) => {
if (ctx.execute.event !== "none") {
await handleEvent(ctx.execute.event, ctr, ctx, ctg);
return resolve();
}
;
if (eventOnly)
return resolve();
if (!ctx.execute.route)
return;
if (ctx.executeCode && ctx.execute.route.type === "websocket" && ctx.execute.route.onConnect) {
try {
await Promise.resolve(ctx.execute.route.onConnect(ctr));
} catch (err) {
ctx.error = err;
ctx.execute.event = "wsConnectError";
await runPageLogic();
}
return resolve();
}
return resolve();
});
await runPageLogic();
})();
}
}
export {
handleWSOpen as default
};