import Request from '/admin/support/request'
import { ref, watch } from 'vue'
import Message from '/admin/support/message'

export function useExcelDownload() {
  const http = new Request()
  const isSuccess = ref(false)
  const loading = ref<boolean>(false)
  const afterDownload = ref()
  function download(path: string, data: object = {}) {
    loading.value = true
    http
      .setResponseType('blob')
      .init()
      .get(path + '/export', data)
      .then(r => {
        if (r.headers['content-type'] === 'application/json') {
          const blob = new Blob([r.data], { type: r.headers['content-type'] })
          const blobReader = new Response(blob).json()
          blobReader.then(res => {
            if (res.code === 1e4) {
              Message.success(res.message)
            } else {
              Message.error(res.message)
            }
          })
        } else {
          const downloadLink = document.createElement('a')
          const blob = new Blob([r.data], { type: r.headers['content-type'] })
          downloadLink.href = URL.createObjectURL(blob)
          downloadLink.download = r.headers.filename
          document.body.appendChild(downloadLink)
          downloadLink.click()
          document.body.removeChild(downloadLink)
        }
      })
      .finally(() => {
        loading.value = false
      })
  }

  const success = (func: Function) => {
    watch(isSuccess, function () {
      isSuccess.value = false
      func()
    })
  }

  return { download, success, loading, afterDownload }
}