yz-yuki-plugin 2.0.7-0 → 2.0.7-2

package/CHANGELOG.md

@@ -1,4 +1,5 @@
 # 2.0.7
+* 添加白名单关键词过滤功能
 * 新增B站视频解析
 
 # 2.0.6

package/README.md

@@ -2,7 +2,7 @@
 
 # YUKI-PLUGIN
 
-- 一个适用于 `Yunzai 系列机器人框架` 的B站动态和微博动态订阅推送的插件
+- 一个适用于 `Yunzai 系列机器人框架` 的B站动态、B站视频链接解析和微博动态订阅推送的插件
 
 - 支持 群聊/私聊 订阅B站动态和微博动态，支持定时推送，支持手动触发推送，支持简单查询B站/微博用户信息。
 
@@ -160,6 +160,7 @@ https://m.weibo.cn/u/7643376782 # 7643376782 为崩坏星穹铁道博主uid
 请使用 `#优纪帮助`或 `/yuki帮助` 获取完整帮助
 
 - [x] B站动态
+- [x] B站视频链接解析
 - [x] 微博动态
 
 
@@ -190,6 +191,7 @@ https://m.weibo.cn/u/7643376782 # 7643376782 为崩坏星穹铁道博主uid
 | 删除B站ck | 删除手动获取的B站cookie，权限：Master | `#删除B站本地ck` |
 | 查看B站ck | 查看当前启用的B站ck,仅限私聊 | `#我的B站ck` |
 | 刷新B站临时ck | 重新获取并刷新redis缓存的未绑定自己的B站ck而自动获取的 临时B站cookie | `#刷新B站临时ck` |
+| B站视频链接解析 | 解析B站视频链接，支持av号、BV号、app分享链接，官方短链 | `链接xxxx` |
 ||||
 | **微博功能** | ------------------------- | ---------- |
 | 添加微博推送 | 检测博主的微博动态进行推送，权限：Master。可选分类：视频、图文、文章、转发，不加分类则默认全部 | `#订阅微博推送uid` `#订阅微博推送 图文 uid` |

package/defaultConfig/bilibili/config.yaml

@@ -43,6 +43,15 @@ pushContentLineLimit: 5
 # 是否展示定时任务的日志，0 不显示 1 显示
 pushTaskLog: 1
 
+# 白名单关键词，命中即推送，不在白名单则不推送。
+# 白名单优先级高于黑名单，即命中白名单即推送，不在白名单则不推送。
+# 白名单为空则不启用白名单功能。
+# 配置示例：
+# whiteWordslist:
+#   - 白名单关键词1
+#   - 白名单关键词2
+whiteWordslist:
+
 # 包含关键词不推送
 banWords:
   - 关键词1

package/defaultConfig/weibo/config.yaml

@@ -38,6 +38,15 @@ pushContentLineLimit: 5
 # 是否展示定时任务的日志，0 不显示 1 显示
 pushTaskLog: 1
 
+# 白名单关键词，命中即推送，不在白名单则不推送。
+# 白名单优先级高于黑名单，即命中白名单即推送，不在白名单则不推送。
+# 白名单为空则不启用白名单功能。
+# 配置示例：
+# whiteWordslist:
+#   - 白名单关键词1
+#   - 白名单关键词2
+whiteWordslist:
+
 # 包含关键词不推送
 banWords:
   - 关键词1

package/lib/apps/bilibili.js

