59 lines
2.0 KiB
JavaScript
59 lines
2.0 KiB
JavaScript
|
import { ElMessage } from "element-plus";
|
|||
|
|
import dayjs from "dayjs";
|
|||
|
|
import {getFileUrl} from "../../utils/index.js";
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @param {string} url - 请求的 API 地址（GET）
|
|||
|
|
* @param {Object} [option] - 配置项（可选）
|
|||
|
|
* @param {string} [option.name] - 下载文件的自定义文件名（不含后缀），默认为当前时间戳
|
|||
|
|
* @param {string} [option.type] - Blob 对象的 MIME 类型，默认为 Excel 类型
|
|||
|
|
* @param {Object} [option.params] - 请求时携带的查询参数对象
|
|||
|
|
* @returns {Promise<any>} 如果下载成功则 resolve 响应数据，失败则 reject 错误信息
|
|||
|
|
*/
|
|||
|
|
export default function useDownloadBlob(
|
|||
|
|
url,
|
|||
|
|
option = { name: "", type: "", params: {} }
|
|||
|
|
) {
|
|||
|
|
const FILE_URL = getFileUrl();
|
|||
|
|
return new Promise((resolve, reject) => {
|
|||
|
|
const finalUrl = url.indexOf(FILE_URL) === -1 ? FILE_URL + url : url;
|
|||
|
|
Object.entries(option.params).forEach(([key, value]) => {
|
|||
|
|
finalUrl.searchParams.append(key, value);
|
|||
|
|
});
|
|||
|
|
fetch(finalUrl, {
|
|||
|
|
method: "GET",
|
|||
|
|
mode: "cors",
|
|||
|
|
headers: {
|
|||
|
|
"Content-Type": "application/json",
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
.then((response) => {
|
|||
|
|
if (!response.ok) {
|
|||
|
|
throw new Error("Network response was not ok");
|
|||
|
|
}
|
|||
|
|
return response.blob();
|
|||
|
|
})
|
|||
|
|
.then((blob) => {
|
|||
|
|
// 保持原有下载逻辑
|
|||
|
|
const finalBlob = new Blob([blob], {
|
|||
|
|
type: option.type || "application/vnd.ms-excel",
|
|||
|
|
});
|
|||
|
|
const downloadElement = document.createElement("a");
|
|||
|
|
const href = window.URL.createObjectURL(finalBlob);
|
|||
|
|
downloadElement.style.display = "none";
|
|||
|
|
downloadElement.href = href;
|
|||
|
|
downloadElement.download =
|
|||
|
|
option.name || dayjs().format("YYYY-MM-DD HH:mm:ss");
|
|||
|
|
document.body.appendChild(downloadElement);
|
|||
|
|
downloadElement.click();
|
|||
|
|
document.body.removeChild(downloadElement);
|
|||
|
|
window.URL.revokeObjectURL(href);
|
|||
|
|
resolve({ data: finalBlob });
|
|||
|
|
})
|
|||
|
|
.catch((err) => {
|
|||
|
|
ElMessage.error("导出失败");
|
|||
|
|
reject(err);
|
|||
|
|
});
|
|||
|
|
});
|
|||
|
|
}
|