51 lines
1.5 KiB
TypeScript
51 lines
1.5 KiB
TypeScript
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 }
|
|
}
|