|
1 | 1 | import Signature from '../internal/types/Signature' |
2 | 2 | import MessagesConfig from '../internal/types/MessagesConfig' |
3 | 3 | import MessageInitializer from '../internal/types/MessageInitializer' |
| 4 | +import MessageCreator from '../internal/types/MessageCreator' |
| 5 | +import buildMessageCreator from '../internal/buildMessageCreator'; |
4 | 6 |
|
5 | 7 | type Func<A extends any[], R> = (...args: A) => R |
6 | 8 |
|
7 | | -type MessageCreator<K, A extends any[], P, M> = { |
8 | | - readonly type: K, |
9 | | - |
10 | | - (...args: A): { |
11 | | - type: K, |
12 | | - payload?: P, |
13 | | - meta?: M |
14 | | - } |
15 | | -} |
16 | | - |
17 | 9 | type Arguments<A extends any[], I extends MessageInitializer<any>> |
18 | 10 | = I extends Func<A, any> |
19 | 11 | ? Signature<I> |
@@ -45,38 +37,7 @@ function defineMessages<T extends MessagesConfig>(config: T): |
45 | 37 | for (let i = 0; i < keys.length; ++i) { |
46 | 38 | const key = keys[i] |
47 | 39 |
|
48 | | - if (typeof config[key] === 'function') { |
49 | | - const getPayload = config[key] as Function |
50 | | - |
51 | | - ret[key] = function (/* arguments */) { |
52 | | - const payload = getPayload.apply(null, arguments) |
53 | | - |
54 | | - return { type: key, payload } |
55 | | - } |
56 | | - } else { |
57 | | - const |
58 | | - messageConfig = config[key] as { payload?: Function, meta?: Function }, |
59 | | - getPayload = messageConfig.payload, |
60 | | - getMeta = messageConfig.meta |
61 | | - |
62 | | - ret[key] = function (/* arguments */) { |
63 | | - const msg: any = { type: key } |
64 | | - |
65 | | - if (getPayload) { |
66 | | - msg.payload = getPayload.apply(null, arguments) |
67 | | - } |
68 | | - |
69 | | - if (getMeta) { |
70 | | - msg.meta = getMeta.apply(null, arguments) |
71 | | - } |
72 | | - |
73 | | - return msg |
74 | | - } |
75 | | - } |
76 | | - |
77 | | - Object.defineProperty(ret[key], 'type', { |
78 | | - value: key |
79 | | - }) |
| 40 | + ret[key] = buildMessageCreator(key, config[key]) |
80 | 41 | } |
81 | 42 |
|
82 | 43 | return ret |
|
0 commit comments