mky-vent-base/src/hooks/system/useMethods.ts

import { defHttp } from '/@/utils/http/axios';
import { useMessage } from '/@/hooks/web/useMessage';
import { useGlobSetting } from '/@/hooks/setting';
import { message } from 'ant-design-vue';

const { createMessage, createWarningModal } = useMessage();
const glob = useGlobSetting();

/**
 * 导出文件xlsx的mime-type
 */
export const XLSX_MIME_TYPE = 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet';
/**
 * 导出文件xlsx的文件后缀
 */
export const XLSX_FILE_SUFFIX = '.xlsx';

export function useMethods() {
  /**
   * 导出xls
   * @param name
   * @param url
   */
  async function exportXls(name, url, params, isXlsx = false) {
    // 生成唯一loading key，避免多操作冲突
    const loadingKey = `export-xls-${Date.now()}`;
    // 显示loading，duration:0 表示不自动关闭
    message.loading({ content: '正在导出文件，请稍等...', key: loadingKey, duration: 0 });
    try {
      const data = await defHttp.get({ url: url, params: params, responseType: 'blob', timeout: 5 * 60 * 1000 }, { isTransformResponse: false });
      if (!data) {
        createMessage.warning('文件下载失败');
        return;
      }
      if (!name || typeof name != 'string') {
        name = '导出文件';
      }
      const blobOptions = { type: 'application/vnd.ms-excel' };
      let fileSuffix = '.xls';
      if (isXlsx === true) {
        blobOptions['type'] = XLSX_MIME_TYPE;
        fileSuffix = XLSX_FILE_SUFFIX;
      }
      if (typeof window.navigator.msSaveBlob !== 'undefined') {
        window.navigator.msSaveBlob(new Blob([data], blobOptions), name + fileSuffix);
      } else {
        const url = window.URL.createObjectURL(new Blob([data], blobOptions));
        const link = document.createElement('a');
        link.style.display = 'none';
        link.href = url;
        link.setAttribute('download', name + fileSuffix);
        document.body.appendChild(link);
        link.click();
        document.body.removeChild(link); //下载完成移除元素
        window.URL.revokeObjectURL(url); //释放掉blob对象
      }
      // 导出成功提示
      message.success({ content: '文件导出成功!', key: loadingKey, duration: 2 });
    } catch (error) {
      if (error && error['code'] == 'ECONNABORTED') {
        if (url.includes('/export/historydata') || url.includes('/ventanalyAlarmLog/exportXls')) {
          createMessage.error('导出的数据量太大，请选择合理的时间范围和间隔时间分批导出！');
        } else {
          createMessage.error('导出的数据量太大，导出失败！');
        }
      } else {
        createMessage.error('文件导出失败!');
      }
    } finally {
      message.destroy(loadingKey);
    }
  }

  /**
   * 导出xls post
   * @param name
   * @param url
   */
  async function exportXlsPost(name, url, params, isXlsx = false) {
    const loadingKey = 'export-xls-post';
    message.loading({ content: '正在导出，请稍等...', key: loadingKey, duration: 0 });
    defHttp
      .post({ url: url, params: params, timeout: 5 * 60 * 1000 }, { isTransformResponse: false })
      .then((data) => {
        if (data.code == 200 && data.result) {
          const messageArr = data.result.split('/');
          const fileUrl = encodeURIComponent(messageArr[messageArr.length - 1]);
          if (fileUrl) {
            const baseApiUrl = glob.domainUrl;
            // 下载文件
            const a = document.createElement('a');
            // 定义下载名称
            a.download = name;
            // 隐藏标签
            a.style.display = 'none';
            // 设置文件路径
            a.href = `${baseApiUrl}/sys/common/static/${fileUrl}`;
            // 将创建的标签插入dom
            document.body.appendChild(a);
            // 点击标签，执行下载
            a.click();
            // 将标签从dom移除
            document.body.removeChild(a);
            message.success({ content: '导出成功!', key: loadingKey, duration: 2 });
          } else {
            message.error({ content: '下载失败!', key: loadingKey, duration: 2 });
          }
        } else {
          message.error({ content: '下载失败!', key: loadingKey, duration: 2 });
        }
      })
      .catch((error) => {
        if (error && error['code'] == 'ECONNABORTED') {
          if (url.includes('/export/historydata') || url.includes('/ventanalyAlarmLog/exportXls')) {
            createMessage.error('导出的数据量太大，请选择合理的时间范围和间隔时间分批导出！');
          } else {
            createMessage.error('导出的数据量太大，导出失败！');
          }
        } else {
          message.error({ content: '下载失败!', key: loadingKey, duration: 2 });
        }
      })
      .finally(() => {
        message.destroy(loadingKey);
      });
  }

  async function exportXlsPost1(name, url, params, isXlsx = false) {
    const loadingKey = `export-xls-post1-${Date.now()}`;
    message.loading({ content: '正在导出文件，请稍等...', key: loadingKey, duration: 0 });

    try {
      const data = await defHttp.post({ url: url, data: params, responseType: 'blob' }, { isTransformResponse: false });
      if (!data) {
        createMessage.warning('文件下载失败');
        return;
      }
      if (!name || typeof name != 'string') {
        name = '导出文件';
      }
      const blobOptions = { type: 'application/vnd.ms-excel' };
      let fileSuffix = '.xls';
      if (isXlsx === true) {
        blobOptions['type'] = XLSX_MIME_TYPE;
        fileSuffix = XLSX_FILE_SUFFIX;
      }
      if (typeof window.navigator.msSaveBlob !== 'undefined') {
        window.navigator.msSaveBlob(new Blob([data], blobOptions), name + fileSuffix);
      } else {
        const url = window.URL.createObjectURL(new Blob([data], blobOptions));
        const link = document.createElement('a');
        link.style.display = 'none';
        link.href = url;
        link.setAttribute('download', name + fileSuffix);
        document.body.appendChild(link);
        link.click();
        document.body.removeChild(link); //下载完成移除元素
        window.URL.revokeObjectURL(url); //释放掉blob对象
      }
      message.success({ content: '文件导出成功!', key: loadingKey, duration: 2 });
    } catch (error) {
      if (error && error['code'] == 'ECONNABORTED') {
        if (url.includes('/export/historydata') || url.includes('/ventanalyAlarmLog/exportXls')) {
          createMessage.error('导出的数据量太大，请选择合理的时间范围和间隔时间分批导出！');
        } else {
          createMessage.error('导出的数据量太大，导出失败！');
        }
      } else {
        message.error({ content: '下载失败!', key: loadingKey, duration: 2 });
      }
    } finally {
      message.destroy(loadingKey);
    }
  }

  /**
   * 导入xls
   * @param data 导入的数据
   * @param url
   * @param success 成功后的回调
   */
  async function importXls(data, url, success) {
    const loadingKey = `import-xls-${Date.now()}`;
    message.loading({ content: '正在导入文件，请稍等...', key: loadingKey, duration: 0 });

    const isReturn = (fileInfo) => {
      try {
        if (fileInfo.code === 201) {
          const {
            message,
            result: { msg, fileUrl, fileName },
          } = fileInfo;
          const href = glob.uploadUrl + fileUrl;
          createWarningModal({
            title: message,
            centered: false,
            content: `<div>
                                <span>${msg}</span><br/> 
                                <span>具体详情请<a href = ${href} download = ${fileName}> 点击下载 </a> </span> 
                              </div>`,
          });
        } else if (fileInfo.code === 500 || fileInfo.code === 510) {
          createMessage.error(fileInfo.message || `${data.file.name} 导入失败`);
        } else {
          createMessage.success(fileInfo.message || `${data.file.name} 文件上传成功`);
        }
      } catch (error) {
        console.log('导入的数据异常', error);
        createMessage.error('文件导入异常!');
      } finally {
        // 关闭loading
        message.destroy(loadingKey);
        typeof success === 'function' ? success(fileInfo) : '';
      }
    };
    try {
      await defHttp.uploadFile({ url }, { file: data.file }, { success: isReturn });
    } catch (error) {
      // 异常时强制关闭loading
      message.destroy(loadingKey);
      createMessage.error('文件导入失败!');
      console.error('导入失败:', error);
    }
  }

  return {
    handleExportXls: (name: string, url: string, params?: object) => exportXls(name, url, params),
    handleExportXlsPost: (name: string, url: string, params?: object) => exportXlsPost(name, url, params),
    exportXlsGetBlob: (name: string, url: string, params?: object) => exportXls(name, url, params),
    exportXlsPostBlob: (name: string, url: string, params?: object) => exportXlsPost1(name, url, params),
    handleImportXls: (data, url, success) => importXls(data, url, success),
    handleExportXlsx: (name: string, url: string, params?: object) => exportXls(name, url, params, true),
  };
}