feat(upload-client): export isReadyPoll helper to wait for the file…

… readiness
uploadcare · Oct 30, 2023 · 12cf9b1 · 12cf9b1
1 parent 4a717e1
commit 12cf9b1
Show file tree

Hide file tree

Showing 6 changed files with 32 additions and 22 deletions.
diff --git a/packages/upload-client/src/UploadClient.ts b/packages/upload-client/src/UploadClient.ts
@@ -27,6 +27,7 @@ import { FileFromOptions, uploadFile } from './uploadFile'
 import { FileInfo, GroupId, GroupInfo, Token, Url, Uuid } from './api/types'
 import { Settings, Sliceable, SupportedFileInput } from './types'
 import { GroupFromOptions, uploadFileGroup } from './uploadFileGroup'
+import { IsReadyPoolOptions, isReadyPoll } from './tools/isReadyPoll'
 
 /** Populate options with settings. */
 const populateOptionsWithSettings = <T>(
@@ -155,4 +156,13 @@ export default class UploadClient {
 
     return uploadFileGroup(data, populateOptionsWithSettings(options, settings))
   }
+
+  isReadyPoll(
+    uuid: Uuid,
+    options: Partial<IsReadyPoolOptions> = {}
+  ): Promise<FileInfo> {
+    const settings = this.getSettings()
+
+    return isReadyPoll(uuid, populateOptionsWithSettings(options, settings))
+  }
 }
diff --git a/packages/upload-client/src/index.ts b/packages/upload-client/src/index.ts
@@ -103,3 +103,4 @@ export {
   ComputableProgressInfo,
   UnknownProgressInfo
 } from './api/types'
+export { isReadyPoll } from './tools/isReadyPoll'
diff --git a/packages/upload-client/src/tools/isReadyPoll.ts b/packages/upload-client/src/tools/isReadyPoll.ts
@@ -2,12 +2,12 @@ import info from '../api/info'
 import {
   ComputableProgressInfo,
   FileInfo,
-  ProgressCallback
+  ProgressCallback,
+  Uuid
 } from '../api/types'
 import { CustomUserAgent, poll } from '@uploadcare/api-client-utils'
 
-type ArgsIsReadyPool = {
-  file: string
+export type IsReadyPoolOptions = {
   publicKey: string
   baseURL?: string
   source?: string
@@ -19,21 +19,23 @@ type ArgsIsReadyPool = {
   signal?: AbortSignal
 }
 
-function isReadyPoll({
-  file,
-  publicKey,
-  baseURL,
-  source,
-  integration,
-  userAgent,
-  retryThrottledRequestMaxTimes,
-  retryNetworkErrorMaxTimes,
-  signal,
-  onProgress
-}: ArgsIsReadyPool): FileInfo | PromiseLike<FileInfo> {
+function isReadyPoll(
+  uuid: Uuid,
+  {
+    publicKey,
+    baseURL,
+    source,
+    integration,
+    userAgent,
+    retryThrottledRequestMaxTimes,
+    retryNetworkErrorMaxTimes,
+    signal,
+    onProgress
+  }: IsReadyPoolOptions
+): Promise<FileInfo> {
   return poll<FileInfo>({
     check: (signal) =>
-      info(file, {
+      info(uuid, {
         publicKey,
         baseURL,
         signal,

diff --git a/packages/upload-client/src/uploadFile/uploadDirect.ts b/packages/upload-client/src/uploadFile/uploadDirect.ts
@@ -78,8 +78,7 @@ export const uploadDirect = (
     metadata
   })
     .then(({ file }) => {
-      return isReadyPoll({
-        file,
+      return isReadyPoll(file, {
         publicKey,
         baseURL,
         source,

diff --git a/packages/upload-client/src/uploadFile/uploadFromUrl.ts b/packages/upload-client/src/uploadFile/uploadFromUrl.ts
@@ -233,8 +233,7 @@ export const uploadFromUrl = (
       return result
     })
     .then((result) =>
-      isReadyPoll({
-        file: result.uuid,
+      isReadyPoll(result.uuid, {
         publicKey,
         baseURL,
         integration,

diff --git a/packages/upload-client/src/uploadFile/uploadMultipart.ts b/packages/upload-client/src/uploadFile/uploadMultipart.ts
@@ -181,8 +181,7 @@ export const uploadMultipart = async (
       if (fileInfo.isReady) {
         return fileInfo
       } else {
-        return isReadyPoll({
-          file: fileInfo.uuid,
+        return isReadyPoll(fileInfo.uuid, {
           publicKey,
           baseURL,
           source,