import axios from 'axios';
import FileSaver from 'file-saver';
import JSZip from 'jszip';
import errorCode from '@/utils/errorCode';
import { blobValidate } from '@/utils/ruoyi';
import { LoadingInstance } from 'element-plus/es/components/loading/src/loading';
import { globalHeaders } from '@/utils/request';

const baseURL = import.meta.env.VITE_APP_BASE_API;
let downloadLoadingInstance: LoadingInstance;
export default {
  async oss(ossId: string | number) {
    const url = baseURL + '/resource/oss/download/' + ossId;
    downloadLoadingInstance = ElLoading.service({ text: '正在下载数据，请稍候', background: 'rgba(0, 0, 0, 0.7)' });
    try {
      const res = await axios({
        method: 'get',
        url: url,
        responseType: 'blob',
        headers: globalHeaders()
      });
      const isBlob = blobValidate(res.data);
      if (isBlob) {
        const blob = new Blob([res.data], { type: 'application/octet-stream' });
        FileSaver.saveAs(blob, decodeURIComponent(res.headers['download-filename'] as string));
      } else {
        this.printErrMsg(res.data);
      }
      downloadLoadingInstance.close();
    } catch (r) {
      console.error(r);
      ElMessage.error('下载文件出现错误，请联系管理员！');
      downloadLoadingInstance.close();
    }
  },
  async zip(url: string, name: string) {
    url = baseURL + url;
    downloadLoadingInstance = ElLoading.service({ text: '正在下载数据，请稍候', background: 'rgba(0, 0, 0, 0.7)' });
    try {
      const res = await axios({
        method: 'get',
        url: url,
        responseType: 'blob',
        headers: globalHeaders()
      });
      const isBlob = blobValidate(res.data);
      if (isBlob) {
        const blob = new Blob([res.data], { type: 'application/zip' });
        FileSaver.saveAs(blob, name);
      } else {
        this.printErrMsg(res.data);
      }
      downloadLoadingInstance.close();
    } catch (r) {
      console.error(r);
      ElMessage.error('下载文件出现错误，请联系管理员！');
      downloadLoadingInstance.close();
    }
  },
  async printErrMsg(data: any) {
    const resText = await data.text();
    const rspObj = JSON.parse(resText);
    const errMsg = errorCode[rspObj.code] || rspObj.msg || errorCode['default'];
    ElMessage.error(errMsg);
  },
  async direct(fileUrl: string, filename?: string) {
    downloadLoadingInstance = ElLoading.service({ text: '正在下载文件，请稍候...', background: 'rgba(0, 0, 0, 0.7)' });
    try {
      const res = await axios({
        method: 'get',
        url: fileUrl,
        responseType: 'blob',
        headers: globalHeaders() // 可根据是否跨域决定是否保留
      });
      const blob = new Blob([res.data], { type: 'application/octet-stream' });
      const name = filename || decodeURIComponent(fileUrl.split('/').pop()!);
      FileSaver.saveAs(blob, name);
    } catch (error) {
      console.error(error);
      ElMessage.error('下载文件失败，请检查链接或联系管理员');
    } finally {
      downloadLoadingInstance.close();
    }
  },
  /**
   * 下载多个文件并打包成 zip
   * @param files 文件信息列表，包含 { url, name }
   * @param zipName 压缩包名称
   */
  async downloadFilesAsZip(
    files: any[],
    options: {
      urlKey?: string; // 默认为 'url'
      nameKey?: string; // 默认为 'name'
      zipName?: string; // 默认为 '打包下载.zip'
    } = {}
  ) {
    const { urlKey = 'url', nameKey = 'name', zipName = '打包下载.zip' } = options;

    if (!files.length) {
      ElMessage.warning('没有可下载的文件');
      return;
    }

    const zip = new JSZip();
    const folder = zip.folder('files');
    const loading = ElLoading.service({ text: '正在打包文件，请稍候...', background: 'rgba(0,0,0,0.7)' });

    try {
      for (const file of files) {
        const fileUrl = file[urlKey];
        const fileName = file[nameKey] || '未命名文件';

        if (!fileUrl) continue;

        const res = await axios.get(fileUrl, { responseType: 'blob' });
        folder?.file(fileName, res.data);
      }

      const content = await zip.generateAsync({ type: 'blob' });
      FileSaver.saveAs(content, zipName);
    } catch (error) {
      console.error(error);
      ElMessage.error('打包下载失败，请联系管理员');
    } finally {
      loading.close();
    }
  }
};