31 lines
1.0 KiB
JavaScript
31 lines
1.0 KiB
JavaScript
|
'use strict';
|
||
|
|
|
||
|
|
var setPrototypeOf = require('es5-ext/object/set-prototype-of')
|
||
|
|
, contains = require('es5-ext/string/#/contains')
|
||
|
|
, d = require('d')
|
||
|
|
, Iterator = require('./')
|
||
|
|
|
||
|
|
, defineProperty = Object.defineProperty
|
||
|
|
, ArrayIterator;
|
||
|
|
|
||
|
|
ArrayIterator = module.exports = function (arr, kind) {
|
||
|
|
if (!(this instanceof ArrayIterator)) return new ArrayIterator(arr, kind);
|
||
|
|
Iterator.call(this, arr);
|
||
|
|
if (!kind) kind = 'value';
|
||
|
|
else if (contains.call(kind, 'key+value')) kind = 'key+value';
|
||
|
|
else if (contains.call(kind, 'key')) kind = 'key';
|
||
|
|
else kind = 'value';
|
||
|
|
defineProperty(this, '__kind__', d('', kind));
|
||
|
|
};
|
||
|
|
if (setPrototypeOf) setPrototypeOf(ArrayIterator, Iterator);
|
||
|
|
|
||
|
|
ArrayIterator.prototype = Object.create(Iterator.prototype, {
|
||
|
|
constructor: d(ArrayIterator),
|
||
|
|
_resolve: d(function (i) {
|
||
|
|
if (this.__kind__ === 'value') return this.__list__[i];
|
||
|
|
if (this.__kind__ === 'key+value') return [i, this.__list__[i]];
|
||
|
|
return i;
|
||
|
|
}),
|
||
|
|
toString: d(function () { return '[object Array Iterator]'; })
|
||
|
|
});
|