feat(overmind-vue): pass callback to other hooks as well

cerebral · Jan 23, 2021 · 8b2d584 · 8b2d584
1 parent f674c8b
commit 8b2d584
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 8 deletions.
diff --git a/packages/node_modules/overmind-vue/src/index.ts b/packages/node_modules/overmind-vue/src/index.ts
@@ -274,20 +274,30 @@ export function createStateHook<Config extends IConfiguration>(): StateHook<Conf
   }) as any
 }
 
-export function createActionsHook<Config extends IConfiguration>() {
-  return (): Overmind<Config>["actions"] => {
+export interface ActionsHook<Config extends IConfiguration> {
+  (): Ref<Overmind<Config>["actions"]>;
+  <CB extends (actions: Overmind<Config>["actions"]) => object>(cb: CB): CB extends (actions: Overmind<Config>["actions"]) => infer O ? O extends object ? Ref<O> : never : never;
+}
+
+export function createActionsHook<Config extends IConfiguration>(): ActionsHook<Config> {
+  return ((cb?: any): Overmind<Config>["actions"] => {
     const overmindInstance = inject<any>('overmind')
 
-    return overmindInstance.actions
-  }
+    return cb ? cb(overmindInstance.actions) : overmindInstance.actions
+  }) as any
 }
 
-export function createEffectsHook<Config extends IConfiguration>() {
-  return (): Overmind<Config>["effects"] => {
+export interface EffectsHook<Config extends IConfiguration> {
+  (): Ref<Overmind<Config>["effects"]>;
+  <CB extends (effects: Overmind<Config>["effects"]) => object>(cb: CB): CB extends (effects: Overmind<Config>["effects"]) => infer O ? O extends object ? Ref<O> : never : never;
+}
+
+export function createEffectsHook<Config extends IConfiguration>(): EffectsHook<Config> {
+  return ((cb?: any): Overmind<Config>["effects"] => {
     const overmindInstance = inject<any>('overmind')
 
-    return overmindInstance.effects
-  }
+    return cb ? cb(overmindInstance.effects) : overmindInstance.effects
+  }) as any
 }
 
 export function createReactionHook<Config extends IConfiguration>() {

diff --git a/packages/node_modules/overmind/src/internalTypes.ts b/packages/node_modules/overmind/src/internalTypes.ts
@@ -20,6 +20,7 @@ export type NestedPartial<T> = T extends Function
 export type Options = {
   delimiter?: string
   name?: string
+  devEnv?: string
   devtools?: string | boolean
   logProxies?: boolean
   hotReloading?: boolean