From b467873397cf149a8e3f2a0171c543a17669595e Mon Sep 17 00:00:00 2001 From: zhiyu1998 <542716863@qq.com> Date: Wed, 11 Sep 2024 10:38:39 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=8C=9F=20feat:=20=E5=A2=9E=E5=8A=A0?= =?UTF-8?q?=E6=96=87=E4=BB=B6=E5=A4=84=E7=90=86=E5=8A=9F=E8=83=BD=E5=92=8C?= =?UTF-8?q?=E9=94=99=E8=AF=AF=E5=A4=84=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 在 utils/file.js 中添加了 MIME 类型常量、通用错误处理函数以及对文件操作函数的改进。 - 添加了常量对象用于文件类型和 MIME 类型的映射 - 增加了错误处理函数 handleError 用于统一错误处理 - 修改了 checkAndRemoveFile、mkdirIfNotExists、deleteFolderRecursive 等函数，使用 handleError 处理异常 - 优化了 readCurrentDir 函数的参数和返回值类型注释 - 增加了 copyFiles 函数的参数类型注释和返回值类型 - 增加了 toBase64 函数的参数类型注释和返回值类型 - 修改了 getMimeType 函数，使用 path.extname 获取文件扩展名 - 增加了 getMediaFilesAndOthers 函数，用于区分图片、视频和其他 --- utils/file.js | 115 ++++++++++++++++++++++++-------------------------- 1 file changed, 55 insertions(+), 60 deletions(-) diff --git a/utils/file.js b/utils/file.js index 83f656c..62cecc0 100644 --- a/utils/file.js +++ b/utils/file.js @@ -1,9 +1,26 @@ import fs from "node:fs"; import path from "path"; +// 常量提取 +const mimeTypes = { + '.jpg': 'image/jpeg', + '.jpeg': 'image/jpeg', + '.png': 'image/png', + '.gif': 'image/gif', + '.pdf': 'application/pdf', + '.txt': 'text/plain', + // 添加其他文件类型和MIME类型的映射 +}; + +// 通用错误处理函数 +function handleError(err) { + logger.error(err); + throw err; +} + /** * 检查文件是否存在并且删除 - * @param file + * @param {string} file - 文件路径 * @returns {Promise} */ export async function checkAndRemoveFile(file) { @@ -12,14 +29,14 @@ export async function checkAndRemoveFile(file) { await fs.promises.unlink(file); } catch (err) { if (err.code !== 'ENOENT') { - throw err; + handleError(err); } } } /** * 创建文件夹，如果不存在 - * @param dir + * @param {string} dir - 文件夹路径 * @returns {Promise} */ export async function mkdirIfNotExists(dir) { @@ -29,15 +46,15 @@ export async function mkdirIfNotExists(dir) { if (err.code === 'ENOENT') { await fs.promises.mkdir(dir, { recursive: true }); } else { - throw err; + handleError(err); } } } /** * 删除文件夹下所有文件 + * @param {string} folderPath - 文件夹路径 * @returns {Promise} - * @param folderPath */ export async function deleteFolderRecursive(folderPath) { try { @@ -58,37 +75,34 @@ export async function deleteFolderRecursive(folderPath) { await Promise.allSettled(actions); return files.length; } catch (error) { - logger.error(error); + handleError(error); return 0; } } /** * 读取当前文件夹的所有文件和文件夹 - * @param path 路径 - * @param printTree 是否打印树状图 - * @returns {Promise<*>} 返回一个包含文件名的数组 + * @param {string} dirPath - 路径 + * @returns {Promise} 返回一个包含文件名的数组 */ -export async function readCurrentDir(path) { +export async function readCurrentDir(dirPath) { try { - const files = await fs.promises.readdir(path); - return files; + return await fs.promises.readdir(dirPath); } catch (err) { - logger.error(err); + handleError(err); } } /** * 拷贝文件 - * @param srcDir 源文件目录 - * @param destDir 目标文件目录 - * @param specificFiles 过滤文件，不填写就拷贝全部 - * @returns {Promise<*|null>} + * @param {string} srcDir - 源文件目录 + * @param {string} destDir - 目标文件目录 + * @param {string[]} [specificFiles=[]] - 过滤文件，不填写就拷贝全部 + * @returns {Promise} 拷贝的文件列表 */ export async function copyFiles(srcDir, destDir, specificFiles = []) { try { await mkdirIfNotExists(destDir); - const files = await readCurrentDir(srcDir); // 如果 specificFiles 数组为空，拷贝全部文件；否则只拷贝指定文件 @@ -112,46 +126,34 @@ export async function copyFiles(srcDir, destDir, specificFiles = []) { return copiedFiles } catch (error) { - logger.error(error); + handleError(error); return []; } } /** * 转换路径图片为base64格式 - * @param filePath 图片路径 - * @return {Promise} + * @param {string} filePath - 图片路径 + * @returns {Promise} Base64字符串 */ export async function toBase64(filePath) { try { - // 读取文件数据 - const fileData = await fs.readFileSync(filePath); - // 将文件数据转换为Base64字符串 + const fileData = await fs.promises.readFile(filePath); const base64Data = fileData.toString('base64'); // 返回Base64字符串 return `data:${getMimeType(filePath)};base64,${base64Data}`; } catch (error) { - throw new Error(`读取文件时出错: ${error.message}`); + handleError(error); } } /** * 辅助函数：根据文件扩展名获取MIME类型 - * @param filePath - * @return {*|string} + * @param {string} filePath - 文件路径 + * @returns {string} MIME类型 */ function getMimeType(filePath) { - const mimeTypes = { - '.jpg': 'image/jpeg', - '.jpeg': 'image/jpeg', - '.png': 'image/png', - '.gif': 'image/gif', - '.pdf': 'application/pdf', - '.txt': 'text/plain', - // 添加其他文件类型和MIME类型的映射 - }; - - const ext = filePath.substring(filePath.lastIndexOf('.')).toLowerCase(); + const ext = path.extname(filePath).toLowerCase(); return mimeTypes[ext] || 'application/octet-stream'; } @@ -161,8 +163,8 @@ function getMimeType(filePath) { * @returns {Promise} 包含图片和视频文件名的对象 */ export async function getMediaFilesAndOthers(folderPath) { - return new Promise((resolve, reject) => { - // 定义图片和视频的扩展名 + try { + const files = await fs.promises.readdir(folderPath); const imageExtensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp', '.webp']; const videoExtensions = ['.mp4', '.mkv', '.avi', '.mov', '.wmv', '.flv', '.webm']; @@ -171,26 +173,19 @@ export async function getMediaFilesAndOthers(folderPath) { const videos = []; const others = []; - // 读取文件夹中的所有文件 - fs.readdir(folderPath, (err, files) => { - if (err) { - return reject('无法读取文件夹: ' + err); + files.forEach(file => { + const ext = path.extname(file).toLowerCase(); + if (imageExtensions.includes(ext)) { + images.push(file); + } else if (videoExtensions.includes(ext)) { + videos.push(file); + } else { + others.push(file); } - - files.forEach(file => { - const ext = path.extname(file).toLowerCase(); - - if (imageExtensions.includes(ext)) { - images.push(file); - } else if (videoExtensions.includes(ext)) { - videos.push(file); - } else { - others.push(file); - } - }); - - // 返回包含图片和视频的对象 - resolve({ images, videos, others }); }); - }); + + return { images, videos, others }; + } catch (err) { + handleError(err); + } } \ No newline at end of file