| 1 | 'use strict';
|
|---|
| 2 |
|
|---|
| 3 | var $TypeError = require('es-errors/type');
|
|---|
| 4 |
|
|---|
| 5 | var AddValueToKeyedGroup = require('./AddValueToKeyedGroup');
|
|---|
| 6 | var Call = require('./Call');
|
|---|
| 7 | var CanonicalizeKeyedCollectionKey = require('./CanonicalizeKeyedCollectionKey');
|
|---|
| 8 | var GetIterator = require('./GetIterator');
|
|---|
| 9 | var IfAbruptCloseIterator = require('./IfAbruptCloseIterator');
|
|---|
| 10 | var IsCallable = require('./IsCallable');
|
|---|
| 11 | var IteratorClose = require('./IteratorClose');
|
|---|
| 12 | var IteratorStep = require('./IteratorStep');
|
|---|
| 13 | var IteratorValue = require('./IteratorValue');
|
|---|
| 14 | var RequireObjectCoercible = require('./RequireObjectCoercible');
|
|---|
| 15 | var ThrowCompletion = require('./ThrowCompletion');
|
|---|
| 16 | var ToPropertyKey = require('./ToPropertyKey');
|
|---|
| 17 |
|
|---|
| 18 | var maxSafeInteger = require('../helpers/maxSafeInteger');
|
|---|
| 19 |
|
|---|
| 20 | // https://262.ecma-international.org/16.0/#sec-groupby
|
|---|
| 21 |
|
|---|
| 22 | module.exports = function GroupBy(items, callbackfn, keyCoercion) {
|
|---|
| 23 | if (keyCoercion !== 'PROPERTY' && keyCoercion !== 'COLLECTION') {
|
|---|
| 24 | throw new $TypeError('Assertion failed: `keyCoercion` must be `"PROPERTY"` or `"COLLECTION"`');
|
|---|
| 25 | }
|
|---|
| 26 |
|
|---|
| 27 | RequireObjectCoercible(items); // step 1
|
|---|
| 28 |
|
|---|
| 29 | if (!IsCallable(callbackfn)) {
|
|---|
| 30 | throw new $TypeError('callbackfn must be callable'); // step 2
|
|---|
| 31 | }
|
|---|
| 32 |
|
|---|
| 33 | var groups = []; // step 3
|
|---|
| 34 |
|
|---|
| 35 | var iteratorRecord = GetIterator(items, 'SYNC'); // step 4
|
|---|
| 36 |
|
|---|
| 37 | var k = 0; // step 5
|
|---|
| 38 |
|
|---|
| 39 | while (true) { // step 6
|
|---|
| 40 | if (k >= maxSafeInteger) { // step 6.a
|
|---|
| 41 | var error = ThrowCompletion(new $TypeError('k must be less than 2 ** 53 - 1')); // step 6.a.i
|
|---|
| 42 | return IteratorClose(iteratorRecord, error); // step 6.a.ii
|
|---|
| 43 | }
|
|---|
| 44 | var next = IteratorStep(iteratorRecord); // step 6.b
|
|---|
| 45 | if (!next) { // step 6.c
|
|---|
| 46 | return groups; // step 6.c.i
|
|---|
| 47 | }
|
|---|
| 48 |
|
|---|
| 49 | var value = IteratorValue(next); // step 6.dv
|
|---|
| 50 |
|
|---|
| 51 | var key;
|
|---|
| 52 | try {
|
|---|
| 53 | key = Call(callbackfn, void undefined, [value, k]); // step 6.e
|
|---|
| 54 | } catch (e) {
|
|---|
| 55 | IfAbruptCloseIterator(ThrowCompletion(e), iteratorRecord); // step 6.f
|
|---|
| 56 | }
|
|---|
| 57 |
|
|---|
| 58 | if (keyCoercion === 'PROPERTY') { // step 6.g
|
|---|
| 59 | try {
|
|---|
| 60 | key = ToPropertyKey(key); // step 6.g.i
|
|---|
| 61 | } catch (e) {
|
|---|
| 62 | IfAbruptCloseIterator(ThrowCompletion(e), iteratorRecord); // step 6.g.ii
|
|---|
| 63 | }
|
|---|
| 64 | } else { // step 6.h
|
|---|
| 65 | if (keyCoercion !== 'COLLECTION') {
|
|---|
| 66 | throw new $TypeError('keyCoercion must be ~PROPERTY~ or ~COLLECTION~'); // step 6.h.i
|
|---|
| 67 | }
|
|---|
| 68 | key = CanonicalizeKeyedCollectionKey(key); // step 6.h.ii
|
|---|
| 69 | }
|
|---|
| 70 |
|
|---|
| 71 | AddValueToKeyedGroup(groups, key, value); // step 6.i
|
|---|
| 72 |
|
|---|
| 73 | k += 1; // step 6.j
|
|---|
| 74 | }
|
|---|
| 75 | };
|
|---|