fix: useLocalStorage types for functional updates

Fix useLocalStorage types for functional updates
streamich · Jul 10, 2019 · e2b8278 · e2b8278
2 parents 4730da6 + 5f5376c
commit e2b8278
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/useLocalStorage.ts b/src/useLocalStorage.ts
@@ -1,7 +1,10 @@
 import { useEffect, useState } from 'react';
 import { isClient } from './util';
 
-const useLocalStorage = <T>(key: string, initialValue?: T, raw?: boolean): [T, (value: T) => void] => {
+type Dispatch<A> = (value: A) => void;
+type SetStateAction<S> = S | ((prevState: S) => S);
+
+const useLocalStorage = <T>(key: string, initialValue?: T, raw?: boolean): [T, Dispatch<SetStateAction<T>>] => {
   if (!isClient) {
     return [initialValue as T, () => {}];
   }