feat(subscribe): add unsubscribe api,add actions return value from Model

Subscriptions: Function => Function[]

Author: ArrayZoneYour

__test__/middlewares/subscribe.spec.ts

@@ -9,15 +9,16 @@ describe('Subscribe middleware', () => {
     let actions: any
     let count = 0
     const { useStore, subscribe } = Model({ Counter })
-    subscribe('Counter', 'increment', () => (count += 1))
+    subscribe('Counter', ['increment'], () => (count += 1))
     subscribe('Counter', 'add', () => (count += 10))
+    subscribe('Counter', ['increment', 'add'], () => (count += 5))
     testHook(() => {
       ;[, actions] = useStore('Counter')
     })
     await actions.increment()
     await actions.add(1)
     await actions.increment()
     await actions.increment()
-    expect(count).toBe(13)
+    expect(count).toBe(33)
   })
 })

__test__/middlewares/unsubscribe.spec.ts

@@ -0,0 +1,26 @@
+/// <reference path="../index.d.ts" />
+import 'react-testing-library/cleanup-after-each'
+import { Model } from '../../src'
+import { Counter } from '..'
+import { testHook } from 'react-hooks-testing-library'
+
+describe('Subscribe middleware', () => {
+  test('run callback when specific action run', async () => {
+    let actions: any
+    let count = 0
+    const { useStore, subscribe, unsubscribe } = Model({ Counter })
+    subscribe('Counter', ['increment'], () => (count += 1))
+    subscribe('Counter', 'add', () => (count += 10))
+    subscribe('Counter', ['increment', 'add'], () => (count += 5))
+    testHook(() => {
+      ;[, actions] = useStore('Counter')
+    })
+    await actions.increment()
+    unsubscribe('Counter', 'add')
+    await actions.add(1)
+    unsubscribe('Counter', ['add', 'increment'])
+    await actions.increment()
+    await actions.increment()
+    expect(count).toBe(6)
+  })
+})

src/index.d.ts

@@ -126,5 +126,5 @@ type ModelsProps<M extends Models> = {
 }
 
 type Subscriptions = {
-  [key: string]: Function
+  [key: string]: Function[]
 }

src/index.tsx

@@ -20,14 +20,27 @@ const Model = <M extends Models>(models: M, initialState?: Global['State']) => {
     Global.AsyncState[name] = model.asyncState
   })
 
+  const actions = Object.keys(models).reduce(
+    (o, modelName) => ({ ...o, name: getActions(modelName) }),
+    {}
+  )
+
   Global.withDevTools =
     typeof window !== 'undefined' &&
     (window as any).__REDUX_DEVTOOLS_EXTENSION__
   if (Global.withDevTools) {
     Global.devTools = (window as any).__REDUX_DEVTOOLS_EXTENSION__
     Global.devTools.connect()
   }
-  return { useStore, getState, getInitialState, getActions, subscribe } as {
+  return {
+    actions,
+    useStore,
+    getState,
+    getInitialState,
+    getActions,
+    subscribe,
+    unsubscribe
+  } as {
     useStore: <K extends keyof M>(
       name: K,
       depActions?: (keyof Get<M[K], 'actions'>)[]
@@ -39,25 +52,59 @@ const Model = <M extends Models>(models: M, initialState?: Global['State']) => {
     getInitialState: typeof getInitialState
     subscribe: <K extends keyof M>(
       modelName: K,
-      actionName: keyof Get<M[K], 'actions'> | keyof Get<M[K], 'actions'>,
+      actionName: keyof Get<M[K], 'actions'> | (keyof Get<M[K], 'actions'>)[],
       callback: Function
     ) => void
+    unsubscribe: <K extends keyof M>(
+      modelName: K,
+      actionName: keyof Get<M[K], 'actions'> | (keyof Get<M[K], 'actions'>)[]
+    ) => void
+    actions: {
+      [K in keyof M]: Readonly<getConsumerActionsType<Get<M[K], 'actions'>>>
+    }
   }
 }
 
+const unsubscribe = (modelName: string, actions: string | string[]) => {
+  subscribe(modelName, actions, undefined)
+}
+
 const subscribe = (
   modelName: string,
-  actionName: string,
-  callback: Function
+  actions: string | string[],
+  callback?: Function
 ) => {
-  Global.subscriptions[`${modelName}_${actionName}`] = callback
+  if (Array.isArray(actions)) {
+    actions.forEach(actionName => {
+      if (!Global.subscriptions[`${modelName}_${actionName}`]) {
+        Global.subscriptions[`${modelName}_${actionName}`] = []
+      }
+      if (callback) {
+        Global.subscriptions[`${modelName}_${actionName}`].push(callback)
+      } else {
+        Global.subscriptions[`${modelName}_${actionName}`] = []
+      }
+    })
+  } else {
+    if (!Global.subscriptions[`${modelName}_${actions}`]) {
+      Global.subscriptions[`${modelName}_${actions}`] = []
+    }
+    if (callback) {
+      Global.subscriptions[`${modelName}_${actions}`].push(callback)
+    } else {
+      Global.subscriptions[`${modelName}_${actions}`] = []
+    }
+  }
 }
 
 const getState = (modelName: keyof typeof Global.State) => {
   return Global.State[modelName]
 }
 
-const getActions = (modelName: string, baseContext: Partial<Context>) => {
+const getActions = (
+  modelName: string,
+  baseContext: Partial<Context> = { type: 'outer' }
+) => {
   const updaters: any = {}
   Object.entries(Global.Actions[modelName]).forEach(
     ([key, action]) =>

src/middlewares.ts

@@ -60,9 +60,11 @@ const stateUpdater: Middleware = async (context, restMiddlewares) => {
 
 const subscription: Middleware = async (context, restMiddlewares) => {
   const { modelName, actionName, next, Global } = context
-  if (Global.subscriptions[`${modelName}_${actionName}`]) {
-    Global.subscriptions[`${modelName}_${actionName}`]()
-  }
+  const subscriptions = Global.subscriptions[`${modelName}_${actionName}`]
+  subscriptions &&
+    subscriptions.forEach(callback => {
+      callback()
+    })
   await next(restMiddlewares)
 }