mioku-plugin-impact 1.1.0

package/handlers/rank.ts

@@ -0,0 +1,58 @@
+import { renderRankChart, type RankRow } from "../image";
+import { getStrangerNickname, pickJj } from "../utils";
+import type { HandlerContext } from "./types";
+
+export async function handleRank(h: HandlerContext): Promise<void> {
+  const { ctx, db, screenshot, event } = h;
+  const ranking = db.getRanking();
+  if (ranking.length < 5) {
+    await event.reply([ctx.segment.text("目前记录的数据量小于5, 无法显示rank喵")], false);
+    return;
+  }
+
+  if (!screenshot) {
+    await event.reply(
+      [ctx.segment.text("缺少 screenshot 服务，无法生成排行榜图片喵")],
+      false,
+    );
+    return;
+  }
+
+  const top5 = ranking.slice(0, 5);
+  const last5 = ranking.slice(-5);
+
+  const myIndex = ranking.findIndex((r) => r.userId === Number(event.user_id));
+  if (myIndex < 0) {
+    await db.addNewUser(Number(event.user_id));
+    await event.reply(
+      [
+        ctx.segment.at(Number(event.user_id)),
+        ctx.segment.text(
+          `你还没有创建${pickJj()}看不到rank喵, 咱帮你创建了喵, 目前长度是10cm喵`,
+        ),
+      ],
+      false,
+    );
+    return;
+  }
+
+  const selfId = Number(event.self_id);
+  const rows: RankRow[] = await Promise.all(
+    [...top5, ...last5].map(async (r) => ({
+      name: await getStrangerNickname(ctx, selfId, r.userId),
+      userId: r.userId,
+      jjLength: r.jjLength,
+    })),
+  );
+
+  const imagePath = await renderRankChart(screenshot, rows);
+
+  await event.reply(
+    [
+      ctx.segment.at(Number(event.user_id)),
+      ctx.segment.image(imagePath),
+      ctx.segment.text(`你的排名为${myIndex + 1}喵`),
+    ],
+    false,
+  );
+}

package/handlers/suo.ts

@@ -0,0 +1,82 @@
+import { checkCooldown, clearCooldown, markCooldown } from "../state";
+import { getAtUserId, getRandomNum, pickJj, roundTo } from "../utils";
+import type { HandlerContext } from "./types";
+
+export async function handleSuo(h: HandlerContext): Promise<void> {
+  const { ctx, db, cd, config, event } = h;
+  const uid = String(event.user_id);
+  const cdState = checkCooldown(cd.suo, uid, config.suoCd);
+  if (!cdState.ok) {
+    await event.reply(
+      [
+        ctx.segment.at(Number(event.user_id)),
+        ctx.segment.text(
+          `你已经嗦不动了喵, 请等待${roundTo(cdState.remaining, 3)}秒后再嗦喵`,
+        ),
+      ],
+      false,
+    );
+    return;
+  }
+  markCooldown(cd.suo, uid);
+
+  const at = getAtUserId(event.message);
+  if (at == null) {
+    if (db.isUserInTable(Number(uid))) {
+      const delta = getRandomNum();
+      await db.addJjLength(Number(uid), delta);
+      await event.reply(
+        [
+          ctx.segment.at(Number(event.user_id)),
+          ctx.segment.text(
+            `你的${pickJj()}很满意喵, 嗦长了${delta}cm喵, 目前长度为${db.getJjLength(
+              Number(uid),
+            )}cm喵`,
+          ),
+        ],
+        false,
+      );
+    } else {
+      await db.addNewUser(Number(uid));
+      clearCooldown(cd.suo, uid);
+      await event.reply(
+        [
+          ctx.segment.at(Number(event.user_id)),
+          ctx.segment.text(
+            `你还没有创建${pickJj()}喵, 咱帮你创建了喵, 目前长度是10cm喵`,
+          ),
+        ],
+        false,
+      );
+    }
+    return;
+  }
+
+  if (db.isUserInTable(at)) {
+    const delta = getRandomNum();
+    await db.addJjLength(at, delta);
+    await event.reply(
+      [
+        ctx.segment.at(Number(event.user_id)),
+        ctx.segment.text(
+          `对方的${pickJj()}很满意喵, 嗦长了${delta}cm喵, 目前长度为${db.getJjLength(
+            at,
+          )}cm喵`,
+        ),
+      ],
+      false,
+    );
+  } else {
+    await db.addNewUser(at);
+    clearCooldown(cd.suo, uid);
+    await event.reply(
+      [
+        ctx.segment.at(Number(event.user_id)),
+        ctx.segment.text(
+          `TA还没有创建${pickJj()}喵, 咱帮TA创建了喵, 目前长度是10cm喵`,
+        ),
+      ],
+      false,
+    );
+  }
+}

