1 | 'use strict';
|
2 |
|
3 | var $ = require('../internals/export');
|
4 | var aFunction = require('../internals/a-function');
|
5 | var anObject = require('../internals/an-object');
|
6 | var isObject = require('../internals/is-object');
|
7 | var createAsyncIteratorProxy = require('../internals/create-async-iterator-proxy');
|
8 | var getAsyncIteratorMethod = require('../internals/get-async-iterator-method');
|
9 |
|
10 | var AsyncIteratorProxy = createAsyncIteratorProxy(function (arg, Promise) {
|
11 | var state = this;
|
12 | var mapper = state.mapper;
|
13 | var innerIterator, iteratorMethod;
|
14 |
|
15 | return new Promise(function (resolve, reject) {
|
16 | var outerLoop = function () {
|
17 | try {
|
18 | Promise.resolve(anObject(state.next.call(state.iterator, arg))).then(function (step) {
|
19 | try {
|
20 | if (anObject(step).done) {
|
21 | state.done = true;
|
22 | resolve({ done: true, value: undefined });
|
23 | } else {
|
24 | Promise.resolve(mapper(step.value)).then(function (mapped) {
|
25 | try {
|
26 | if (isObject(mapped) && (iteratorMethod = getAsyncIteratorMethod(mapped)) !== undefined) {
|
27 | state.innerIterator = innerIterator = iteratorMethod.call(mapped);
|
28 | state.innerNext = aFunction(innerIterator.next);
|
29 | return innerLoop();
|
30 | } resolve({ done: false, value: mapped });
|
31 | } catch (error2) { reject(error2); }
|
32 | }, reject);
|
33 | }
|
34 | } catch (error1) { reject(error1); }
|
35 | }, reject);
|
36 | } catch (error) { reject(error); }
|
37 | };
|
38 |
|
39 | var innerLoop = function () {
|
40 | if (innerIterator = state.innerIterator) {
|
41 | try {
|
42 | Promise.resolve(anObject(state.innerNext.call(innerIterator))).then(function (result) {
|
43 | try {
|
44 | if (anObject(result).done) {
|
45 | state.innerIterator = state.innerNext = null;
|
46 | outerLoop();
|
47 | } else resolve({ done: false, value: result.value });
|
48 | } catch (error1) { reject(error1); }
|
49 | }, reject);
|
50 | } catch (error) { reject(error); }
|
51 | } else outerLoop();
|
52 | };
|
53 |
|
54 | innerLoop();
|
55 | });
|
56 | });
|
57 |
|
58 | $({ target: 'AsyncIterator', proto: true, real: true }, {
|
59 | flatMap: function flatMap(mapper) {
|
60 | return new AsyncIteratorProxy({
|
61 | iterator: anObject(this),
|
62 | mapper: aFunction(mapper),
|
63 | innerIterator: null,
|
64 | innerNext: null
|
65 | });
|
66 | }
|
67 | });
|