rjweb-server
Version:
Easy and Robust Way to create a Web Server with Many Easy-to-use Features in NodeJS
69 lines (68 loc) • 2.29 kB
JavaScript
import handleEvent from "../handleEvent";
function handleWSConnect(ws, message, ctg) {
const { custom, ctx } = ws.getUserData();
ctg.logger.debug("WebSocket message with bytelen", message.byteLength, "recieved");
ctx.response.content = [Buffer.allocUnsafe(0)];
ctx.body.type = "unknown";
ctx.body.raw = Buffer.from(message);
ctx.body.parsed = "";
ctx.continueSend = true;
ctx.executeSelf = () => true;
ctx.executeCode = true;
ctg.data.incoming.increase(ctx.body.raw.byteLength);
ctg.webSockets.messages.incoming.increase();
ctx.handleError = (err) => {
if (!err)
return;
ctx.error = err;
ctx.execute.event = "wsMessageError";
};
{
(async () => {
const ctr = new ctg.classContexts.wsMessage(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.wsMessageEvent)
continue;
try {
await Promise.resolve(middleware.data.wsMessageEvent(middleware.localContext, () => ctx.executeCode = false, ctr, ctx, ctg));
if (ctx.error)
throw ctx.error;
} catch (err) {
ctx.handleError(err);
break;
}
}
}
await handleEvent("wsMessage", 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.onMessage) {
try {
await Promise.resolve(ctx.execute.route.onMessage(ctr));
} catch (err) {
ctx.error = err;
ctx.execute.event = "wsMessageError";
await runPageLogic();
}
return resolve();
}
return resolve();
});
await runPageLogic();
})();
}
}
export {
handleWSConnect as default
};