39 lines
1.3 KiB
JavaScript
39 lines
1.3 KiB
JavaScript
|
'use strict';
|
||
|
|
||
|
var setPrototypeOf = require('es5-ext/object/set-prototype-of')
|
||
|
, d = require('d')
|
||
|
, Iterator = require('es6-iterator')
|
||
|
, toStringTagSymbol = require('es6-symbol').toStringTag
|
||
|
, kinds = require('./iterator-kinds')
|
||
|
|
||
|
, defineProperties = Object.defineProperties
|
||
|
, unBind = Iterator.prototype._unBind
|
||
|
, MapIterator;
|
||
|
|
||
|
MapIterator = module.exports = function (map, kind) {
|
||
|
if (!(this instanceof MapIterator)) return new MapIterator(map, kind);
|
||
|
Iterator.call(this, map.__mapKeysData__, map);
|
||
|
if (!kind || !kinds[kind]) kind = 'key+value';
|
||
|
defineProperties(this, {
|
||
|
__kind__: d('', kind),
|
||
|
__values__: d('w', map.__mapValuesData__)
|
||
|
});
|
||
|
};
|
||
|
if (setPrototypeOf) setPrototypeOf(MapIterator, Iterator);
|
||
|
|
||
|
MapIterator.prototype = Object.create(Iterator.prototype, {
|
||
|
constructor: d(MapIterator),
|
||
|
_resolve: d(function (i) {
|
||
|
if (this.__kind__ === 'value') return this.__values__[i];
|
||
|
if (this.__kind__ === 'key') return this.__list__[i];
|
||
|
return [this.__list__[i], this.__values__[i]];
|
||
|
}),
|
||
|
_unBind: d(function () {
|
||
|
this.__values__ = null;
|
||
|
unBind.call(this);
|
||
|
}),
|
||
|
toString: d(function () { return '[object Map Iterator]'; })
|
||
|
});
|
||
|
Object.defineProperty(MapIterator.prototype, toStringTagSymbol,
|
||
|
d('c', 'Map Iterator'));
|