toast.promise accept function

fkhadra · Aug 27, 2021 · a38fde4 · a38fde4
1 parent f154fdd
commit a38fde4
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/core/toast.tsx b/src/core/toast.tsx
@@ -128,7 +128,7 @@ interface ToastPromiseParams {
 }
 
 function handlePromise<T>(
-  promise: Promise<T>,
+  promise: Promise<T> | (() => Promise<T>),
   { pending, error, success }: ToastPromiseParams,
   options?: ToastOptions
 ) {
@@ -161,11 +161,12 @@ function handlePromise<T>(
     });
     return result;
   };
-  promise
-    .then(result => resolver('success', success, result))
-    .catch(err => resolver('error', error, err));
+  const p = isFn(promise) ? promise() : promise;
+  p.then(result => resolver('success', success, result)).catch(err =>
+    resolver('error', error, err)
+  );
 
-  return promise;
+  return p;
 }
 
 toast.promise = handlePromise;