mioku-plugin-impact 1.1.0

package/index.ts

@@ -0,0 +1,142 @@
+import type { ConfigService, ScreenshotService } from "mioku";
+import { definePlugin, type MiokiContext } from "mioki";
+import {
+  cloneConfig,
+  DEFAULT_CONFIG,
+  normalizeImpactConfig,
+  type ImpactConfig,
+} from "./configs/base";
+import { initImpactDatabase, type ImpactDatabase } from "./db";
+import { handleDajiao } from "./handlers/dajiao";
+import { handleOpenModule } from "./handlers/open-module";
+import { handlePk } from "./handlers/pk";
+import { handleQueryInjection } from "./handlers/query-injection";
+import { handleQueryJj } from "./handlers/queryjj";
+import { handleRank } from "./handlers/rank";
+import { handleSuo } from "./handlers/suo";
+import { handleYinpa } from "./handlers/yinpa";
+import type { HandlerContext } from "./handlers/types";
+import { parseImpactCommand } from "./router";
+import { createCooldownState } from "./state";
+import { NOT_ALLOW_TEXT, isGroupEvent } from "./utils";
+
+const impactPlugin = definePlugin({
+  name: "impact",
+  version: "1.0.0",
+  description: "让群友们眼前一黑的淫趴插件（牛子比拼），移植自 nonebot_plugin_impact",
+
+  async setup(ctx: MiokiContext) {
+    ctx.logger.info("impact 插件正在初始化...");
+
+    let db: ImpactDatabase;
+    try {
+      db = await initImpactDatabase();
+    } catch (error) {
+      ctx.logger.error(`impact 数据库初始化失败: ${error}`);
+      return () => {
+        ctx.logger.info("impact 插件已卸载");
+      };
+    }
+
+    const cd = createCooldownState();
+    const screenshot = ctx.services?.screenshot as
+      | ScreenshotService
+      | undefined;
+    if (!screenshot) {
+      ctx.logger.warn(
+        "impact: screenshot 服务未启用, jj排行榜 / 注入查询历史图表将不可用",
+      );
+    }
+
+    let config: ImpactConfig = cloneConfig(DEFAULT_CONFIG);
+    const configService = ctx.services?.config as ConfigService | undefined;
+    if (configService) {
+      await configService.registerConfig("impact", "base", config);
+      const persisted = await configService.getConfig("impact", "base");
+      if (persisted) {
+        config = normalizeImpactConfig(persisted);
+      }
+      configService.onConfigChange("impact", "base", (next) => {
+        config = normalizeImpactConfig(next);
+      });
+    } else {
+      ctx.logger.warn("config 服务未加载，impact 插件将使用默认配置");
+    }
+
+    if (config.isalive) {
+      ctx.cron("0 0 * * *", async () => {
+        try {
+          const touched = await db.punishInactiveUsers();
+          if (touched > 0) {
+            ctx.logger.info(`impact: 已对 ${touched} 个不活跃用户随机扣减长度`);
+          }
+        } catch (error) {
+          ctx.logger.error(`impact 不活跃惩罚执行失败: ${error}`);
+        }
+      });
+    }
+
+    ctx.handle("message", async (event: any) => {
+      if (!isGroupEvent(event)) return;
+      const text = ctx.text(event);
+      if (!text) return;
+
+      const cmd = parseImpactCommand(text);
+      if (cmd.type === "none") return;
+
+      const h: HandlerContext = { ctx, db, cd, config, screenshot, event };
+
+      try {
+        // open_module 始终可用，不受群开关限制
+        if (cmd.type === "open_module") {
+          await handleOpenModule(h, cmd.enable);
+          return;
+        }
+
+        // 其余命令需要群开启了银趴
+        if (!db.isGroupAllowed(Number(event.group_id))) {
+          await event.reply(
+            [ctx.segment.at(Number(event.user_id)), ctx.segment.text(NOT_ALLOW_TEXT)],
+            false,
+          );
+          return;
+        }
+
+        switch (cmd.type) {
+          case "pk":
+            return await handlePk(h);
+          case "dajiao":
+            return await handleDajiao(h);
+          case "suo":
+            return await handleSuo(h);
+          case "queryjj":
+            return await handleQueryJj(h);
+          case "rank":
+            return await handleRank(h);
+          case "yinpa":
+            return await handleYinpa(h, cmd.subject);
+          case "query_injection":
+            return await handleQueryInjection(h, cmd.mode);
+        }
+      } catch (error) {
+        ctx.logger.error(`impact 命令 ${cmd.type} 执行失败: ${error}`);
+        try {
+          await event.reply(
+            [ctx.segment.text(`淫趴插件出错了喵: ${String(error)}`)],
+            false,
+          );
+        } catch {
+          // 忽略二次失败
+        }
+      }
+    });
+
+    ctx.logger.info("impact 插件初始化完成");
+
+    return () => {
+      ctx.logger.info("impact 插件已卸载");
+    };
+  },
+});
+
+export default impactPlugin;

