koishi-plugin-cs2-server-query 2.0.0 → 2.2.0

package/lib/index.js

@@ -302,7 +302,7 @@ function apply(ctx, config) {
       const recentCount = (mapStats[group.name]?.[server.map] || []).filter(
         (ts) => Date.now() - ts <= 7 * 24 * 60 * 60 * 1e3
       ).length;
-      const isHot = recentCount >= 20;
+      const isHot = recentCount >= 15;
       return `
       
     <tr>
@@ -419,17 +419,50 @@ function apply(ctx, config) {
     return screenshot;
   }
   __name(generatePushImage, "generatePushImage");
+  async function processMapChange(group, server, newMap) {
+    checkAndAddMapTranslation(newMap, mapTranslations);
+    if (!mapStats[group.name]) mapStats[group.name] = {};
+    if (!mapStats[group.name][newMap]) mapStats[group.name][newMap] = [];
+    mapStats[group.name][newMap].push(Date.now());
+    saveMapStats();
+    const relevantSubs = subscriptions.filter(
+      (sub) => sub.groupName === group.name && sub.mapName === newMap && canPush(server.ip, newMap)
+    );
+    for (const sub of relevantSubs) {
+      const image = await generatePushImage(server, mapTranslations);
+      const targetId = sub.pushType === "私聊" ? sub.userId : sub.channelId;
+      const message = [
+        import_koishi.h.image(image, "image/png"),
+        `connect ${server.ip}`
+      ];
+      await sendMessageToTarget(
+        ctx,
+        targetId,
+        sub.pushType,
+        message,
+        sub.userId
+      );
+      updatePushRecord(server.ip, newMap);
+    }
+  }
+  __name(processMapChange, "processMapChange");
   async function handleOfflineServers(groups2) {
     for (const group of groups2) {
       for (const server of group.servers) {
         if (server.status === "离线") {
           try {
             const info = await (0, import_steam_server_query.queryGameServerInfo)(server.ip);
-            server.name = info.name || `服务器 ${server.ip}`;
-            server.map = info.map;
+            const oldMap = server.map;
+            const newMap = info.map;
+            server.name = info.name || server.name;
+            server.map = newMap;
             server.players = info.players;
             server.maxPlayers = info.maxPlayers;
             server.status = "在线";
+            if (oldMap !== newMap) {
+              server.mapChangeTime = Date.now();
+              await processMapChange(group, server, newMap);
+            }
           } catch (err) {
             server.status = "离线";
           }
@@ -749,14 +782,17 @@ ${mapList}
       }
     }
   }, 6e4);
-  ctx.command("cs2取消订阅 <groupName> <mapInput>", "取消订阅指定社区服务器的地图，需输入以下指定社区名称:EXG 风云社 僵尸乐园 UB社区 ZE国际服 X社区").action(async ({ session }, groupName, mapInput) => {
+  ctx.command("cs2取消订阅 <groupName> <mapInput>", "取消订阅指定社区服务器的地图，需输入以下指定社区名称:EXG 风云社 僵尸乐园 UB社区 ZE国际服 X社区，例：cs2取消订阅 EXG 狮子王 -a").option("all", "-a 取消该地图的所有社区订阅").action(async ({ session, options }, groupName, mapInput) => {
     const userId = session.userId;
-    if (!groupName || !mapInput) return "参数不足，请提供社区名和地图名。";
+    if (options.all) groupName = null;
+    if (!options.all && !groupName) return "参数不足，请提供以下指定社区名称:EXG 风云社 僵尸乐园 UB社区 ZE国际服 X社区，以及地图名，末端可使用 -a 取消所有社区订阅。例：cs2取消订阅 EXG 狮子王 -a";
+    if (!mapInput) return "参数不足，请提供以下指定社区名称:EXG 风云社 僵尸乐园 UB社区 ZE国际服 X社区，以及地图名，末端可使用 -a 取消所有社区订阅。例：cs2取消订阅 EXG 狮子王 -a";
     const matchedMaps = fuzzyMatchMap(mapInput);
     if (matchedMaps.length === 0) return "未找到匹配的地图。";
     if (matchedMaps.length > 1) {
       pendingUnsubscriptions.set(userId, {
-        groupName,
+        groupName: options.all ? "__ALL__" : groupName,
+        // 使用特殊标记表示所有社区
         maps: matchedMaps,
         timestamp: Date.now()
       });
@@ -768,7 +804,7 @@ ${mapList}
 （请在30秒内回复）`;
     }
     const mapName = matchedMaps[0];
