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
59
|
'use strict';
var Promise = require('pinkie-promise');
var objectAssign = require('object-assign');
var bufferStream = require('./buffer-stream');
function getStream(inputStream, opts) {
if (!inputStream) {
return Promise.reject(new Error('Expected a stream'));
}
opts = objectAssign({maxBuffer: Infinity}, opts);
var maxBuffer = opts.maxBuffer;
var stream;
var clean;
var p = new Promise(function (resolve, reject) {
stream = bufferStream(opts);
inputStream.once('error', error);
inputStream.pipe(stream);
stream.on('data', function () {
if (stream.getBufferedLength() > maxBuffer) {
reject(new Error('maxBuffer exceeded'));
}
});
stream.once('error', error);
stream.on('end', resolve);
clean = function () {
// some streams doesn't implement the stream.Readable interface correctly
if (inputStream.unpipe) {
inputStream.unpipe(stream);
}
};
function error(err) {
if (err) { // null check
err.bufferedData = stream.getBufferedValue();
}
reject(err);
}
});
p.then(clean, clean);
return p.then(function () {
return stream.getBufferedValue();
});
}
module.exports = getStream;
module.exports.buffer = function (stream, opts) {
return getStream(stream, objectAssign({}, opts, {encoding: 'buffer'}));
};
module.exports.array = function (stream, opts) {
return getStream(stream, objectAssign({}, opts, {array: true}));
};
|
