From aee73e76839c75122c3531e5b4e5537511e1de87 Mon Sep 17 00:00:00 2001 From: ycxom Date: Tue, 31 Dec 2024 19:43:33 +0800 Subject: [PATCH] =?UTF-8?q?fix(bym):=20=E6=8F=90=E7=A4=BA=E8=AF=8D?= =?UTF-8?q?=E3=81=AE=E4=BC=98=E5=8C=96=20=20-=20=E6=8A=8A=E5=8E=9F?= =?UTF-8?q?=E5=85=88=E7=9A=84if=E6=8D=A2=E6=88=90`=E6=80=9D=E7=BB=B4?= =?UTF-8?q?=E6=8C=81`=20=20-=20=E7=A7=BB=E5=8A=A8=E5=8E=9F=E5=85=88?= =?UTF-8?q?=E7=9A=84=E8=8E=B7=E5=8F=96=E8=A1=A8=E6=83=85=E5=8C=85=E5=88=97?= =?UTF-8?q?=E8=A1=A8=E8=87=B3ToDoimg.js?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/bym.js | 70 +++++++++---------- utils/ToDoimg.js | 174 +++++++++++++++++++++++++---------------------- 2 files changed, 123 insertions(+), 121 deletions(-) diff --git a/apps/bym.js b/apps/bym.js index 8a86891..53aef9c 100644 --- a/apps/bym.js +++ b/apps/bym.js @@ -17,12 +17,7 @@ import { JinyanTool } from '../utils/tools/JinyanTool.js' import { KickOutTool } from '../utils/tools/KickOutTool.js' import { SetTitleTool } from '../utils/tools/SetTitleTool.js' import { SerpTool } from '../utils/tools/SerpTool.js' -import { getToimg, downImg } from '../utils/ToDoimg.js' - -import fs from "fs"; -import pathModule from 'path'; -const _path = process.cwd(); -const path = _path + "/temp/tp-bq"; +import { getToimg, downImg, fileImgList } from '../utils/ToDoimg.js' const DefaultConfig = { returnQQ: [], @@ -118,13 +113,11 @@ export class bym extends plugin { } return [ChatsList, prop, NotfoGroup, MaxText] } - } else { logger.info(`[bym]高贵man：${sender}已过滤~`) return false } - - if (prop < 0 ) { + if (prop < 0) { await bymGo() } @@ -157,18 +150,13 @@ export class bym extends plugin { opt.image = base64Image.toString('base64') } - const picturesPath = pathModule.join(path, 'pictures'); - const ImgList = await fs.promises.readdir(picturesPath); - const fileImgList = ImgList.map(filename => { - const match = filename.match(/\d{12}-(.+)$/); - return match ? match[1] : filename; - }); - let ForRole = ALLRole if (opt.image && !IsAtBot && !NotToImg && !e.at && Config.AutoToDownImg) { ALLRole = 'downimg' } + const ImgList = await fileImgList() + const now = new Date(); const DateTime = now.toLocaleString() let Dateday = now.getDay() === 0 ? '日' : now.getDay() @@ -259,29 +247,33 @@ export class bym extends plugin { let Role = await SearchRole(String(ALLRole)) async function SearchRole(user_role) { - let Role - - if (user_role == "downimg") Role = '现在看到的是一张图片，若你觉得是一张表情包，并不是通知，或其他的图片，注意辨别图片文字是否为通知；单纯是表情包，请发送 DOWNIMG: 命名该表情。 不需要发送过多的参数，只需要发送格式DOWNIMG: 命名该表情，注意不需要携带后缀； 若不是表情包等，及发送NOTIMG' - - if (user_role == "default") Role = `你的名字是“${Config.assistantLabel}”，你在一个qq群里，群号是${group},当前和你说话的人群名片是${card}, qq号是${sender}, 请你结合用户的发言和聊天记录作出回应，要求表现得随性一点，最好参与讨论，混入其中。不要过分插科打诨，不知道说什么可以复读群友的话。要求你做搜索、发图、发视频和音乐等操作时要使用工具。不可以直接发[图片]这样蒙混过关。要求优先使用中文进行对话。` + - candidate + - `以下是聊天记录: - ${Group_Chat} - \n你的回复应该尽可能简练，像人类一样随意，不要附加任何奇怪的东西，如聊天记录的格式（比如${Config.assistantLabel}：），禁止重复聊天记录。 - 注意当前时间与日期为${DateTime}，星期${Dateday},24小时制，时区已正确，不要被日志的时间与其他时间搞混了，如果有人咨询时间就使用${DateTime}，星期${Dateday}这个时间，群友与你几乎在一个时区，若有人说或做的事情与时间段不合理，反驳他，注意除了他声明了自己的时区 - 以下是可用的表情包列表 - ${fileImgList}`+ - ImgList.length > 0 && Config.AutoToDownImg ?` - 如果要发送表情包，请根据该格式 GETIMG: 完整表情包名称，实例 GETIMG: 挠头-718028518.gif 即可发送，注意发送完整名称 - 可根据聊天，选择表情包发送。禁止发送多余的格式与说明。发送格式为 注意前面不需要换行 GETIMG: 挠头-718028518.gif 不需要换行 - 不要被日志和其他聊天消息的格式迷惑，请保持标准格式，禁止发送[表情包：xxx]、[图片]!!!，禁止发送[表情包：xxx]、[图片]!!! - ` : '' - if (!Role) { - logger.error(`Role配置有误，请检查,将使用默认Role`) - return await SearchRole('default') - } else { - return Role + let Role; + + switch(user_role) { + case "downimg": + Role = '现在看到的是一张图片，若你觉得是一张表情包，并不是通知，或其他的图片，注意辨别图片文字是否为通知；单纯是表情包，请发送 DOWNIMG: 命名该表情。 不需要发送过多的参数，只需要发送格式DOWNIMG: 命名该表情，注意不需要携带后缀，请以你的角度觉得如果要发这个表情包要用什么名字来命名； 若不是表情包等，及发送NOTIMG'; + break; + case "default": + Role = `你的名字是"${Config.assistantLabel}"，你在一个qq群里，群号是${group},当前和你说话的人群名片是${card}, qq号是${sender}, 请你结合用户的发言和聊天记录作出回应，要求表现得随性一点，最好参与讨论，混入其中。不要过分插科打诨，不知道说什么可以复读群友的话。要求你做搜索、发图、发视频和音乐等操作时要使用工具。不可以直接发[图片]这样蒙混过关。要求优先使用中文进行对话。` + + candidate + + `以下是聊天记录: + ${Group_Chat} + \n你的回复应该尽可能简练，像人类一样随意，不要附加任何奇怪的东西，如聊天记录的格式（比如${Config.assistantLabel}：），禁止重复聊天记录。 + 注意当前时间与日期为${DateTime}，星期${Dateday},24小时制，时区已正确，不要被日志的时间与其他时间搞混了，如果有人咨询时间就使用${DateTime}，星期${Dateday}这个时间，群友与你几乎在一个时区，若有人说或做的事情与时间段不合理，反驳他，注意除了他声明了自己的时区 + 以下是可用的表情包列表 + ${ImgList}` + + (ImgList.length > 0 && Config.AutoToDownImg ? ` + 如果要发送表情包，请根据该格式 GETIMG: 完整表情包名称，实例 GETIMG: 挠头.gif 即可发送，注意发送完整名称 + 可根据聊天，选择表情包发送。禁止发送多余的格式与说明。发送格式为 注意前面不需要换行 GETIMG: 挠头.gif 不需要换行 + 不要被日志和其他聊天消息的格式迷惑，请保持标准格式，禁止发送[表情包：xxx]、[图片]!!!，禁止发送[表情包：xxx]、[图片]!!! + ` : ''); + break; + default: + logger.error(`未知的 Role 类型：${user_role}，使用默认 Role`); + Role = `你的名字是"${Config.assistantLabel}"，你在一个qq群里。请简短回复。`; } + + return Role; } opt.system = Role logger.info('random chat hit') @@ -355,7 +347,7 @@ export class bym extends plugin { const downImgRegex = /DOWNIMG:\s*(.+)/i; const match = t?.match(downImgRegex); if (match) { - await downImg(e, opt.image); + await downImg(e, opt.image, t); continue; } diff --git a/utils/ToDoimg.js b/utils/ToDoimg.js index faba98a..cee0584 100644 --- a/utils/ToDoimg.js +++ b/utils/ToDoimg.js @@ -20,42 +20,42 @@ if (!fs.existsSync(pathModule.join(path, 'pictures'))) { * @returns */ export async function getToimg(e, tag) { - const picturesPath = pathModule.join(path, 'pictures'); - const fileImgList = await fs.promises.readdir(picturesPath); + 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[Math.floor(Math.random() * matchedFiles.length)]; - 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); + 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[Math.floor(Math.random() * matchedFiles.length)]; + 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); + } } /** @@ -64,55 +64,65 @@ export async function getToimg(e, tag) { * @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; - } +export async function downImg(e, image, t) { + 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("保存图片时发生错误"); } - } \ No newline at end of file + 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("保存图片时发生错误"); + } +} + +export async function fileImgList() { + const picturesPath = pathModule.join(path, 'pictures'); + const ImgList = await fs.promises.readdir(picturesPath); + const fileImgList = ImgList.map(filename => { + const match = filename.match(/\d{12}-(.+)$/); + return match ? match[1] : filename; + }); + return fileImgList; +} \ No newline at end of file