@lingxiteam/ebe-utils 0.0.25 → 0.0.29

package/lib/pcpublic/src/utils/exportCustomData/tool.ts

@@ -0,0 +1,399 @@
+import {
+  closeProgressMsg,
+  openProgressMsg,
+  showProgressNotification,
+} from '@/components/ProgressComp';
+import { APPID } from '@/constants';
+import engineServices from '@/services/api/engine';
+import fileServices from '@/services/api/file';
+import { messageApi } from '@/utils/messageApi';
+import security from '@/utils/Security';
+/**
+ * 文件保存方法
+ * @param data
+ * @param filename
+ */
+const saveBlobFile = ({ data, fileName }: any) => {
+  if (typeof (window as any).chrome !== 'undefined') {
+    // chrome
+    const link = document.createElement('a');
+    let href = window.URL.createObjectURL(new Blob([data]));
+    const format = typeof fileName === 'string' && fileName.split('.').pop();
+    if (/(csv|xls)$/.test(format as string) && typeof data === 'string') {
+      href = `data:text/${format};charset=utf-8,\ufeff${encodeURIComponent(
+        data,
+      )}`;
+    }
+    link.href = href; // eslint-disable-line
+    link.download = fileName;
+    link.click();
+  } else if (typeof (window as any).navigator.msSaveBlob !== 'undefined') {
+    // ie
+    const blob = new Blob([data], {
+      type: 'application/force-download',
+    }); // eslint-disable-line
+    (window as any).navigator.msSaveBlob(blob, fileName);
+  } else {
+    // firefox
+    const file = new File([data], fileName, {
+      type: 'application/force-download',
+    }); // eslint-disable-line
+    window.open(URL.createObjectURL(file));
+  }
+};
+
+const getFileName = (fileContentStr: string, newFileName?: string) => {
+  let fileName: string = '';
+  let suffix: string = '';
+  if (fileContentStr) {
+    fileName =
+      fileContentStr.split('filename=').pop()?.replace(/\"/g, '') || '';
+    const result = fileContentStr.match(
+      /(?:.*filename\*|filename)=(?:([^'"]*)''|("))([^;]+)\2(?:[;`\n]|$)/,
+    );
+    if (result && result[3]) {
+      fileName = result[3].replace(/['"]/g, '').replace(/[+]/g, '%20');
+    }
+    suffix = fileName.split('.').pop() || '';
+    fileName = decodeURIComponent(fileName);
+  }
+
+  if (newFileName) {
+    if (!/\./.test(newFileName)) {
+      fileName = `${newFileName}.${suffix}`;
+    } else {
+      fileName = newFileName;
+    }
+  }
+  return fileName;
+};
+
+const isErrorStatus = (status: number) => {
+  switch (status) {
+    case 400:
+    case 401:
+    case 403:
+    case 404:
+    case 500:
+    case 501:
+    case 502:
+    case 503:
+    case 413:
+      return true;
+    default:
+      return false;
+  }
+};
+
+const exportServiceMap = {
+  sql: engineServices.exportSqlDatasByAsync,
+  object: engineServices.exportInstsByAsync,
+  multiExport: engineServices.exportMultiServiceResultByAsync,
+};
+
+const exportPathMap = {
+  sql: engineServices.exportSqlDatasApiPath(),
+  object: engineServices.exportInstsApiPath(),
+  multiExport: engineServices.exportMultiServiceResultPath(),
+};
+const downloadByXMLHttpRequest = (params: {
+  newFileName?: string;
+  onError?: any;
+  methodPath: string;
+  downloadIndex?: string;
+}) => {
+  const { newFileName, onError, methodPath, downloadIndex } = params;
+
+  if (window.XMLHttpRequest) {
+    const xhr = new XMLHttpRequest(); // ActiveXObject只存在于IE7及以下，无需兼容
+    xhr.open('GET', methodPath, true);
+    xhr.responseType = 'blob'; // 设置接受类型
+    const headers: any = {
+      'Content-Type': 'application/json',
+      // 'X-B-TARGET-ID': pageId,
+      'X-B-AUTH': '1',
+      'APP-ID': APPID,
+    };
+    Object.keys(headers).forEach((header) => {
+      xhr.setRequestHeader(header, headers[header]); // 设置请求头参数
+    });
+    xhr.setRequestHeader(
+      'X-SIGN',
+      security.createHttpSignStr(methodPath, {
+        method: 'GET',
+        headers,
+        body: '',
+        search: '',
+      }),
+    );
+    xhr.withCredentials = true;
+    xhr.onprogress = (evt) => {
+      // 展示下载进度条
+      if (evt.lengthComputable) {
+        const percentComplete = Math.round((evt.loaded * 100) / evt.total);
+        openProgressMsg(downloadIndex, newFileName, percentComplete, '下载');
+      }
+    };
+    xhr.onload = async () => {
+      const data = xhr.response; // 获取响应体数据
+      if (xhr.status === 200) {
+        const fileName: string = getFileName(
+          xhr.getResponseHeader('Content-Disposition') || '',
+          newFileName,
+        );
+        saveBlobFile({
+          data,
+          fileName,
+        });
+      } else if (isErrorStatus(xhr.status)) {
+        messageApi.error('下载失败');
+      } else {
+        const errMessage = data;
+        let errorMessage = '';
+        try {
+          errorMessage = JSON.parse(errMessage)?.message || '下载失败';
+        } catch {
+          errorMessage = '下载失败';
+        }
+        messageApi.error(errorMessage);
+        if (typeof onError === 'function') {
+          onError(errorMessage);
+        }
+      }
+      closeProgressMsg(downloadIndex);
+    };
+    xhr.ontimeout = () => {
+      xhr.abort(); // 取消请求
+    };
+    xhr.send();
+  }
+};
+
+/**
+ * 批量文件下载
+ * @param fileId
+ * @param newFileName
+ * @param onError
+ * @param messageApi
+ * @param zip
+ */
+const batchDownloadFileByIds = (params: any) => {
+  const { fileIds, zip } = params;
+
+  downloadByXMLHttpRequest({
+    methodPath: fileServices.batchDownloadFileByIds(fileIds, { zip }),
+    downloadIndex: `${fileIds}_${Math.random()}`,
+    ...params,
+  });
+};
+// 异步导出文件并显示进度
+export const exportFileShowProgressAsync = async (exportParams: any) => {
+  const {
+    fileOrigin,
+    fileName,
+    params = {},
+    onSuccess,
+    onFail,
+    downloadIndex,
+  } = exportParams;
+  let timeoutId: number | null = null;
+  let percent = 0;
+  let isStop = false;
+  // @ts-ignore
+  const serviceApi = exportServiceMap[fileOrigin];
+  const applyId = await serviceApi(params, {});
+  const messageConfig = {
+    title: fileName ? `导出-${fileName}` : undefined,
+    key: downloadIndex,
+    onCancel: () => {
+      if (timeoutId) {
+        isStop = true;
+        clearTimeout(timeoutId);
+      }
+    },
+  };
+  const requestProgress = async (isPolled = true) => {
+    try {
+      const { fileId, statusCd, requestJson, responseJson, failReason } =
+        await engineServices.getImportExportApply({ applyId }, {});
+      // 点击取消时候刚好触发了下一次接口，接口未回调时
+      if (isStop) return -1;
+      let requestObj: {
+        descriptor?: {
+          [key: string]: any;
+        };
+      } = {};
+      let responseObj: {
+        total?: number;
+        currentCount?: number;
+      } = {};
+      try {
+        requestObj = JSON.parse(requestJson) || {};
+        responseObj = JSON.parse(responseJson) || {};
+      } catch (e) {
+        console.log(e);
+      }
+      let showLoading: boolean = false;
+      const { total = 0, currentCount = 0 } = responseObj;
+      if (!responseJson || (percent === 100 && statusCd === 1)) {
+        // 上一次导出进程还未结束，无法返回进度
+        // 已经导出完全但进程还未结束
+        showLoading = true;
+      } else {
+        percent = +((+currentCount / +total) * 100).toFixed(2);
+      }
+      showProgressNotification({
+        ...messageConfig,
+        message: showLoading
+          ? undefined
+          : `总记录${total}条，已生成数据${currentCount}条，剩余${
+              total - currentCount
+            }条完成到处。`,
+        percent,
+        showLoading,
+        loadingText:
+          percent === 100
+            ? 'export.process.downloading'
+            : 'export.process.handling',
+      });
+      if (statusCd === 1 && isPolled) {
+        polledRequest();
+      } else if (statusCd === 10) {
+        onSuccess?.();
+        showProgressNotification({
+          ...messageConfig,
+          percent: 100,
+          btn: () => null,
+          duration: 1,
+        });
+        if (fileId) {
+          const newFileName: string | undefined =
+            fileName || requestObj?.descriptor?.filename;
+          batchDownloadFileByIds({
+            fileIds: fileId,
+            newFileName,
+            onError: (errorMessage: any) => {
+              messageApi.error(errorMessage);
+            },
+          });
+        }
+      } else if (statusCd === -1) {
+        showProgressNotification({
+          ...messageConfig,
+          type: 'error',
+          percent,
+          message: failReason || '导出失败，稍后请重试。',
+          btn: () => null,
+          duration: 2,
+        });
+        onFail?.();
+      }
+      return statusCd;
+    } catch (e) {
+      onFail?.();
+      return -1;
+    }
+  };
+  const polledRequest = (requestFlag = false) => {
+    if (applyId && (timeoutId || requestFlag)) {
+      timeoutId = window.setTimeout(async () => {
+        await requestProgress();
+      }, 1500);
+    }
+  };
+  const status = await requestProgress(false);
+  if (status === 1) {
+    // 如果未导出则定时轮询
+    polledRequest(true);
+  }
+};
+// 导出文件显示进度条
+export const exportFileShowProgress = async (exportParams: {
+  fileOrigin: any;
+  methodType?: 'POST' | undefined;
+  fileName: any;
+  params?: {} | undefined;
+  onSuccess?: any;
+  onFail?: any;
+  downloadIndex: any;
+  async: any;
+}) => {
+  const {
+    fileOrigin,
+    methodType = 'POST',
+    fileName,
+    params = {},
+    onSuccess,
+    onFail,
+    downloadIndex,
+    async,
+  } = exportParams;
+
+  const _downloadIndex = downloadIndex || `${fileName}_${Math.random()}`;
+  if (async === 'async') {
+    await exportFileShowProgressAsync({
+      ...exportParams,
+      downloadIndex: _downloadIndex,
+    });
+    return;
+  }
+
+  if (fileOrigin && window.XMLHttpRequest) {
+    const xhr = new XMLHttpRequest(); // ActiveXObject只存在于IE7及以下，无需兼容
+    // @ts-ignore
+    const methodPath = exportPathMap[fileOrigin];
+    xhr.open(methodType, methodPath, true);
+    xhr.responseType = 'blob'; // 设置接受类型
+    const headers: any = {
+      'Content-Type': 'application/json',
+      //   'X-B-TARGET-ID': pageId,
+      'X-B-AUTH': '1',
+      'APP-ID': APPID,
+    };
+    Object.keys(headers).forEach((header) => {
+      xhr.setRequestHeader(header, headers[header]); // 设置请求头参数
+    });
+    xhr.setRequestHeader(
+      'X-SIGN',
+      security.createHttpSignStr(methodPath, {
+        method: methodType,
+        headers,
+        body: JSON.stringify(params) || '',
+        search: '',
+      }),
+    );
+    xhr.withCredentials = true;
+    xhr.onprogress = (evt) => {
+      // 展示下载进度条
+      if (evt.lengthComputable) {
+        const percentComplete = Math.round((evt.loaded * 100) / evt.total);
+        openProgressMsg(_downloadIndex, fileName, percentComplete, '导出');
+      }
+    };
+    xhr.onload = async () => {
+      const data = xhr.response; // 获取响应体数据
+      if (xhr.status === 200) {
+        saveBlobFile({
+          data,
+          fileName: getFileName(
+            xhr.getResponseHeader('Content-Disposition') || '',
+            fileName,
+          ),
+        });
+        if (typeof onSuccess === 'function') {
+          onSuccess();
+        }
+      } else if (typeof onFail === 'function') {
+        onFail();
+      } else {
+        messageApi.error('导出失败');
+      }
+      closeProgressMsg(_downloadIndex);
+    };
+    xhr.ontimeout = () => {
+      closeProgressMsg(_downloadIndex);
+      xhr.abort(); // 取消请求
+    };
+    xhr.send(JSON.stringify(params));
+  }
+};

package/lib/pcpublic/src/utils/messageApi.ts

@@ -68,26 +68,31 @@ const transformContent = (value: string) => {
   }
   return finalContent;
 };
-const messageApi = (
-  type: 'info' | 'success' | 'error' | 'warning' | 'loading' | 'warn',
-  content: string,
-  duration?: number,
-) => {
-  switch (type) {
-    case 'info':
-      return message.info(transformContent(content), duration);
-    case 'success':
-      return message.success(transformContent(content), duration);
-    case 'error':
-      return message.error(transformContent(content), duration);
-    case 'warning':
-    case 'warn':
-      return message.warning(transformContent(content), duration);
-    case 'loading':
-      return message.loading(transformContent(content), duration);
-    default:
-      return message.info(transformContent(content), duration);
-  }
+const messageApi = {
+  info: (content: string, duration?: number) => {
+    return message.info(transformContent(content), duration);
+  },
+  success: (content: string, duration?: number) => {
+    return message.success(transformContent(content), duration);
+  },
+  error: (content: string, duration?: number) => {
+    return message.error(transformContent(content), duration);
+  },
+  warning: (content: string, duration?: number) => {
+    return message.warning(transformContent(content), duration);
+  },
+  warn: (content: string, duration?: number) => {
+    return message.warning(transformContent(content), duration);
+  },
+  loading: (content: string, duration?: number) => {
+    return message.loading(transformContent(content), duration);
+  },
+  destroy: () => {
+    return message.destroy();
+  },
+  hidden: () => {
+    return message.destroy();
+  },
 };
 
 const MessageApiModal = Modal;