iterate-stream.js 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116
  1. "use strict";
  2. function _awaitAsyncGenerator(value) { return new _AwaitValue(value); }
  3. function _wrapAsyncGenerator(fn) { return function () { return new _AsyncGenerator(fn.apply(this, arguments)); }; }
  4. function _AsyncGenerator(gen) { var front, back; function send(key, arg) { return new Promise(function (resolve, reject) { var request = { key: key, arg: arg, resolve: resolve, reject: reject, next: null }; if (back) { back = back.next = request; } else { front = back = request; resume(key, arg); } }); } function resume(key, arg) { try { var result = gen[key](arg); var value = result.value; var wrappedAwait = value instanceof _AwaitValue; Promise.resolve(wrappedAwait ? value.wrapped : value).then(function (arg) { if (wrappedAwait) { resume(key === "return" ? "return" : "next", arg); return; } settle(result.done ? "return" : "normal", arg); }, function (err) { resume("throw", err); }); } catch (err) { settle("throw", err); } } function settle(type, value) { switch (type) { case "return": front.resolve({ value: value, done: true }); break; case "throw": front.reject(value); break; default: front.resolve({ value: value, done: false }); break; } front = front.next; if (front) { resume(front.key, front.arg); } else { back = null; } } this._invoke = send; if (typeof gen.return !== "function") { this.return = undefined; } }
  5. if (typeof Symbol === "function" && Symbol.asyncIterator) { _AsyncGenerator.prototype[Symbol.asyncIterator] = function () { return this; }; }
  6. _AsyncGenerator.prototype.next = function (arg) { return this._invoke("next", arg); };
  7. _AsyncGenerator.prototype.throw = function (arg) { return this._invoke("throw", arg); };
  8. _AsyncGenerator.prototype.return = function (arg) { return this._invoke("return", arg); };
  9. function _AwaitValue(value) { this.wrapped = value; }
  10. module.exports = /*#__PURE__*/function () {
  11. var _iterateStream = _wrapAsyncGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee(stream) {
  12. var contents, resolveStreamEndedPromise, streamEndedPromise, ended, error, data;
  13. return regeneratorRuntime.wrap(function _callee$(_context) {
  14. while (1) {
  15. switch (_context.prev = _context.next) {
  16. case 0:
  17. contents = [];
  18. stream.on('data', function (data) {
  19. return contents.push(data);
  20. });
  21. streamEndedPromise = new Promise(function (resolve) {
  22. return resolveStreamEndedPromise = resolve;
  23. });
  24. ended = false;
  25. stream.on('end', function () {
  26. ended = true;
  27. resolveStreamEndedPromise();
  28. });
  29. error = false;
  30. stream.on('error', function (err) {
  31. error = err;
  32. resolveStreamEndedPromise();
  33. });
  34. case 7:
  35. if (!(!ended || contents.length > 0)) {
  36. _context.next = 22;
  37. break;
  38. }
  39. if (!(contents.length === 0)) {
  40. _context.next = 14;
  41. break;
  42. }
  43. stream.resume(); // eslint-disable-next-line no-await-in-loop
  44. _context.next = 12;
  45. return _awaitAsyncGenerator(Promise.race([once(stream, 'data'), streamEndedPromise]));
  46. case 12:
  47. _context.next = 18;
  48. break;
  49. case 14:
  50. stream.pause();
  51. data = contents.shift();
  52. _context.next = 18;
  53. return data;
  54. case 18:
  55. if (!error) {
  56. _context.next = 20;
  57. break;
  58. }
  59. throw error;
  60. case 20:
  61. _context.next = 7;
  62. break;
  63. case 22:
  64. resolveStreamEndedPromise();
  65. case 23:
  66. case "end":
  67. return _context.stop();
  68. }
  69. }
  70. }, _callee);
  71. }));
  72. function iterateStream(_x) {
  73. return _iterateStream.apply(this, arguments);
  74. }
  75. return iterateStream;
  76. }();
  77. function once(eventEmitter, type) {
  78. // TODO: Use require('events').once when node v10 is dropped
  79. return new Promise(function (resolve) {
  80. var fired = false;
  81. var handler = function handler() {
  82. if (!fired) {
  83. fired = true;
  84. eventEmitter.removeListener(type, handler);
  85. resolve();
  86. }
  87. };
  88. eventEmitter.addListener(type, handler);
  89. });
  90. }
  91. //# sourceMappingURL=iterate-stream.js.map