64 lines
1.8 KiB
JavaScript
64 lines
1.8 KiB
JavaScript
|
import { ElMessage } from 'element-plus';
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 下载文件下载工具函数
|
|||
|
|
* @param {string} fileUrl - 文件地址
|
|||
|
|
* @param {string} [fileName] - 可选，指定文件名
|
|||
|
|
* @param {Object} [headers] - 可选，请求头（如携带token）
|
|||
|
|
*/
|
|||
|
|
export const downloadFile = async (fileUrl, fileName, headers = {}) => {
|
|||
|
|
try {
|
|||
|
|
// 发起请求获取文件数据
|
|||
|
|
const response = await fetch(fileUrl, {
|
|||
|
|
method: 'GET',
|
|||
|
|
headers: {
|
|||
|
|
'Content-Type': 'application/octet-stream',
|
|||
|
|
...headers
|
|||
|
|
},
|
|||
|
|
credentials: 'include' // 允许携带cookie
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
if (!response.ok) {
|
|||
|
|
throw new Error(`下载失败: ${response.statusText}`);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 将响应转换为blob对象
|
|||
|
|
const blob = await response.blob();
|
|||
|
|
|
|||
|
|
// 创建临时URL
|
|||
|
|
const url = URL.createObjectURL(blob);
|
|||
|
|
|
|||
|
|
// 处理文件名
|
|||
|
|
let downloadName = fileName;
|
|||
|
|
if (!downloadName) {
|
|||
|
|
// 从响应头获取文件名
|
|||
|
|
const contentDisposition = response.headers.get('content-disposition');
|
|||
|
|
if (contentDisposition) {
|
|||
|
|
const match = contentDisposition.match(/filename[^;=\n]*=((['"]).*?\2|[^;\n]*)/);
|
|||
|
|
if (match && match[1]) {
|
|||
|
|
downloadName = decodeURIComponent(match[1].replace(/['"]/g, ''));
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
// 从URL提取文件名
|
|||
|
|
downloadName = fileUrl.split('/').pop()?.split('?')[0] || 'download_file';
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建a标签并触发下载
|
|||
|
|
const link = document.createElement('a');
|
|||
|
|
link.href = url;
|
|||
|
|
link.download = downloadName;
|
|||
|
|
link.style.display = 'none'; // 隐藏a标签
|
|||
|
|
document.body.appendChild(link);
|
|||
|
|
link.click();
|
|||
|
|
|
|||
|
|
// 清理资源
|
|||
|
|
setTimeout(() => {
|
|||
|
|
document.body.removeChild(link);
|
|||
|
|
URL.revokeObjectURL(url); // 释放临时URL
|
|||
|
|
}, 100);
|
|||
|
|
} catch (error) {
|
|||
|
|
ElMessage.error(`文件下载失败: ${error.message}`);
|
|||
|
|
}
|
|||
|
|
};
|