import axios from 'axios'
import { ElLoading, ElMessage } from 'element-plus'
import { saveAs } from 'file-saver'
import { getToken } from '@/utils/auth'
import errorCode from '@/utils/errorCode'
import { blobValidate } from '@/utils/ruoyi'
import { getJavaApi } from '@/utils/java-api'
import { resolveAttachmentUrl, attachmentFileName, alternateImageAttachmentUrl, validateAttachmentBlob } from '@/utils/attachment'

const baseURL = import.meta.env.VITE_APP_BASE_API
let downloadLoadingInstance

export default {
  name(name, isDelete = true) {
    var url = baseURL + "/common/download?fileName=" + encodeURIComponent(name) + "&delete=" + isDelete
    axios({
      method: 'get',
      url: url,
      responseType: 'blob',
      headers: { 'Authorization': 'Bearer ' + getToken() }
    }).then((res) => {
      const isBlob = blobValidate(res.data)
      if (isBlob) {
        const blob = new Blob([res.data])
        this.saveAs(blob, decodeURIComponent(res.headers['download-filename']))
      } else {
        this.printErrMsg(res.data)
      }
    })
  },
  resource(resource) {
    var url = baseURL + "/common/download/resource?resource=" + encodeURIComponent(resource)
    axios({
      method: 'get',
      url: url,
      responseType: 'blob',
      headers: { 'Authorization': 'Bearer ' + getToken() }
    }).then((res) => {
      const isBlob = blobValidate(res.data)
      if (isBlob) {
        const blob = new Blob([res.data])
        this.saveAs(blob, decodeURIComponent(res.headers['download-filename']))
      } else {
        this.printErrMsg(res.data)
      }
    })
  },
  zip(url, name) {
    var url = baseURL + url
    downloadLoadingInstance = ElLoading.service({ text: "正在下载数据，请稍候", background: "rgba(0, 0, 0, 0.7)", })
    axios({
      method: 'get',
      url: url,
      responseType: 'blob',
      headers: { 'Authorization': 'Bearer ' + getToken() }
    }).then((res) => {
      const isBlob = blobValidate(res.data)
      if (isBlob) {
        const blob = new Blob([res.data], { type: 'application/zip' })
        this.saveAs(blob, name)
      } else {
        this.printErrMsg(res.data)
      }
      downloadLoadingInstance.close()
    }).catch((r) => {
      console.error(r)
      ElMessage.error('下载文件出现错误，请联系管理员！')
      downloadLoadingInstance.close()
    })
  },
  saveAs(text, name, opts) {
    if (typeof text === 'string') return this.url(text, name, opts)
    return saveAs(text, name, opts)
  },
  attachment(file, name) {
    try {
      return this.url(resolveAttachmentUrl(file?.fileUrl, file?.type), name || file?.fileName, undefined,
        ![1, 2].includes(Number(file?.type)))
    } catch (error) {
      ElMessage.error(error.message)
      return Promise.resolve(false)
    }
  },
  async url(path, name, opts, allowAlternative = true) {
    try {
      const url = resolveAttachmentUrl(path)
      const backend = new URL(getJavaApi(), window.location.origin)
      const target = new URL(url, window.location.origin)
      const isBackend = target.origin === backend.origin
        && (target.pathname === backend.pathname || target.pathname.startsWith(backend.pathname.replace(/\/$/, '') + '/'))
      const filename = attachmentFileName(url, name)
      // 外部附件沿用浏览器下载，兼容不允许跨域读取的文件服务。
      if (!isBackend && /^https?:$/.test(target.protocol)) {
        saveAs(url, filename, opts)
        return true
      }
      const headers = isBackend && getToken() ? { Authorization: 'Bearer ' + getToken() } : {}
      let response
      try {
        response = await axios.get(url, { responseType: 'blob', headers })
      } catch (error) {
        const alternative = isBackend && allowAlternative && error.response?.status === 404
          ? alternateImageAttachmentUrl(url) : null
        if (!alternative) throw error
        response = await axios.get(alternative, { responseType: 'blob', headers })
      }
      const { data } = response
      await validateAttachmentBlob(data, filename)
      saveAs(data, filename, opts)
      return true
    } catch (error) {
      const status = error.response?.status
      ElMessage.error(status === 404 ? '附件不存在或已被删除' : status === 401 || status === 403
        ? '登录已过期或没有附件下载权限' : error.message || '附件下载失败，请稍后重试')
      return false
    }
  },
  async printErrMsg(data) {
    const resText = await data.text()
    const rspObj = JSON.parse(resText)
    const errMsg = errorCode[rspObj.code] || rspObj.msg || errorCode['default']
    ElMessage.error(errMsg)
  }
}