Types fix, useAsync hook + removing negation

src/index.ts

@@ -1,19 +1,37 @@
-import {DependencyList, useEffect, useState} from 'react'
+import { DependencyList, useRef, useEffect, useState, useReducer, useMemo } from 'react'
+
+export function useAsyncMemo<T>(factory: () => Promise<T>, deps?: DependencyList, initial?: T) {
+  let [val, setVal] = useState(initial)
 
-export function useAsyncMemo<T>(factory: () => Promise<T> | undefined | null, deps: DependencyList, initial: T = undefined): T {
-  const [val, setVal] = useState<T>(initial)
   useEffect(() => {
-    let cancel = false
-    const promise = factory()
-    if (promise === undefined || promise === null) return
-    promise.then((val) => {
-      if (!cancel) {
-        setVal(val)
+    let pending = true
+    factory().then(res => pending && setVal(res))
+
+    return () => {
+      pending = false
+    }
+  }, deps)
+
+  return val
+}
+
+export function useAsync<T>(factory: () => Promise<T>, deps?: DependencyList, initial?: T) {
+  let forceUpdate = useReducer(x => !x, false)[1]
+  let val = useRef(initial)
+
+  useMemo(() => val.current = initial, deps)
+  useEffect(() => {
+    let pending = true
+    factory().then(res => {
+      if(pending) {
+        val.current = res
+        forceUpdate()
       }
     })
     return () => {
-      cancel = true
+      pending = false
     }
   }, deps)
-  return val
+
+  return val.current
 }