heroku-debug/tmp/node_modules/get-stream/index.js

debugging plugin for the CLI

'use strict';

module.exports = function (stream, opts) {
	if (!stream) {
		return Promise.reject(new Error('Expected a stream'));
	}

	opts = opts || {};

	var ret = '';

	return new Promise(function (resolve, reject) {
		stream.setEncoding(opts.encoding || 'utf8');

		stream.on('readable', function () {
			var chunk;

			while ((chunk = stream.read())) {
				ret += chunk;
			}
		});

		stream.on('error', reject);

		stream.on('end', function () {
			resolve(ret);
		});
	});
};

module.exports.buffer = function (stream) {
	if (!stream) {
		return Promise.reject(new Error('Expected a stream'));
	}

	var ret = [];
	var len = 0;

	return new Promise(function (resolve, reject) {
		stream.on('readable', function () {
			var chunk;

			while ((chunk = stream.read())) {
				ret.push(chunk);
				len += chunk.length;
			}
		});

		stream.on('error', reject);

		stream.on('end', function () {
			resolve(Buffer.concat(ret, len));
		});
	});
};