2 var Promise = require('pinkie-promise');
3 var objectAssign = require('object-assign');
4 var bufferStream = require('./buffer-stream');
6 function getStream(inputStream, opts) {
8 return Promise.reject(new Error('Expected a stream'));
11 opts = objectAssign({maxBuffer: Infinity}, opts);
12 var maxBuffer = opts.maxBuffer;
16 var p = new Promise(function (resolve, reject) {
17 stream = bufferStream(opts);
18 inputStream.once('error', error);
19 inputStream.pipe(stream);
21 stream.on('data', function () {
22 if (stream.getBufferedLength() > maxBuffer) {
23 reject(new Error('maxBuffer exceeded'));
26 stream.once('error', error);
27 stream.on('end', resolve);
30 // some streams doesn't implement the stream.Readable interface correctly
31 if (inputStream.unpipe) {
32 inputStream.unpipe(stream);
37 if (err) { // null check
38 err.bufferedData = stream.getBufferedValue();
46 return p.then(function () {
47 return stream.getBufferedValue();
51 module.exports = getStream;
53 module.exports.buffer = function (stream, opts) {
54 return getStream(stream, objectAssign({}, opts, {encoding: 'buffer'}));
57 module.exports.array = function (stream, opts) {
58 return getStream(stream, objectAssign({}, opts, {array: true}));