43 lines
1.7 KiB
JavaScript
43 lines
1.7 KiB
JavaScript
import { identity } from "../../function/identity.mjs";
|
|
import { isArrayLike } from "../predicate/isArrayLike.mjs";
|
|
import { iteratee } from "../util/iteratee.mjs";
|
|
//#region src/compat/array/partition.ts
|
|
/**
|
|
* Creates an array of elements split into two groups, the first of which contains elements
|
|
* `predicate` returns truthy for, the second of which contains elements
|
|
* `predicate` returns falsy for. The predicate is invoked with one argument: (value).
|
|
*
|
|
* @template T
|
|
* @param {ArrayLike<T> | T | null | undefined} source - The array or object to iterate over.
|
|
* @param {((item: T, index: number, arr: any) => unknown) | Partial<T> | [keyof T, unknown] | PropertyKey} [predicate=identity] - The function invoked per iteration.
|
|
* @returns {[T[], T[]]} - Returns the array of grouped elements.
|
|
*
|
|
* @example
|
|
* partition([{ a: 1 }, { a: 2 }, { b: 1 }], 'a');
|
|
* // => [[{ a: 1 }, { a: 2 }], [{ b: 1 }]]
|
|
*
|
|
* partition([{ a: 1 }, { a: 2 }, { b: 1 }], { b: 1 });
|
|
* // => [[{ b: 1 }], [{ a: 1 }, { a: 2 }]]
|
|
*
|
|
* partition({ item1: { a: 0, b: true }, item2: { a: 1, b: true }, item3: { a: 2, b: false }}, { b: false })
|
|
* // => [[{ a: 2, b: false }], [{ a: 0, b: true }, { a: 1, b: true }]]
|
|
*
|
|
* partition([{ a: 1 }, { a: 2 }, { a: 3 }], ['a', 2]);
|
|
* // => [[{ a: 2 }], [{ a: 1 }, { a: 3 }]]
|
|
*/
|
|
function partition(source, predicate = identity) {
|
|
if (!source) return [[], []];
|
|
const collection = isArrayLike(source) ? source : Object.values(source);
|
|
predicate = iteratee(predicate);
|
|
const matched = [];
|
|
const unmatched = [];
|
|
for (let i = 0; i < collection.length; i++) {
|
|
const value = collection[i];
|
|
if (predicate(value)) matched.push(value);
|
|
else unmatched.push(value);
|
|
}
|
|
return [matched, unmatched];
|
|
}
|
|
//#endregion
|
|
export { partition };
|