package/package.json

@@ -0,0 +1,126 @@
+{
+  "name": "mioku-plugin-impact",
+  "version": "1.1.0",
+  "description": "让群友们眼前一黑的淫趴插件",
+  "main": "index.ts",
+  "type": "module",
+  "keywords": [
+    "mioku",
+    "impact",
+    "牛子"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/mioku-lab/mioku.git"
+  },
+  "mioku": {
+    "services": [
+      "screenshot",
+      "help",
+      "config"
+    ],
+    "accessHooks": [
+      {
+        "id": "开启淫趴",
+        "match": "/^(?:开启|禁止|开始|关闭)(?:淫趴|银趴)\\s*$/",
+        "description": "群内开启或关闭淫趴功能"
+      },
+      {
+        "id": "日透",
+        "match": "/^(?:日群友|透群友|日群主|透群主|日管理|透管理)(?:\\s|$|@)/",
+        "description": "日 / 透 系列命令"
+      },
+      {
+        "id": "打胶",
+        "match": "/^(?:打胶|开导)\\s*$/",
+        "description": "打胶 / 开导"
+      },
+      {
+        "id": "嗦牛子",
+        "match": "/^嗦牛子(?:\\s|$|@)/",
+        "description": "嗦自己或被 @ 群友的牛子"
+      },
+      {
+        "id": "pk",
+        "match": "/^(?:pk|PK|对决)(?:\\s|@|$)/",
+        "description": "牛子对决"
+      },
+      {
+        "id": "查询",
+        "match": "/^查询(?:\\s|@|$)/",
+        "description": "查询自己或被 @ 群友的牛子长度"
+      },
+      {
+        "id": "注入查询",
+        "match": "/^(?:注入查询|摄入查询|射入查询)/",
+        "description": "查询当日或历史被注入量"
+      },
+      {
+        "id": "jj排行榜",
+        "match": "/^(?:jj|JJ)(?:排行榜|排名|榜单|rank|RANK)(?:\\s|$)/",
+        "description": "牛子长度排行榜"
+      }
+    ],
+    "help": {
+      "title": "淫趴 ",
+      "description": "让群友们眼前一黑的牛子比拼插件",
+      "commands": [
+        {
+          "cmd": "开启淫趴 / 禁止淫趴",
+          "desc": "在本群开启或关闭淫趴功能（管理员/群主/主人可用）",
+          "usage": "开启淫趴 / 关闭银趴",
+          "role": "admin"
+        },
+        {
+          "cmd": "打胶 / 开导",
+          "desc": "给自己的牛子增加一段随机长度",
+          "role": "member"
+        },
+        {
+          "cmd": "嗦牛子",
+          "desc": "嗦自己或被 @ 的群友的牛子，增加随机长度",
+          "usage": "嗦牛子 / 嗦牛子 @某人",
+          "role": "member"
+        },
+        {
+          "cmd": "pk / 对决",
+          "desc": "和 @ 的群友进行牛子对决，胜者抢走败者的随机长度",
+          "usage": "pk @某人",
+          "role": "member"
+        },
+        {
+          "cmd": "查询",
+          "desc": "查询自己或 @ 的群友的牛子长度",
+          "usage": "查询 / 查询 @某人",
+          "role": "member"
+        },
+        {
+          "cmd": "jj排行榜",
+          "desc": "查看牛子长度前五与倒数五名的图表",
+          "usage": "jj排行榜 / jjrank",
+          "role": "member"
+        },
+        {
+          "cmd": "日群友 / 透群友 / 透群主 / 透管理",
+          "desc": "随机或定向选取一位幸运儿进行注入，按命令决定身份",
+          "usage": "日群友 / 透群友 @某人 / 透群主 / 透管理",
+          "role": "member"
+        },
+        {
+          "cmd": "注入查询",
+          "desc": "查询当日或全部历史被注入量，加 历史/全部 看走势图",
+          "usage": "注入查询 / 注入查询 历史 / 注入查询 @某人 全部",
+          "role": "member"
+        }
+      ]
+    }
+  },
+  "peerDependencies": {
+    "mioku": "^0.8.0",
+    "mioki": "^0.16.0"
+  },
+  "devDependencies": {
+    "mioku": "workspace:*",
+    "mioki": "^0.16.0"
+  }
+}

package/router.ts

@@ -0,0 +1,94 @@
+export type ImpactCommand =
+  | { type: "pk" }
+  | { type: "dajiao" }
+  | { type: "suo" }
+  | { type: "queryjj" }
+  | { type: "rank" }
+  | { type: "yinpa"; subject: "member" | "owner" | "admin" }
+  | { type: "open_module"; enable: boolean }
+  | { type: "query_injection"; mode: "today" | "history" }
+  | { type: "none" };
+
+const YINPA_RE =
+  /^(?:日群友|透群友|日群主|透群主|日管理|透管理)(?:\s|$|@)/u;
+
+const OPEN_MODULE_RE =
+  /^(?:开始银趴|关闭银趴|开启淫趴|禁止淫趴|开启银趴|禁止银趴)\s*$/u;
+
+const RANK_HEADS = new Set([
+  "jj排行榜",
+  "jj排名",
+  "jj榜单",
+  "jjrank",
+  "JJ排行榜",
+  "JJ排名",
+  "JJ榜单",
+  "JJrank",
+  "JJRANK",
+]);
+
+/**
+ * 把消息文本解析成 impact 命令。
+ * 仅看消息开头（去掉前导空白），命中失败返回 { type: "none" }。
+ */
+export function parseImpactCommand(text: string): ImpactCommand {
+  const head = text.replace(/^\s+/, "");
+  if (!head) return { type: "none" };
+
+  // 银趴开关：完整匹配
+  const open = head.match(OPEN_MODULE_RE);
+  if (open) {
+    const word = open[0];
+    const enable = word.startsWith("开启") || word.startsWith("开始");
+    return { type: "open_module", enable };
+  }
+
+  // 透/日 系列
+  const yinpa = head.match(YINPA_RE);
+  if (yinpa) {
+    const word = yinpa[0];
+    const subject: "owner" | "admin" | "member" = word.includes("群主")
+      ? "owner"
+      : word.includes("管理")
+      ? "admin"
+      : "member";
+    return { type: "yinpa", subject };
+  }
+
+  // pk / 对决：要求形如 "pk @某人" 或 "pk"。具体的 at 校验放到 handler。
+  if (/^(pk|PK|对决)(\s|@|$)/.test(head)) {
+    return { type: "pk" };
+  }
+
+  // 打胶 / 开导：完整匹配
+  if (/^(打胶|开导)\s*$/.test(head)) {
+    return { type: "dajiao" };
+  }
+
+  // 嗦牛子
+  if (/^嗦牛子(\s|@|$)/.test(head)) {
+    return { type: "suo" };
+  }
+
+  // 注入查询 / 摄入查询 / 射入查询
+  const inj = head.match(/^(?:注入查询|摄入查询|射入查询)(.*)$/u);
+  if (inj) {
+    const rest = inj[1] || "";
+    const mode = /历史|全部/.test(rest) ? "history" : "today";
+    return { type: "query_injection", mode };
+  }
+
+  // 排行榜
+  for (const word of RANK_HEADS) {
+    if (head === word || head.startsWith(`${word} `) || head.startsWith(`${word}\n`)) {
+      return { type: "rank" };
+    }
+  }
+
+  // 只匹配 "查询" + 空白/at/结束，避免误吞太多文本
+  if (/^查询(\s|@|$)/.test(head)) {
+    return { type: "queryjj" };
+  }
+
+  return { type: "none" };
+}

package/state.ts

@@ -0,0 +1,35 @@
+import type { CooldownState } from "./types";
+import { nowSeconds } from "./utils";
+
+export function createCooldownState(): CooldownState {
+  return {
+    pk: new Map(),
+    dj: new Map(),
+    suo: new Map(),
+    fuck: new Map(),
+  };
+}
+
+/**
+ * 检查 CD 是否已过；返回 { ok, remaining }，remaining 单位秒。
+ * ok 为 true 表示允许执行，调用方应自行 markUsed。
+ */
+export function checkCooldown(
+  bucket: Map<string, number>,
+  uid: string,
+  cdSeconds: number,
+): { ok: boolean; remaining: number } {
+  const last = bucket.get(uid);
+  if (last == null) return { ok: true, remaining: 0 };
+  const elapsed = nowSeconds() - last;
+  if (elapsed >= cdSeconds) return { ok: true, remaining: 0 };
+  return { ok: false, remaining: cdSeconds - elapsed };
+}
+
+export function markCooldown(bucket: Map<string, number>, uid: string): void {
+  bucket.set(uid, nowSeconds());
+}
+
+export function clearCooldown(bucket: Map<string, number>, uid: string): void {
+  bucket.delete(uid);
+}

