12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455 |
- 'use strict';
- var $ = require('../internals/export');
- var call = require('../internals/function-call');
- var aCallable = require('../internals/a-callable');
- var anObject = require('../internals/an-object');
- var getIteratorDirect = require('../internals/get-iterator-direct');
- var getIteratorFlattenable = require('../internals/get-iterator-flattenable');
- var createIteratorProxy = require('../internals/iterator-create-proxy');
- var iteratorClose = require('../internals/iterator-close');
- var IS_PURE = require('../internals/is-pure');
- var iteratorHelperWithoutClosingOnEarlyError = require('../internals/iterator-helper-without-closing-on-early-error');
- var flatMapWithoutClosingOnEarlyError = !IS_PURE && iteratorHelperWithoutClosingOnEarlyError('flatMap', TypeError);
- var IteratorProxy = createIteratorProxy(function () {
- var iterator = this.iterator;
- var mapper = this.mapper;
- var result, inner;
- while (true) {
- if (inner = this.inner) try {
- result = anObject(call(inner.next, inner.iterator));
- if (!result.done) return result.value;
- this.inner = null;
- } catch (error) { iteratorClose(iterator, 'throw', error); }
- result = anObject(call(this.next, iterator));
- if (this.done = !!result.done) return;
- try {
- this.inner = getIteratorFlattenable(mapper(result.value, this.counter++), false);
- } catch (error) { iteratorClose(iterator, 'throw', error); }
- }
- });
- // `Iterator.prototype.flatMap` method
- // https://tc39.es/ecma262/#sec-iterator.prototype.flatmap
- $({ target: 'Iterator', proto: true, real: true, forced: IS_PURE || flatMapWithoutClosingOnEarlyError }, {
- flatMap: function flatMap(mapper) {
- anObject(this);
- try {
- aCallable(mapper);
- } catch (error) {
- iteratorClose(this, 'throw', error);
- }
- if (flatMapWithoutClosingOnEarlyError) return call(flatMapWithoutClosingOnEarlyError, this, mapper);
- return new IteratorProxy(getIteratorDirect(this), {
- mapper: mapper,
- inner: null
- });
- }
- });
|