添加initialAction参数支持;更新测试用例

Author: Saber2pr

README.md

@@ -7,15 +7,16 @@ npm i @saber2pr/use-reducer
 # API
 
 ```ts
-const state = {
-  div1: 100,
-  div2: 1
+export const initState = {
+  current: 100,
+  delta: 1
 }
 
 export const [getState, useReducer] = createStore(state)
 
 // useReducer
-const [state, dispatch] = useReducer(reducer, initState)
+const [state, dispatch] = useReducer(reducer, initialState)
+// const [state, dispatch] = useReducer(reducer, initialState, initialAction)
 
 // getState
 const globalState = getState()

src/core/createStore.ts

@@ -12,7 +12,8 @@ export type dispatch<A> = (action: A) => void
 
 export type useReducer<S> = <A>(
   reducer: Reducer<S, A>,
-  initialState: S
+  initialState: S,
+  initialAction?: A
 ) => [S, dispatch<A>]
 
 /**
@@ -29,15 +30,19 @@ export function createStore<S>(
   const __STORE: S = initialState
   const getState: getState<S> = () => __STORE
 
-  const useReducer: useReducer<S[keyof S]> = (reducer, initialState) => {
-    const [state, setState] = React.useState(initialState)
-    return [
-      state,
-      action =>
-        setState(
-          (__STORE[reducer.name] = reducer(state || initialState, action))
-        )
-    ]
+  const useReducer: useReducer<S[keyof S]> = (
+    reducer,
+    initialState,
+    initialAction?
+  ) => {
+    const initState = initialAction
+      ? reducer(initialState, initialAction)
+      : initialState
+
+    const [state, setState] = React.useState(initState)
+    __STORE[reducer.name] = state
+
+    return [state, action => setState(reducer(state, action))]
   }
 
   return [getState, useReducer]

src/test/Delta.tsx

@@ -3,7 +3,7 @@ import { useReducer } from './store/store'
 import { delta } from './store/delta'
 
 export const Delta = ({ init }: { init: number }) => {
-  const [state, dispatch] = useReducer(delta, init)
+  const [state, dispatch] = useReducer(delta, init, { type: 'change' })
   return (
     <div>
       delta:{state}