koishi-plugin-cat-raising 1.3.9 → 1.4.0

package/lib/index.d.ts

@@ -1,5 +1,21 @@
 import { Context, Schema } from 'koishi';
 export declare const name = "cat-raising";
+export declare const inject: {
+    required: string[];
+    optional: any[];
+};
+declare module 'koishi' {
+    interface Tables {
+        cat_raising_forward_history: CatRaisingForwardHistory;
+    }
+}
+export interface CatRaisingForwardHistory {
+    id: number;
+    roomId: string;
+    originalMessageId: string;
+    forwardTime: Date;
+    shenjinCount: number;
+}
 /** B站 access_key 配置项 */
 export interface BiliAccessKeyConfig {
     /** Bilibili access_key */

package/lib/index.js

@@ -32,13 +32,19 @@ var src_exports = {};
 __export(src_exports, {
   Config: () => Config,
   apply: () => apply,
+  inject: () => inject,
   name: () => name
 });
 module.exports = __toCommonJS(src_exports);
 var import_koishi = require("koishi");
 var crypto = __toESM(require("crypto"));
+var net = __toESM(require("net"));
 var import_url = require("url");
 var name = "cat-raising";
+var inject = {
+  required: ["database"],
+  optional: []
+};
 var Config = import_koishi.Schema.intersect([
   import_koishi.Schema.object({
     targetQQ: import_koishi.Schema.string().description("转发目标：填 QQ 或 群号").role("link").required(),
@@ -314,11 +320,97 @@ async function fetchBilibiliInfo(ctx, roomId) {
 }
 __name(fetchBilibiliInfo, "fetchBilibiliInfo");
 function apply(ctx, config) {
+  ctx.model.extend("cat_raising_forward_history", {
+    id: "unsigned",
+    roomId: "string",
+    originalMessageId: "string",
+    forwardTime: "timestamp",
+    shenjinCount: "double"
+  }, {
+    autoInc: true
+  });
   const forwardedHistory = [];
   const warningMessageMap = /* @__PURE__ */ new Map();
   const pendingDanmakuTimersByMessage = /* @__PURE__ */ new Map();
   const pendingDanmakuTimersByRoom = /* @__PURE__ */ new Map();
   const pendingDanmakuRoomMap = /* @__PURE__ */ new Map();
+  ctx.command("cat", "🐱喵喵喵");
+  ctx.command("cat.status", "查看插件状态").action(async () => {
+    let latencyMsg = "";
+    try {
+      const ms = await tcpPing("api.live.bilibili.com", 80);
+      latencyMsg = `B站API延迟: ${ms}ms`;
+    } catch (e) {
+      latencyMsg = `B站API延迟: 失败 (${e.message || "Unknown"})`;
+    }
+    return `🐱喵喵喵
+${latencyMsg}`;
+  });
+  ctx.command("cat.total", "统计最近24小时内的神金数量").action(async () => {
+    const oneDayAgo = new Date(Date.now() - 24 * 60 * 60 * 1e3);
+    const records = await ctx.database.select("cat_raising_forward_history").where((row) => import_koishi.$.gte(row.forwardTime, oneDayAgo)).execute();
+    const count = records.length;
+    const stats = /* @__PURE__ */ new Map();
+    const now = /* @__PURE__ */ new Date();
+    for (let i = 23; i >= 0; i--) {
+      const d = new Date(now.getTime() - i * 60 * 60 * 1e3);
+      const key = `${d.getHours().toString().padStart(2, "0")}:00`;
+      stats.set(key, 0);
+    }
+    for (const record of records) {
+      const d = new Date(record.forwardTime);
+      const key = `${d.getHours().toString().padStart(2, "0")}:00`;
+      if (stats.has(key)) {
+        stats.set(key, stats.get(key) + 1);
+      }
+    }
+    const labels = Array.from(stats.keys());
+    const data = Array.from(stats.values());
+    const chartConfig = {
+      type: "line",
+      data: {
+        labels,
+        datasets: [{
+          label: "转发数量",
+          data,
+          borderColor: "rgb(75, 192, 192)",
+          tension: 0.1,
+          fill: false
+        }]
+      },
+      options: {
+        title: {
+          display: true,
+          text: `最近24小时内的神金数: ${count}`
+        },
+        scales: {
+          yAxes: [{
+            ticks: {
+              beginAtZero: true,
+              stepSize: 1
+            }
+          }]
+        }
+      }
+    };
+    try {
+      const response = await ctx.http.post("https://quickchart.io/chart", {
+        chart: chartConfig,
+        width: 800,
+        height: 400,
+        backgroundColor: "white"
+      }, {
+        responseType: "arraybuffer"
+      });
+      return (0, import_koishi.h)("message", {}, [
+        import_koishi.h.text(`最近24小时内的神金数: ${count}`),
+        import_koishi.h.image(response, "image/png")
+      ]);
+    } catch (err) {
+      ctx.logger.error("Error generating chart:", err);
+      return `最近24小时内的神金数: ${count} (图表生成失败)`;
+    }
+  });
   ctx.on("message", async (session) => {
     const groupConfig = config.monitorGroups.find((g) => g.groupId === session.channelId);
     if (!groupConfig) return;
@@ -379,9 +471,17 @@ function apply(ctx, config) {
           helperMessageId,
           // 存储辅助消息ID用于撤回联动
           roomId,
-          dateTime
+          dateTime,
+          forwardTime: Date.now()
         });
         if (forwardedHistory.length > config.historySize) forwardedHistory.shift();
+        const shenjinCount = parsedEvent.rewards.reduce((sum, r) => sum + r.amount, 0);
+        ctx.database.create("cat_raising_forward_history", {
+          roomId,
+          originalMessageId: session.messageId,
+          forwardTime: /* @__PURE__ */ new Date(),
+          shenjinCount
+        });
       } catch (error) {
         session.send("🐱 - 转发失败，请检查目标QQ/群号配置是否正确");
         ctx.logger.error("[转发] 失败:", error);
@@ -460,6 +560,7 @@ function apply(ctx, config) {
         ctx.logger.warn(`[撤回] 转发消息 (ID: ${entry.forwardedMessageId}) 失败:`, e);
       } finally {
         forwardedHistory.splice(entryIndex, 1);
+        ctx.database.remove("cat_raising_forward_history", { originalMessageId }).catch((err) => ctx.logger.warn(`[撤回] 删除数据库记录失败:`, err));
         ctx.logger.info(`[撤回] 已联动撤回与源消息 ${originalMessageId} (直播间: ${entry.roomId}) 相关的转发。`);
       }
     }
@@ -481,9 +582,30 @@ function sleep(ms) {
   return new Promise((resolve) => setTimeout(resolve, ms));
 }
 __name(sleep, "sleep");
+function tcpPing(host, port = 80, timeout = 2e3) {
+  return new Promise((resolve, reject) => {
+    const start = Date.now();
+    const socket = new net.Socket();
+    socket.setTimeout(timeout, () => {
+      socket.destroy();
+      reject(new Error("Timeout"));
+    });
+    socket.connect(port, host, () => {
+      const ms = Date.now() - start;
+      socket.destroy();
+      resolve(ms);
+    });
+    socket.on("error", (err) => {
+      socket.destroy();
+      reject(err);
+    });
+  });
+}
+__name(tcpPing, "tcpPing");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   Config,
   apply,
+  inject,
   name
 });

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-cat-raising",
   "description": "",
-  "version": "1.3.9",
+  "version": "1.4.0",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [