feat(in-app/v1): safe create object url

leancloud · Mar 20, 2024 · db374dc · db374dc
1 parent 0712722
commit db374dc
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/in-app/v1/src/utils/useUpload.ts b/in-app/v1/src/utils/useUpload.ts
@@ -44,7 +44,7 @@ export function useUpload({ onError }: UseUploadOptions = {}): UseUploadResult {
       const fileInfo: FileInfo<number> = {
         key,
         name: file.name,
-        url: URL.createObjectURL(file),
+        url: createObjectURL(file),
         progress: 0,
       };
 
@@ -80,7 +80,7 @@ export function useUpload({ onError }: UseUploadOptions = {}): UseUploadResult {
         .then((file) => {
           update({
             id: file.id,
-            // url: file.url,
+            url: fileInfo.url ?? file.url,
             mime: file.mime,
             progress: undefined,
           });
@@ -102,3 +102,9 @@ export function useUpload({ onError }: UseUploadOptions = {}): UseUploadResult {
 
   return { files, isUploading: uploadingCount > 0, upload, remove, removeAll };
 }
+
+function createObjectURL(obj: Blob) {
+  if (typeof URL === 'function' && typeof URL.createObjectURL === 'function') {
+    return URL.createObjectURL(obj);
+  }
+}