@@ -503,24 +503,26 @@ message.use(async (e) => {
         logger?.info(`优纪B站视频链接解析配置文件已设置关闭，解析终止。`);
         return;
     }
-    const VideoIDStr = e.msg
-        .match(/(b23\.tv\/[a-zA-Z0-9]+)|(www\.bilibili\.com\/video\/)?(av\d+|BV[a-zA-Z0-9]+)/)?.[0]
-        .replace(/^www\.bilibili\.com\/video\//, '');
-    let vedioID;
-    if (VideoIDStr && new RegExp(/^b23\.tv\/[a-zA-Z0-9]+/).test(VideoIDStr)) {
-        const tvUrlID = VideoIDStr.replace(/^b23\.tv\//g, '');
-        const bvidStr = await new BilibiliWebDataFetcher(e).getBVIDByShortUrl(tvUrlID);
-        vedioID = { bvid: bvidStr };
-    }
-    else if (VideoIDStr && new RegExp(/^av\d+/).test(VideoIDStr)) {
-        const aid = VideoIDStr.replace(/^av/g, '');
-        vedioID = { aid: Number(aid) };
-    }
-    else if (VideoIDStr && new RegExp(/|^BV[a-zA-Z0-9]+/).test(VideoIDStr)) {
-        vedioID = { bvid: VideoIDStr };
+    const videoIDMatch = e.msg.match(/(b23\.tv\/([a-zA-Z0-9]+))|(www\.bilibili\.com\/video\/)?(av\d+|BV[a-zA-Z0-9]+)/);
+    let videoID;
+    if (videoIDMatch) {
+        if (videoIDMatch[2]) {
+            // 匹配 b23.tv/ 后面的部分
+            const bvidStr = await new BilibiliWebDataFetcher(e).getBVIDByShortUrl(videoIDMatch[2]);
+            videoID = { bvid: bvidStr };
+        }
+        else if (videoIDMatch[4].startsWith('av')) {
+            // 匹配 av 开头的部分
+            const aid = videoIDMatch[4].replace(/^av/, '');
+            videoID = { aid: Number(aid) };
+        }
+        else if (videoIDMatch[4].startsWith('BV')) {
+            // 匹配 BV 开头的部分
+            videoID = { bvid: videoIDMatch[4] };
+        }
     }
     e.reply('优纪酱解析中，请稍后~');
-    const res = await new BilibiliWebDataFetcher(e).getBiliVideoInfoByAid_BV(vedioID);
+    const res = await new BilibiliWebDataFetcher(e).getBiliVideoInfoByAid_BV(videoID);
     if (res?.statusText !== 'OK') {
         e.reply('诶嘿，出了点网络问题，等会再试试吧~');
         return;
@@ -585,11 +587,13 @@ message.use(async (e) => {
             `\n--------------------`,
             `\n${formatNumber(data?.stat?.view)}播放 • ${formatNumber(data?.stat?.danmaku)}弹幕 • ${formatNumber(data?.stat?.reply)}评论 `,
             `\n${formatNumber(data?.stat?.like)}点赞 • ${formatNumber(data?.stat?.coin)}投币 • ${formatNumber(data?.stat?.favorite)}收藏`,
-            `\n${formatNumber(data?.stat?.share)}分享`
+            `\n${formatNumber(data?.stat?.share)}分享`,
+            `\n--------------------`,
+            `\n链接：b23.tv/${data?.bvid}`
         ];
         e.reply(message);
     }
-}, [/(b23\.tv\/[a-zA-Z0-9]+)|(www\.bilibili\.com\/video\/)?(av\d+|BV[a-zA-Z0-9]+)/]);
+}, [/(b23\.tv\/([a-zA-Z0-9]+))|(www\.bilibili\.com\/video\/)?(av\d+|BV[a-zA-Z0-9]+)/]);
 const YukiBli = message.ok;
 
 export { YukiBli };

package/lib/models/bilibili/bilibili.main.task.js

@@ -184,12 +184,23 @@ class BiliTask {
         if (pushMsgMode === 'PIC') {
             const { data, uid } = await BiliQuery.formatDynamicData(pushDynamicData); // 处理动态数据
             const extentData = { ...data };
+            const getWhiteWords = biliConfigData?.whiteWordslist;
             const getBanWords = biliConfigData?.banWords;
+            if (getWhiteWords && Array.isArray(getWhiteWords) && getWhiteWords.length > 0) {
+                // 构建白名单关键字正则表达式，转义特殊字符
+                const whiteWords = new RegExp(getWhiteWords.map(word => word.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')).join('|'), 'g');
+                if (!whiteWords.test(`${extentData?.title}${extentData?.content}`)) {
+                    return; // 如果动态消息不在白名单中，则直接返回
+                }
+            }
+            else if (getWhiteWords && !Array.isArray(getWhiteWords)) {
+                logger.error(`B站动态：Yaml配置文件中，whiteWordslist 字段格式不是数组格式，请检查！`);
+            }
             if (getBanWords && Array.isArray(getBanWords) && getBanWords.length > 0) {
                 // 构建屏蔽关键字正则表达式，转义特殊字符
                 const banWords = new RegExp(getBanWords.map(word => word.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')).join('|'), 'g');
                 if (banWords.test(`${extentData?.title}${extentData?.content}`)) {
-                    return 'return'; // 如果动态消息包含屏蔽关键字，则直接返回
+                    return; // 如果动态消息包含屏蔽关键字，则直接返回
                 }
             }
             else if (getBanWords && !Array.isArray(getBanWords)) {
@@ -223,7 +234,18 @@ class BiliTask {
             if (dynamicMsg === undefined || dynamicMsg === 'continue') {
                 return 'return'; // 如果动态消息构建失败，则直接返回
             }
+            const getWhiteWords = biliConfigData?.whiteWordslist;
             const getBanWords = biliConfigData?.banWords;
+            if (getWhiteWords && Array.isArray(getWhiteWords) && getWhiteWords.length > 0) {
+                // 构建白名单关键字正则表达式，转义特殊字符
+                const whiteWords = new RegExp(getWhiteWords.map(word => word.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')).join('|'), 'g');
+                if (!whiteWords.test(dynamicMsg.msg.join(''))) {
+                    return; // 如果动态消息不在白名单中，则直接返回
+                }
+            }
+            else if (getWhiteWords && !Array.isArray(getWhiteWords)) {
+                logger.error(`B站动态：Yaml配置文件中，whiteWordslist 字段格式不是数组格式，请检查！`);
+            }
             if (getBanWords && Array.isArray(getBanWords) && getBanWords.length > 0) {
                 // 构建屏蔽关键字正则表达式，转义特殊字符
                 const banWords = new RegExp(getBanWords.map(word => word.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')).join('|'), 'g');

package/lib/models/weibo/weibo.main.task.js

@@ -149,7 +149,18 @@ class WeiboTask {
             return; // 如果已经发送过，则直接返回
         if (!!weiboConfigData.pushMsgMode) {
             const { data, uid } = await WeiboQuery.formatDynamicData(pushDynamicData); // 处理动态数据
+            const getWhiteWords = weiboConfigData?.whiteWordslist;
             const getBanWords = weiboConfigData?.banWords;
+            if (getWhiteWords && Array.isArray(getWhiteWords) && getWhiteWords.length > 0) {
+                // 构建白名单关键字正则表达式，转义特殊字符
+                const whiteWords = new RegExp(getWhiteWords.map(word => word.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')).join('|'), 'g');
+                if (!whiteWords.test(`${data?.title}${data?.content}`)) {
+                    return; // 如果动态消息不在白名单中，则直接返回
+                }
+            }
+            else if (getWhiteWords && !Array.isArray(getWhiteWords)) {
+                logger.error(`微博动态：Yaml配置文件中，whiteWordslist 字段格式不是数组格式，请检查！`);
+            }
             if (getBanWords && Array.isArray(getBanWords) && getBanWords.length > 0) {
                 // 构建屏蔽关键字正则表达式，转义特殊字符
                 const banWords = new RegExp(getBanWords.map(word => word.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')).join('|'), 'g');
@@ -189,7 +200,18 @@ class WeiboTask {
             if (dynamicMsg === undefined || dynamicMsg === 'continue') {
                 return 'return'; // 如果动态消息构建失败或内部资源获取失败，则直接返回
             }
+            const getWhiteWords = weiboConfigData?.whiteWordslist;
             const getBanWords = weiboConfigData?.banWords;
+            if (getWhiteWords && Array.isArray(getWhiteWords) && getWhiteWords.length > 0) {
+                // 构建白名单关键字正则表达式，转义特殊字符
+                const whiteWords = new RegExp(getWhiteWords.map(word => word.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')).join('|'), 'g');
+                if (!whiteWords.test(dynamicMsg.msg.join(''))) {
+                    return; // 如果动态消息不在白名单中，则直接返回
+                }
+            }
+            else if (getWhiteWords && !Array.isArray(getWhiteWords)) {
+                logger.error(`微博动态：Yaml配置文件中，whiteWordslist 字段格式不是数组格式，请检查！`);
+            }
             if (getBanWords && Array.isArray(getBanWords) && getBanWords.length > 0) {
                 // 构建屏蔽关键字正则表达式，转义特殊字符
                 const banWords = new RegExp(getBanWords.map(word => word.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')).join('|'), 'g');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "yz-yuki-plugin",
-  "version": "2.0.7-0",
+  "version": "2.0.7-2",
   "description": "优纪插件，yunzaijs 关于 微博推送、B站推送 等功能的拓展插件",
   "author": "snowtafir",
   "type": "module",