feat: Typings for useDefault

streamich · Jan 20, 2020 · fa0f53b · fa0f53b
2 parents 1ec99eb + 83f4c46
commit fa0f53b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/useDefault.ts b/src/useDefault.ts
@@ -1,13 +1,13 @@
 import { useState } from 'react';
 
-const useDefault = (defaultValue, initialValue): [any, (nextValue?: any) => void] => {
-  const [value, setValue] = useState(initialValue);
+const useDefault = <TStateType>(defaultValue: TStateType, initialValue: TStateType | (() => TStateType)) => {
+  const [value, setValue] = useState<TStateType | undefined | null>(initialValue);
 
   if (value === undefined || value === null) {
-    return [defaultValue, setValue];
+    return [defaultValue, setValue] as const;
   }
 
-  return [value, setValue];
+  return [value, setValue] as const;
 };
 
 export default useDefault;