-
Notifications
You must be signed in to change notification settings - Fork 4
/
curry.ts
76 lines (73 loc) · 2.31 KB
/
curry.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
import type { AnyFn } from '../type';
type Curry1<T1, R> = {
(): Curry1<T1, R>;
(t1: T1): R;
};
type Curry2<T1, T2, R> = {
(): Curry2<T1, T2, R>;
(t1: T1): Curry1<T2, R>;
(t1: T1, t2: T2): R;
};
type Curry3<T1, T2, T3, R> = {
(): Curry3<T1, T2, T3, R>;
(t1: T1): Curry2<T2, T3, R>;
(t1: T1, t2: T2): Curry1<T3, R>;
(t1: T1, t2: T2, t3: T3): R;
};
type Curry4<T1, T2, T3, T4, R> = {
(): Curry4<T1, T2, T3, T4, R>;
(t1: T1): Curry3<T2, T3, T4, R>;
(t1: T1, t2: T2): Curry2<T3, T4, R>;
(t1: T1, t2: T2, t3: T3): Curry1<T4, R>;
(t1: T1, t2: T2, t3: T3, t4: T4): R;
};
type Curry5<T1, T2, T3, T4, T5, R> = {
(): Curry5<T1, T2, T3, T4, T5, R>;
(t1: T1): Curry4<T2, T3, T4, T5, R>;
(t1: T1, t2: T2): Curry3<T3, T4, T5, R>;
(t1: T1, t2: T2, t3: T3): Curry2<T4, T5, R>;
(t1: T1, t2: T2, t3: T3, t4: T4): Curry1<T5, R>;
(t1: T1, t2: T2, t3: T3, t4: T4, t5: T5): R;
};
type Curry6<T1, T2, T3, T4, T5, T6, R> = {
(): Curry6<T1, T2, T3, T4, T5, T6, R>;
(t1: T1): Curry5<T2, T3, T4, T5, T6, R>;
(t1: T1, t2: T2): Curry4<T3, T4, T5, T6, R>;
(t1: T1, t2: T2, t3: T3): Curry3<T4, T5, T6, R>;
(t1: T1, t2: T2, t3: T3, t4: T4): Curry2<T5, T6, R>;
(t1: T1, t2: T2, t3: T3, t4: T4, t5: T5): Curry1<T6, R>;
(t1: T1, t2: T2, t3: T3, t4: T4, t5: T5, t6: T6): R;
};
/**
* 柯里化
* @param func
* @param depth
* @category Function
*/
function curry<T1, R>(func: (t1: T1) => R, depth?: number): Curry1<T1, R>;
function curry<T1, T2, R>(func: (t1: T1, t2: T2) => R, depth?: number): Curry2<T1, T2, R>;
function curry<T1, T2, T3, R>(
func: (t1: T1, t2: T2, t3: T3) => R,
depth?: number,
): Curry3<T1, T2, T3, R>;
function curry<T1, T2, T3, T4, R>(
func: (t1: T1, t2: T2, t3: T3, t4: T4) => R,
depth?: number,
): Curry4<T1, T2, T3, T4, R>;
function curry<T1, T2, T3, T4, T5, R>(
func: (t1: T1, t2: T2, t3: T3, t4: T4, t5: T5) => R,
depth?: number,
): Curry5<T1, T2, T3, T4, T5, R>;
function curry<T1, T2, T3, T4, T5, T6, R>(
func: (t1: T1, t2: T2, t3: T3, t4: T4, t5: T5, t6: T6) => R,
depth?: number,
): Curry6<T1, T2, T3, T4, T5, T6, R>;
function curry(fn: AnyFn, depth: number = fn.length): AnyFn {
return function (...args) {
if (args.length && args.length < depth) {
return curry(fn.bind(null, ...args), depth - args.length);
}
return fn.call(null, ...args);
};
}
export default curry;