koishi-plugin-bilibili-notify 1.0.6-alpha.0 → 1.0.8

package/lib/comRegister.d.ts

@@ -1,9 +1,11 @@
 import { Bot, Context, Logger, Schema, Session } from "koishi";
+import { Notifier } from "@koishijs/plugin-notifier";
 declare class ComRegister {
     static inject: string[];
     logger: Logger;
     config: ComRegister.Config;
     num: number;
+    subNotifier: Notifier;
     subManager: {
         id: number;
         uid: string;
@@ -19,8 +21,10 @@ declare class ComRegister {
     constructor(ctx: Context, config: ComRegister.Config);
     dynamicDetect(ctx: Context, bot: Bot<Context>, guildId: string, uid: string): () => Promise<void | string[]>;
     liveDetect(ctx: Context, bot: Bot<Context>, guildId: string, roomId: string): () => Promise<void>;
+    subShow(): string;
     checkIfNeedSub(comNeed: boolean, subType: string, session: Session, data?: any): Promise<boolean>;
-    getSubFromDatabase(ctx: Context): Promise<void>;
+    updateSubNotifier(ctx: Context): void;
+    getSubFromDatabase(ctx: Context): Promise<string[]>;
     unsubSingle(ctx: Context, id: string, type: number): string;
     checkIfIsLogin(ctx: Context): Promise<boolean>;
 }

package/lib/comRegister.js

@@ -17,6 +17,7 @@ class ComRegister {
     logger;
     config;
     num = 0;
+    subNotifier;
     subManager = [];
     // QQ群机器人
     qqBot;
@@ -175,13 +176,13 @@ class ComRegister {
                         }]);
                     // 销毁定时器
                     dispose();
+                    // 订阅之前的订阅
+                    await this.getSubFromDatabase(ctx);
                     // 清除控制台通知
                     ctx.biliAPI.disposeNotifier();
                     // 发送成功登录推送
                     await session.send('登录成功！');
-                    // 订阅之前的订阅
-                    await this.getSubFromDatabase(ctx);
-                    // 调用bili show指令
+                    // bili show
                     await session.execute('bili show');
                     return;
                 }