-    return handleUnsubscription(userId, groupName, mapName);
+    return handleUnsubscription(userId, options.all ? "__ALL__" : groupName, mapName);
   });
   ctx.middleware(async (session, next) => {
     const pendingUnsub = pendingUnsubscriptions.get(session.userId);
@@ -789,16 +825,18 @@ ${mapList}
     }
     return next();
   });
-  function handleUnsubscription(userId, groupName, mapName) {
+  function handleUnsubscription(userId, groupIdentifier, mapName) {
     const initialLength = subscriptions.length;
-    subscriptions = subscriptions.filter(
-      (sub) => !(sub.userId === userId && sub.groupName === groupName && sub.mapName === mapName)
-    );
+    subscriptions = subscriptions.filter((sub) => {
+      const isTargetMap = sub.mapName === mapName;
+      const isTargetGroup = groupIdentifier === "__ALL__" ? true : sub.groupName === groupIdentifier;
+      return !(sub.userId === userId && isTargetMap && isTargetGroup);
+    });
     if (subscriptions.length === initialLength) {
-      return `未找到社区 "${groupName}" 中关于地图 "${mapName}" 的订阅。`;
+      return groupIdentifier === "__ALL__" ? `未找到关于地图 "${mapName}" 的任何订阅。` : `未找到社区 "${groupIdentifier}" 中关于地图 "${mapName}" 的订阅。`;
     }
     saveSubscriptions();
-    return `已成功取消订阅社区 "${groupName}" 的地图 "${mapName}"。`;
+    return groupIdentifier === "__ALL__" ? `已成功取消所有社区中地图 "${mapName}" 的订阅。` : `已成功取消社区 "${groupIdentifier}" 中地图 "${mapName}" 的订阅。`;
   }
   __name(handleUnsubscription, "handleUnsubscription");
   ctx.setInterval(() => {
@@ -809,18 +847,235 @@ ${mapList}
       }
     }
   }, 6e4);
