koishi-plugin-kkk 3.7.2 → 3.9.0

package/lib/karin/module/db/index.js

@@ -14,13 +14,17 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.cleanOldDynamicCache = exports.bootstrapDatabases = exports.statisticsDBInstance = exports.bilibiliDBInstance = exports.douyinDBInstance = exports.statisticsDB = exports.bilibiliDB = exports.douyinDB = exports.initAllDatabases = exports.getStatisticsDB = exports.getBilibiliDB = exports.getDouyinDB = void 0;
+exports.cleanOldDynamicCache = exports.bootstrapDatabases = exports.statisticsDBInstance = exports.bilibiliDBInstance = exports.douyinDBInstance = exports.statisticsDB = exports.bilibiliDB = exports.douyinDB = exports.initAllDatabases = exports.getStatisticsDB = exports.getBilibiliDB = exports.getDouyinDB = exports.setDatabaseContext = exports.extendModels = exports.TABLE = void 0;
 const bilibili_1 = require("./bilibili");
 const douyin_1 = require("./douyin");
+const model_1 = require("./model");
 const statistics_1 = require("./statistics");
 __exportStar(require("./bilibili"), exports);
 __exportStar(require("./douyin"), exports);
 __exportStar(require("./statistics"), exports);
+var model_2 = require("./model");
+Object.defineProperty(exports, "TABLE", { enumerable: true, get: function () { return model_2.TABLE; } });
+Object.defineProperty(exports, "extendModels", { enumerable: true, get: function () { return model_2.extendModels; } });
 /** 抖音数据库实例 */
 let douyinDB = null;
 let douyinInitializing = false;
@@ -30,6 +34,26 @@ let bilibiliInitializing = false;
 /** 统计数据库实例 */
 let statisticsDB = null;
 let statisticsInitializing = false;
+/**
+ * 数据库用的是 **Koishi 原生数据库服务**（`ctx.database`），
+ * 插件自己不再开 sqlite 文件。这里存的就是插件启动时传进来的那个 ctx。
+ */
+let dbCtx = null;
+/**
+ * 绑定数据库上下文并注册表结构。
+ * 插件 apply 阶段调用一次；之后所有数据库实例都从这个 ctx 上取 `ctx.database`。
+ */
+const setDatabaseContext = (ctx) => {
+    dbCtx = ctx;
+    (0, model_1.extendModels)(ctx);
+};
+exports.setDatabaseContext = setDatabaseContext;
+/** 取数据库上下文；还没绑定就抛一个说得清楚的错 */
+const requireCtx = (name) => {
+    if (!dbCtx)
+        throw new Error('[' + name + '] 数据库尚未初始化，请先等待插件启动完成');
+    return dbCtx;
+};
 /**
  * 获取或初始化 DouyinDB 实例（单例模式）
  * @returns DouyinDB实例
@@ -44,7 +68,7 @@ const getDouyinDB = async () => {
     }
     douyinInitializing = true;
     try {
-        douyinDB = await new douyin_1.DouyinDBBase().init();
+        douyinDB = await new douyin_1.DouyinDBBase(requireCtx('DouyinDB')).init();
         return douyinDB;
     }
     finally {
@@ -66,7 +90,7 @@ const getBilibiliDB = async () => {
     }
     bilibiliInitializing = true;
     try {
-        bilibiliDB = await new bilibili_1.BilibiliDBBase().init();
+        bilibiliDB = await new bilibili_1.BilibiliDBBase(requireCtx('BilibiliDB')).init();
         return bilibiliDB;
     }
     finally {
@@ -88,7 +112,7 @@ const getStatisticsDB = async () => {
     }
     statisticsInitializing = true;
     try {
-        statisticsDB = await new statistics_1.StatisticsDBBase().init();
+        statisticsDB = await new statistics_1.StatisticsDBBase(requireCtx('StatisticsDB')).init();
         return statisticsDB;
     }
     finally {
@@ -138,7 +162,8 @@ exports.douyinDBInstance = douyinDB;
 exports.bilibiliDBInstance = bilibiliDB;
 exports.statisticsDBInstance = statisticsDB;
 /** 初始化并填充所有数据库实例（插件启动时调用一次） */
