LendAndRegret/node_modules/es-toolkit/dist/compat/array/flatten.mjs
2026-05-02 17:27:43 +08:00

33 lines
1.0 KiB
JavaScript

import { isArrayLike } from '../predicate/isArrayLike.mjs';
function flatten(value, depth = 1) {
const result = [];
const flooredDepth = Math.floor(depth);
if (!isArrayLike(value)) {
return result;
}
const recursive = (arr, currentDepth) => {
for (let i = 0; i < arr.length; i++) {
const item = arr[i];
if (currentDepth < flooredDepth &&
(Array.isArray(item) ||
Boolean(item?.[Symbol.isConcatSpreadable]) ||
(item !== null && typeof item === 'object' && Object.prototype.toString.call(item) === '[object Arguments]'))) {
if (Array.isArray(item)) {
recursive(item, currentDepth + 1);
}
else {
recursive(Array.from(item), currentDepth + 1);
}
}
else {
result.push(item);
}
}
};
recursive(Array.from(value), 0);
return result;
}
export { flatten };