-  ctx.command("cs2我的订阅", "列出当前用户订阅的地图").action(({ session }) => {
+  async function generateSubscriptionImage(subscriptions2, mapTranslations2) {
+    const browser = await import_puppeteer.default.launch({ args: ["--no-sandbox", "--disable-setuid-sandbox"] });
+    const page = await browser.newPage();
+    const difficultyColors2 = {
+      "未知": "#3C3C3C",
+      "简单": "#33cc00",
+      "普通": "#ffcc00",
+      "困难": "#cc6600",
+      "高难": "#cc6699",
+      "极难": "#cc0099",
+      "史诗": "#ff33cc",
+      "梦魇": "#cc3300",
+      "绝境": "#8b1c62"
+    };
+    const generateItemsHTML = /* @__PURE__ */ __name(() => {
+      return subscriptions2.map((sub, index) => {
+        const translation = mapTranslations2[sub.mapName] || { translation: "未翻译", difficulty: "未知" };
+        const diffColor = difficultyColors2[translation.difficulty] || "#3C3C3C";
+        return `
+        <div class="subscription-item">
+          <div class="item-header">
+            <span class="item-index">#${index + 1}</span>
+            <span class="community-name">${sub.groupName}</span>
+            <span class="push-type ${sub.pushType === "私聊" ? "private" : "group"}">
+              ${sub.pushType}
+            </span>
+          </div>
+          <div class="item-body">
+            <div class="map-name">
+              <span class="en-name">${sub.mapName}</span>
+              <span class="cn-name">${translation.translation}</span>
+            </div>
+            <div class="meta-info">
+              <span class="difficulty-tag" style="background: ${diffColor}">
+                ${translation.difficulty}
+              </span>
+            </div>
+          </div>
+        </div>
+      `;
+      }).join("");
+    }, "generateItemsHTML");
+    const generateStatsHTML = /* @__PURE__ */ __name(() => {
+      const stats = {};
+      subscriptions2.forEach((sub) => {
+        const diff = mapTranslations2[sub.mapName]?.difficulty || "未知";
+        stats[diff] = (stats[diff] || 0) + 1;
+      });
+      return Object.entries(stats).map(([diff, count]) => `
+      <div class="stat-item">
+        <span class="diff-tag" style="background: ${difficultyColors2[diff]}">${diff}</span>
+        <span class="diff-count">${count} 个订阅</span>
+      </div>
+    `).join("");
+    }, "generateStatsHTML");
+    const html = `
+    <!DOCTYPE html>
+    <html>
+    <head>
+      <style>
+        body {
+          margin: 0;
+          padding: 40px;
+          background: url('${config.pushBackgroundImage}') no-repeat center center/cover;
+          font-family: 'Microsoft YaHei', sans-serif;
+        }
+        .container {
+          background: rgba(0, 0, 0, 0.85);
+          border-radius: 15px;
+          padding: 30px;
+          backdrop-filter: blur(8px);
+          box-shadow: 0 0 30px rgba(0,0,0,0.5);
+        }
+        .title {
+          color: #fff;
+          font-size: 28px;
+          margin-bottom: 25px;
+          text-align: center;
+          text-shadow: 0 2px 4px rgba(0,0,0,0.5);
+        }
+        .subscription-list {
+          display: grid;
+          gap: 20px;
+        }
+        .subscription-item {
+          background: rgba(255,255,255,0.1);
+          border-radius: 12px;
+          padding: 20px;
+          transition: transform 0.2s;
+        }
+        .item-header {
+          display: flex;
+          align-items: center;
+          margin-bottom: 15px;
+          border-bottom: 1px solid rgba(255,255,255,0.1);
+          padding-bottom: 12px;
+        }
+        .item-index {
+          font-size: 18px;
+          color: #FFD700;
+          margin-right: 15px;
+        }
+        .community-name {
+          font-size: 20px;
+          color: #00ff9d;
+          flex-grow: 1;
+        }
+        .push-type {
+          padding: 4px 12px;
+          border-radius: 6px;
+          font-size: 14px;
+        }
+        .push-type.private {
+          background: rgba(0, 255, 157, 0.15);
+          color: #00ff9d;
+        }
+        .push-type.group {
+          background: rgba(255, 215, 0, 0.15);
+          color: #FFD700;
+        }
+        .map-name {
+          margin-bottom: 12px;
+        }
+        .en-name {
+          font-size: 18px;
+          color: #fff;
+          display: block;
+          margin-bottom: 5px;
+        }
+        .cn-name {
+          font-size: 16px;
+          color: #aaa;
+        }
+        .meta-info {
+          display: flex;
+          gap: 15px;
+          align-items: center;
+        }
+        .difficulty-tag {
+          padding: 4px 12px;
+          border-radius: 6px;
+          font-size: 14px;
+        }
+        .channel-id {
+          color: #888;
+          font-size: 14px;
+        }
+        .stats-section {
+          margin-top: 30px;
+          padding-top: 25px;
+          border-top: 1px solid rgba(255,255,255,0.1);
+        }
+        .stats-title {
+          color: #fff;
+          font-size: 20px;
+          margin-bottom: 15px;
+        }
+        .stat-item {
+          display: flex;
+          align-items: center;
+          gap: 10px;
+          margin-bottom: 10px;
+        }
+        .diff-tag {
+          padding: 4px 12px;
+          border-radius: 6px;
+          font-size: 14px;
+          min-width: 60px;
+          text-align: center;
+        }
+        .diff-count {
+          color: #ddd;
+          font-size: 14px;
+        }
+      </style>
+    </head>
+    <body>
+      <div class="container">
+        <h1 class="title">我的 CS2 地图订阅（共 ${subscriptions2.length} 个）</h1>
+        
+        <div class="subscription-list">
+          ${generateItemsHTML()}
+        </div>
+
+        <div class="stats-section">
+          <h2 class="stats-title">难度分布统计</h2>
+          ${generateStatsHTML()}
+        </div>
+      </div>
+    </body>
+    </html>
+  `;
+    await page.setContent(html);
+    await page.setViewport({ width: 800, height: 600 + subscriptions2.length * 120 });
+    const screenshot = await page.screenshot({
+      type: "png",
+      fullPage: true,
+      encoding: "binary"
+    });
+    await browser.close();
+    return screenshot;
+  }
+  __name(generateSubscriptionImage, "generateSubscriptionImage");
+  ctx.command("cs2我的订阅", "显示当前用户的订阅列表（图片版）").action(async ({ session }) => {
     const userId = session.userId;
     const userSubscriptions = subscriptions.filter((sub) => sub.userId === userId);
     if (userSubscriptions.length === 0) {
-      return "您当前没有订阅任何地图。";
+      return "🎮 您当前没有活跃的地图订阅。";
+    }
+    try {
+      const imageBuffer = await generateSubscriptionImage(userSubscriptions, mapTranslations);
+      return [
+        import_koishi.h.image(imageBuffer, "image/png"),
+        '\n💡 提示：使用 "cs2取消订阅 社区名 地图名" 可取消订阅'
+      ];
+    } catch (err) {
+      ctx.logger("cs2-server-query").error("生成订阅图片失败:", err);
+      return "生成订阅列表失败，请稍后重试。";
     }
-    const subscriptionList = userSubscriptions.map((sub) => {
-      return `社区: ${sub.groupName}, 地图: ${sub.mapName}, 推送类型: ${sub.pushType}`;
-    }).join("\n");
-    return `您当前订阅的地图列表：
-${subscriptionList}`;
   });
+  function calculateDifficultyStats(subs) {
+    const stats = {};
+    subs.forEach((sub) => {
+      const difficulty = mapTranslations[sub.mapName]?.difficulty || "未标记";
+      stats[difficulty] = (stats[difficulty] || 0) + 1;
+    });
+    return "🎚️ 难度分布：\n" + Object.entries(stats).map(([diff, count]) => `▫️ ${diff.padEnd(6)}：${count} 个`).join("\n");
+  }
+  __name(calculateDifficultyStats, "calculateDifficultyStats");
   async function sendMessageToTarget(ctx2, targetId, pushType, message, userId) {
     try {
       if (pushType === "群聊" && userId) {
@@ -853,42 +1108,15 @@ ${subscriptionList}`;
           try {
             const info = await (0, import_steam_server_query.queryGameServerInfo)(server.ip);
             const newMap = info.map;
+            const oldMap = server.map;
             if (server.map !== newMap) {
               server.mapChangeTime = Date.now();
               server.map = newMap;
-              checkAndAddMapTranslation(newMap, mapTranslations);
-              const relevantSubscriptions = subscriptions.filter((sub) => sub.groupName === group.name);
-              for (const sub of relevantSubscriptions) {
-                const mapNameToCompare = sub.mapName;
-                const translatedMapName = mapTranslations[mapNameToCompare]?.translation;
-                let matchPercentage = 0;
-                if (server.map !== newMap) {
-                  server.mapChangeTime = Date.now();
-                  server.map = newMap;
-                  if (!mapStats[group.name]) {
-                    mapStats[group.name] = {};
-                  }
-                  if (!mapStats[group.name][newMap]) {
-                    mapStats[group.name][newMap] = [];
-                  }
-                  mapStats[group.name][newMap].push(Date.now());
-                  saveMapStats();
-                }
-                if (newMap === mapNameToCompare && canPush(server.ip, mapNameToCompare)) {
-                  const imageBuffer = await generatePushImage(server, mapTranslations);
-                  const message = `connect ${server.ip}`;
-                  const targetId = sub.pushType === "私聊" ? sub.userId : sub.channelId;
-                  const combinedMessage = [
-                    import_koishi.h.image(imageBuffer, "image/png"),
-                    // 图片
-                    message
-                    // 文字
-                  ];
-                  await sendMessageToTarget(ctx, targetId, sub.pushType, combinedMessage, sub.userId);
-                  updatePushRecord(server.ip, mapNameToCompare);
-                }
-              }
+              await processMapChange(group, server, newMap);
             }
+            server.status = "在线";
+            server.players = info.players;
+            server.maxPlayers = info.maxPlayers;
           } catch (err) {
             server.status = "离线";
           }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-cs2-server-query",
   "description": "自用,不推荐下载",
-  "version": "2.0.0",
+  "version": "2.2.0",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [