rjweb-server
Version:
Easy and Robust Way to create a Web Server with Many Easy-to-use Features in NodeJS
70 lines (69 loc) • 2.22 kB
JavaScript
import handleEvent from "../handleEvent";
function handleWSClose(ws, message, ctg) {
const { custom, ctx } = ws.getUserData();
ctg.logger.debug("WebSocket connection closed");
ctx.body.type = "unknown";
ctx.body.raw = Buffer.from(message);
ctx.body.parsed = "";
ctx.executeSelf = () => true;
ctx.executeCode = true;
ctx.events.send("requestAborted");
ctg.data.incoming.increase();
ctx.handleError = (err) => {
if (!err)
return;
ctx.error = err;
ctx.execute.event = "wsCloseError";
};
for (const { ref, refListener } of ctx.refListeners) {
ref["removeOnChange"](refListener);
}
{
(async () => {
const ctr = new ctg.classContexts.wsClose(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.wsCloseEvent)
continue;
try {
await Promise.resolve(middleware.data.wsCloseEvent(middleware.localContext, () => ctx.executeCode = false, ctr, ctx, ctg));
if (ctx.error)
throw ctx.error;
} catch (err) {
ctx.handleError(err);
break;
}
}
}
await handleEvent("wsClose", 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.onClose) {
try {
await Promise.resolve(ctx.execute.route.onClose(ctr));
} catch (err) {
ctx.error = err;
ctx.execute.event = "wsCloseError";
await runPageLogic();
}
return resolve();
}
return resolve();
});
await runPageLogic();
})();
}
}
export {
handleWSClose as default
};