[FEATURE] Passing a function for draft or value for direct state usage (

#76)

* Enables passing a direct value to the useCallback hook for using it as simple state

* Fixed typing error

src/index.ts

@@ -12,14 +12,17 @@ export type ImmerHook<S> = [S, Updater<S>];
 
 export function useImmer<S = any>(
   initialValue: S | (() => S)
-): [S, (f: (draft: Draft<S>) => void | S) => void];
+): [S, (f: ((draft: Draft<S> | S) => void) | S) => void];
 
 export function useImmer(initialValue: any) {
   const [val, updateValue] = useState(initialValue);
   return [
     val,
     useCallback(updater => {
-      updateValue(produce(updater));
+      if(typeof updater === "function")
+        updateValue(produce(updater));
+      else
+        updateValue(updater);
     }, [])
   ];
 }