koishi-plugin-onebot-verifier 1.0.9 → 1.1.0

package/lib/index.d.ts

@@ -21,6 +21,13 @@ export interface Config {
         minLevel?: number;
         action?: 'accept' | 'reject';
     }[];
+    syncNotify?: boolean;
+    specialRules?: {
+        guildId: string;
+        enabled: boolean;
+        mode: 'vote' | 'captcha';
+    }[];
+    voteRatio?: string;
 }
 export declare const Config: Schema<Config>;
 export declare function apply(ctx: Context, config?: Config): void;

package/lib/index.js

@@ -73,12 +73,25 @@ var Config = import_koishi.Schema.intersect([
         import_koishi.Schema.const("accept").description("同意"),
         import_koishi.Schema.const("reject").description("拒绝")
       ]).description("操作")
-    })).description("加群验证配置").role("table")
-  }).description("加群请求配置")
+    })).description("普通验证配置").role("table")
+  }).description("加群请求配置"),
+  import_koishi.Schema.object({
+    syncNotify: import_koishi.Schema.boolean().description("同步通知目标").default(true),
+    specialRules: import_koishi.Schema.array(import_koishi.Schema.object({
+      guildId: import_koishi.Schema.string().description("群号").required(),
+      mode: import_koishi.Schema.union([
+        import_koishi.Schema.const("vote").description("投票"),
+        import_koishi.Schema.const("captcha").description("验证码")
+      ]).description("模式").default("vote"),
+      enabled: import_koishi.Schema.boolean().description("前置规则").default(true)
+    })).description("配置列表").role("table"),
+    voteRatio: import_koishi.Schema.string().description("投票比例").default("3:2")
+  }).description("特殊验证配置")
 ]);
 function apply(ctx, config = {}) {
   const logger = new import_koishi.Logger("onebot-verifier");
   const activeTasks = /* @__PURE__ */ new Map();
+  const activeCaptchas = /* @__PURE__ */ new Map();
   const inviterMap = /* @__PURE__ */ new Map();
   const getComment = /* @__PURE__ */ __name((comment) => {
     if (!comment) return "";
@@ -130,6 +143,7 @@ function apply(ctx, config = {}) {
       if (adminId) infoLines.push(`管理：${adminInfo?.name ? `${adminInfo.name}(${adminId})` : adminId}`);
       if (session.guildId) infoLines.push(`群组：${groupInfo?.name ? `${groupInfo.name}(${session.guildId})` : session.guildId}`);
       if (eventData.comment) infoLines.push(`验证信息：${eventData.comment}`);
+      if (status === "waiting") infoLines.push(`使用"y/n"回复本消息，以同意/拒绝该请求`);
       const content = infoLines.join("\n");
       const msgIds = await (targetType === "private" ? session.bot.sendPrivateMessage(targetId, content) : session.bot.sendMessage(targetId, content)) || [];
       return msgIds;
@@ -194,9 +208,29 @@ function apply(ctx, config = {}) {
             return;
           }
         }
+        const specialRule = config.specialRules?.find((r) => String(r.guildId) === String(session.guildId) && r.enabled);
+        if (specialRule) {
+          if (specialRule.mode === "vote") {
+            const [yesStr, noStr] = config.voteRatio.split(":");
+            const targetYes = parseInt(yesStr) || 0;
+            const targetNo = parseInt(noStr) || 0;
+            let msgIds = [];
+            if (config.syncNotify !== false) msgIds = await sendNotice(session, kind, "waiting");
+            if (msgIds.length > 0) {
+              const task = { session, kind, messages: msgIds, specialMode: "vote", voteTarget: { yes: targetYes, no: targetNo }, votes: { yes: /* @__PURE__ */ new Set(), no: /* @__PURE__ */ new Set() } };
+              msgIds.forEach((id) => activeTasks.set(id, task));
+            }
+            return;
+          }
+          if (specialRule.mode === "captcha") {
+            await executeAction(session, kind, true, "验证码验证，自动通过");
+            if (config.syncNotify !== false) await sendNotice(session, kind, "auto_pass");
+            return;
+          }
+        }
         if (config.verifyMode && config.verifyMode !== "manual") {
           const isApprove = config.verifyMode === "accept";
-          await executeAction(session, kind, isApprove, "等待超时，自动处理");
+          await executeAction(session, kind, isApprove, "默认规则，自动处理");
           await sendNotice(session, kind, isApprove ? "auto_pass" : "auto_reject");
           return;
         }
@@ -255,10 +289,57 @@ function apply(ctx, config = {}) {
       logger.error(`处理失败: ${error}`);
     }
   }, "hookEvent");
+  const handleSpecialVote = /* @__PURE__ */ __name(async (session, task, isApprove, extraInfo, targetType, targetId) => {
+    if (!task.voteTarget || !task.votes) return;
+    const voterId = session.userId;
+    if (!voterId) return;
+    task.votes.yes.delete(voterId);
+    task.votes.no.delete(voterId);
+    if (isApprove) {
+      task.votes.yes.add(voterId);
+    } else {
+      task.votes.no.add(voterId);
+    }
+    if (config.debugMode) logger.info(`[投票] 赞成: ${task.votes.yes.size}/${task.voteTarget.yes} | 反对: ${task.votes.no.size}/${task.voteTarget.no}`);
+    let thresholdMet = false;
+    let finalVerdict = false;
+    if (task.voteTarget.yes > 0 && task.votes.yes.size >= task.voteTarget.yes) {
+      thresholdMet = true;
+      finalVerdict = true;
+    } else if (task.voteTarget.no > 0 && task.votes.no.size >= task.voteTarget.no) {
+      thresholdMet = true;
+      finalVerdict = false;
+    }
+    if (!thresholdMet) return;
+    task.messages.forEach((msg) => activeTasks.delete(msg));
+    const isSuccess = await executeAction(task.session, task.kind, finalVerdict, finalVerdict ? "" : extraInfo);
+    const replyText = isSuccess ? `已${finalVerdict ? "通过" : "拒绝"}该投票` : `处理投票失败`;
+    if (session.bot) await (targetType === "private" ? session.bot.sendPrivateMessage(targetId, replyText) : session.bot.sendMessage(targetId, replyText)).catch(() => {
+    });
+  }, "handleSpecialVote");
   ctx.on("friend-request", hookEvent("friend"));
   ctx.on("guild-request", hookEvent("guild"));
   ctx.on("guild-member-request", hookEvent("member"));
   ctx.on("guild-added", hookEvent("guild"));
+  ctx.on("guild-member-added", async (session) => {
+    if (!config.specialRules || !session.guildId || !session.userId) return;
+    const rule = config.specialRules.find((r) => String(r.guildId) === String(session.guildId) && r.enabled);
+    if (rule?.mode === "captcha") {
+      const a = Math.floor(Math.random() * 20) + 1;
+      const b = Math.floor(Math.random() * 20) + 1;
+      const answer = (a + b).toString();
+      const captchaKey = `${session.guildId}:${session.userId}`;
+      await session.send(`<at id="${session.userId}"/> 请在 60 秒内回复计算结果，以进行验证：${a} + ${b} =`);
+      const timer = setTimeout(async () => {
+        if (activeCaptchas.has(captchaKey)) {
+          activeCaptchas.delete(captchaKey);
+          await session.send(`<at id="${session.userId}"/> 验证失败，将被移出本群。`);
+          await session.onebot?.setGroupKick(session.guildId, session.userId, false);
+        }
+      }, 6e4);
+      activeCaptchas.set(captchaKey, { guildId: session.guildId, userId: session.userId, answer, timer });
+    }
+  });
   ctx.on("guild-removed", async (session) => {
     if (session.event?._data?.sub_type === "kick_me") {
       const gid = session.guildId;
@@ -277,7 +358,21 @@ function apply(ctx, config = {}) {
     await sendNotice(session, "removed");
   });
   ctx.middleware(async (session, next) => {
-    if (typeof session.content !== "string" || !session.quote?.id) return next();
+    if (typeof session.content !== "string") return next();
+    if (session.guildId && session.userId) {
+      const captchaKey = `${session.guildId}:${session.userId}`;
+      const captcha = activeCaptchas.get(captchaKey);
+      if (captcha) {
+        const input2 = session.content.trim();
+        if (input2 === captcha.answer) {
+          clearTimeout(captcha.timer);
+          activeCaptchas.delete(captchaKey);
+          await session.send(`<at id="${session.userId}"/> 验证成功，欢迎加入本群！`);
+          return;
+        }
+      }
+    }
+    if (!session.quote?.id) return next();
     const task = activeTasks.get(session.quote.id);
     if (!task) return next();
     const [targetType, targetId] = (config.notifyTarget || "").split(":");
@@ -286,11 +381,15 @@ function apply(ctx, config = {}) {
     const input = session.content.replace(/<(quote|at)\s+[^>]*\/>/gi, "").trim();
     const cmdMatch = input.match(/^(y|n|通过|拒绝)(?:\s+(.*))?$/i);
     if (!cmdMatch) return next();
-    if (task.timer) clearTimeout(task.timer);
-    task.messages.forEach((msg) => activeTasks.delete(msg));
     const isApprove = ["y", "通过"].includes(cmdMatch[1].toLowerCase());
     const extraInfo = cmdMatch[2]?.trim() || "";
     if (config.debugMode) logger.info(`[操作] 收到指令: ${isApprove ? "同意" : "拒绝"}`);
+    if (task.specialMode === "vote") {
+      await handleSpecialVote(session, task, isApprove, extraInfo, targetType, targetId);
+      return;
+    }
+    if (task.timer) clearTimeout(task.timer);
+    task.messages.forEach((msg) => activeTasks.delete(msg));
     const isSuccess = await executeAction(task.session, task.kind, isApprove, isApprove ? "" : extraInfo, isApprove && task.kind === "friend" ? extraInfo : "");
     const replyText = isSuccess ? `已${isApprove ? "通过" : "拒绝"}该请求` : `处理请求失败`;
     if (session.bot) await (targetType === "private" ? session.bot.sendPrivateMessage(targetId, replyText) : session.bot.sendMessage(targetId, replyText)).catch(() => {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-onebot-verifier",
   "description": "适用于 Onebot 的审核插件，支持自动审核好友/加群/邀请请求",
-  "version": "1.0.9",
+  "version": "1.1.0",
   "contributors": [
     "Yis_Rime <yis_rime@outlook.com>"
   ],