esnext.array.group-by-to-map.js 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. 'use strict';
  2. var $ = require('../internals/export');
  3. var getBuiltIn = require('../internals/get-built-in');
  4. var bind = require('../internals/function-bind-context');
  5. var uncurryThis = require('../internals/function-uncurry-this');
  6. var IndexedObject = require('../internals/indexed-object');
  7. var toObject = require('../internals/to-object');
  8. var lengthOfArrayLike = require('../internals/length-of-array-like');
  9. var addToUnscopables = require('../internals/add-to-unscopables');
  10. var Map = getBuiltIn('Map');
  11. var MapPrototype = Map.prototype;
  12. var mapGet = uncurryThis(MapPrototype.get);
  13. var mapHas = uncurryThis(MapPrototype.has);
  14. var mapSet = uncurryThis(MapPrototype.set);
  15. var push = uncurryThis([].push);
  16. // `Array.prototype.groupByToMap` method
  17. // https://github.com/tc39/proposal-array-grouping
  18. $({ target: 'Array', proto: true }, {
  19. groupByToMap: function groupByToMap(callbackfn /* , thisArg */) {
  20. var O = toObject(this);
  21. var self = IndexedObject(O);
  22. var boundFunction = bind(callbackfn, arguments.length > 1 ? arguments[1] : undefined);
  23. var map = new Map();
  24. var length = lengthOfArrayLike(self);
  25. var index = 0;
  26. var key, value;
  27. for (;length > index; index++) {
  28. value = self[index];
  29. key = boundFunction(value, index, O);
  30. if (mapHas(map, key)) push(mapGet(map, key), value);
  31. else mapSet(map, key, [value]);
  32. } return map;
  33. }
  34. });
  35. addToUnscopables('groupByToMap');