15815213711
2024-08-26 67b8b6731811983447e053d4396b3708c14dfe3c
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
48
49
50
51
52
53
54
55
56
57
58
'use strict';
const {Transform, PassThrough} = require('stream');
const zlib = require('zlib');
const mimicResponse = require('mimic-response');
 
module.exports = response => {
    const contentEncoding = (response.headers['content-encoding'] || '').toLowerCase();
 
    if (!['gzip', 'deflate', 'br'].includes(contentEncoding)) {
        return response;
    }
 
    // TODO: Remove this when targeting Node.js 12.
    const isBrotli = contentEncoding === 'br';
    if (isBrotli && typeof zlib.createBrotliDecompress !== 'function') {
        response.destroy(new Error('Brotli is not supported on Node.js < 12'));
        return response;
    }
 
    let isEmpty = true;
 
    const checker = new Transform({
        transform(data, _encoding, callback) {
            isEmpty = false;
 
            callback(null, data);
        },
 
        flush(callback) {
            callback();
        }
    });
 
    const finalStream = new PassThrough({
        autoDestroy: false,
        destroy(error, callback) {
            response.destroy();
 
            callback(error);
        }
    });
 
    const decompressStream = isBrotli ? zlib.createBrotliDecompress() : zlib.createUnzip();
 
    decompressStream.once('error', error => {
        if (isEmpty && !response.readable) {
            finalStream.end();
            return;
        }
 
        finalStream.destroy(error);
    });
 
    mimicResponse(response, finalStream);
    response.pipe(checker).pipe(decompressStream).pipe(finalStream);
 
    return finalStream;
};