fix: make res.setHeader, res.removeHeader, res.writeHead, res.end, res.write methods return values

This commit is contained in:
Dorian Niemiec 2025-01-02 22:30:53 +01:00
parent f09a174dca
commit a5f57be4df

View file

@ -84,12 +84,12 @@ module.exports = function (req, res, logFacilities, config, next) {
res.setHeader = function (name, value) { res.setHeader = function (name, value) {
writtenHeaders[name.toLowerCase()] = value; writtenHeaders[name.toLowerCase()] = value;
originalSetHeader(name, value); return originalSetHeader(name, value);
}; };
res.removeHeader = function (name) { res.removeHeader = function (name) {
delete writtenHeaders[name.toLowerCase()]; delete writtenHeaders[name.toLowerCase()];
originalRemoveHeader(name); return originalRemoveHeader(name);
}; };
res.writeHead = function (statusCode, statusCodeDescription, headers) { res.writeHead = function (statusCode, statusCodeDescription, headers) {
@ -102,13 +102,13 @@ module.exports = function (req, res, logFacilities, config, next) {
writtenStatusCode = statusCode; writtenStatusCode = statusCode;
originalSetHeader("X-SVRJS-Cache", "MISS"); originalSetHeader("X-SVRJS-Cache", "MISS");
if (headers || typeof statusCodeDescription !== "object") { if (headers || typeof statusCodeDescription !== "object") {
originalWriteHead( return originalWriteHead(
writtenStatusCode, writtenStatusCode,
statusCodeDescription, statusCodeDescription,
writtenHeaders writtenHeaders
); );
} else { } else {
originalWriteHead(writtenStatusCode, writtenHeaders); return originalWriteHead(writtenStatusCode, writtenHeaders);
} }
}; };
@ -193,7 +193,7 @@ module.exports = function (req, res, logFacilities, config, next) {
} }
} }
originalEnd(chunk, encoding, callback); return originalEnd(chunk, encoding, callback);
}; };
if (req.method != "HEAD") { if (req.method != "HEAD") {
@ -220,7 +220,7 @@ module.exports = function (req, res, logFacilities, config, next) {
} }
} }
originalWrite(chunk, encoding, callback); return originalWrite(chunk, encoding, callback);
}; };
res.on("pipe", (src) => { res.on("pipe", (src) => {