feat(core): added support for specifying multiple mutations in triggers

andrewcourtice · Apr 16, 2021 · 8797e7b · 8797e7b
1 parent 2d060ed
commit 8797e7b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/core/src/index.ts b/core/src/index.ts
@@ -106,9 +106,9 @@ export function createStore<T extends object = any>(name: string, data: T, optio
     };
 
     const getMutationHook = (eventName: string) => {
-        return <TPayload = any, TResult = any>(mutationName: string, handler: MutationHookHandler<TPayload, TResult>) => {
+        return <TPayload = any, TResult = any>(mutationName: string | string[], handler: MutationHookHandler<TPayload, TResult>) => {
             return store.on(eventName, (event?: EventPayload<MutationEventData<TPayload, TResult>>) => {
-                if (event && event.data.mutation === mutationName) {
+                if (event && ([] as string[]).concat(mutationName).includes(event.data.mutation)) {
                     handler(event.data);
                 }
             })

diff --git a/core/src/types.ts b/core/src/types.ts
@@ -65,9 +65,9 @@ export interface Store<TState> extends StoreBase<TState> {
     on(event: string, handler: EventHandler): EventListener;
     once(event: string, handler: EventHandler): EventListener;
     destroy(): void;
-    onBeforeMutation<TPayload = any, TResult = any>(mutationName: string, handler: MutationHookHandler<TPayload, TResult>): EventListener;
-    onAfterMutation<TPayload = any, TResult = any>(mutationName: string, handler: MutationHookHandler<TPayload, TResult>): EventListener;
-    onMutationError<TPayload = any, TResult = any>(mutationName: string, handler: MutationHookHandler<TPayload, TResult>): EventListener;
+    onBeforeMutation<TPayload = any, TResult = any>(mutationName: string | string[], handler: MutationHookHandler<TPayload, TResult>): EventListener;
+    onAfterMutation<TPayload = any, TResult = any>(mutationName: string | string[], handler: MutationHookHandler<TPayload, TResult>): EventListener;
+    onMutationError<TPayload = any, TResult = any>(mutationName: string | string[], handler: MutationHookHandler<TPayload, TResult>): EventListener;
 };
 
 export interface HarlemPlugin {