koishi-plugin-bilibili-notify 1.2.10 → 1.2.11-alpha.1

package/lib/biliAPI.d.ts

@@ -1,4 +1,4 @@
-import { Context, Service } from "koishi";
+import { Context, Schema, Service } from "koishi";
 import { CookieJar } from 'tough-cookie';
 import { Notifier } from "@koishijs/plugin-notifier";
 declare module 'koishi' {
@@ -10,11 +10,12 @@ declare class BiliAPI extends Service {
     static inject: string[];
     jar: CookieJar;
     client: any;
+    apiConfig: BiliAPI.Config;
     loginData: any;
     loginNotifier: Notifier;
     refreshCookieTimer: Function;
     loginInfoIsLoaded: boolean;
-    constructor(ctx: Context);
+    constructor(ctx: Context, config: BiliAPI.Config);
     protected start(): void | Promise<void>;
     protected stop(): void | Promise<void>;
     getServerUTCTime(): Promise<number>;
@@ -41,4 +42,10 @@ declare class BiliAPI extends Service {
     enableRefreshCookiesDetect(): void;
     checkIfTokenNeedRefresh(refreshToken: string, csrf: string, times?: number): Promise<void>;
 }
+declare namespace BiliAPI {
+    interface Config {
+        userAgent: string;
+    }
+    const Config: Schema<Config>;
+}
 export default BiliAPI;

package/lib/biliAPI.js

@@ -26,19 +26,20 @@ class BiliAPI extends koishi_1.Service {
     static inject = ['database', 'wbi', 'notifier'];
     jar;
     client;
+    apiConfig;
     loginData;
     loginNotifier;
     refreshCookieTimer;
     loginInfoIsLoaded = false;
-    constructor(ctx) {
+    constructor(ctx, config) {
         super(ctx, 'biliAPI');
+        this.apiConfig = config;
     }
     start() {
         // 创建新的http客户端(axios)
         this.createNewClient();
         // 从数据库加载cookies
         this.loadCookiesFromDatabase();
-        // logger
         // this.logger.info('工作中')
     }
     stop() {
@@ -167,7 +168,8 @@ class BiliAPI extends koishi_1.Service {
             jar: this.jar,
             headers: {
                 'Content-Type': 'application/json',
-                'User-Agent': this.getRandomUserAgent(),
+                'User-Agent': this.apiConfig.userAgent !== 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36' ?
+                    this.apiConfig.userAgent : this.getRandomUserAgent(),
                 'Origin': 'https://www.bilibili.com',
                 'Referer': 'https://www.bilibili.com/'
             }
@@ -389,4 +391,9 @@ class BiliAPI extends koishi_1.Service {
         // 没有问题，cookies已更新完成
     }
 }
+(function (BiliAPI) {
+    BiliAPI.Config = koishi_1.Schema.object({
+        userAgent: koishi_1.Schema.string()
+    });
+})(BiliAPI || (BiliAPI = {}));
 exports.default = BiliAPI;

package/lib/comRegister.d.ts

@@ -38,6 +38,7 @@ declare class ComRegister {
     checkIfLoginInfoIsLoaded(ctx: Context): Promise<unknown>;
     getSubFromDatabase(ctx: Context): Promise<void>;
     unsubSingle(ctx: Context, id: string, type: number): string;
+    unsubAll(ctx: Context, bot: Bot<Context>, uid: string): void;
     checkIfIsLogin(ctx: Context): Promise<boolean>;
 }
 declare namespace ComRegister {

package/lib/comRegister.js

@@ -364,7 +364,7 @@ class ComRegister {
                 content = await ctx.biliAPI.getUserInfo(mid);
             }
             catch (e) {
-                return 'bili sub getUserInfo() 发生了错误，错误为：' + e.toString();
+                return 'bili sub getUserInfo() 发生了错误，错误为：' + e.message;
             }
             // 判断是否有其他问题
             if (content.code !== 0) {
@@ -501,7 +501,7 @@ class ComRegister {
                 userData = data;
             }
             catch (e) {
-                this.logger.error('bili sub指令 getMasterInfo() 发生了错误，错误为：' + e.toString());
+                this.logger.error('bili sub指令 getMasterInfo() 发生了错误，错误为：' + e.message);
                 return '订阅出错啦，请重试';
             }
             // 需要订阅直播
@@ -577,7 +577,7 @@ class ComRegister {
                 content = await ctx.biliAPI.getLiveRoomInfo(roomId);
             }
             catch (e) {
-                return 'bili status指令 getLiveRoomInfo() 发生了错误，错误为：' + e.toString();
+                return 'bili status指令 getLiveRoomInfo() 发生了错误，错误为：' + e.message;
             }
             const { data } = content;
             let userData;
@@ -586,7 +586,7 @@ class ComRegister {
                 userData = userInfo;
             }
             catch (e) {
-                return 'bili status指令 getMasterInfo() 发生了错误，错误为：' + e.toString();
+                return 'bili status指令 getMasterInfo() 发生了错误，错误为：' + e.message;
             }
             // B站出问题了
             if (content.code !== 0) {
@@ -757,7 +757,7 @@ class ComRegister {
                 }
                 catch (e) {
                     if (i === attempts - 1) { // 已尝试三次
-                        this.logger.error(`发送群组ID:${guildId}消息失败！原因: ` + e.toString());
+                        this.logger.error(`发送群组ID:${guildId}消息失败！原因: ` + e.message);
                         this.sendPrivateMsg(bot, `发送群组ID:${guildId}消息失败，请检查机器人状态`);
                     }
                 }
@@ -782,7 +782,7 @@ class ComRegister {
                 content = await ctx.biliAPI.getUserSpaceDynamic(uid);
             }
             catch (e) {
-                return this.logger.error('dynamicDetect getUserSpaceDynamic() 发生了错误，错误为：' + e.toString());
+                return this.logger.error('dynamicDetect getUserSpaceDynamic() 发生了错误，错误为：' + e.message);
             }
             // 判断是否出现其他问题
             if (content.code !== 0) {
@@ -794,9 +794,7 @@ class ComRegister {
                     default: await this.sendPrivateMsg(bot, '获取动态信息错误，错误码为：' + content.code + '，错误为：' + content.message); // 未知错误
                 }
                 // 取消订阅
-                this.unsubSingle(ctx, uid, 1); /* 1为取消动态订阅 */
-                // 发送取消订阅消息
-                await this.sendPrivateMsg(bot, `UID:${uid}，已取消订阅动态`);
+                this.unsubAll(ctx, bot, uid);
                 // 结束循环
                 return;
             }
@@ -822,8 +820,11 @@ class ComRegister {
                         try {
                             // 渲染图片
                             const { pic: gimgPic, buffer: gimgBuffer } = await ctx.gimg.generateDynamicImg(items[num]);
+                            // 赋值
                             pic = gimgPic;
                             buffer = gimgBuffer;
+                            // 成功则跳出循环
+                            break;
                         }
                         catch (e) {
                             // 直播开播动态，不做处理
@@ -840,7 +841,7 @@ class ComRegister {
                             }
                             // 未知错误
                             if (i === attempts - 1) {
-                                this.logger.error('dynamicDetect generateDynamicImg() 推送卡片发送失败，原因：' + e.toString());
+                                this.logger.error('dynamicDetect generateDynamicImg() 推送卡片发送失败，原因：' + e.message);
                                 // 发送私聊消息并重启服务
                                 return await this.sendPrivateMsgAndRebootService(ctx, bot, '插件可能出现某些未知错误，请尝试重启插件，如果仍然发生该错误，请带着日志向作者反馈');
                             }
@@ -854,10 +855,13 @@ class ComRegister {
                         // pic存在，使用的是render模式
                         await this.sendMsg(ctx, guildId, bot, pic + (0, jsx_runtime_1.jsx)(jsx_runtime_1.Fragment, { children: dUrl }));
                     }
-                    else {
+                    else if (buffer) {
                         this.logger.info('推送动态中，使用page模式');
                         // pic不存在，说明使用的是page模式
-                        await this.sendMsg(ctx, guildId, bot, (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [(0, jsx_runtime_1.jsx)("img", { src: 'data:image/png;base64,' + buffer.toString('base64') }), " ", dUrl] }));
+                        await this.sendMsg(ctx, guildId, bot, (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [koishi_1.h.image(buffer, 'image/png'), " ", dUrl] }));
+                    }
+                    else {
+                        this.logger.info(items[num].modules.module_author.name + '发布了一条动态，但是推送失败');
                     }
                     // 更新时间点为最新发布动态的发布时间
                     switch (num) {
@@ -899,7 +903,7 @@ class ComRegister {
                 }
                 catch (e) {
                     if (i === attempts - 1) { // 已尝试三次
-                        this.logger.error('liveDetect generateLiveImg() 推送卡片生成失败，原因：' + e.toString());
+                        this.logger.error('liveDetect generateLiveImg() 推送卡片生成失败，原因：' + e.message);
                         return await this.sendPrivateMsgAndRebootService(ctx, bot, '插件可能出现某些未知错误，请尝试重启插件，如果仍然发生该错误，请带着日志向作者反馈');
                     }
                 }
@@ -911,7 +915,7 @@ class ComRegister {
                 return await this.sendMsg(ctx, guildId, bot, pic + msg);
             }
             // pic不存在，说明使用的是page模式
-            await this.sendMsg(ctx, guildId, bot, (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [(0, jsx_runtime_1.jsx)("img", { src: 'data:image/png;base64,' + buffer.toString('base64') }), " ", atAll && (0, jsx_runtime_1.jsx)("at", { type: "all" }), " ", liveStartMsg && liveStartMsg] }));
+            await this.sendMsg(ctx, guildId, bot, (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [koishi_1.h.image(buffer, 'image/png'), " ", atAll && (0, jsx_runtime_1.jsx)("at", { type: "all" }), " ", liveStartMsg && liveStartMsg] }));
         };
         return async () => {
             try {
@@ -930,7 +934,7 @@ class ComRegister {
                         break;
                     }
                     catch (e) {
-                        this.logger.error('liveDetect getLiveRoomInfo 发生了错误，错误为：' + e.toString());
+                        this.logger.error('liveDetect getLiveRoomInfo 发生了错误，错误为：' + e.message);
                         if (i === attempts - 1) { // 已尝试三次
                             return await this.sendPrivateMsgAndRebootService(ctx, bot, '插件可能出现某些未知错误，请尝试重启插件，如果仍然发生该错误，请带着日志向作者反馈');
                         }
@@ -952,7 +956,7 @@ class ComRegister {
                             break;
                         }
                         catch (e) {
-                            this.logger.error('liveDetect getMasterInfo() 发生了错误，错误为：' + e.toString());
+                            this.logger.error('liveDetect getMasterInfo() 发生了错误，错误为：' + e.message);
                             if (i === attempts - 1) { // 已尝试三次
                                 return await this.sendPrivateMsgAndRebootService(ctx, bot, '插件可能出现某些未知错误，请尝试重启插件，如果仍然发生该错误，请带着日志向作者反馈');
                             }
@@ -1022,7 +1026,7 @@ class ComRegister {
                                     break;
                                 }
                                 catch (e) {
-                                    this.logger.error('liveDetect open getMasterInfo() 发生了错误，错误为：' + e.toString());
+                                    this.logger.error('liveDetect open getMasterInfo() 发生了错误，错误为：' + e.message);
                                     if (i === attempts - 1) { // 已尝试三次
                                         return await this.sendPrivateMsgAndRebootService(ctx, bot, '插件可能出现某些未知错误，请尝试重启插件，如果仍然发生该错误，请带着日志向作者反馈');
                                     }
@@ -1229,7 +1233,7 @@ class ComRegister {
                     break;
                 }
                 catch (e) {
-                    this.logger.error('getSubFromDatabase() getUserInfo() 发生了错误，错误为：' + e.toString());
+                    this.logger.error('getSubFromDatabase() getUserInfo() 发生了错误，错误为：' + e.message);
                     if (i === attempts - 1) { // 已尝试三次
                         return await this.sendPrivateMsgAndRebootService(ctx, bot, '你的网络可能出现了某些问题，请检查后重启插件');
                     }
@@ -1368,6 +1372,25 @@ class ComRegister {
             this.updateSubNotifier(ctx);
         }
     }
+    unsubAll(ctx, bot, uid) {
+        this.subManager.filter(sub => sub.uid === uid).map(async (sub, i) => {
+            // 取消全部订阅 执行dispose方法，销毁定时器
+            if (sub.dynamic)
+                this.subManager[i].dynamicDispose();
+            if (sub.live)
+                this.subManager[i].liveDispose();
+            // 从数据库中删除订阅
+            await ctx.database.remove('bilibili', { uid: this.subManager[i].uid });
+            // 将该订阅对象从订阅管理对象中移除
+            this.subManager.splice(i, 1);
+            // id--
+            this.num--;
+            // 发送成功通知
+            this.sendPrivateMsg(bot, `UID:${uid}，已取消订阅该用户`);
+            // 更新控制台提示
+            this.updateSubNotifier(ctx);
+        });
+    }
     async checkIfIsLogin(ctx) {
         if ((await ctx.database.get('loginBili', 1)).length !== 0) { // 数据库中有数据
             // 检查cookie中是否有值

package/lib/index.d.ts

@@ -13,6 +13,7 @@ export interface Config {
     basicSettings: {};
     unlockSubLimits: boolean;
     renderType: 'render' | 'page';
+    userAgent: string;
     dynamic: {};
     dynamicUrl: boolean;
     dynamicCheckNumber: number;

package/lib/index.js

@@ -76,6 +76,9 @@ exports.Config = koishi_1.Schema.object({
         .role('')
         .default('render')
         .description('渲染类型，默认为render模式，渲染速度更快，但会出现乱码问题，若出现乱码问题，请切换到page模式。若使用自定义字体，建议选择render模式'),
+    userAgent: koishi_1.Schema.string()
+        .default('Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36')
+        .description('设置请求头User-Agen，请求出现-352时可以尝试修改'),
     dynamic: koishi_1.Schema.object({}).description('动态推送设置'),
     dynamicUrl: koishi_1.Schema.boolean()
         .default(false)
@@ -172,7 +175,9 @@ class ServerManager extends koishi_1.Service {
         this.registerPlugin();
     }
     registerPlugin = () => {
-        const biliApi = this.ctx.plugin(biliAPI_1.default);
+        const biliApi = this.ctx.plugin(biliAPI_1.default, {
+            userAgent: globalConfig.userAgent
+        });
         const generateImg = this.ctx.plugin(generateImg_1.default, {
             renderType: this.renderType,
             filter: globalConfig.filter,

package/package.json

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

package/readme.md

@@ -144,6 +144,7 @@
 - ver 1.2.8 修复例如像UP主籽岷使用webp格式的头像，下播通知无法发出的bug
 - ver 1.2.9-alpha.0 bug测试版本，请跳过
 - ver 1.2.10 修复插件启动一段时间后一直报错的问题，更新了UA列表，新增了更多日志输出
+- ver 1.2.11-alpha.0 新增自定义UA的设置。动态推送出错时，现在会直接取消订阅该UP主而不是取消订阅动态
 
 ## 交流群