feat: improve useSetState() typings

Fix useSetState callback parameter type
streamich · May 30, 2019 · fad4440 · fad4440
2 parents e63ef6e + eeb05e9
commit fad4440
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/useSetState.ts b/src/useSetState.ts
@@ -1,6 +1,6 @@
 import { useState } from 'react';
 
-const useSetState = <T extends object>(initialState: T = {} as T): [T, (patch: Partial<T> | (() => void)) => void] => {
+const useSetState = <T extends object>(initialState: T = {} as T): [T, (patch: Partial<T> | ((prevState: T) => Partial<T>)) => void] => {
   const [state, set] = useState<T>(initialState);
   const setState = patch => {
     set(prevState => Object.assign({}, prevState, patch instanceof Function ? patch(prevState) : patch));