23 lines
624 B
JavaScript
23 lines
624 B
JavaScript
"use strict";
|
|
|
|
Object.defineProperty(exports, "__esModule", {
|
|
value: true
|
|
});
|
|
exports.default = wrapMiddleware;
|
|
function wrapMiddleware(middleware) {
|
|
return function (req, res, next) {
|
|
if (req.ws !== null && req.ws !== undefined) {
|
|
req.wsHandled = true;
|
|
try {
|
|
/* Unpack the `.ws` property and call the actual handler. */
|
|
middleware(req.ws, req, next);
|
|
} catch (err) {
|
|
/* If an error is thrown, let's send that on to any error handling */
|
|
next(err);
|
|
}
|
|
} else {
|
|
/* This wasn't a WebSocket request, so skip this middleware. */
|
|
next();
|
|
}
|
|
};
|
|
} |