package/types.ts

@@ -0,0 +1,17 @@
+export interface EjaculationRecord {
+  /** YYYY-MM-DD */
+  date: string;
+  volume: number;
+}
+
+export interface RankEntry {
+  userId: number;
+  jjLength: number;
+}
+
+export interface CooldownState {
+  pk: Map<string, number>;
+  dj: Map<string, number>;
+  suo: Map<string, number>;
+  fuck: Map<string, number>;
+}

package/utils.ts

@@ -0,0 +1,99 @@
+import type { MiokiContext } from "mioki";
+
+export const JJ_VARIABLES = ["牛子", "牛牛", "丁丁", "JJ"] as const;
+
+export const NOT_ALLOW_TEXT =
+  '群内还未开启淫趴游戏, 请管理员或群主发送"开启淫趴"以开启该功能';
+
+export function nowSeconds(): number {
+  return Math.floor(Date.now() / 1000);
+}
+
+export function getToday(): string {
+  const d = new Date();
+  const pad = (n: number) => String(n).padStart(2, "0");
+  return `${d.getFullYear()}-${pad(d.getMonth() + 1)}-${pad(d.getDate())}`;
+}
+
+export function pickJj(): string {
+  return JJ_VARIABLES[Math.floor(Math.random() * JJ_VARIABLES.length)];
+}
+
+/**
+ * 获取一个随机数：0.1 概率落在 [1, 2)，0.9 概率落在 [0, 1)。保留 3 位小数。
+ */
+export function getRandomNum(): number {
+  const seed = Math.random();
+  const value = seed > 0.1 ? Math.random() : 1 + Math.random();
+  return Math.round(value * 1000) / 1000;
+}
+
+export function roundTo(value: number, digits: number): number {
+  const f = 10 ** digits;
+  return Math.round(value * f) / f;
+}
+
+/**
+ * 提取消息里第一个 @ 的 QQ 号，忽略 @全体。
+ * 找不到返回 undefined。
+ */
+export function getAtUserId(message: any): number | undefined {
+  if (!Array.isArray(message)) return undefined;
+  for (const seg of message) {
+    if (seg?.type !== "at") continue;
+    const raw = seg?.qq ?? seg?.data?.qq;
+    if (raw == null || raw === "all") continue;
+    const qq = Number(raw);
+    if (Number.isFinite(qq)) return qq;
+  }
+  return undefined;
+}
+
+export function isGroupEvent(event: any): boolean {
+  return event?.message_type === "group" && event?.group_id != null;
+}
+
+export function isGroupAdmin(event: any): boolean {
+  const role = event?.sender?.role;
+  return role === "admin" || role === "owner";
+}
+
+export function getSenderName(event: any): string {
+  const card = String(event?.sender?.card || "").trim();
+  if (card) return card;
+  const nickname = String(event?.sender?.nickname || "").trim();
+  if (nickname) return nickname;
+  return String(event?.user_id ?? "群友");
+}
+
+export async function getStrangerNickname(
+  ctx: MiokiContext,
+  selfId: number,
+  userId: number,
+): Promise<string> {
+  try {
+    const info = (await ctx
+      .pickBot(selfId)
+      .api("get_stranger_info", { user_id: userId, no_cache: false })) as
+      | { nickname?: string }
+      | undefined;
+    const name = String(info?.nickname || "").trim();
+    if (name) return name;
+  } catch {
+    // 接口不可用就回退到 QQ 号
+  }
+  return String(userId);
+}
+
+export function getAvatarUrl(userId: number): string {
+  return `https://q1.qlogo.cn/g?b=qq&nk=${userId}&s=640`;
+}
+
+export function escapeHtml(value: string): string {
+  return String(value)
+    .replace(/&/g, "&amp;")
+    .replace(/</g, "&lt;")
+    .replace(/>/g, "&gt;")
+    .replace(/"/g, "&quot;")
+    .replace(/'/g, "&#39;");
+}