es-toolkit
Version:
A state-of-the-art, high-performance JavaScript utility library with a small bundle size and strong type annotations.
39 lines (38 loc) • 1.5 kB
JavaScript
const require_flatMapDeep = require("../../array/flatMapDeep.js");
const require_lazy = require("../_internal/lazy.js");
//#region src/fp/array/flatMapDeep.ts
/**
* Creates a function that maps each element and recursively flattens the mapped values.
*
* The iteratee receives each value and index. The returned function is lazy-capable inside
* {@link pipe}; a trailing short-circuiting operator can stop before later input values are mapped.
*
* @template T - The type of elements in the input array.
* @template U - The type returned by the iteratee before recursive flattening.
* @param iteratee - Called with each value and index to produce values to flatten.
* @returns A function that maps a readonly array to a deeply flattened array.
*
* @example
* import { flatMapDeep, pipe } from 'es-toolkit/fp';
*
* pipe([1, 2], flatMapDeep(value => [[value, value * 10]]));
* // => [1, 10, 2, 20]
*/
function flatMapDeep(iteratee) {
function flatMapDeepEager(array) {
return require_flatMapDeep.flatMapDeep(array, (item, index) => iteratee(item, index));
}
const flatMapDeepLazy = require_lazy.createLazyFunction((value, index, emit) => {
emitDeep(iteratee(value, index), emit);
});
return require_lazy.combineEagerAndLazyFunctions(flatMapDeepEager, flatMapDeepLazy);
}
function emitDeep(value, emit) {
if (Array.isArray(value)) {
for (let index = 0; index < value.length; index++) emitDeep(value[index], emit);
return;
}
emit(value);
}
//#endregion
exports.flatMapDeep = flatMapDeep;