42 lines
1.9 KiB
JavaScript
42 lines
1.9 KiB
JavaScript
|
import { isSymbol } from "../predicate/isSymbol.mjs";
|
||
|
|
import { range } from "../../math/range.mjs";
|
||
|
|
import { isArrayLike } from "../predicate/isArrayLike.mjs";
|
||
|
|
import { iteratee } from "../util/iteratee.mjs";
|
||
|
|
import { identity } from "../function/identity.mjs";
|
||
|
|
import { keysIn } from "./keysIn.mjs";
|
||
|
|
import { getSymbolsIn } from "../_internal/getSymbolsIn.mjs";
|
||
|
|
//#region src/compat/object/pickBy.ts
|
||
|
|
/**
|
||
|
|
* Creates a new object composed of the properties that satisfy the predicate function.
|
||
|
|
*
|
||
|
|
* This function takes an object and a predicate function, and returns a new object that
|
||
|
|
* includes only the properties for which the predicate function returns true.
|
||
|
|
*
|
||
|
|
* @template T - The type of object.
|
||
|
|
* @param {Record<string, T> | Record<number, T> | object | null | undefined} obj - The object to pick properties from.
|
||
|
|
* @param {ValueKeyIterateeTypeGuard<T, S> | ValueKeyIteratee<T[keyof T]> | ValueKeyIteratee<T>} [shouldPick] - A predicate function that determines
|
||
|
|
* whether a property should be picked. It takes the property's key and value as arguments and returns `true`
|
||
|
|
* if the property should be picked, and `false` otherwise.
|
||
|
|
* @returns {Record<string, S> | Record<number, S> | Partial<T>} A new object with the properties that satisfy the predicate function.
|
||
|
|
*
|
||
|
|
* @example
|
||
|
|
* const obj = { a: 1, b: 'pick', c: 3 };
|
||
|
|
* const shouldPick = (value) => typeof value === 'string';
|
||
|
|
* const result = pickBy(obj, shouldPick);
|
||
|
|
* // result will be { b: 'pick' }
|
||
|
|
*/
|
||
|
|
function pickBy(obj, shouldPick) {
|
||
|
|
if (obj == null) return {};
|
||
|
|
const predicate = iteratee(shouldPick ?? identity);
|
||
|
|
const result = {};
|
||
|
|
const keys = isArrayLike(obj) ? range(0, obj.length) : [...keysIn(obj), ...getSymbolsIn(obj)];
|
||
|
|
for (let i = 0; i < keys.length; i++) {
|
||
|
|
const key = isSymbol(keys[i]) ? keys[i] : keys[i].toString();
|
||
|
|
const value = obj[key];
|
||
|
|
if (predicate(value, key, obj)) result[key] = value;
|
||
|
|
}
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
//#endregion
|
||
|
|
export { pickBy };
|