-
-
Notifications
You must be signed in to change notification settings - Fork 1
/
SendHeadersMiddleware.js
47 lines (43 loc) · 1.37 KB
/
SendHeadersMiddleware.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
import { PassThrough } from 'stream';
/** @typedef {import('../types').IMiddleware} IMiddleware */
/** @typedef {import('../types').MiddlewareFunction} MiddlewareFunction */
/** @typedef {import('../types').MiddlewareFunctionParams} MiddlewareFunctionParams */
/** @typedef {import('../types').MiddlewareFunctionResult} MiddlewareFunctionResult */
/**
* @typedef {Object} SendHeadersMiddlewareOptions
* @prop {boolean} [setStatus=false]
* Automatically set `200` or `204` status if not set
*/
/** @implements {IMiddleware} */
export default class SendHeadersMiddleware {
/** @param {SendHeadersMiddlewareOptions} options */
constructor(options = {}) {
this.setStatus = options.setStatus === true;
}
/**
* @param {MiddlewareFunctionParams} params
* @return {MiddlewareFunctionResult}
*/
execute({ res }) {
const newWritable = new PassThrough();
const destination = res.replaceStream(newWritable);
newWritable.once('data', () => {
if (!res.headersSent) {
if (this.setStatus && res.status == null) {
res.status = 200;
}
res.sendHeaders(false);
}
});
newWritable.on('end', () => {
if (!res.headersSent) {
if (this.setStatus && res.status == null) {
res.status = 204;
}
res.sendHeaders(false);
}
});
newWritable.pipe(destination);
return 'continue';
}
}