标签导航:

文件下载重命名时扩展名被误识别怎么办?

解决文件下载重命名时扩展名误识别问题

下载文件并重命名时,如果文件名包含多个点号(.),浏览器或系统可能将最后一个点号后的部分误认为是文件扩展名,导致扩展名错误。例如,“document.2023.pdf”可能被误识别为“.pdf”,而非“document.2023.pdf”。本文分析此问题成因并提供解决方案。

之前的方案尝试通过计数文件名中的点号来解决,并在文件名后追加从URL中提取的扩展名。然而,如果文件名已包含扩展名,则会重复添加,例如“xxxx.zip”变为“xxxx.zip.zip”。

更可靠的方法是利用浏览器响应头中的Content-Type信息确定文件扩展名。Content-Type例如application/pdf、application/zip等,可据此确定正确的扩展名。

改进后的代码如下:

export const downloadFile = (fileUrl, fileName, errorCallback) => {
    try {
        const xhr = new XMLHttpRequest();
        xhr.open("GET", fileUrl, true);
        xhr.responseType = "blob";
        xhr.send();
        xhr.onload = function () {
            if (this.status === 200 || this.status === 304) {
                const contentType = xhr.getResponseHeader("Content-Type");
                let extension = "";
                if (contentType) {
                    // 根据Content-Type确定扩展名,可根据实际情况扩展
                    if (contentType.includes("pdf")) {
                        extension = ".pdf";
                    } else if (contentType.includes("zip")) {
                        extension = ".zip";
                    } else if (contentType.includes("txt")) {
                        extension = ".txt";
                    } // ... 添加其他类型判断
                }

                const blob = new Blob([this.response], { type: contentType });
                const finalFileName = fileName + extension; // 追加扩展名
                if ("msSaveOrOpenBlob" in navigator) {
                    navigator.msSaveOrOpenBlob(blob, finalFileName);
                    return;
                }
                const url = URL.createObjectURL(blob);
                simulateClick(blob, finalFileName, "_blank");
                URL.revokeObjectURL(url);
            }
        };
        xhr.onerror = function () {
            errorCallback && errorCallback();
        };
    } catch (error) {
        errorCallback && errorCallback();
    }
};

此代码根据Content-Type确定扩展名,避免了依赖文件名中点号数量的缺陷,更可靠地处理文件下载和重命名。 需根据实际情况扩展Content-Type判断逻辑,以支持更多文件类型。