iterator-create-proxy.js 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. 'use strict';
  2. var call = require('../internals/function-call');
  3. var anObject = require('../internals/an-object');
  4. var create = require('../internals/object-create');
  5. var createNonEnumerableProperty = require('../internals/create-non-enumerable-property');
  6. var defineBuiltIns = require('../internals/define-built-ins');
  7. var wellKnownSymbol = require('../internals/well-known-symbol');
  8. var InternalStateModule = require('../internals/internal-state');
  9. var getMethod = require('../internals/get-method');
  10. var IteratorPrototype = require('../internals/iterators-core').IteratorPrototype;
  11. var ITERATOR_PROXY = 'IteratorProxy';
  12. var setInternalState = InternalStateModule.set;
  13. var getInternalState = InternalStateModule.getterFor(ITERATOR_PROXY);
  14. var TO_STRING_TAG = wellKnownSymbol('toStringTag');
  15. module.exports = function (nextHandler, IS_ITERATOR) {
  16. var IteratorProxy = function Iterator(record, state) {
  17. if (state) {
  18. state.iterator = record.iterator;
  19. state.next = record.next;
  20. } else state = record;
  21. state.type = ITERATOR_PROXY;
  22. state.done = false;
  23. state.ignoreArg = !IS_ITERATOR;
  24. setInternalState(this, state);
  25. };
  26. IteratorProxy.prototype = defineBuiltIns(create(IteratorPrototype), {
  27. next: function next(arg) {
  28. var state = getInternalState(this);
  29. var args = arguments.length ? [state.ignoreArg ? undefined : arg] : IS_ITERATOR ? [] : [undefined];
  30. state.ignoreArg = false;
  31. var result = state.done ? undefined : call(nextHandler, state, args);
  32. return { done: state.done, value: result };
  33. },
  34. 'return': function (value) {
  35. var state = getInternalState(this);
  36. var iterator = state.iterator;
  37. state.done = true;
  38. var $$return = getMethod(iterator, 'return');
  39. return { done: true, value: $$return ? anObject(call($$return, iterator, value)).value : value };
  40. },
  41. 'throw': function (value) {
  42. var state = getInternalState(this);
  43. var iterator = state.iterator;
  44. state.done = true;
  45. var $$throw = getMethod(iterator, 'throw');
  46. if ($$throw) return call($$throw, iterator, value);
  47. throw value;
  48. }
  49. });
  50. if (!IS_ITERATOR) {
  51. createNonEnumerableProperty(IteratorProxy.prototype, TO_STRING_TAG, 'Generator');
  52. }
  53. return IteratorProxy;
  54. };