package/handlers/types.ts

@@ -0,0 +1,15 @@
+import type { ScreenshotService } from "mioku";
+import type { MiokiContext } from "mioki";
+import type { ImpactConfig } from "../configs/base";
+import type { ImpactDatabase } from "../db";
+import type { CooldownState } from "../types";
+
+export interface HandlerContext {
+  ctx: MiokiContext;
+  db: ImpactDatabase;
+  cd: CooldownState;
+  config: ImpactConfig;
+  /** 截图服务可能未启用，依赖图片输出的命令需要降级或拒绝 */
+  screenshot?: ScreenshotService;
+  event: any;
+}

package/handlers/yinpa.ts

@@ -0,0 +1,179 @@
+import { isOwner } from "mioki";
+import { checkCooldown, clearCooldown, markCooldown } from "../state";
+import {
+  getAtUserId,
+  getAvatarUrl,
+  getSenderName,
+  roundTo,
+} from "../utils";
+import type { HandlerContext } from "./types";
+
+interface PrepMember {
+  user_id: number;
+  role: "owner" | "admin" | "member";
+  card?: string;
+  nickname?: string;
+}
+
+const SLEEP_BEFORE_REPLY_MS = 2000;
+
+/**
+ * 透/日 系列命令。subject 由命令文本决定：
+ *   含"群主" → owner，含"管理" → admin，否则 → member。
+ * 行为与 nonebot_plugin_impact.handle.Impart.yinpa 对齐。
+ */
+export async function handleYinpa(
+  h: HandlerContext,
+  subject: "owner" | "admin" | "member",
+): Promise<void> {
+  const { ctx, db, cd, config, event } = h;
+  const uid = Number(event.user_id);
+  const uidKey = String(uid);
+
+  const cdState = checkCooldown(cd.fuck, uidKey, config.fuckCd);
+  // 主人不受 CD 限制
+  if (!cdState.ok && !isOwner(event)) {
+    await event.reply(
+      [
+        ctx.segment.at(uid),
+        ctx.segment.text(
+          `你已经榨不出来任何东西了, 请先休息${roundTo(cdState.remaining, 3)}秒`,
+        ),
+      ],
+      false,
+    );
+    return;
+  }
+  markCooldown(cd.fuck, uidKey);
+
+  const reqUserCard = getSenderName(event);
+  const selfId = Number(event.self_id);
+  let prepList: PrepMember[];
+  try {
+    const raw = await ctx
+      .pickBot(selfId)
+      .api("get_group_member_list", { group_id: Number(event.group_id) });
+    prepList = (raw as PrepMember[]) ?? [];
+  } catch (err) {
+    clearCooldown(cd.fuck, uidKey);
+    ctx.logger.error(`impact: 获取群成员列表失败: ${err}`);
+    await event.reply([ctx.segment.text("喵喵喵? 拿不到群成员列表了！")], false);
+    return;
+  }
+
+  let lucky: number | null;
+  if (subject === "owner") {
+    lucky = await pickOwner(prepList, uid, reqUserCard, ctx, event);
+  } else if (subject === "admin") {
+    lucky = await pickAdmin(prepList, uid, reqUserCard, ctx, event);
+  } else {
+    lucky = await pickMember(prepList, uid, reqUserCard, ctx, event);
+  }
+  if (lucky == null) {
+    // pickXxx 内部已 finish 并清掉 CD
+    clearCooldown(cd.fuck, uidKey);
+    return;
+  }
+
+  // 1.0 ~ 100.0 ml 随机，保留 3 位
+  const volume = roundTo(1 + Math.random() * 99, 3);
+  await db.insertEjaculation(lucky, volume);
+
+  await sleep(SLEEP_BEFORE_REPLY_MS);
+
+  await db.updateActivity(lucky);
+  await db.updateActivity(uid);
+
+  const luckyMember = prepList.find((p) => p.user_id === lucky);
+  const luckyCard =
+    String(luckyMember?.card || "").trim() ||
+    String(luckyMember?.nickname || "").trim() ||
+    "群友";
+  const todayTotal = db.getTodayEjaculationVolume(lucky);
+  const seconds = 1 + Math.floor(Math.random() * 20);
+
+  await event.reply(
+    [
+      ctx.segment.text(
+        `好欸！${reqUserCard}(${uid})用时${seconds}秒 \n给 ${luckyCard}(${lucky}) 注入了${volume}毫升的脱氧核糖核酸, 当日总注入量为：${todayTotal}毫升\n`,
+      ),
+      ctx.segment.image(getAvatarUrl(lucky)),
+    ],
+    false,
+  );
+}
+
+async function pickOwner(
+  prepList: PrepMember[],
+  uid: number,
+  reqUserCard: string,
+  ctx: any,
+  event: any,
+): Promise<number | null> {
+  // 找不到群主时回退到自己
+  const owner = prepList.find((p) => p.role === "owner");
+  const lucky = owner?.user_id ?? uid;
+  if (lucky === uid) {
+    await event.reply([ctx.segment.text("你透你自己?")], false);
+    return null;
+  }
+  await event.reply(
+    [ctx.segment.text(`现在咱将把群主\n送给${reqUserCard}色色！`)],
+    false,
+  );
+  return lucky;
+}
+
+async function pickAdmin(
+  prepList: PrepMember[],
+  uid: number,
+  reqUserCard: string,
+  ctx: any,
+  event: any,
+): Promise<number | null> {
+  let admins = prepList.filter((p) => p.role === "admin").map((p) => p.user_id);
+  if (admins.includes(uid)) {
+    // 自己是管理的话移除自己
+    admins = admins.filter((id) => id !== uid);
+  }
+  if (admins.length === 0) {
+    await event.reply([ctx.segment.text("喵喵喵? 找不到群管理!")], false);
+    return null;
+  }
+  const lucky = admins[Math.floor(Math.random() * admins.length)];
+  await event.reply(
+    [ctx.segment.text(`现在咱将随机抽取一位幸运管理\n送给${reqUserCard}色色！`)],
+    false,
+  );
+  return lucky;
+}
+
+async function pickMember(
+  prepList: PrepMember[],
+  uid: number,
+  reqUserCard: string,
+  ctx: any,
+  event: any,
+): Promise<number | null> {
+  const at = getAtUserId(event.message);
+  if (at != null) {
+    // 有 @ 就直接指定目标，不播报抽取提示
+    return at;
+  }
+  let candidates = prepList.map((p) => p.user_id);
+  candidates = candidates.filter((id) => id !== uid);
+  if (candidates.length === 0) {
+    await event.reply([ctx.segment.text("喵喵喵? 群里没有别的群友了!")], false);
+    return null;
+  }
+  const lucky = candidates[Math.floor(Math.random() * candidates.length)];
+  await event.reply(
+    [ctx.segment.text(`现在咱将随机抽取一位幸运群友\n送给${reqUserCard}色色！`)],
+    false,
+  );
+  return lucky;
+}
+
+function sleep(ms: number): Promise<void> {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}

package/image.ts

@@ -0,0 +1,335 @@
+import type { ScreenshotService } from "mioku";
+import { escapeHtml, getAvatarUrl } from "./utils";
+
+export interface RankRow {
+  name: string;
+  userId: number;
+  jjLength: number;
+}
+
+export interface InjectionPoint {
+  date: string;
+  volume: number;
+}
+
+const COLORS = [
+  "#1f77b4",
+  "#2ca02c",
+  "#d62728",
+  "#ff7f0e",
+  "#9467bd",
+  "#17becf",
+  "#bcbd22",
+  "#e377c2",
+  "#ff6347",
+  "#ffd700",
+];
+
+const BASE_CSS = `
+  * { box-sizing: border-box; margin: 0; padding: 0; }
+  body {
+    font-family: "MiSans", "PingFang SC", "Microsoft YaHei",
+      "Noto Sans CJK SC", "Hiragino Sans GB", "Apple Color Emoji",
+      "Noto Color Emoji", sans-serif;
+    background: #fff;
+    color: #1d1d1f;
+  }
+`;
+
+/**
+ * 渲染牛子排行榜：横向条形图（前五正向延伸，后五负向延伸）。
+ */
+export async function renderRankChart(
+  screenshot: ScreenshotService,
+  rows: RankRow[],
+): Promise<string> {
+  const width = 960;
+  const rowHeight = 56;
+  const headerHeight = 96;
+  const padding = 32;
+  const labelWidth = 220;
+  const trackPad = 32;
+  const trackWidth = width - padding * 2 - labelWidth - trackPad;
+  const height = headerHeight + rowHeight * rows.length + padding;
+
+  const maxAbs = Math.max(
+    1,
+    ...rows.map((r) => Math.abs(r.jjLength)),
+  );
+
+  const items = rows
+    .map((row, i) => {
+      const color = COLORS[i % COLORS.length];
+      const ratio = Math.min(1, Math.abs(row.jjLength) / maxAbs);
+      const barWidth = ratio * (trackWidth / 2 - 6);
+      const isPositive = row.jjLength >= 0;
+      const trackLeft = padding + labelWidth + trackPad;
+      const center = trackLeft + trackWidth / 2;
+      const barLeft = isPositive ? center : center - barWidth;
+      const labelText = `${row.jjLength.toFixed(2)} cm`;
+      return `
+        <div class="row" style="top:${headerHeight + i * rowHeight}px;">
+          <img class="avatar" src="${escapeHtml(getAvatarUrl(row.userId))}" />
+          <div class="name">${escapeHtml(truncate(row.name, 9))}</div>
+          <div class="track" style="left:${trackLeft}px; width:${trackWidth}px;">
+            <div class="axis"></div>
+            <div class="bar" style="
+              left:${barLeft}px;
+              width:${barWidth}px;
+              background:${color};
+            "></div>
+            <div class="value" style="${
+              isPositive
+                ? `left:${barLeft + barWidth + 8}px;`
+                : `left:${barLeft - 70}px; text-align:right;`
+            }">${labelText}</div>
+          </div>
+        </div>
+      `;
+    })
+    .join("");
+
+  const html = `<!DOCTYPE html><html lang="zh-CN"><head><meta charset="utf-8" />
+    <style>
+      ${BASE_CSS}
+      .canvas {
+        position: relative;
+        width: ${width}px;
+        height: ${height}px;
+        background: #fafafa;
+      }
+      .title {
+        position: absolute;
+        left: 0; top: 28px;
+        width: 100%;
+        text-align: center;
+        font-size: 32px;
+        font-weight: 700;
+        color: #1d1d1f;
+      }
+      .subtitle {
+        position: absolute;
+        left: 0; top: 64px;
+        width: 100%;
+        text-align: center;
+        font-size: 16px;
+        color: #6e6e73;
+      }
+      .row {
+        position: absolute;
+        left: ${padding}px;
+        right: ${padding}px;
+        height: ${rowHeight - 8}px;
+      }
+      .avatar {
+        position: absolute;
+        left: 0; top: 0;
+        width: ${rowHeight - 12}px;
+        height: ${rowHeight - 12}px;
+        border-radius: 50%;
+        object-fit: cover;
+        border: 1px solid #e5e5ea;
+        background: #f2f2f7;
+      }
+      .name {
+        position: absolute;
+        left: 60px;
+        top: 50%;
+        transform: translateY(-50%);
+        width: ${labelWidth - 60}px;
+        font-size: 18px;
+        color: #1d1d1f;
+        white-space: nowrap;
+        overflow: hidden;
+        text-overflow: ellipsis;
+      }
+      .track {
+        position: absolute;
+        top: 50%;
+        height: ${rowHeight - 18}px;
+        transform: translateY(-50%);
+      }
+      .axis {
+        position: absolute;
+        left: 50%;
+        top: -4px;
+        bottom: -4px;
+        width: 2px;
+        background: #d2d2d7;
+      }
+      .bar {
+        position: absolute;
+        top: 0;
+        bottom: 0;
+        border-radius: 4px;
+        opacity: 0.92;
+      }
+      .value {
+        position: absolute;
+        top: 50%;
+        transform: translateY(-50%);
+        font-size: 14px;
+        color: #1d1d1f;
+        white-space: nowrap;
+        width: 70px;
+      }
+    </style>
+  </head>
+  <body>
+    <div class="canvas">
+      <div class="title">JJ 排行榜</div>
+      <div class="subtitle">Top 5 与 Bottom 5（单位：cm）</div>
+      ${items}
+    </div>
+  </body></html>`;
+
+  return screenshot.screenshot(html, {
+    width,
+    height,
+    fullPage: false,
+    type: "png",
+  });
+}
+
+/**
+ * 渲染历史注入折线图。
+ */
+export async function renderInjectionChart(
+  screenshot: ScreenshotService,
+  points: InjectionPoint[],
+  totalVolume: number,
+): Promise<string> {
+  const width = 1024;
+  const height = 560;
+  const padding = { top: 80, right: 48, bottom: 80, left: 64 };
+  const plotW = width - padding.left - padding.right;
+  const plotH = height - padding.top - padding.bottom;
+
+  const maxValue = Math.max(1, ...points.map((p) => p.volume));
+  const niceMax = niceCeil(maxValue);
+
+  const ticksY = 5;
+  const yLabels = Array.from({ length: ticksY + 1 }, (_, i) => {
+    const v = (niceMax / ticksY) * i;
+    return {
+      value: v.toFixed(0),
+      y: padding.top + plotH - (v / niceMax) * plotH,
+    };
+  });
+
+  const stepX = points.length > 1 ? plotW / (points.length - 1) : 0;
+
+  const dots = points.map((p, i) => {
+    const x = padding.left + stepX * i;
+    const y = padding.top + plotH - (p.volume / niceMax) * plotH;
+    return { x, y, point: p };
+  });
+
+  const linePath = dots
+    .map((d, i) => `${i === 0 ? "M" : "L"} ${d.x.toFixed(2)} ${d.y.toFixed(2)}`)
+    .join(" ");
+
+  const dotsHtml = dots
+    .map(
+      (d) => `
+        <circle cx="${d.x}" cy="${d.y}" r="5" fill="#1f77b4" />
+        <text x="${d.x}" y="${d.y - 12}" text-anchor="middle" font-size="13" fill="#1d1d1f">${d.point.volume}</text>
+      `,
+    )
+    .join("");
+
+  const sampledLabels = sampleLabels(dots, 8);
+  const xLabelsHtml = sampledLabels
+    .map(
+      (d) => `
+        <text x="${d.x}" y="${padding.top + plotH + 22}" text-anchor="middle" font-size="13" fill="#6e6e73" transform="rotate(-30 ${d.x} ${padding.top + plotH + 22})">${escapeHtml(d.point.date)}</text>
+      `,
+    )
+    .join("");
+
+  const yAxisHtml = yLabels
+    .map(
+      (l) => `
+        <line x1="${padding.left}" y1="${l.y}" x2="${padding.left + plotW}" y2="${l.y}" stroke="#e5e5ea" stroke-dasharray="4 4" />
+        <text x="${padding.left - 8}" y="${l.y + 4}" text-anchor="end" font-size="13" fill="#6e6e73">${l.value}</text>
+      `,
+    )
+    .join("");
+
+  const html = `<!DOCTYPE html><html lang="zh-CN"><head><meta charset="utf-8" />
+    <style>
+      ${BASE_CSS}
+      .canvas {
+        position: relative;
+        width: ${width}px;
+        height: ${height}px;
+        background: #fafafa;
+      }
+      .title {
+        position: absolute;
+        left: 0; top: 24px;
+        width: 100%;
+        text-align: center;
+        font-size: 28px;
+        font-weight: 700;
+        color: #1d1d1f;
+      }
+      .subtitle {
+        position: absolute;
+        left: 0; top: 56px;
+        width: 100%;
+        text-align: center;
+        font-size: 14px;
+        color: #6e6e73;
+      }
+      svg { position: absolute; left: 0; top: 0; }
+    </style>
+  </head>
+  <body>
+    <div class="canvas">
+      <div class="title">历史注入走势</div>
+      <div class="subtitle">总量 ${totalVolume.toFixed(2)} ml · 共 ${points.length} 天有记录</div>
+      <svg width="${width}" height="${height}">
+        ${yAxisHtml}
+        <line x1="${padding.left}" y1="${padding.top + plotH}" x2="${padding.left + plotW}" y2="${padding.top + plotH}" stroke="#1d1d1f" stroke-width="2" />
+        <line x1="${padding.left}" y1="${padding.top}" x2="${padding.left}" y2="${padding.top + plotH}" stroke="#1d1d1f" stroke-width="2" />
+        <path d="${linePath}" fill="none" stroke="#1f77b4" stroke-width="2.5" />
+        ${dotsHtml}
+        ${xLabelsHtml}
+      </svg>
+    </div>
+  </body></html>`;
+
+  return screenshot.screenshot(html, {
+    width,
+    height,
+    fullPage: false,
+    type: "png",
+  });
+}
+
+function truncate(text: string, max: number): string {
+  return text.length > max ? `${text.slice(0, max)}…` : text;
+}
+
+function niceCeil(v: number): number {
+  if (v <= 0) return 1;
+  const exp = Math.pow(10, Math.floor(Math.log10(v)));
+  const base = v / exp;
+  let mult: number;
+  if (base <= 1) mult = 1;
+  else if (base <= 2) mult = 2;
+  else if (base <= 5) mult = 5;
+  else mult = 10;
+  return mult * exp;
+}
+
+function sampleLabels<T>(items: T[], maxLabels: number): T[] {
+  if (items.length <= maxLabels) return items;
+  const step = (items.length - 1) / (maxLabels - 1);
+  const out: T[] = [];
+  for (let i = 0; i < maxLabels; i++) {
+    out.push(items[Math.round(i * step)]);
+  }
+  return out;
+}