SPDM/src/api/request.ts

import axios from 'axios';
import { ElMessage } from 'element-plus';

const env = import.meta.env;
const w: any = window;
const $wujie: any = w.$wujie;

const service = axios.create({
  baseURL: env.VITE_API_BASE_URL,
  timeout: 60000,
  headers: {
    'Content-Type': 'application/json;application/xml;charset=utf-8;',
  },
});

service.interceptors.request.use(
  (config) => {
    config.headers['company'] = 'carsafe';
    config.headers['jobNumber'] = $wujie?.props?.USER_ID || '1980235559149838337';
    config.headers['token'] = $wujie?.props?.TOKEN || '';
    config.headers['userId'] = $wujie?.props?.USER_ID || '1980235559149838337';
    config.headers['tenantId'] = $wujie?.props?.TENANT_ID || '1979091834410176514';
    return config;
  },
  (error) => {
    return Promise.reject(error);
  }
);

service.interceptors.response.use(
  (res) => {
    // 判断响应类型，如果是 blob，返回完整响应对象
    if (res.config.responseType === 'blob') {
      return res;
    }
    // 普通接口返回原来的 res.data
    if (res.data.code !== 200) {
      ElMessage.error(res.data.message);
    }
    return res.data;
  },
  (error) => {
    ElMessage.error('系统繁忙，请稍后再试');
    return Promise.reject(error);
  }
);

const get = (url: string, params = {}) => {
  return service.get(url, { params });
};

const post = (url: string, data = {}) => {
  return service.post(url, data);
};

const upload = (url: string, formData = {}, onProgress?: any) => {
  return service.post(url, formData, {
    headers: {
      'Content-Type': 'multipart/form-data',
    },
    onUploadProgress: (event: any) => {
      if (onProgress) {
        onProgress({
          loaded: event.loaded,
          total: event.total,
        });
      }
    },
  });
};

const download = (url: string, params = {}, filename = 'download') => {
  return service
    .post(url, params, {
      responseType: 'blob',
    })
    .then((res: any) => {
      const contentDisposition =
        res.headers?.['content-disposition'] ||
        res.headers?.['Content-Disposition'] ||
        '';
      let fileName = filename;

      if (contentDisposition) {
        const regex = /(filename\*?)\s*=\s*(?:UTF-8'')?["']?([^;"']+)["']?/ig;
        let match: RegExpExecArray | null = null;
        while ((match = regex.exec(contentDisposition)) !== null) {
          const key = match[1];
          const val = match[2];
          if (!val) continue;
          if (key.endsWith('*')) {
            if (val.toLowerCase() === 'utf-8') {
              continue;
            }
            try {
              fileName = decodeURIComponent(val);
            } catch {
              fileName = val;
            }
          } else {
            fileName = val.replace(/^["']|["']$/g, '');
          }
          break;
        }
      }

      const blob = new Blob([res.data]);
      const link = document.createElement('a');
      link.href = window.URL.createObjectURL(blob);
      link.download = fileName;
      document.body.appendChild(link);
      link.click();
      document.body.removeChild(link);
      window.URL.revokeObjectURL(link.href);
    })
    .catch(() => {
      ElMessage.error('下载失败');
    });
};

export {
  get,
  post,
  upload,
  download,
};