koishi-plugin-bilibili-notify 1.2.2-alpha.0 → 1.2.2-beta.1

package/lib/biliAPI.d.ts

@@ -13,6 +13,7 @@ declare class BiliAPI extends Service {
     loginData: any;
     loginNotifier: Notifier;
     refreshCookieTimer: Function;
+    loginInfoIsLoaded: boolean;
     constructor(ctx: Context);
     protected start(): void | Promise<void>;
     getServerUTCTime(): Promise<number>;
@@ -29,6 +30,7 @@ declare class BiliAPI extends Service {
     createNewClient(): void;
     getTimeOfUTC8(): number;
     getCookies(): string;
+    getLoginInfoIsLoaded(): boolean;
     getLoginInfoFromDB(): Promise<{
         cookies: any;
         refresh_token: string;

package/lib/biliAPI.js

@@ -29,6 +29,7 @@ class BiliAPI extends koishi_1.Service {
     loginData;
     loginNotifier;
     refreshCookieTimer;
+    loginInfoIsLoaded = false;
     constructor(ctx) {
         super(ctx, 'biliAPI');
     }
@@ -165,6 +166,9 @@ class BiliAPI extends koishi_1.Service {
         cookies = JSON.stringify(this.jar.serializeSync().cookies);
         return cookies;
     }
+    getLoginInfoIsLoaded() {
+        return this.loginInfoIsLoaded;
+    }
     async getLoginInfoFromDB() {
         // 读取数据库获取cookies
         const data = (await this.ctx.database.get('loginBili', 1))[0];
@@ -229,6 +233,8 @@ class BiliAPI extends koishi_1.Service {
             });
             this.jar.setCookieSync(cookie, `http${cookie.secure ? 's' : ''}://${cookie.domain}${cookie.path}`, {});
         });
+        // Login info is loaded
+        this.loginInfoIsLoaded = true;
         // restart plugin check
         this.checkIfTokenNeedRefresh(refresh_token, csrf);
         // enable refresh cookies detect
@@ -252,7 +258,7 @@ class BiliAPI extends koishi_1.Service {
             }).value;
             // 检查是否需要更新
             this.checkIfTokenNeedRefresh(refresh_token, csrf);
-        }, 43200000);
+        }, 3600000);
     }
     async checkIfTokenNeedRefresh(refreshToken, csrf, times = 3) {
         // 定义数据

package/lib/comRegister.d.ts

@@ -24,6 +24,7 @@ declare class ComRegister {
     redBot: Bot<Context>;
     telegramBot: Bot<Context>;
     satoriBot: Bot<Context>;
+    chronocatBot: Bot<Context>;
     constructor(ctx: Context, config: ComRegister.Config);
     dynamicDetect(ctx: Context, bot: Bot<Context>, uid: string, guildId: Array<string>): () => Promise<void>;
     sendMsg(targets: Array<string>, bot: Bot<Context>, content: any): Promise<void>;
@@ -31,6 +32,7 @@ declare class ComRegister {
     subShow(): string;
     checkIfNeedSub(comNeed: boolean, subType: string, session: Session, data?: any): Promise<boolean>;
     updateSubNotifier(ctx: Context): void;
+    checkIfLoginInfoIsLoaded(ctx: Context): Promise<unknown>;
     getSubFromDatabase(ctx: Context): Promise<void>;
     unsubSingle(ctx: Context, id: string, type: number): string;
     checkIfIsLogin(ctx: Context): Promise<boolean>;

package/lib/comRegister.js

@@ -33,6 +33,8 @@ class ComRegister {
     telegramBot;
     // Satori机器人
     satoriBot;
+    // Chronocat机器人
+    chronocatBot;
     constructor(ctx, config) {
         this.logger = ctx.logger('commandRegister');
         this.config = config;
@@ -57,6 +59,9 @@ class ComRegister {
                 case 'satori':
                     this.satoriBot = bot;
                     break;
+                case 'chronocat':
+                    this.chronocatBot = bot;
+                    break;
             }
         });
         // 从数据库获取订阅
@@ -146,11 +151,17 @@ class ComRegister {
             })
 
         testCom
-            .subcommand('.refresh')
-            .usage('测试cookie刷新方法')
-            .example('test refresh')
+            .subcommand('.livestop', '发送下播提示语测试')
+            .usage('发送下播提示语测试')
+            .example('test livestop')
             .action(async ({ session }) => {
-                ctx.biliAPI.test_refresh_token()
+                const { data } = await ctx.biliAPI.getMasterInfo('194484313')
+                console.log(data);
+                await session.send(
+                    <>
+                        <img width="10px" height="10px" src="https://koishi.chat/logo.png"/>
+                    </>
+                )
             }) */
         const biliCom = ctx.command('bili', 'bili-notify插件相关指令', { permissions: ['authority:3'] });
         biliCom.subcommand('.login', '登录B站之后才可以进行之后的操作')
@@ -311,6 +322,7 @@ class ComRegister {
                 case 'onebot':
                 case 'telegram':
                 case 'satori':
+                case 'chronocat':
                 case 'qq':
                 case 'qqguild': break;
                 default: return '暂不支持该平台';
@@ -404,6 +416,10 @@ class ComRegister {
                         okGuild = await checkIfGuildHasJoined(this.satoriBot);
                         break;
                     }
+                    case 'chronocat': {
+                        okGuild = await checkIfGuildHasJoined(this.chronocatBot);
+                        break;
+                    }
                     default: {
                         // 发送错误提示并返回
                         session.send('您尚未配置任何QQ群相关机器人，不能对QQ群进行操作');
@@ -517,6 +533,9 @@ class ComRegister {
                 case 'satori':
                     bot = this.satoriBot;
                     break;
+                case 'chronocat':
+                    bot = this.chronocatBot;
+                    break;
                 default: {
                     this.logger.warn(`${uid}非法调用 dynamic 指令，不支持该平台`);
                     return '非法调用';
@@ -563,6 +582,9 @@ class ComRegister {
                 case 'satori':
                     bot = this.satoriBot;
                     break;
+                case 'chronocat':
+                    bot = this.chronocatBot;
+                    break;
                 default: {
                     this.logger.warn(`${roomId}非法调用 dynamic 指令，不支持该平台`);
                     return `${roomId}非法调用 dynamic 指令`;
@@ -616,6 +638,19 @@ class ComRegister {
             // pic不存在，说明使用的是page模式
             await session.send(koishi_1.h.image(buffer, 'image/png'));
         });
+        biliCom
+            .subcommand('.bot', '查询当前拥有的机器人信息', { hidden: true })
+            .usage('查询当前拥有的机器人信息')
+            .example('bili bot 查询当前拥有的机器人信息')
+            .action(() => {
+            this.logger.info('开始输出BOT信息');
+            ctx.bots.forEach(bot => {
+                this.logger.info('--------------------------------');
+                this.logger.info('平台：' + bot.platform);
+                this.logger.info('名称：' + bot.user.name);
+                this.logger.info('--------------------------------');
+            });
+        });
     }
     dynamicDetect(ctx, bot, uid, guildId) {
         let firstSubscription = true;
@@ -740,18 +775,34 @@ class ComRegister {
         let uData;
         // 相当于锁的作用，防止上一个循环没处理完
         let flag = true;
-        const sendLiveNotifyCard = async (data, uData, liveType) => {
+        const sendLiveNotifyCard = async (data, uData, liveType, liveStartMsg, atAll) => {
             let attempts = 3;
             for (let i = 0; i < attempts; i++) {
                 try {
                     // 获取直播通知卡片
                     const { pic, buffer } = await ctx.gimg.generateLiveImg(data, uData, liveType);
                     // 推送直播信息
-                    // pic 存在，使用的是render模式
-                    if (pic)
-                        return await this.sendMsg(guildId, bot, pic);
-                    // pic不存在，说明使用的是page模式
-                    await this.sendMsg(guildId, bot, koishi_1.h.image(buffer, 'image/png'));
+                    if (!liveStartMsg) {
+                        // pic 存在，使用的是render模式
+                        if (pic)
+                            return await this.sendMsg(guildId, bot, pic);
+                        // pic不存在，说明使用的是page模式
+                        await this.sendMsg(guildId, bot, koishi_1.h.image(buffer, 'image/png'));
+                    }
+                    else if (liveStartMsg && atAll) {
+                        // pic 存在，使用的是render模式
+                        if (pic)
+                            return await this.sendMsg(guildId, bot, pic + (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [(0, jsx_runtime_1.jsx)("at", { type: "all" }), " ", liveStartMsg, " "] }));
+                        // pic不存在，说明使用的是page模式
+                        await this.sendMsg(guildId, bot, koishi_1.h.image(buffer, 'image/png' + ' ' + (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [(0, jsx_runtime_1.jsx)("at", { type: "all" }), " ", liveStartMsg] })));
+                    }
+                    else {
+                        // pic 存在，使用的是render模式
+                        if (pic)
+                            return await this.sendMsg(guildId, bot, pic + liveStartMsg);
+                        // pic不存在，说明使用的是page模式
+                        await this.sendMsg(guildId, bot, koishi_1.h.image(buffer, 'image/png' + ' ' + liveStartMsg));
+                    }
                     // 成功则跳出循环
                     break;
                 }
@@ -869,16 +920,16 @@ class ComRegister {
                             // 定义开播通知语
                             let liveStartMsg = this.config.customLiveStart
                                 .replace('-name', uData.info.uname)
-                                .replace('-time', await ctx.gimg.getTimeDifference(liveTime));
-                            // 发送直播通知卡片
-                            await sendLiveNotifyCard(data, uData, LiveType.StartBroadcasting);
+                                .replace('-time', await ctx.gimg.getTimeDifference(liveTime))
+                                .replace('-link', `https://live.bilibili.com/${data.short_id === 0 ? data.room_id : data.short_id}`);
                             // 判断是否需要@全体成员
                             if (this.config.liveStartAtAll) {
                                 // 发送@全体成员通知
-                                await this.sendMsg(guildId, bot, (0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [(0, jsx_runtime_1.jsx)("at", { type: "all" }), " ", liveStartMsg, " "] }));
+                                await sendLiveNotifyCard(data, uData, LiveType.StartBroadcasting, liveStartMsg, true);
                             }
                             else {
-                                await this.sendMsg(guildId, bot, liveStartMsg);
+                                // 发送直播通知卡片
+                                await sendLiveNotifyCard(data, uData, LiveType.StartBroadcasting, liveStartMsg);
                             }
                         }
                         else { // 还在直播
@@ -966,24 +1017,33 @@ class ComRegister {
         // 设置更新后的提示
         this.subNotifier = ctx.notifier.create(table);
     }
+    async checkIfLoginInfoIsLoaded(ctx) {
+        return new Promise(resolve => {
+            const check = () => {
+                if (!ctx.biliAPI.getLoginInfoIsLoaded()) {
+                    ctx.setTimeout(check, 500);
+                }
+                else {
+                    resolve('success');
+                }
+            };
+            check();
+        });
+    }
     async getSubFromDatabase(ctx) {
-        this.logger.info('开始执行数据库读取操作');
-        // 检查登录状态
-        const isLogin = await this.checkIfIsLogin(ctx);
-        // log
-        this.logger.info(`登录状态:${isLogin}`);
+        // 判断登录信息是否已加载完毕
+        await this.checkIfLoginInfoIsLoaded(ctx);
         // 如果未登录，则直接返回
-        if (!(await this.checkIfIsLogin(ctx)))
+        if (!(await this.checkIfIsLogin(ctx))) {
+            // log
+            this.logger.info(`账号未登录，请登录`);
             return;
-        this.logger.info('已登录账号');
+        }
         // 已存在订阅管理对象，不再进行订阅操作
         if (this.subManager.length !== 0)
             return;
-        this.logger.info('不存在订阅管理对象');
         // 从数据库中获取数据
         const subData = await ctx.database.get('bilibili', { id: { $gt: 0 } });
-        this.logger.info('已从数据库获取到数据，数据为：');
-        this.logger.info(subData);
         // 设定订阅数量
         this.num = subData.length;
         // 如果订阅数量超过三个则数据库被非法修改
@@ -1028,6 +1088,9 @@ class ComRegister {
                 case 'satori':
                     bot = this.satoriBot;
                     break;
+                case 'chronocat':
+                    bot = this.chronocatBot;
+                    break;
                 default: {
                     // 本条数据被篡改，删除该条订阅
                     ctx.database.remove('bilibili', { id: sub.id });
@@ -1047,8 +1110,6 @@ class ComRegister {
                 try {
                     // 获取用户信息
                     content = await ctx.biliAPI.getUserInfo(sub.uid);
-                    // log
-                    this.logger.info(`UID:${sub.uid} 获取到用户信息`);
                     // 成功则跳出循环
                     break;
                 }
@@ -1094,8 +1155,6 @@ class ComRegister {
                 this.logger.info(`UID:${sub.uid} 房间号被篡改，自动取消订阅`);
                 return;
             }
-            // log
-            this.logger.info(`UID:${sub.uid} 开始构建订阅对象`);
             // 构建订阅对象
             let subManagerItem = {
                 id: sub.id,
@@ -1108,34 +1167,24 @@ class ComRegister {
                 liveDispose: null,
                 dynamicDispose: null
             };
-            // log
-            this.logger.info(`UID:${sub.uid} 订阅对象构建成功，开始进行订阅操作`);
             // 判断需要订阅的服务
             if (sub.dynamic) { // 需要订阅动态
                 // 开始循环检测
                 const dispose = ctx.setInterval(this.dynamicDetect(ctx, bot, sub.uid, targetArr), this.config.dynamicLoopTime * 1000);
                 // 保存销毁函数
                 subManagerItem.dynamicDispose = dispose;
-                // log
-                this.logger.info(`UID:${sub.uid} 成功订阅动态`);
             }
             if (sub.live) { // 需要订阅直播
                 // 开始循环检测
                 const dispose = ctx.setInterval(this.liveDetect(ctx, bot, sub.room_id, targetArr), this.config.liveLoopTime * 1000);
                 // 保存销毁函数
                 subManagerItem.liveDispose = dispose;
-                // log
-                this.logger.info(`UID:${sub.uid} 成功订阅直播`);
             }
             // 保存新订阅对象
             this.subManager.push(subManagerItem);
-            // log
-            this.logger.info(`UID:${sub.uid} 成功保存订阅对象`);
         }
         // 在控制台中显示订阅对象
         this.updateSubNotifier(ctx);
-        // log
-        this.logger.info(`数据库读取操作已完成`);
     }
     unsubSingle(ctx, id /* UID或RoomId */, type /* 0取消Live订阅，1取消Dynamic订阅 */) {
         let index;

package/lib/index.js

@@ -80,8 +80,8 @@ exports.Config = koishi_1.Schema.object({
         .default(1)
         .description('设定隔多长时间推送一次直播状态，单位为小时，默认为一小时'),
     customLiveStart: koishi_1.Schema.string()
-        .default('-name开播啦')
-        .description('自定义开播提示语，-name代表UP昵称。例如-name开播啦，会发送为xxxUP开播啦'),
+        .default('-name开播啦 -link')
+        .description('自定义开播提示语，-name代表UP昵称，-link代表直播间链接（如果使用的是QQ官方机器人，请不要使用）。例如-name开播啦，会发送为xxxUP开播啦'),
     customLiveEnd: koishi_1.Schema.string()
         .default('-name下播啦，本次直播了-time')
         .description('自定义下播提示语，-name代表UP昵称，-time代表开播时长。例如-name下播啦，本次直播了-time，会发送为xxxUP下播啦，直播时长为xx小时xx分钟xx秒'),

package/package.json

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

package/readme.md

@@ -116,6 +116,8 @@
 - ver 1.2.0 添加屏蔽转发动态功能，添加发送动态卡片时附带文本信息和动态链接功能，支持订阅哔哩哔哩番剧出差
 - ver 1.2.1 现已支持Satori平台(实验性)
 - ver 1.2.2-alpha.0 bug测试
+- ver 1.2.2-beta.0 修复重启koishi后，提示没有任何订阅的bug，新增对chronocat的支持(实验性)
+- ver 1.2.2-beta.1 现已支持直播开播发送链接(实验性)
 
 ## 交流群
 801338523 使用问题或bug都可以在群里提出