feat(bym): 重构图片处理逻辑，新增 ToDoimg 工具模块

2025-12-17 13:57:10 +00:00 · 2024-12-31 14:53:13 +08:00 · 2024-12-31 14:53:13 +08:00 · dc1e280f06
commit dc1e280f06
parent aec89ef10f
2 changed files with 120 additions and 100 deletions
--- a/utils/ToDoimg.js
+++ b/utils/ToDoimg.js
@ -0,0 +1,116 @@
+import pathModule from 'path';
+import { fileTypeFromBuffer } from 'file-type';
+import moment from 'moment';
+const _path = process.cwd();
+const path = _path + "/temp/tp-bq";
+
+// 没文件夹就创建一个
+if (!fs.existsSync(path)) {
+  fs.mkdirSync(path, { recursive: true })
+}
+if (!fs.existsSync(pathModule.join(path, 'pictures'))) {
+  fs.mkdirSync(pathModule.join(path, 'pictures'), { recursive: true })
+}
+
+/**
+ * 
+ * @param {*} e - 输入的消息
+ * @param {*} tag - 表情包标签
+ * @returns 
+ */
+export async function getToimg(e, tag) {
+    const picturesPath = pathModule.join(path, 'pictures');
+    const fileImgList = await fs.promises.readdir(picturesPath);
+
+    try {
+        const sanitizedTag = tag
+            .replace(/[\u200B-\u200D\uFEFF]/g, '')
+            .replace(/[\[\]]/g, '')
+            .trim()
+            .replace(/[^a-zA-Z0-9\u4e00-\u9fa5-_.]/g, '-');
+        let matchedFiles = fileImgList.filter(file => file === sanitizedTag);
+        if (matchedFiles.length === 0) {
+            matchedFiles = fileImgList.filter(file => file.startsWith(sanitizedTag));
+        }
+        if (matchedFiles.length === 0) {
+            matchedFiles = fileImgList.filter(file => file.includes(sanitizedTag));
+        }
+        if (matchedFiles.length === 0) {
+            logger.warn(`未找到匹配的表情包: ${sanitizedTag}`);
+            return;
+        }
+        const selectedFile = matchedFiles[0];
+        const picPath = pathModule.join(picturesPath, selectedFile);
+        try {
+            await fs.promises.access(picPath);
+        } catch {
+            logger.warn(`找不到指定的表情包文件: ${picPath}`);
+            return;
+        }
+        e.reply(segment.image('file:///' + picPath));
+
+        logger.info(`发送表情包: ${picPath}`);
+        return false;
+    } catch (error) {
+        logger.error('Error in getToimg:', error);
+    }
+}
+
+/**
+ * 
+ * @param {*} e - 输入的消息
+ * @param {*} image - 图片Base64
+ * @returns 
+ */
+export async function downImg(e, image) {
+    try {
+      let reply;
+      if (e.source) {
+        if (e.isGroup) {
+          reply = (await e.group.getChatHistory(e.source.seq, 1)).pop()?.message;
+        } else {
+          reply = (await e.friend.getChatHistory(e.source.time, 1)).pop()?.message;
+        }
+        if (reply) {
+          for (let val of reply) {
+            if (val.type === "image") {
+              e.img = [val.url];
+              break;
+            }
+          }
+        }
+      }
+      if (!e.img && !image) {
+        return false;
+      }
+      let kWordReg = /^#?(DOWNIMG:)\s*(.*)/i;
+      t = t.replace(/[\u200B-\u200D\uFEFF]/g, '').trim();
+      const match = kWordReg.exec(t);
+      if (!match) {
+        logger.error('DOWNIMG command format invalid:', t);
+        return;
+      }
+      let rawmsg = match[2] || "defaultTag";
+      let kWord = rawmsg.replace(/，|,|、| |。/g, "-").replace(/--+/g, "-").replace(/^-|-$|--/g, "").trim() || "defaultTag";
+      if (image) {
+        const imageBuffer = Buffer.from(image, 'base64');
+        const type = await fileTypeFromBuffer(imageBuffer);
+        let picType = 'png';
+        if (type && type.ext) {
+          picType = type.ext;
+        }
+        const currentTime = moment().format("YYMMDDHHmmss");
+        const safeTag = kWord.replace(/[^a-zA-Z0-9\u4e00-\u9fa5-_]/g, '-');
+        const picPath = pathModule.join(path, 'pictures', `${currentTime}-${safeTag.substring(0, 200)}.${picType}`);
+        logger.mark("DOWNIMG：", picPath);
+        if (!fs.existsSync(pathModule.join(path, 'pictures'))) {
+          fs.mkdirSync(pathModule.join(path, 'pictures'), { recursive: true });
+        }
+        fs.writeFileSync(picPath, imageBuffer);
+        logger.info(`图片已保存，标签为：${kWord}`);
+      }
+    } catch (error) {
+      logger.error('Error in downImg:', error);
+      logger.error("保存图片时发生错误");
+    }
+  }