@@ -198,6 +199,9 @@ class ComRegister {
             // 若用户UID为空则直接返回
             if (!uid)
                 return '用户UID不能为空';
+            // -d -l两个选项不能同时存在
+            if (options.dynamic && options.live)
+                return '需要取消订阅该UP主请直接使用指令bili unsub 用户UID';
             // 定义是否存在
             let exist;
             await Promise.all(this.subManager.map(async (sub, i) => {
@@ -220,11 +224,13 @@ class ComRegister {
                     // 从数据库中删除订阅
                     await ctx.database.remove('bilibili', { uid: this.subManager[i].uid });
                     // 将该订阅对象从订阅管理对象中移除
-                    this.subManager = this.subManager.splice(i, i);
+                    this.subManager.splice(i, 1);
                     // id--
                     this.num--;
                     // 发送成功通知
                     session.send('已取消订阅该用户');
+                    // 更新控制台提示
+                    this.updateSubNotifier(ctx);
                     // 将存在flag设置为true
                     exist = true;
                 }
@@ -236,13 +242,9 @@ class ComRegister {
             .subcommand('.show', '展示订阅对象')
             .usage('展示订阅对象')
             .example('bili show')
-            .action(async ({ session }) => {
-            let table = ``;
-            this.subManager.forEach(sub => {
-                table += `UID:${sub.uid}  ${sub.dynamic ? '已订阅动态' : ''}  ${sub.live ? '已订阅直播' : ''}` + '\n';
-            });
-            !table && session.send('没有订阅任何UP');
-            table && session.send(table);
+            .action(() => {
+            const subTable = this.subShow();
+            return subTable;
         });
         ctx.command('bili')
             .subcommand('.sub <mid:string> [guildId:string]', '订阅用户动态和直播通知')
@@ -382,6 +384,15 @@ class ComRegister {
                 // 发送订阅消息通知
                 await bot.sendMessage(sub.targetId, `订阅${userData.info.uname}动态通知`);
             }
+            // 新增订阅展示到控制台
+            // 获取subTable
+            const subTable = this.subShow();
+            // 判断之前是否存在Notifier
+            this.subNotifier && this.subNotifier.dispose();
+            this.subNotifier = ctx.notifier.create({
+                type: 'primary',
+                content: subTable
+            });
         });
         ctx.command('bili')
             .subcommand('.dynamic <uid:string> <guildId:string>', '订阅用户动态推送', { hidden: true })
@@ -503,7 +514,7 @@ class ComRegister {
         let firstSubscription = true;
         let timePoint;
         // Test code
-        let timer = 0;
+        // let timer = 0
         return async () => {
             // Test code
             /* this.logger.info('timer:' + timer++)
@@ -748,6 +759,14 @@ class ComRegister {
             }
         };
     }
+    subShow() {
+        // 在控制台中显示订阅对象
+        let table = ``;
+        this.subManager.forEach(sub => {
+            table += `UID:${sub.uid}  ${sub.dynamic ? '已订阅动态' : ''}  ${sub.live ? '已订阅直播' : ''}` + '\n';
+        });
+        return table ? table : '没有订阅任何UP';
+    }
     async checkIfNeedSub(comNeed, subType, session, data) {
         if (comNeed) {
             if (subType === '直播' && !data.live_room) {
@@ -783,16 +802,30 @@ class ComRegister {
             }
         }
     }
+    updateSubNotifier(ctx) {
+        // 更新控制台提示
+        this.subNotifier && this.subNotifier.dispose();
+        // 获取subTable
+        const subTable = this.subShow();
+        // 设置更新后的提示
+        this.subNotifier = ctx.notifier.create({
+            type: 'primary',
+            content: subTable
+        });
+    }
     async getSubFromDatabase(ctx) {
-        if (!(await this.checkIfIsLogin(ctx))) { // 如果未登录，则直接返回
+        // 如果未登录，则直接返回
+        if (!(await this.checkIfIsLogin(ctx)))
+            return;
+        // 已存在订阅管理对象，不再进行订阅操作
+        if (this.subManager.length !== 0)
             return;
-        }
         // 从数据库中获取数据
         const subData = await ctx.database.get('bilibili', { id: { $gt: 0 } });
         // 设定订阅数量
         this.num = subData.length;
         // 如果订阅数量超过三个则被非法修改数据库
-        // 向管理员发送重新订阅通知
+        // 在控制台提示重新订阅
         if (this.num > 3) {
             ctx.notifier.create({
                 type: 'danger',
@@ -803,7 +836,14 @@ class ComRegister {
         // 定义Bot
         let bot;
         // 循环遍历
-        subData.forEach(async (sub) => {
+        for (const sub of subData) {
+            // 判断是否存在没有任何订阅的数据
+            if (!sub.dynamic && !sub.live) { // 存在未订阅任何项目的数据
+                // 删除该条数据
+                ctx.database.remove('bilibili', { id: sub.id });
+                // 跳过下面的步骤
+                continue;
+            }
             // 拿到对应bot
             switch (sub.platform) {
                 case 'qq': bot = this.qqBot;
@@ -880,67 +920,72 @@ class ComRegister {
             }
             // 保存新订阅对象
             this.subManager.push(subManagerItem);
-            // 发送订阅成功通知
-        });
+        }
+        // 在控制台中显示订阅对象
+        this.updateSubNotifier(ctx);
     }
     unsubSingle(ctx, id /* UID或RoomId */, type /* 0取消Live订阅，1取消Dynamic订阅 */) {
         let index;
-        switch (type) {
-            case 0: { // 取消Live订阅
-                index = this.subManager.findIndex(sub => sub.roomId === id);
-                if (index === -1)
-                    return '未订阅该用户，无需取消订阅';
-                // 取消订阅
-                this.subManager[index].live && this.subManager[index].liveDispose();
-                // 如果没有对这个UP的任何订阅，则移除
-                if (!this.subManager[index].dynamic) {
-                    // 获取要删除行的id
-                    const id = this.subManager[index].id;
-                    // 从管理对象中移除
-                    this.subManager = this.subManager.splice(index, index);
-                    // 从数据库中删除
-                    ctx.database.remove('bilibili', [id]);
-                    // num--
-                    this.num--;
-                    return '已取消订阅该用户';
+        const checkIfNoSubExist = (index) => {
+            if (!this.subManager[index].dynamic && !this.subManager[index].live) {
+                // 获取要删除行的id
+                const id = this.subManager[index].id;
+                // 从管理对象中移除
+                this.subManager.splice(index, 1);
+                // 从数据库中删除
+                ctx.database.remove('bilibili', [id]);
+                // num--
+                this.num--;
+                return '已取消订阅该用户';
+            }
+            return null;
+        };
+        try {
+            switch (type) {
+                case 0: { // 取消Live订阅
+                    index = this.subManager.findIndex(sub => sub.roomId === id);
+                    if (index === -1)
+                        return '未订阅该用户，无需取消订阅';
+                    // 取消订阅
+                    this.subManager[index].live && this.subManager[index].liveDispose();
+                    this.subManager[index].liveDispose = null;
+                    this.subManager[index].live = false;
+                    // 如果没有对这个UP的任何订阅，则移除
+                    const info = checkIfNoSubExist(index);
+                    if (info)
+                        return info;
+                    // 更新数据库
+                    ctx.database.upsert('bilibili', [{
+                            id: +`${this.subManager[index].id}`,
+                            live: 0
+                        }]);
+                    return '已取消订阅Live';
                 }
-                this.subManager[index].liveDispose = null;
-                this.subManager[index].live = false;
-                // 更新数据库
-                ctx.database.upsert('bilibili', [{
-                        id: +`${this.subManager[index].id}`,
-                        live: 0
-                    }]);
-                return '已取消订阅Live';
-            }
-            case 1: { // 取消Dynamic订阅
-                index = this.subManager.findIndex(sub => sub.uid === id);
-                if (index === -1)
-                    return '未订阅该用户，无需取消订阅';
-                // 取消订阅
-                this.subManager[index].dynamic && this.subManager[index].dynamicDispose();
-                // 如果没有对这个UP的任何订阅，则移除
-                if (!this.subManager[index].live) {
-                    // 获取要删除行的id
-                    const id = this.subManager[index].id;
-                    // 从管理对象中移除
-                    this.subManager = this.subManager.splice(index, index);
-                    // 从数据库中删除
-                    ctx.database.remove('bilibili', [id]);
-                    // num--
-                    this.num--;
-                    return '已取消订阅该用户';
+                case 1: { // 取消Dynamic订阅
+                    index = this.subManager.findIndex(sub => sub.uid === id);
+                    if (index === -1)
+                        return '未订阅该用户，无需取消订阅';
+                    // 取消订阅
+                    this.subManager[index].dynamic && this.subManager[index].dynamicDispose();
+                    this.subManager[index].dynamicDispose = null;
+                    this.subManager[index].dynamic = false;
+                    // 如果没有对这个UP的任何订阅，则移除
+                    const info = checkIfNoSubExist(index);
+                    if (info)
+                        return info;
+                    // 更新数据库
+                    ctx.database.upsert('bilibili', [{
+                            id: +`${this.subManager[index].id}`,
+                            dynamic: 0
+                        }]);
+                    return '已取消订阅Dynamic';
                 }
-                this.subManager[index].dynamicDispose = null;
-                this.subManager[index].dynamic = false;
-                // 更新数据库
-                ctx.database.upsert('bilibili', [{
-                        id: +`${this.subManager[index].id}`,
-                        dynamic: 0
-                    }]);
-                return '已取消订阅Dynamic';
             }
         }
+        finally {
+            // 执行完该方法后，保证执行一次updateSubNotifier()
+            this.updateSubNotifier(ctx);
+        }
     }
     async checkIfIsLogin(ctx) {
         if ((await ctx.database.get('loginBili', 1)).length !== 0) { // 数据库中有数据

package/package.json

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

package/readme.md

@@ -81,7 +81,9 @@
 - ver 1.0.3 修复了一些bug，提供用户自己选择推送卡片字体样式的选项
 - ver 1.0.4 修复了重复推送的bug，提供用户选择推送卡片渲染方式的选项
 - ver 1.0.5 修复了用户非法篡改数据库内容可能导致程序异常运行的bug，修复了UP主开播动态推送空白卡片的bug
-- ver 1.0.6 修复了转发动态转发信息出现undefined的bug，优化了动态推送的逻辑
+- ver 1.0.6 修复了转发动态转发信息出现undefined的bug，修复了再次登录订阅显示错误的bug，优化了动态推送的逻辑
+- ver 1.0.7 修复了在已登录情况下，再次登录会导致重复订阅和提示用户未订阅任何UP主的提示（实际上已订阅）的bug，新增了订阅对象在控制台的显示，优化了bili show指令的逻辑
+- ver 1.0.8 修复了取消订阅的bug
 
 ## 感谢