feat: 封装下载hooks

src/hooks/business/download.ts

@@ -0,0 +1,102 @@
+import { localStg } from '@/utils/storage';
+import { getServiceBaseURL } from '@/utils/service';
+import { errorCodeRecord } from '@/constants/common';
+
+export function useDownload() {
+  const isHttpProxy = import.meta.env.DEV && import.meta.env.VITE_HTTP_PROXY === 'Y';
+  const { baseURL } = getServiceBaseURL(import.meta.env, isHttpProxy);
+
+  function downloadByData(data: BlobPart, filename: string, type: string = 'application/octet-stream') {
+    const blobData = [data];
+    const blob = new Blob(blobData, { type });
+
+    const blobURL = window.URL.createObjectURL(blob);
+    const tempLink = document.createElement('a');
+    tempLink.style.display = 'none';
+    tempLink.href = blobURL;
+    tempLink.setAttribute('download', filename);
+    if (typeof tempLink.download === 'undefined') {
+      tempLink.setAttribute('target', '_blank');
+    }
+    document.body.appendChild(tempLink);
+    tempLink.click();
+    document.body.removeChild(tempLink);
+    window.URL.revokeObjectURL(blobURL);
+  }
+
+  function download(url: string, params: any, fileName: string) {
+    window.$loading?.startLoading();
+    const token = localStg.get('token');
+    const clientId = import.meta.env.VITE_APP_CLIENT_ID;
+    const now = new Date().getTime();
+    const formData = new FormData();
+    Object.keys(params).forEach(key => formData.append(key, params[key]));
+    fetch(`${baseURL}${url}?t=${now}`, {
+      method: 'post',
+      body: formData,
+      headers: {
+        Authorization: `Bearer ${token}`,
+        Clientid: clientId!,
+        'Content-Type': 'application/x-www-form-urlencoded'
+      }
+    })
+      .then(async response => {
+        if (response.headers.get('Content-Type')?.includes('application/json')) {
+          const res = await response.json();
+          const code = res.code as CommonType.ErrorCode;
+          throw new Error(errorCodeRecord[code] || res.msg || errorCodeRecord.default);
+        }
+        return response.blob();
+      })
+      .then(data => downloadByData(data, fileName, 'application/zip'))
+      .catch(err => window.$message?.error(err.message))
+      .finally(() => window.$loading?.endLoading());
+  }
+
+  function oss(ossId: CommonType.IdType) {
+    window.$loading?.startLoading();
+    const token = localStg.get('token');
+    const clientId = import.meta.env.VITE_APP_CLIENT_ID;
+    const url = `/resource/oss/download/${ossId}`;
+    const now = new Date().getTime();
+    let fileName = String(`${ossId}-${now}`);
+    fetch(`${baseURL}${url}?t=${now}`, {
+      method: 'get',
+      headers: {
+        Authorization: `Bearer ${token}`,
+        Clientid: clientId!
+      }
+    })
+      .then(async response => {
+        fileName = String(response.headers.get('Download-Filename'));
+        return response.blob();
+      })
+      .then(data => downloadByData(data, fileName))
+      .catch(err => window.$message?.error(err.message))
+      .finally(() => window.$loading?.endLoading());
+  }
+
+  function zip(url: string, fileName: string) {
+    window.$loading?.startLoading();
+    const token = localStg.get('token');
+    const clientId = import.meta.env.VITE_APP_CLIENT_ID;
+    const now = new Date().getTime();
+    fetch(`${baseURL}${url}${url.includes('?') ? '&' : '?'}t=${now}`, {
+      method: 'get',
+      headers: {
+        Authorization: `Bearer ${token}`,
+        Clientid: clientId!
+      }
+    })
+      .then(async response => response.blob())
+      .then(data => downloadByData(data, fileName, 'application/zip'))
+      .catch(err => window.$message?.error(err.message))
+      .finally(() => window.$loading?.endLoading());
+  }
+
+  return {
+    oss,
+    zip,
+    download
+  };
+}