0508修改

src/utils/download.ts

@@ -1,12 +1,38 @@
-const download0 = (data: Blob, fileName: string, mineType: string) => {
+export interface DownloadSource {
+  data: BlobPart
+  fileName?: string
+}
+
+const resolveDownloadPayload = (source: BlobPart | DownloadSource, fallbackFileName?: string) => {
+  if (
+    source &&
+    typeof source === 'object' &&
+    !(source instanceof Blob) &&
+    !(source instanceof ArrayBuffer) &&
+    'data' in source
+  ) {
+    const resolvedSource = source as DownloadSource
+    return {
+      data: resolvedSource.data,
+      fileName: resolvedSource.fileName || fallbackFileName || 'download'
+    }
+  }
+  return {
+    data: source as BlobPart,
+    fileName: fallbackFileName || 'download'
+  }
+}
+
+const download0 = (source: BlobPart | DownloadSource, fileName: string | undefined, mineType: string) => {
+  const payload = resolveDownloadPayload(source, fileName)
   // 创建 blob
-  const blob = new Blob([data], { type: mineType })
+  const blob = new Blob([payload.data], { type: mineType })
   // 创建 href 超链接，点击进行下载
   window.URL = window.URL || window.webkitURL
   const href = URL.createObjectURL(blob)
   const downA = document.createElement('a')
   downA.href = href
-  downA.download = fileName
+  downA.download = payload.fileName
   downA.click()
   // 销毁超连接
   window.URL.revokeObjectURL(href)
@@ -14,27 +40,27 @@ const download0 = (data: Blob, fileName: string, mineType: string) => {
 
 const download = {
   // 下载 Excel 方法
-  excel: (data: Blob, fileName: string) => {
+  excel: (data: BlobPart | DownloadSource, fileName?: string) => {
     download0(data, fileName, 'application/vnd.ms-excel')
   },
   // 下载 Word 方法
-  word: (data: Blob, fileName: string) => {
+  word: (data: BlobPart | DownloadSource, fileName?: string) => {
     download0(data, fileName, 'application/msword')
   },
   // 下载 Zip 方法
-  zip: (data: Blob, fileName: string) => {
+  zip: (data: BlobPart | DownloadSource, fileName?: string) => {
     download0(data, fileName, 'application/zip')
   },
   // 下载 Html 方法
-  html: (data: Blob, fileName: string) => {
+  html: (data: BlobPart | DownloadSource, fileName?: string) => {
     download0(data, fileName, 'text/html')
   },
   // 下载 Markdown 方法
-  markdown: (data: Blob, fileName: string) => {
+  markdown: (data: BlobPart | DownloadSource, fileName?: string) => {
     download0(data, fileName, 'text/markdown')
   },
   // 下载 Json 方法
-  json: (data: Blob, fileName: string) => {
+  json: (data: BlobPart | DownloadSource, fileName?: string) => {
     download0(data, fileName, 'application/json')
   },
   // 下载图片（允许跨域）