-const bootstrapDatabases = async () => {
+const bootstrapDatabases = async (ctx) => {
+    (0, exports.setDatabaseContext)(ctx);
     const result = await (0, exports.initAllDatabases)();
     douyinDBReal = result.douyinDB;
     bilibiliDBReal = result.bilibiliDB;

package/lib/karin/module/db/model.js

@@ -0,0 +1,200 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.extendModels = exports.TABLE = void 0;
+/** 表名常量表 */
+exports.TABLE = {
+    douyinBot: 'kkk.douyin.bot',
+    douyinGroup: 'kkk.douyin.group',
+    douyinUser: 'kkk.douyin.user',
+    douyinSubscription: 'kkk.douyin.subscription',
+    douyinAwemeCache: 'kkk.douyin.awemeCache',
+    douyinFilterWord: 'kkk.douyin.filterWord',
+    douyinFilterTag: 'kkk.douyin.filterTag',
+    douyinListSnapshot: 'kkk.douyin.listSnapshot',
+    bilibiliBot: 'kkk.bilibili.bot',
+    bilibiliGroup: 'kkk.bilibili.group',
+    bilibiliUser: 'kkk.bilibili.user',
+    bilibiliSubscription: 'kkk.bilibili.subscription',
+    bilibiliDynamicCache: 'kkk.bilibili.dynamicCache',
+    bilibiliFilterWord: 'kkk.bilibili.filterWord',
+    bilibiliFilterTag: 'kkk.bilibili.filterTag',
+    statsParse: 'kkk.statistics.parse',
+    statsHistory: 'kkk.statistics.history',
+    statsGlobal: 'kkk.statistics.global',
+    statsGroupHistory: 'kkk.statistics.groupHistory',
+    statsHour: 'kkk.statistics.hour',
+    statsWorkType: 'kkk.statistics.workType',
+    statsMetric: 'kkk.statistics.metric',
+};
+/* ------------------------------ 建表 ------------------------------ */
+/** 进程内只建一次（插件热重载时 apply 会再跑一遍，重复 extend 没意义） */
+let modelsReady = false;
+/**
+ * 注册插件用到的所有表。
+ * 在数据库实例初始化之前调用一次即可；实际建表由 Koishi 的数据库服务负责。
+ */
+const extendModels = (ctx) => {
+    if (modelsReady)
+        return;
+    modelsReady = true;
+    // ---------- 抖音 ----------
+    ctx.model.extend(exports.TABLE.douyinBot, {
+        id: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { primary: 'id' });
+    ctx.model.extend(exports.TABLE.douyinGroup, {
+        id: 'string',
+        botId: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { primary: ['id', 'botId'] });
+    ctx.model.extend(exports.TABLE.douyinUser, {
+        sec_uid: 'string',
+        short_id: 'string',
+        remark: 'string',
+        living: 'boolean',
+        filterMode: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { primary: 'sec_uid' });
+    ctx.model.extend(exports.TABLE.douyinSubscription, {
+        groupId: 'string',
+        sec_uid: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { primary: ['groupId', 'sec_uid'] });
+    ctx.model.extend(exports.TABLE.douyinAwemeCache, {
+        id: 'unsigned',
+        aweme_id: 'string',
+        sec_uid: 'string',
+        groupId: 'string',
+        pushType: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { autoInc: true, unique: [['aweme_id', 'sec_uid', 'groupId', 'pushType']] });
+    ctx.model.extend(exports.TABLE.douyinFilterWord, {
+        id: 'unsigned',
+        sec_uid: 'string',
+        word: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { autoInc: true, unique: [['sec_uid', 'word']] });
+    ctx.model.extend(exports.TABLE.douyinFilterTag, {
+        id: 'unsigned',
+        sec_uid: 'string',
+        tag: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { autoInc: true, unique: [['sec_uid', 'tag']] });
+    ctx.model.extend(exports.TABLE.douyinListSnapshot, {
+        id: 'unsigned',
+        sec_uid: 'string',
+        pushType: 'string',
+        aweme_id: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { autoInc: true, unique: [['sec_uid', 'pushType', 'aweme_id']] });
+    // ---------- B站 ----------
+    ctx.model.extend(exports.TABLE.bilibiliBot, {
+        id: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { primary: 'id' });
+    ctx.model.extend(exports.TABLE.bilibiliGroup, {
+        id: 'string',
+        botId: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { primary: ['id', 'botId'] });
+    ctx.model.extend(exports.TABLE.bilibiliUser, {
+        host_mid: 'unsigned',
+        remark: 'string',
+        filterMode: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { primary: 'host_mid' });
+    ctx.model.extend(exports.TABLE.bilibiliSubscription, {
+        groupId: 'string',
+        host_mid: 'unsigned',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { primary: ['groupId', 'host_mid'] });
+    ctx.model.extend(exports.TABLE.bilibiliDynamicCache, {
+        id: 'unsigned',
+        dynamic_id: 'string',
+        host_mid: 'unsigned',
+        groupId: 'string',
+        dynamic_type: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { autoInc: true, unique: [['dynamic_id', 'host_mid', 'groupId']] });
+    ctx.model.extend(exports.TABLE.bilibiliFilterWord, {
+        id: 'unsigned',
+        host_mid: 'unsigned',
+        word: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { autoInc: true, unique: [['host_mid', 'word']] });
+    ctx.model.extend(exports.TABLE.bilibiliFilterTag, {
+        id: 'unsigned',
+        host_mid: 'unsigned',
+        tag: 'string',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { autoInc: true, unique: [['host_mid', 'tag']] });
+    // ---------- 统计 ----------
+    ctx.model.extend(exports.TABLE.statsParse, {
+        id: 'unsigned',
+        groupId: 'string',
+        userId: 'string',
+        platform: 'string',
+        parseCount: 'unsigned',
+        createdAt: 'string',
+        updatedAt: 'string',
+    }, { autoInc: true, unique: [['groupId', 'userId', 'platform']] });
+    ctx.model.extend(exports.TABLE.statsHistory, {
+        id: 'unsigned',
+        date: 'string',
+        totalParses: 'unsigned',
+        douyin: 'unsigned',
+        bilibili: 'unsigned',
+        kuaishou: 'unsigned',
+        xiaohongshu: 'unsigned',
+        createdAt: 'string',
+    }, { autoInc: true, unique: [['date']] });
+    ctx.model.extend(exports.TABLE.statsGlobal, {
+        key: 'string',
+        value: 'unsigned',
+        updatedAt: 'string',
+    }, { primary: 'key' });
+    ctx.model.extend(exports.TABLE.statsGroupHistory, {
+        groupId: 'string',
+        date: 'string',
+        platform: 'string',
+        parseCount: 'unsigned',
+        updatedAt: 'string',
+    }, { primary: ['groupId', 'date', 'platform'] });
+    ctx.model.extend(exports.TABLE.statsHour, {
+        groupId: 'string',
+        hour: 'unsigned',
+        platform: 'string',
+        parseCount: 'unsigned',
+        updatedAt: 'string',
+    }, { primary: ['groupId', 'hour', 'platform'] });
+    ctx.model.extend(exports.TABLE.statsWorkType, {
+        groupId: 'string',
+        platform: 'string',
+        workType: 'string',
+        parseCount: 'unsigned',
+        updatedAt: 'string',
+    }, { primary: ['groupId', 'platform', 'workType'] });
+    ctx.model.extend(exports.TABLE.statsMetric, {
+        groupId: 'string',
+        metric: 'string',
+        bucket: 'string',
+        parseCount: 'unsigned',
+        updatedAt: 'string',
+    }, { primary: ['groupId', 'metric', 'bucket'] });
+};
+exports.extendModels = extendModels;