koishi-plugin-dwrgbwiki 1.1.0 → 1.3.0

package/index.js

@@ -5,13 +5,31 @@ const lib = require('./lib')
 
 const HELP = lib.buildHelp()
 
+const MAIN_HELP = [
+  '第五人格BWIKI数据查询机器人',
+  '',
+  '支持指令：',
+  '  胜率 <角色名> [赛季] [周数] [排序要素] [段位]',
+  '    查询角色胜率，例：胜率 小女孩',
+  '  地图 <地图名> [赛季] [周数] [段位] [阵营]',
+  '    查询地图胜率，例：地图 军工厂',
+  '  模仿者 <角色名> [日期] [模式]',
+  '    查询模仿者游戏数据，例：模仿者 侦探',
+  '  勋章 <角色名> [日期] [模式]',
+  '    查询角色勋章分数线，例：勋章 小女孩',
+  '',
+  '发送“胜率”等单独指令可查看对应完整帮助',
+  '数据均根据第五人格官方小程序数据计算',
+].join('\n')
+
 module.exports = {
   name: 'dwrgbwiki',
   description: '第五人格BWIKI编辑组测试用组件，纯AI生成喵',
+  inject: ['puppeteer'],
 
   Config: Schema.object({
-    dataSource: Schema.string()
-      .description('数据源地址，可在两个镜像站点之间切换：' + lib.KNOWN_SOURCES.join(' / '))
+    dataSource: Schema.union(lib.KNOWN_SOURCES)
+      .description('数据源地址，可在两个镜像站点之间切换')
       .default('https://website.homeworkkun.top'),
     tryAlternate: Schema.boolean()
       .description('主数据源失败时自动尝试备用数据源')
@@ -29,25 +47,76 @@ module.exports = {
       .description('勋章日增长率天数')
       .default(7),
     width: Schema.number().min(400).max(4000)
-      .description('图片宽度（px）')
-      .default(1000),
+      .description('图片宽度（px），其他指令未单独设置时使用该值')
+      .default(1300),
+    mapWidth: Schema.number().min(0).max(4000)
+      .description('地图指令图片宽度（px），0=使用默认宽度')
+      .default(0),
+    mimicWidth: Schema.number().min(0).max(4000)
+      .description('模仿者指令图片宽度（px），0=使用默认宽度')
+      .default(0),
+    badgeWidth: Schema.number().min(0).max(4000)
+      .description('勋章指令图片宽度（px），0=使用默认宽度')
+      .default(0),
     scale: Schema.number().min(1).max(4)
       .description('截图倍率（越大越清晰）')
       .default(2),
     timeout: Schema.number().min(3000).max(60000)
       .description('网络请求超时（毫秒）')
       .default(15000),
+    heartbeatEnable: Schema.boolean()
+      .description('启用心跳包定时发送功能')
+      .default(false),
+    heartbeatServer: Schema.string()
+      .description('心跳服务端地址（如 http://192.168.1.10:8080）')
+      .default('http://localhost:8080'),
+    heartbeatUid: Schema.string()
+      .description('心跳 uid（可在指令中修改）')
+      .default(''),
+    heartbeatInterval: Schema.number().min(5).max(86400)
+      .description('心跳发送间隔（秒）')
+      .default(60),
   }),
 
   apply(ctx, config) {
+    ctx.command('BWIKI机器人', '显示当前支持的指令列表')
+      .action(async ({ session }) => {
+        return helpImage(ctx, config, MAIN_HELP)
+      })
+
+    const heartbeat = {
+      uid: config.heartbeatUid || '',
+      timer: null,
+    }
+    const sendHeartbeat = async () => {
+      if (!heartbeat.uid || !config.heartbeatServer) return
+      const url = config.heartbeatServer.replace(/\/+$/, '') + '/api/heartbeat?uid=' + encodeURIComponent(heartbeat.uid)
+      const ctrl = new AbortController()
+      const timer = setTimeout(() => ctrl.abort(), 10000)
+      try {
+        await fetch(url, { signal: ctrl.signal })
+      } catch (e) {
+        ctx.logger('dwrgbwiki').warn('[heartbeat] 发送失败: ' + (e && e.message ? e.message : e))
+      } finally {
+        clearTimeout(timer)
+      }
+    }
+    const startHeartbeat = () => {
+      if (heartbeat.timer) return
+      if (!config.heartbeatInterval || config.heartbeatInterval < 5) config.heartbeatInterval = 60
+      heartbeat.timer = ctx.setInterval(() => sendHeartbeat(), config.heartbeatInterval * 1000)
+      sendHeartbeat()
+    }
+    if (config.heartbeatEnable) startHeartbeat()
+
     ctx.command('胜率 <角色名> [条件:text]', '查询角色胜率并生成图片')
       .usage(HELP)
       .example('胜率 小女孩')
       .example('胜率 小女孩 第43赛季 第3周 使用率')
       .example('胜率 小女孩 43 3 胜率')
       .action(async ({ session }, name, cond) => {
-        if (!name || !name.trim()) return HELP
-        return run(ctx, config, lib.collect, lib.buildHtml, name.trim(), cond)
+        if (!name || !name.trim()) return helpImage(ctx, config, HELP)
+        return run(ctx, config, lib.collect, lib.buildHtml, name.trim(), cond, 'width')
       })
 
     ctx.command('地图 <地图名> [条件:text]', '查询地图胜率并生成图片')
@@ -56,8 +125,8 @@ module.exports = {
       .example('地图 红教堂 S44 3 求生者')
       .example('地图 月亮河公园 全赛季')
       .action(async ({ session }, name, cond) => {
-        if (!name || !name.trim()) return MAP_HELP
-        return run(ctx, config, lib.collectMap, lib.buildMapHtml, name.trim(), cond)
+        if (!name || !name.trim()) return helpImage(ctx, config, MAP_HELP)
+        return run(ctx, config, lib.collectMap, lib.buildMapHtml, name.trim(), cond, 'mapWidth')
       })
 
     ctx.command('模仿者 <角色名> [条件:text]', '查询模仿者游戏角色数据并生成图片')
@@ -65,8 +134,8 @@ module.exports = {
       .example('模仿者 侦探')
       .example('模仿者 药剂师 260531 低级场')
       .action(async ({ session }, name, cond) => {
-        if (!name || !name.trim()) return MIMIC_HELP
-        return run(ctx, config, lib.collectMimic, lib.buildMimicHtml, name.trim(), cond)
+        if (!name || !name.trim()) return helpImage(ctx, config, MIMIC_HELP)
+        return run(ctx, config, lib.collectMimic, lib.buildMimicHtml, name.trim(), cond, 'mimicWidth')
       })
 
     ctx.command('勋章 <角色名> [条件:text]', '查询角色勋章分数线并生成图片')
@@ -75,25 +144,50 @@ module.exports = {
       .example('勋章 小女孩 20250917 7日增长率')
       .example('勋章 小女孩 赛季同比增长率')
       .action(async ({ session }, name, cond) => {
-        if (!name || !name.trim()) return BADGE_HELP
-        return run(ctx, config, lib.collectBadge, lib.buildBadgeHtml, name.trim(), cond)
+        if (!name || !name.trim()) return helpImage(ctx, config, BADGE_HELP)
+        return run(ctx, config, lib.collectBadge, lib.buildBadgeHtml, name.trim(), cond, 'badgeWidth')
       })
   },
 }
 
-async function run(ctx, config, collectFn, buildFn, name, cond) {
+async function run(ctx, config, collectFn, buildFn, name, cond, widthKey) {
   if (!ctx.puppeteer) {
     return '未检测到 puppeteer 服务，请先安装并启用 koishi-plugin-puppeteer'
   }
   try {
     const result = await collectFn(config, name, cond)
-    if (!result.ok) return result.error
-    const html = buildFn(result, { ...config, avatarBase: result.base })
+    if (!result.ok) return errorImage(ctx, config, result.error)
+    const width = (config[widthKey] && config[widthKey] > 0) ? config[widthKey] : config.width
+    const html = buildFn(result, { ...config, avatarBase: result.base, width })
+    const image = await capture(ctx, html, { ...config, width })
+    return [h.image(image, 'image/png')]
+  } catch (e) {
+    ctx.logger('dwrgbwiki').warn(e)
+    return errorImage(ctx, config, '查询失败：' + (e && e.message ? e.message : e))
+  }
+}
+
+async function helpImage(ctx, config, text) {
+  if (!ctx.puppeteer) return text
+  try {
+    const html = lib.buildHelpHtml(text, { ...config, width: config.width })
+    const image = await capture(ctx, html, config)
+    return [h.image(image, 'image/png')]
+  } catch (e) {
+    ctx.logger('dwrgbwiki').warn(e)
+    return text
+  }
+}
+
+async function errorImage(ctx, config, text) {
+  if (!ctx.puppeteer || !text) return text
+  try {
+    const html = lib.buildHelpHtml(text, { ...config, width: config.width })
     const image = await capture(ctx, html, config)
-    return [h.image(image)]
+    return [h.image(image, 'image/png')]
   } catch (e) {
     ctx.logger('dwrgbwiki').warn(e)
-    return '查询失败：' + (e && e.message ? e.message : e)
+    return text
   }
 }
 

package/lib.js

@@ -940,7 +940,8 @@ function buildHtml(data, config) {
   const zoneLeft = '<div class="stat-block">' + sixRows + '</div>' + changeBlock +
     '<div class="zone-sep"></div>' +
     '<div class="stat-block"><div class="comp-title" style="color:var(--accent);margin-bottom:4px">各段位去平胜率（距平均）</div>' +
-    '<div class="allrank-grid">' + partsItems + '</div></div>'
+    '<div class="allrank-grid">' + partsItems + '</div></div>' +
+    '<div class="credit">@夜莺小姐档案馆</div>'
 
   let zoneRight
   if (badge) {
@@ -1036,6 +1037,7 @@ body { background:var(--bg); color:var(--text); font-family:-apple-system,BlinkM
 .rk-name { white-space:nowrap; overflow:hidden; text-overflow:ellipsis; }
 .rk-val { text-align:right; font-variant-numeric:tabular-nums; font-weight:600; font-size:16px; }
 .rank-header .rk-pos, .rank-header .rk-val { color:var(--text-secondary); font-weight:600; }
+.credit { text-align:center; color:var(--text-secondary); font-size:12px; margin-top:14px; opacity:.7; }
 .stat-block { display:flex; flex-direction:column; gap:6px; }
 .stat-block + .stat-block { margin-top:12px; }
 .s-row { display:flex; align-items:center; justify-content:space-between; padding:8px 12px; background:rgba(255,255,255,.02); border:1px solid rgba(255,255,255,.04); border-radius:6px; gap:10px; }
@@ -1337,6 +1339,84 @@ function buildHelp() {
   ].join('\n')
 }
 
+function buildHelpHtml(text, config) {
+  const width = config.width || 1300
+  const cardW = Math.min(width, 880)
+
+  let hero = '', body = '', titleTaken = false
+  String(text).split('\n').forEach(line => {
+    const raw = String(line).replace(/\r$/, '')
+    const trimmed = raw.trim()
+    if (!titleTaken) {
+      titleTaken = true
+      hero =
+        '<div class="hero-icon">🤖</div>' +
+        '<div class="hero-title">' + esc(trimmed) + '</div>' +
+        '<div class="hero-sub">第五人格 BWIKI 数据查询 · 数据均根据官方小程序数据计算</div>'
+      return
+    }
+    if (!trimmed) { body += '<div class="gap"></div>'; return }
+    if (/^(支持指令|用法|示例)：/.test(trimmed)) {
+      body += '<div class="sec">' + esc(trimmed) + '</div>'
+      return
+    }
+    if (/^[ ]{4,}/.test(raw)) {
+      body += '<div class="desc">' + esc(trimmed) + '</div>'
+      return
+    }
+    if (/^[ ]{2}/.test(raw)) {
+      body += '<div class="code">' + esc(trimmed) + '</div>'
+      return
+    }
+    if (/^(排序要素|段位|日期|模式|阵营|赛季)/.test(trimmed) && trimmed.includes('：')) {
+      body += '<div class="param"><span class="param-key">' + esc(trimmed.split('：')[0] + '：') + '</span>' + esc(trimmed.slice(trimmed.indexOf('：') + 1)) + '</div>'
+      return
+    }
+    body += '<div class="line">' + esc(trimmed) + '</div>'
+  })
+
+  return `<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+<meta charset="UTF-8">
+<style>
+* { margin:0; padding:0; box-sizing:border-box; }
+body { font-family:-apple-system,BlinkMacSystemFont,'Segoe UI','Microsoft YaHei',sans-serif; padding:28px 16px; color:#c9d1d9;
+  background:radial-gradient(1200px 500px at 50% -10%, #1b2a4a 0%, #10182b 45%, #0a0e17 100%); }
+.page { max-width:${cardW}px; margin:0 auto; }
+.hero { display:flex; flex-direction:column; align-items:center; gap:10px; text-align:center; margin-bottom:22px; }
+.hero-icon { width:74px; height:74px; border-radius:22px; display:flex; align-items:center; justify-content:center; font-size:40px;
+  background:linear-gradient(135deg, rgba(88,166,255,.18), rgba(210,153,29,.16)); border:1px solid rgba(88,166,255,.35);
+  box-shadow:0 0 36px rgba(88,166,255,.18), inset 0 1px 0 rgba(255,255,255,.08); }
+.hero-title { font-size:30px; font-weight:800; letter-spacing:1px;
+  background:linear-gradient(90deg, #79c0ff, #ffd76d); -webkit-background-clip:text; background-clip:text; -webkit-text-fill-color:transparent; }
+.hero-sub { font-size:13px; color:#8b949e; }
+.card { background:rgba(22,27,34,.72); border:1px solid rgba(255,255,255,.09); border-radius:22px; padding:26px 30px;
+  backdrop-filter:blur(8px); box-shadow:0 24px 64px rgba(0,0,0,.45); }
+.gap { height:8px; }
+.sec { font-size:16px; font-weight:700; color:#ffd76d; margin:14px 0 8px; display:flex; align-items:center; gap:10px; }
+.sec::before { content:''; width:4px; height:16px; border-radius:2px; background:linear-gradient(#ffd76d, #ff8c00); flex-shrink:0; }
+.sec:first-child { margin-top:0; }
+.line { font-size:14px; line-height:1.9; color:#c9d1d9; }
+.param { font-size:13.5px; line-height:1.9; color:#c9d1d9; padding:5px 14px; background:rgba(255,255,255,.03); border:1px solid rgba(255,255,255,.06); border-radius:8px; margin:5px 0; }
+.param-key { color:#ffd76d; font-weight:600; }
+.desc { font-size:13px; line-height:1.8; color:#8b949e; padding:2px 0 2px 18px; }
+.code { font-family:'SF Mono',Consolas,'Courier New',monospace; font-size:13.5px; line-height:1.7; color:#79c0ff;
+  background:rgba(88,166,255,.08); border:1px solid rgba(88,166,255,.2); border-radius:8px; padding:6px 14px; margin:5px 0;
+  word-break:break-all; }
+.foot { text-align:center; color:#8b949e; font-size:12px; margin-top:14px; opacity:.8; }
+</style>
+</head>
+<body>
+<div class="page">
+  <div class="hero">${hero}</div>
+  <div class="card">${body}</div>
+  <div class="foot">@夜莺小姐档案馆 · 数据均根据第五人格官方小程序数据计算</div>
+</div>
+</body>
+</html>`
+}
+
 module.exports = {
   KNOWN_SOURCES,
   METRICS,
@@ -1360,5 +1440,6 @@ module.exports = {
   buildMimicHtml,
   buildBadgeHtml,
   buildHelp,
+  buildHelpHtml,
   findSimilar,
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-dwrgbwiki",
-  "version": "1.1.0",
-  "description": "第五人格BWIKI编辑组测试用组件，纯AI生成喵",
+  "version": "1.3.0",
+  "description": "第五人格BWIKI编辑组测试用插件，所有代码由DeepSeek V4 Flash编写",
   "main": "index.js",
   "files": [
     "index.js",
@@ -21,7 +21,7 @@
   },
   "koishi": {
     "description": {
-      "zh": "第五人格BWIKI编辑组测试用组件，纯AI生成喵"
+      "zh": "第五人格BWIKI编辑组测试用插件，所有代码由DeepSeek V4 Flash编写"
     }
   }
 }

package/readme.md

@@ -9,12 +9,15 @@
 ## 用法
 
 ```
+BWIKI机器人
 胜率 <角色名> [赛季] [周数] [排序要素] [段位]
 地图 <地图名> [赛季] [周数] [段位] [阵营]
 模仿者 <角色名> [日期] [模式]
 勋章 <角色名> [日期] [模式]
 ```
 
+发送 `BWIKI机器人` 可查看当前支持的指令列表图片；发送 `胜率` 等单独指令可查看对应完整帮助。
+
 ### 胜率（角色胜率）
 
 示例：
@@ -83,7 +86,10 @@
 | `mapDefaultCamp` | 地图查询默认阵营（1=监管者，2=求生者） | `1` |
 | `showBadge` | 显示勋章分数据 | `true` |
 | `badgeDays` | 勋章日增长率天数 | `7` |
-| `width` | 图片宽度（px） | `1000` |
+| `width` | 图片宽度（px），其他指令未单独设置时使用该值 | `1300` |
+| `mapWidth` | 地图指令图片宽度（px），0=使用默认宽度 | `0` |
+| `mimicWidth` | 模仿者指令图片宽度（px），0=使用默认宽度 | `0` |
+| `badgeWidth` | 勋章指令图片宽度（px），0=使用默认宽度 | `0` |
 | `scale` | 截图倍率 | `2` |
 | `timeout` | 网络请求超时（毫秒） | `15000` |