60 lines
1.9 KiB
JavaScript
60 lines
1.9 KiB
JavaScript
|
import { isArrayLike } from "../predicate/isArrayLike.mjs";
|
||
|
|
import { get } from "./get.mjs";
|
||
|
|
import { has } from "./has.mjs";
|
||
|
|
import { isNil } from "../predicate/isNil.mjs";
|
||
|
|
import { set } from "./set.mjs";
|
||
|
|
//#region src/compat/object/pick.ts
|
||
|
|
/**
|
||
|
|
* Creates a new object composed of the picked object properties.
|
||
|
|
*
|
||
|
|
* This function takes an object and an array of keys, and returns a new object that
|
||
|
|
* includes only the properties corresponding to the specified keys.
|
||
|
|
*
|
||
|
|
* @template T - The type of object.
|
||
|
|
* @template U - The type of keys to pick.
|
||
|
|
* @param {T | any | null | undefined} object - The object to pick keys from.
|
||
|
|
* @param {...Array<Many<U>> | Array<Many<PropertyPath>>} props - An array of keys to be picked from the object. received keys goes through a flattening process before being used.
|
||
|
|
* @returns {Pick<T, U> | Partial<T>} A new object with the specified keys picked.
|
||
|
|
*
|
||
|
|
* @example
|
||
|
|
* const obj = { a: 1, b: 2, c: 3 };
|
||
|
|
* const result = pick(obj, ['a', 'c']);
|
||
|
|
* // result will be { a: 1, c: 3 }
|
||
|
|
*
|
||
|
|
* // each path can be passed individually as an argument
|
||
|
|
* const obj = { a: 1, b: 2, c: 3 };
|
||
|
|
* const result = pick(obj, 'a', 'c');
|
||
|
|
*
|
||
|
|
* // pick a key over a path
|
||
|
|
* const obj = { 'a.b': 1, a: { b: 2 } };
|
||
|
|
* const result = pick(obj, 'a.b');
|
||
|
|
* // result will be { 'a.b': 1 }
|
||
|
|
*/
|
||
|
|
function pick(obj, ...keysArr) {
|
||
|
|
if (isNil(obj)) return {};
|
||
|
|
const result = {};
|
||
|
|
for (let i = 0; i < keysArr.length; i++) {
|
||
|
|
let keys = keysArr[i];
|
||
|
|
switch (typeof keys) {
|
||
|
|
case "object":
|
||
|
|
if (!Array.isArray(keys)) if (isArrayLike(keys)) keys = Array.from(keys);
|
||
|
|
else keys = [keys];
|
||
|
|
break;
|
||
|
|
case "string":
|
||
|
|
case "symbol":
|
||
|
|
case "number":
|
||
|
|
keys = [keys];
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
for (const key of keys) {
|
||
|
|
const value = get(obj, key);
|
||
|
|
if (value === void 0 && !has(obj, key)) continue;
|
||
|
|
if (typeof key === "string" && Object.hasOwn(obj, key)) result[key] = value;
|
||
|
|
else set(result, key, value);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
//#endregion
|
||
|
|
export { pick };
|