feat(action): added root strategies

andrewcourtice · Feb 17, 2022 · 407f2a0 · 407f2a0
1 parent cf49768
commit 407f2a0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/extensions/action/src/index.ts b/extensions/action/src/index.ts
@@ -51,7 +51,7 @@ export const ABORT_STRATEGY = {
 
 export default function actionsExtension<TState extends BaseState>(options?: Partial<Options>) {
     const {
-        strategies,
+        strategies: rootStrategies,
     } = {
         ...options,
         strategies: {
@@ -116,10 +116,17 @@ export default function actionsExtension<TState extends BaseState>(options?: Par
             const {
                 parallel,
                 autoClearErrors,
+                strategies,
             } = {
                 parallel: false,
                 autoClearErrors: true,
+
                 ...options,
+
+                strategies: {
+                    ...rootStrategies,
+                    ...options?.strategies,
+                },
             } as ActionOptions;
 
             const mutate = (mutator: Mutator<TState, undefined, void>) => _store.write(name, SENDER, mutator);
@@ -158,6 +165,7 @@ export default function actionsExtension<TState extends BaseState>(options?: Par
                         result = await body(providedPayload, mutate, controller, onAbort);
 
                         emit(EVENTS.action.success);
+
                         incrementRunCount(name);
                         resolve(result);
                     } catch (error) {

diff --git a/extensions/action/src/types.ts b/extensions/action/src/types.ts
@@ -14,9 +14,11 @@ export type ActionHookHandler<TPayload, TResult> = (data: ActionEventData<TPaylo
 export type ActionAbortStrategy = (name: string, id: symbol, resolve: (value?: any) => void, reject: (reason: unknown) => void, reason?: unknown) => void;
 
 export interface Options {
-    strategies: {
-        abort: ActionAbortStrategy
-    }
+    strategies: ActionStrategies;
+}
+
+export interface ActionStrategies {
+    abort: ActionAbortStrategy;
 }
 
 export interface ActionAbortStrategies {
@@ -38,6 +40,7 @@ export interface ActionOptions {
     parallel: boolean;
     autoClearErrors: boolean;
     suppressAbortErrors: boolean;
+    strategies: ActionStrategies;
 }
 
 export interface ActionEventData<TPayload = any, TResult = any> {