diff options
Diffstat (limited to 'src/node_modules/es-abstract/helpers/getIteratorMethod.js')
-rw-r--r-- | src/node_modules/es-abstract/helpers/getIteratorMethod.js | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/src/node_modules/es-abstract/helpers/getIteratorMethod.js b/src/node_modules/es-abstract/helpers/getIteratorMethod.js new file mode 100644 index 0000000..02f932c --- /dev/null +++ b/src/node_modules/es-abstract/helpers/getIteratorMethod.js @@ -0,0 +1,45 @@ +'use strict'; + +var hasSymbols = require('has-symbols')(); +var GetIntrinsic = require('../GetIntrinsic'); +var callBound = require('./callBound'); + +var $iterator = GetIntrinsic('%Symbol.iterator%', true); +var $stringSlice = callBound('String.prototype.slice'); + +module.exports = function getIteratorMethod(ES, iterable) { + var usingIterator; + if (hasSymbols) { + usingIterator = ES.GetMethod(iterable, $iterator); + } else if (ES.IsArray(iterable)) { + usingIterator = function () { + var i = -1; + var arr = this; // eslint-disable-line no-invalid-this + return { + next: function () { + i += 1; + return { + done: i >= arr.length, + value: arr[i] + }; + } + }; + }; + } else if (ES.Type(iterable) === 'String') { + usingIterator = function () { + var i = 0; + return { + next: function () { + var nextIndex = ES.AdvanceStringIndex(iterable, i, true); + var value = $stringSlice(iterable, i, nextIndex); + i = nextIndex; + return { + done: nextIndex > iterable.length, + value: value + }; + } + }; + }; + } + return usingIterator; +}; |