koishi-plugin-chat-analyse 1.3.6 → 1.3.7

package/lib/index.js

@@ -1697,7 +1697,7 @@ var Renderer = class {
     const selectedPalette = colorfulPalettes[Math.floor(Math.random() * colorfulPalettes.length)];
     const shuffledColors = [...selectedPalette].sort(() => 0.5 - Math.random());
     const seriesColors = series.map((_, index) => shuffledColors[index % shuffledColors.length]);
-    const width = 600, height = 300;
+    const width = 600, height = 320;
     const padding = { top: 15, right: 15, bottom: 60, left: 25 };
     const chartWidth = width - padding.left - padding.right;
     const chartHeight = height - padding.top - padding.bottom;
@@ -2318,64 +2318,66 @@ var Analyse = class {
       });
     }
     if (this.config.enableSimilarActivity) {
-      cmd.subcommand("simiactive", "相似活跃分析").usage("分析你和群友的活跃度，找出谁和你的活跃度最相似。").option("hours", "-n <hours:number> 指定时长", { fallback: 24 }).option("separate", "-p 分时分析").action(async ({ session, options }) => {
+      cmd.subcommand("simiactive", "相似活跃分析").usage("分析你和群友的活跃规律，找出谁和你的作息最相似。").option("hours", "-n <hours:number> 指定时长", { fallback: 24 }).option("separate", "-p 分时分析").action(async ({ session, options }) => {
         if (!session.guildId) return "请在群组中使用此命令";
         try {
-          const until = /* @__PURE__ */ new Date();
-          let since;
-          let points;
-          let title;
-          let labels;
-          let daysToAnalyze = 0;
-          if (options.separate) {
-            points = options.hours;
-            since = new Date(until.getTime() - options.hours * import_koishi6.Time.hour);
-            title = `${options.hours}小时相似活跃分析`;
-            labels = Array.from({ length: points }, (_, i) => String(new Date(until.getTime() - (points - 1 - i) * import_koishi6.Time.hour).getHours()));
-          } else {
-            daysToAnalyze = Math.floor(options.hours / 24);
-            if (daysToAnalyze < 1) return "请指定至少 24 小时时长";
-            const analysisDurationHours = daysToAnalyze * 24;
-            since = new Date(until.getTime() - analysisDurationHours * import_koishi6.Time.hour);
-            points = 24;
-            title = `${daysToAnalyze}天相似活跃分析`;
-            labels = Array.from({ length: 24 }, (_, i) => String(i));
-          }
           const guildUsers = await this.ctx.database.get("analyse_user", { channelId: session.guildId });
           if (guildUsers.length < 2) return "暂无用户数据";
           const selfUser = guildUsers.find((u) => u.userId === session.userId);
           const guildUserUids = guildUsers.map((u) => u.uid);
           const uidToNameMap = new Map(guildUsers.map((u) => [u.uid, u.userName]));
-          const records = await this.ctx.database.get("analyse_rank", { uid: { $in: guildUserUids }, timestamp: { $gte: since } });
-          if (records.length === 0) return "暂无统计数据";
-          const activityVectors = /* @__PURE__ */ new Map();
-          guildUserUids.forEach((uid) => activityVectors.set(uid, Array(points).fill(0)));
-          records.forEach((stat) => {
-            let index;
-            if (options.separate) {
-              const diff = until.getTime() - stat.timestamp.getTime();
-              index = points - 1 - Math.floor(diff / import_koishi6.Time.hour);
-            } else {
-              index = stat.timestamp.getHours();
-            }
-            if (index >= 0 && index < points) {
-              activityVectors.get(stat.uid)[index] += stat.count;
-            }
-          });
-          if (!options.separate && daysToAnalyze > 0) activityVectors.forEach((vector) => {
-            for (let i = 0; i < vector.length; i++) vector[i] = vector[i] / daysToAnalyze;
-          });
+          const until = /* @__PURE__ */ new Date();
+          let analysisConfig;
+          if (options.separate) {
+            const { hours } = options;
+            analysisConfig = {
+              points: hours,
+              since: new Date(until.getTime() - hours * import_koishi6.Time.hour),
+              title: `${hours}小时相似活跃分析`,
+              labels: Array.from({ length: hours }, (_, i) => String(new Date(until.getTime() - (hours - 1 - i) * import_koishi6.Time.hour).getHours())),
+              getIndex: /* @__PURE__ */ __name((timestamp) => {
+                const diff = until.getTime() - timestamp.getTime();
+                const index = hours - 1 - Math.floor(diff / import_koishi6.Time.hour);
+                return index >= 0 && index < hours ? index : -1;
+              }, "getIndex"),
+              reorderVector: /* @__PURE__ */ __name((vec) => vec, "reorderVector")
+            };
+          } else {
+            const daysToAnalyse = Math.floor(options.hours / 24);
+            if (daysToAnalyse < 1) return "分析时长请指定至少 1 天";
+            const hoursToAnalyse = daysToAnalyse * 24;
+            const currentHour = until.getHours();
+            const labels = Array.from({ length: 24 }, (_, i) => String((currentHour - (23 - i) + 24) % 24));
+            analysisConfig = {
+              points: 24,
+              since: new Date(until.getTime() - hoursToAnalyse * import_koishi6.Time.hour),
+              title: `${daysToAnalyse}天相似活跃分析`,
+              labels,
+              getIndex: /* @__PURE__ */ __name((timestamp) => timestamp.getHours(), "getIndex"),
+              reorderVector: /* @__PURE__ */ __name((vector) => labels.map((label) => vector[parseInt(label)]), "reorderVector")
+            };
+          }
+          const records = await this.ctx.database.get("analyse_rank", { uid: { $in: guildUserUids }, timestamp: { $gte: analysisConfig.since } });
+          if (!records.length) return "暂无统计数据";
+          const activityVectors = new Map(guildUserUids.map((uid) => [uid, Array(analysisConfig.points).fill(0)]));
+          for (const record of records) {
+            const index = analysisConfig.getIndex(record.timestamp);
+            if (index !== -1) activityVectors.get(record.uid)[index] += record.count;
+          }
           const selfVector = activityVectors.get(selfUser.uid);
-          if (!selfVector || selfVector.every((v) => v === 0)) return "暂无统计数据";
-          const similarities = guildUserUids.filter((uid) => uid !== selfUser.uid && !activityVectors.get(uid).every((v) => v === 0)).map((uid) => ({ uid, score: cosineSimilarity(selfVector, activityVectors.get(uid)) })).sort((a, b) => b.score - a.score);
-          if (similarities.length === 0) return "暂无相似用户";
+          const similarities = guildUserUids.filter((uid) => uid !== selfUser.uid && activityVectors.get(uid).some((v) => v !== 0)).map((uid) => ({
+            uid,
+            score: cosineSimilarity(selfVector, activityVectors.get(uid))
+          })).sort((a, b) => b.score - a.score);
+          if (!similarities.length) return "暂无相似用户";
           const top5 = similarities.slice(0, 5);
-          const series = [{ name: uidToNameMap.get(selfUser.uid) || "您", data: selfVector }];
-          top5.forEach((sim) => {
+          const series = [{ name: uidToNameMap.get(selfUser.uid) || "您", data: analysisConfig.reorderVector(selfVector) }];
+          for (const sim of top5) {
             const name2 = uidToNameMap.get(sim.uid) || `UID ${sim.uid}`;
-            series.push({ name: `${name2} (${(sim.score * 100).toFixed(1)}%)`, data: activityVectors.get(sim.uid) });
-          });
-          const imageGenerator = this.renderer.renderLineChart({ title, time: /* @__PURE__ */ new Date(), series, labels });
+            const data = analysisConfig.reorderVector(activityVectors.get(sim.uid));
+            series.push({ name: `${name2} (${(sim.score * 100).toFixed(1)}%)`, data });
+          }
+          const imageGenerator = this.renderer.renderLineChart({ title: analysisConfig.title, time: /* @__PURE__ */ new Date(), series, labels: analysisConfig.labels });
           for await (const buffer of imageGenerator) await session.send(import_koishi6.h.image(buffer, "image/png"));
         } catch (error) {
           this.ctx.logger.error("生成作息分析图片失败:", error);

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chat-analyse",
   "description": "强大而全面的聊天数据分析，支持统计命令，发言，消息类型，活跃度，支持发言排行和生成词云",
-  "version": "1.3.6",
+  "version": "1.3.7",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],