fix(send-type): apply custom headers when passed a stream (#198)

* Fixes #197: apply custom headers when passed a stream * Use setHeader instead of writeHead to fix Chrome issue * Use for…in loop as suggested by Luke I don’t have a preference either way as long as all the headers are sent ;) Co-authored-by: Luke Edwards <[email protected]> --------- Co-authored-by: Luke Edwards <[email protected]>
lukeed · Dec 4, 2023 · 1cf649c · 1cf649c
1 parent 2ce10df
commit 1cf649c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/packages/send-type/index.js b/packages/send-type/index.js
@@ -12,7 +12,10 @@ module.exports = function (res, code=200, data='', headers={}) {
 	let type = obj[TYPE] || res.getHeader(TYPE);
 
 	if (!!data && typeof data.pipe === 'function') {
-		res.setHeader(TYPE, type || OSTREAM);
+		obj[TYPE] = type || OSTREAM;
+		for (k in obj) {
+		  res.setHeader(k, obj[k]);
+		}
 		return data.pipe(res);
 	}