async-iterator-create-proxy.js 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970
  1. 'use strict';
  2. var path = require('../internals/path');
  3. var aFunction = require('../internals/a-function');
  4. var anObject = require('../internals/an-object');
  5. var create = require('../internals/object-create');
  6. var createNonEnumerableProperty = require('../internals/create-non-enumerable-property');
  7. var redefineAll = require('../internals/redefine-all');
  8. var wellKnownSymbol = require('../internals/well-known-symbol');
  9. var InternalStateModule = require('../internals/internal-state');
  10. var getBuiltIn = require('../internals/get-built-in');
  11. var Promise = getBuiltIn('Promise');
  12. var setInternalState = InternalStateModule.set;
  13. var getInternalState = InternalStateModule.get;
  14. var TO_STRING_TAG = wellKnownSymbol('toStringTag');
  15. module.exports = function (nextHandler, IS_ITERATOR) {
  16. var AsyncIteratorProxy = function AsyncIterator(state) {
  17. state.next = aFunction(state.iterator.next);
  18. state.done = false;
  19. state.ignoreArgument = !IS_ITERATOR;
  20. setInternalState(this, state);
  21. };
  22. AsyncIteratorProxy.prototype = redefineAll(create(path.AsyncIterator.prototype), {
  23. next: function next(arg) {
  24. var that = this;
  25. var hasArgument = !!arguments.length;
  26. return new Promise(function (resolve) {
  27. var state = getInternalState(that);
  28. var args = hasArgument ? [state.ignoreArgument ? undefined : arg] : IS_ITERATOR ? [] : [undefined];
  29. state.ignoreArgument = false;
  30. resolve(state.done ? { done: true, value: undefined } : anObject(nextHandler.call(state, Promise, args)));
  31. });
  32. },
  33. 'return': function (value) {
  34. var that = this;
  35. return new Promise(function (resolve, reject) {
  36. var state = getInternalState(that);
  37. var iterator = state.iterator;
  38. state.done = true;
  39. var $$return = iterator['return'];
  40. if ($$return === undefined) return resolve({ done: true, value: value });
  41. Promise.resolve($$return.call(iterator, value)).then(function (result) {
  42. anObject(result);
  43. resolve({ done: true, value: value });
  44. }, reject);
  45. });
  46. },
  47. 'throw': function (value) {
  48. var that = this;
  49. return new Promise(function (resolve, reject) {
  50. var state = getInternalState(that);
  51. var iterator = state.iterator;
  52. state.done = true;
  53. var $$throw = iterator['throw'];
  54. if ($$throw === undefined) return reject(value);
  55. resolve($$throw.call(iterator, value));
  56. });
  57. }
  58. });
  59. if (!IS_ITERATOR) {
  60. createNonEnumerableProperty(AsyncIteratorProxy.prototype, TO_STRING_TAG, 'Generator');
  61. }
  62. return AsyncIteratorProxy;
  63. };