fix(types): correct the type of actions returned by next model

Author: ArrayZoneYour

package.json

@@ -1,6 +1,6 @@
 {
   "name": "react-model",
-  "version": "2.7.0",
+  "version": "2.7.1",
   "description": "The State management library for React",
   "main": "./dist/react-model.js",
   "umd:main": "./dist/react-model.umd.js",

src/global.ts

@@ -1,6 +1,7 @@
 const State = {}
 const Actions = {}
 const AsyncState = {}
+const Middlewares = {}
 // Communicate between Provider-Consumer and Hooks
 const Setter: Setter = {
   // classSetter stores the setState from Provider
@@ -21,6 +22,7 @@ let uid = Math.random() // The unique id of hooks
 export default {
   Actions,
   AsyncState,
+  Middlewares,
   Setter,
   State,
   devTools,

src/index.d.ts

@@ -24,6 +24,9 @@ interface Global {
   AsyncState: {
     [modelName: string]: undefined | ((context?: any) => Promise<Partial<any>>)
   }
+  Middlewares: {
+    [modelName: string]: Middleware[]
+  }
   subscriptions: Subscriptions
   Setter: Setter
   devTools: any
@@ -48,12 +51,12 @@ type Action<S = {}, P = any, ActionKeys = {}> = (
   | Promise<void>
 
 // v3.0 Action
-type NextAction<S = {}, P = any, ActionKeys = {}> = (
+type NextAction<S = {}, P = any, ActionKeys = {}, ExtContext = {}> = (
   params: P,
   context: {
     state: S
     actions: getConsumerNextActionsType<NextActions<S, ActionKeys>>
-  }
+  } & ExtContext
 ) =>
   | Partial<S>
   | Promise<Partial<S>>
@@ -103,6 +106,7 @@ interface InnerContext<S = {}> extends BaseContext<S> {
 
 type Context<S = {}> = (InnerContext<S>) & {
   next: Function
+  modelMiddlewares?: Middleware[]
 }
 
 type Middleware<S = {}> = (C: Context<S>, M: Middleware<S>[]) => void
@@ -150,7 +154,9 @@ interface APIs<M extends Models> {
     modelName: K
   ) => M[K] extends ModelType
     ? Readonly<getConsumerActionsType<Get<M[K], 'actions'>>>
-    : undefined
+    : M[K] extends API
+    ? M[K]['actions']
+    : unknown
   getInitialState: <T extends any>(
     context?: T | undefined
   ) => Promise<{
@@ -166,7 +172,11 @@ interface APIs<M extends Models> {
     actionName: keyof Get<M[K], 'actions'> | Array<keyof Get<M[K], 'actions'>>
   ) => void
   actions: {
-    [K in keyof M]: Readonly<getConsumerActionsType<Get<M[K], 'actions'>>>
+    [K in keyof M]: M[K] extends ModelType
+      ? Readonly<getConsumerActionsType<Get<M[K], 'actions'>>>
+      : M[K] extends API
+      ? M[K]['actions']
+      : unknown
   }
 }
 
@@ -179,14 +189,16 @@ type ModelType<InitStateType = any, ActionKeys = any> = {
 }
 
 // v3.0
-type NextModelType<InitStateType = any, ActionKeys = any> = {
+type NextModelType<InitStateType = any, ActionKeys = any, ExtContext = {}> = {
   actions: {
     [P in keyof ActionKeys]: NextAction<
       InitStateType,
       ActionKeys[P],
-      ActionKeys
+      ActionKeys,
+      ExtContext
     >
   }
+  middlewares?: Middleware[]
   state: InitStateType
   asyncState?: (context?: any) => Promise<Partial<InitStateType>>
 }

src/index.tsx

@@ -31,6 +31,9 @@ function Model<M extends Models, MT extends NextModelType>(
     Global.uid += 1
     const hash = '__' + Global.uid
     Global.State[hash] = models.state
+    if (models.middlewares) {
+      Global.Middlewares[hash] = models.middlewares
+    }
     const nextActions: Actions = Object.entries(models.actions).reduce(
       (o: { [name: string]: Action }, [name, action]) => {
         o[name] = async (state, actions, params) => {
@@ -63,6 +66,9 @@ function Model<M extends Models, MT extends NextModelType>(
     }
     Object.entries(models).forEach(([name, model]) => {
       if (!isAPI(model)) {
+        console.warn(
+          'we recommend you to use NextModel now, document link: https://github.com/byte-fe/react-model#model'
+        )
         if (!Global.State[name]) {
           Global.State[name] = model.state
         }
@@ -156,7 +162,11 @@ const getActions = (
           ...baseContext,
           Global
         }
-        await applyMiddlewares(actionMiddlewares, context)
+        if (Global.Middlewares[modelName]) {
+          await applyMiddlewares(Global.Middlewares[modelName], context)
+        } else {
+          await applyMiddlewares(actionMiddlewares, context)
+        }
       })
   )
   return updaters

tsconfig.json

@@ -15,6 +15,6 @@
     "sourceMap": false,
     "strict": true
   },
-  "include": ["src/**/*"],
+  "include": ["./src/*"],
   "exclude": ["node_modules"]
 }