joannangx/visionaria_app

'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}));
};