koishi-plugin-bilibili-notify 1.0.4-rc.1 → 1.0.5-alpha.0

package/lib/biliAPI.js

@@ -181,7 +181,6 @@ class BiliAPI extends koishi_1.Service {
         // 定义CSRF Token
         let csrf;
         cookies.forEach(cookieData => {
-            // console.log(cookieData);
             // 获取key为bili_jct的值
             if (cookieData.key === 'bili_jct')
                 csrf = cookieData.value;

package/lib/comRegister.js

@@ -574,8 +574,20 @@ class ComRegister {
                     let attempts = 3;
                     for (let i = 0; i < attempts; i++) {
                         try {
+                            // 定义变量
+                            let pic;
+                            let buffer;
                             // 获取动态推送图片
-                            const { pic, buffer } = await ctx.gimg.generateDynamicImg(items[num]);
+                            try {
+                                const { pic: gimgPic, buffer: gimgBuffer } = await ctx.gimg.generateDynamicImg(items[num]);
+                                pic = gimgPic;
+                                buffer = gimgBuffer;
+                            }
+                            catch (e) {
+                                // 直播开播动态，不做处理
+                                if (e.message === '直播开播动态，不做处理')
+                                    break;
+                            }
                             // 如果pic存在，则直接返回pic
                             if (pic)
                                 return await bot.sendMessage(guildId, pic);
@@ -630,12 +642,10 @@ class ComRegister {
         }
         return async () => {
             try {
-                // console.log('start before' + ' ' + flag);
                 // 如果flag为false则说明前面的代码还未执行完，则直接返回
                 if (!flag)
                     return;
                 flag && (flag = false);
-                // console.log('start after' + ' ' + flag);
                 // 发送请求检测直播状态
                 let content;
                 try {
@@ -733,10 +743,8 @@ class ComRegister {
                 }
             }
             finally {
-                // console.log('end before' + ' ' + flag);
                 // 执行完方法体不论如何都把flag设置为true
                 flag = true;
-                // console.log('end after' + ' ' + flag);
             }
         };
     }
@@ -785,17 +793,67 @@ class ComRegister {
         this.num = subData.length;
         // 如果订阅数量超过三个则被非法修改数据库
         // 向管理员发送重新订阅通知
-        if (this.num > 3)
+        if (this.num > 3) {
+            ctx.notifier.create({
+                type: 'danger',
+                content: '数据库被非法修改，请你删除bilibili表的所有内容后重启插件'
+            });
             return;
+        }
         // 定义Bot
         let bot;
         // 循环遍历
-        subData.forEach(sub => {
+        subData.forEach(async (sub) => {
             // 拿到对应bot
             switch (sub.platform) {
                 case 'qq': bot = this.qqBot;
                 case 'qqguild': bot = this.qqguildBot;
             }
+            // 判断数据库是否被篡改
+            // 获取用户信息
+            let content;
+            let attempts = 3;
+            for (let i = 0; i < attempts; i++) {
+                try {
+                    content = await ctx.biliAPI.getUserInfo(sub.uid);
+                    break;
+                }
+                catch (e) {
+                    this.logger.error('getSubFromDatabase() getUserInfo() 本次网络请求失败');
+                    if (i === attempts - 1) { // 已尝试三次
+                        return bot.sendMessage(sub.targetId, '你的网络可能出现了某些问题，请检查后重启插件');
+                    }
+                }
+            }
+            // 获取data
+            const { data } = content;
+            // 定义函数删除数据和发送提示
+            const deleteSub = async () => {
+                // 从数据库删除该条数据
+                await ctx.database.remove('bilibili', { id: sub.id });
+                // 给用户发送提示
+                bot.sendMessage(sub.targetId, `UID:${sub.uid} 数据库内容被篡改，已取消对该UP主的订阅`);
+            };
+            // 判断是否有其他问题
+            if (content.code !== 0) {
+                switch (content.code) {
+                    case -352:
+                    case -403:
+                        bot.sendMessage(sub.targetId, '你的登录信息已过期，请重新登录Bilibili');
+                        return;
+                    case -400:
+                    case -404:
+                    default:
+                        deleteSub();
+                        return;
+                }
+            }
+            // 检测房间号是否被篡改
+            if (sub.live && (!data.live_room || data.live_room.roomid.toString() !== sub.room_id)) {
+                // 房间号被篡改，删除该订阅
+                deleteSub();
+                return;
+            }
             // 构建订阅对象
             let subManagerItem = {
                 id: sub.id,

package/lib/generateImg.js

@@ -481,7 +481,7 @@ class GenerateImg extends koishi_1.Service {
                 case DYNAMIC_TYPE_UGC_SEASON: return [`${upName}更新了合集，我暂时无法渲染，请自行查看`, link];
                 case DYNAMIC_TYPE_NONE: return [`${upName}发布了一条无效动态`, link];
                 // 直播开播，不做处理
-                case DYNAMIC_TYPE_LIVE_RCMD: return ['', ''];
+                case DYNAMIC_TYPE_LIVE_RCMD: throw new Error('直播开播动态，不做处理');
                 case DYNAMIC_TYPE_SUBSCRIPTION_NEW:
                 case DYNAMIC_TYPE_BANNER:
                 case DYNAMIC_TYPE_SUBSCRIPTION:
@@ -494,6 +494,7 @@ class GenerateImg extends koishi_1.Service {
             }
             return [main, link, forwardInfo];
         }
+        // 获取动态主要内容
         const [main, link] = await getDynamicMajor(data, false);
         // 加载字体
         const fontURL = (0, url_1.pathToFileURL)((0, path_1.resolve)(__dirname, './HYZhengYuan-55W.ttf'));

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-bilibili-notify",
   "description": "Koishi bilibili notify plugin",
-  "version": "1.0.4-rc.1",
+  "version": "1.0.5-alpha.0",
   "contributors": [
     "Akokko <admin@akokko.com>"
   ],

package/readme.md

@@ -80,6 +80,7 @@
 - ver 1.0.2 修复时间bug和字体乱码问题
 - ver 1.0.3 修复了一些bug，提供用户自己选择推送卡片字体样式的选项
 - ver 1.0.4 修复了重复推送的bug，提供用户选择推送卡片渲染方式的选项
+- ver 1.0.5 修复了用户非法篡改数据库内容可能导致程序异常运行的bug，修复了UP主开播动态推送空白卡片的bug
 
 ## 感谢