feat:优化代码

yzr041215 · Dec 17, 2023 · 9fc7686 · 9fc7686
1 parent 67b8957
commit 9fc7686
Showing 1 changed file with 45 additions and 10 deletions.
diff --git a/src/utils/common.js b/src/utils/common.js
@@ -73,17 +73,48 @@ export async function clipboardImage(src, callback) {
     const data = await fetch(src)
     const blob = await data.blob()
 
-    if (blob.type != 'image/png') {
-      return alert('当前图片类型不支持复制')
-    }
+    // navigator.clipboard.write 仅支持 png 图片
+    if (blob.type == 'image/png') {
+      await navigator.clipboard.write([
+        new ClipboardItem({
+          [blob.type]: blob
+        })
+      ])
 
-    await navigator.clipboard.write([
-      new ClipboardItem({
-        [blob.type]: blob
-      })
-    ])
+      return callback()
+    }
 
-    callback()
+    const objectURL = URL.createObjectURL(blob)
+
+    const img = new Image()
+    img.src = URL.createObjectURL(blob)
+    img.onload = () => {
+      const canvas = document.createElement('canvas')
+      const ctx = canvas.getContext('2d')
+
+      canvas.width = img.width
+      canvas.height = img.height
+      ctx.drawImage(img, 0, 0)
+
+      canvas.toBlob(
+        (blob) => {
+          const data = [new ClipboardItem({ [blob.type]: blob })]
+
+          navigator.clipboard
+            .write(data)
+            .then(() => {
+              callback()
+            })
+            .catch((err) => {
+              console.error(err)
+            })
+
+          URL.revokeObjectURL(objectURL)
+        },
+        'image/png',
+        1
+      )
+    }
   } catch (err) {
     console.error(err.name, err.message)
   }
@@ -129,7 +160,9 @@ export function getVideoImage(file, time = 1) {
   return new Promise((resolve) => {
     let video = document.createElement('video')
 
-    video.src = URL.createObjectURL(file)
+    const objectURL = URL.createObjectURL(file)
+
+    video.src = objectURL
     video.currentTime = time
     video.autoplay = true
     video.muted = true
@@ -156,6 +189,8 @@ export function getVideoImage(file, time = 1) {
           lastModified: Date.now()
         })
 
+        URL.revokeObjectURL(objectURL)
+
         resolve(data)
       }, 'image/jpeg')
     }