42 lines
1.6 KiB
JavaScript
42 lines
1.6 KiB
JavaScript
|
import { isFunction } from "../../predicate/isFunction.mjs";
|
||
|
|
import { isNil } from "../../predicate/isNil.mjs";
|
||
|
|
import "../../predicate/index.mjs";
|
||
|
|
import { isArrayLike } from "../predicate/isArrayLike.mjs";
|
||
|
|
import { get } from "../object/get.mjs";
|
||
|
|
//#region src/compat/array/invokeMap.ts
|
||
|
|
/**
|
||
|
|
* Invokes the method at path of each element in collection.
|
||
|
|
*
|
||
|
|
* @template T, R
|
||
|
|
* @param {ArrayLike<T> | Record<string, T> | null | undefined} collection - The collection to iterate over.
|
||
|
|
* @param {string | ((...args: any[]) => R)} path - The path of the method to invoke or the method to invoke.
|
||
|
|
* @param {...any[]} args - The arguments to invoke each method with.
|
||
|
|
* @returns {Array<R | undefined>} Returns the array of results.
|
||
|
|
*
|
||
|
|
* @example
|
||
|
|
* invokeMap([[5, 1, 7], [3, 2, 1]], 'sort');
|
||
|
|
* // => [[1, 5, 7], [1, 2, 3]]
|
||
|
|
*/
|
||
|
|
function invokeMap(collection, path, ...args) {
|
||
|
|
if (isNil(collection)) return [];
|
||
|
|
const values = isArrayLike(collection) ? Array.from(collection) : Object.values(collection);
|
||
|
|
const result = [];
|
||
|
|
for (let i = 0; i < values.length; i++) {
|
||
|
|
const value = values[i];
|
||
|
|
if (isFunction(path)) {
|
||
|
|
result.push(path.apply(value, args));
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
const method = get(value, path);
|
||
|
|
let thisContext = value;
|
||
|
|
if (Array.isArray(path)) {
|
||
|
|
const pathExceptLast = path.slice(0, -1);
|
||
|
|
if (pathExceptLast.length > 0) thisContext = get(value, pathExceptLast);
|
||
|
|
} else if (typeof path === "string" && path.includes(".")) thisContext = get(value, path.split(".").slice(0, -1).join("."));
|
||
|
|
result.push(method == null ? void 0 : method.apply(thisContext, args));
|
||
|
|
}
|
||
|
|
return result;
|
||
|
|
}
|
||
|
|
//#endregion
|
||
|
|
export { invokeMap };
|