koishi-plugin-cs2-server-query 1.0.0

package/lib/index.d.ts

@@ -0,0 +1,10 @@
+import { Context, Schema } from 'koishi';
+export declare const name = "cs2-server-query";
+export interface Config {
+    mapTranslationFile: string;
+    backgroundImage: string;
+    dataFile: string;
+    commandMappingFile: string;
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -0,0 +1,347 @@
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __export = (target, all) => {
+  for (var name2 in all)
+    __defProp(target, name2, { get: all[name2], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var src_exports = {};
+__export(src_exports, {
+  Config: () => Config,
+  apply: () => apply,
+  name: () => name
+});
+module.exports = __toCommonJS(src_exports);
+var import_koishi = require("koishi");
+var import_steam_server_query = require("steam-server-query");
+var import_fs = __toESM(require("fs"));
+var import_path = __toESM(require("path"));
+var import_puppeteer = __toESM(require("puppeteer"));
+var name = "cs2-server-query";
+var Config = import_koishi.Schema.object({
+  mapTranslationFile: import_koishi.Schema.string().default("map-translation.json"),
+  backgroundImage: import_koishi.Schema.string().description("背景图片URL (建议尺寸 800x600)").default("image/background.jpg"),
+  dataFile: import_koishi.Schema.string().default("server-data.json"),
+  commandMappingFile: import_koishi.Schema.string().default("command-mapping.json")
+});
+function apply(ctx, config) {
+  const dataFilePath = import_path.default.resolve(__dirname, config.dataFile);
+  const commandMappingFilePath = import_path.default.resolve(__dirname, config.commandMappingFile);
+  let groups = [];
+  let mapTranslations = {};
+  let commandMappings = {};
+  const mapTranslationFilePath = import_path.default.resolve(__dirname, config.mapTranslationFile);
+  if (import_fs.default.existsSync(mapTranslationFilePath)) {
+    mapTranslations = JSON.parse(import_fs.default.readFileSync(mapTranslationFilePath, "utf-8"));
+  }
+  if (import_fs.default.existsSync(dataFilePath)) {
+    groups = JSON.parse(import_fs.default.readFileSync(dataFilePath, "utf-8"));
+  }
+  if (import_fs.default.existsSync(commandMappingFilePath)) {
+    commandMappings = JSON.parse(import_fs.default.readFileSync(commandMappingFilePath, "utf-8"));
+  }
+  function saveMapTranslations() {
+    import_fs.default.writeFileSync(mapTranslationFilePath, JSON.stringify(mapTranslations, null, 2));
+  }
+  __name(saveMapTranslations, "saveMapTranslations");
+  function saveData() {
+    import_fs.default.writeFileSync(dataFilePath, JSON.stringify(groups, null, 2));
+  }
+  __name(saveData, "saveData");
+  function saveCommandMappings() {
+    import_fs.default.writeFileSync(commandMappingFilePath, JSON.stringify(commandMappings, null, 2));
+  }
+  __name(saveCommandMappings, "saveCommandMappings");
+  const difficultyColors = {
+    "简单": "#00ff00",
+    "普通": "#7fff00",
+    "困难": "#ff7f00",
+    "高难": "#ff4f00",
+    "史诗": "#ff2f00",
+    "梦魇": "#ff0f00",
+    "绝境": "#ff0000"
+  };
+  async function generateServerImage(group, mapTranslations2) {
+    const browser = await import_puppeteer.default.launch({ args: ["--no-sandbox", "--disable-setuid-sandbox"] });
+    const page = await browser.newPage();
+    const totalPlayers = group.servers.reduce((sum, server) => sum + (server.status === "在线" ? server.players : 0), 0);
+    const totalMaxPlayers = group.servers.reduce((sum, server) => sum + (server.status === "在线" ? server.maxPlayers : 0), 0);
+    const baseHeight = 150;
+    const rowHeight = 50;
+    const totalHeight = baseHeight + rowHeight * group.servers.length;
+    const html = `
+      <!DOCTYPE html>
+      <html>
+      <head>
+        <style>
+          body { 
+            margin: 0;
+            padding: 20px;
+            background: url('${config.backgroundImage}') no-repeat center center fixed;
+            background-size: cover;
+            height: 100vh;
+            position: relative;
+          }
+          .overlay {
+            background: rgba(0, 0, 0, 0.7);
+            border-radius: 10px;
+            padding: 20px;
+            color: white;
+            font-family: Arial, sans-serif;
+          }
+          h2 {
+            color: #fff;
+            margin: 0 0 15px 0;
+            border-bottom: 2px solid #fff;
+            padding-bottom: 10px;
+          }
+          table {
+            width: 100%;
+            border-collapse: collapse;
+            background: rgba(255, 255, 255, 0.1);
+          }
+          th, td {
+            padding: 12px;
+            text-align: left;
+            border-bottom: 1px solid rgba(255, 255, 255, 0.2);
+          }
+          th {
+            background: rgba(0, 0, 0, 0.5);
+          }
+          tr:hover {
+            background: rgba(255, 255, 255, 0.05);
+          }
+          .map-name {
+            color: #00ff9d;
+          }
+          .status-online {
+            color: #00ff00;
+          }
+          .status-offline {
+            color: #ff0000;
+          }
+          .difficulty {
+            display: inline-block;
+            padding: 2px 6px;
+            border-radius: 4px;
+            color: white;
+            font-size: 12px;
+            margin-left: 5px;
+          }
+        </style>
+      </head>
+      <body>
+        <div class="overlay">
+          <h2>${group.name} - 服务器列表</h2>
+          <p>总玩家数: ${totalPlayers}/${totalMaxPlayers}</p>
+          <table>
+            <tr>
+              <th>编号</th>
+              <th>服务器名称</th>
+              <th>地图</th>
+              <th>玩家</th>
+              <th>延迟</th>
+              <th>状态</th>
+              <th>IP 地址</th>
+            </tr>
+            ${group.servers.map((server, index) => {
+      const translation = mapTranslations2[server.map];
+      const difficulty = translation?.difficulty;
+      const difficultyColor = difficulty ? difficultyColors[difficulty] : "#000000";
+      return `
+                <tr>
+                  <td>${index + 1}</td>
+                  <td>${server.name}</td>
+                  <td>
+                    <span class="map-name">${server.map}</span>
+                    ${translation ? `<br>(${translation.translation})` : ""}
+                    ${difficulty ? `<span class="difficulty" style="background-color: ${difficultyColor};">${difficulty}</span>` : ""}
+                  </td>
+                  <td>${server.status === "在线" ? `${server.players}/${server.maxPlayers}` : "-"}</td>
+                  <td>${server.status === "在线" ? `${server.latency}ms` : "-"}</td>
+                  <td class="status-${server.status === "在线" ? "online" : "offline"}">${server.status}</td>
+                  <td><code>${server.ip}</code></td>
+                </tr>
+              `;
+    }).join("")}
+          </table>
+        </div>
+      </body>
+      </html>
+    `;
+    await page.setContent(html);
+    await page.setViewport({ width: 1280, height: totalHeight });
+    const screenshot = await page.screenshot({
+      type: "png",
+      fullPage: true,
+      encoding: "binary"
+    });
+    await browser.close();
+    return screenshot;
+  }
+  __name(generateServerImage, "generateServerImage");
+  ctx.middleware(async (session, next) => {
+    const content = session.content.trim();
+    const commandMatch = content.match(/^(.+?)(\s+(\d+))?$/);
+    if (!commandMatch) return next();
+    const command = commandMatch[1];
+    const serverIndex = commandMatch[3] ? parseInt(commandMatch[3]) : null;
+    const groupName = commandMappings[command];
+    if (!groupName) return next();
+    const group = groups.find((g) => g.name === groupName);
+    if (!group) return "未找到该组";
+    for (const server of group.servers) {
+      try {
+        const info = await (0, import_steam_server_query.queryGameServerInfo)(server.ip);
+        server.name = info.name || server.name;
+        server.map = info.map;
+        server.players = info.players;
+        server.maxPlayers = info.maxPlayers;
+        server.status = "在线";
+      } catch (err) {
+        server.status = "离线";
+      }
+    }
+    saveData();
+    if (serverIndex !== null) {
+      if (isNaN(serverIndex) || serverIndex < 1 || serverIndex > group.servers.length) {
+        return "无效的服务器编号";
+      }
+      const server = group.servers[serverIndex - 1];
+      return `connect ${server.ip}`;
+    } else {
+      try {
+        const imageBuffer = await generateServerImage(group, mapTranslations);
+        return import_koishi.h.image(imageBuffer, "image/png");
+      } catch (err) {
+        ctx.logger("cs2-server-query").error(err);
+        return "生成服务器列表图片失败，请检查后台日志";
+      }
+    }
+  });
+  ctx.command("cs2组创建 <name>", "添加一个新组").action((_, name2) => {
+    if (!name2) return "请提供组名。";
+    if (groups.some((group) => group.name === name2)) return "组已存在。";
+    groups.push({ name: name2, servers: [] });
+    saveData();
+    return `组 "${name2}" 添加成功。`;
+  });
+  ctx.command("cs2组列表", "列出所有组").action(() => {
+    if (groups.length === 0) return "未找到任何组。";
+    return `组列表：${groups.map((group) => group.name).join(", ")}`;
+  });
+  ctx.command("cs2添加 <groupName> <ip>", "向组中添加一个服务器").action(async (_, groupName, ip) => {
+    const group = groups.find((group2) => group2.name === groupName);
+    if (!group) return "未找到该组。";
+    if (group.servers.some((server) => server.ip === ip)) return "该服务器已存在于该组中。";
+    try {
+      const info = await (0, import_steam_server_query.queryGameServerInfo)(ip);
+      const serverName = info.name || `服务器 ${ip}`;
+      group.servers.push({
+        name: serverName,
+        ip,
+        map: info.map,
+        players: info.players,
+        maxPlayers: info.maxPlayers,
+        latency: 0,
+        status: "在线"
+        // 默认状态为在线
+      });
+      saveData();
+      return `服务器 "${serverName}" (${ip}) 已成功添加到组 "${groupName}" 中。`;
+    } catch (err) {
+      const serverName = `服务器 ${ip}`;
+      group.servers.push({
+        name: serverName,
+        ip,
+        map: "未知",
+        players: 0,
+        maxPlayers: 0,
+        latency: 0,
+        status: "离线"
+      });
+      saveData();
+      return `服务器 "${serverName}" (${ip}) 已添加到组 "${groupName}" 中，但当前状态为离线。`;
+    }
+  });
+  ctx.command("cs2组删除 <groupName>", "删除一个组").action((_, groupName) => {
+    const index = groups.findIndex((group) => group.name === groupName);
+    if (index === -1) return "未找到该组。";
+    groups.splice(index, 1);
+    saveData();
+    return `组 "${groupName}" 删除成功。`;
+  });
+  ctx.command("cs2删除 <groupName> <serverIndex>", "从组中删除一个服务器").action((_, groupName, serverIndex) => {
+    const group = groups.find((group2) => group2.name === groupName);
+    if (!group) return "未找到该组。";
+    const index = parseInt(serverIndex) - 1;
+    if (isNaN(index) || index < 0 || index >= group.servers.length) return "无效的服务器编号。";
+    group.servers.splice(index, 1);
+    saveData();
+    return `服务器 ${serverIndex} 已从组 "${groupName}" 中删除。`;
+  });
+  ctx.command("cs2地图翻译 <mapName> <translation> [difficulty]", "添加地图翻译").action((_, mapName, translation, difficulty) => {
+    mapTranslations[mapName] = { translation, difficulty };
+    saveMapTranslations();
+    return `地图 "${mapName}" 的翻译已成功添加。`;
+  });
+  ctx.command("cs2地图翻译删除 <mapName>", "删除地图翻译").action((_, mapName) => {
+    if (!mapTranslations[mapName]) return "未找到该地图的翻译。";
+    delete mapTranslations[mapName];
+    saveMapTranslations();
+    return `地图 "${mapName}" 的翻译已删除。`;
+  });
+  ctx.command("cs2更新地图翻译 <mapName> <translation> [difficulty]", "更新地图翻译").action((_, mapName, translation, difficulty) => {
+    if (!mapTranslations[mapName]) return "未找到该地图的翻译。";
+    mapTranslations[mapName] = { translation, difficulty };
+    saveMapTranslations();
+    return `地图 "${mapName}" 的翻译已更新。`;
+  });
+  ctx.command("cs2指令添加 <command> <groupName>", "添加一个指令映射").action((_, command, groupName) => {
+    if (commandMappings[command]) return `指令 "${command}" 已存在。`;
+    if (!groups.some((group) => group.name === groupName)) return `组 "${groupName}" 不存在。`;
+    commandMappings[command] = groupName;
+    saveCommandMappings();
+    return `指令 "${command}" 已成功映射到组 "${groupName}"。`;
+  });
+  ctx.command("cs2指令删除 <command>", "删除一个指令映射").action((_, command) => {
+    if (!commandMappings[command]) return `指令 "${command}" 不存在。`;
+    delete commandMappings[command];
+    saveCommandMappings();
+    return `指令 "${command}" 已删除。`;
+  });
+  ctx.command("cs2指令列表", "列出所有指令映射").action(() => {
+    if (Object.keys(commandMappings).length === 0) return "未找到任何指令映射。";
+    return `指令映射列表：${Object.entries(commandMappings).map(([command, groupName]) => `${command} -> ${groupName}`).join(", ")}`;
+  });
+}
+__name(apply, "apply");
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  Config,
+  apply,
+  name
+});

package/package.json

@@ -0,0 +1,31 @@
+{
+  "name": "koishi-plugin-cs2-server-query",
+  "description": "自用,不推荐下载",
+  "version": "1.0.0",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "puppeteer",
+    "steam-server-query",
+    "dist"
+  ],
+  
+  "license": "MIT",
+  "keywords": [
+    "chatbot",
+    "koishi",
+    "plugin"
+  ],
+  "peerDependencies": {
+    "koishi": "^4.18.7"
+  },
+  "dependencies": {
+    "steam-server-query": "^1.1.3",
+    "puppeteer": "^24.2.0"
+  },
+  "devDependencies": {
+    "esbuild": "^0.19.0",
+    "typescript": "^5.0.0"
+  }
+}

package/readme.md

@@ -0,0 +1,5 @@
+# koishi-plugin-cs2-server-query
+
+[![npm](https://img.shields.io/npm/v/koishi-plugin-cs2-server-query?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-cs2-server-query)
+
+自用CS2社区服查询插件