You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
776 B
JavaScript
32 lines
776 B
JavaScript
5 years ago
|
"use strict";
|
||
|
|
||
|
const Promise = require("bluebird");
|
||
|
const through2 = require("through2");
|
||
|
const assureArray = require("assure-array");
|
||
|
|
||
|
function wrapStreamHandler(stream, handler, callback, item) {
|
||
|
return Promise.try(() => {
|
||
|
return handler(item);
|
||
|
}).then((result) => {
|
||
|
return assureArray(result);
|
||
|
}).each((newItem) => {
|
||
|
stream.push(newItem);
|
||
|
}).then(() => {
|
||
|
callback();
|
||
|
}).catch((err) => {
|
||
|
callback(err);
|
||
|
});
|
||
|
}
|
||
|
|
||
|
module.exports = function stream(handler, flushHandler) {
|
||
|
let flushHandlerWrapper = (flushHandler == null)
|
||
|
? undefined
|
||
|
: function (callback) {
|
||
|
wrapStreamHandler(this, flushHandler, callback);
|
||
|
};
|
||
|
|
||
|
return through2.obj(function (item, _encoding, callback) {
|
||
|
wrapStreamHandler(this, handler, callback, item);
|
||
|
}, flushHandlerWrapper);
|
||
|
};
|