updeep-remeda/lib/util/curry.js

53 lines
1.2 KiB
JavaScript

/* eslint no-shadow:0 */
export function curry1(fn) {
return function curried(a, b, c) {
const n = arguments.length;
if (n >= 1) return fn(a, b, c);
return curried;
};
}
export function curry2(fn) {
return function curried(a, b, c, d) {
const n = arguments.length;
if (n >= 2) return fn(a, b, c, d);
if (n === 1) return curry1((b, c, d) => fn(a, b, c, d));
return curried;
};
}
export function curry3(fn) {
return function curried(a, b, c, d, e) {
const n = arguments.length;
if (n >= 3) return fn(a, b, c, d, e);
if (n === 2) return curry1((c, d, e) => fn(a, b, c, d, e));
if (n === 1) return curry2((b, c, d, e) => fn(a, b, c, d, e));
return curried;
};
}
export function curry4(fn) {
return function curried(a, b, c, d, e, f) {
const n = arguments.length;
if (n >= 4) return fn(a, b, c, d, e, f);
if (n === 3) return curry1((d, e, f) => fn(a, b, c, d, e, f));
if (n === 2) return curry2((c, d, e, f) => fn(a, b, c, d, e, f));
if (n === 1) return curry3((b, c, d, e, f) => fn(a, b, c, d, e, f));
return curried;
};
}
export default function curry(fn, length = fn.length) {
return [
fn,
curry1,
curry2,
curry3,
curry4,
][length](fn);
}