koishi-plugin-group-control 0.2.10 → 0.2.11

package/lib/database.d.ts

@@ -19,11 +19,16 @@ export interface GroupBotStatus {
     guildId: string;
     botEnabled: boolean;
 }
+export interface SmallGroupWhitelist {
+    platform: string;
+    guildId: string;
+}
 declare module 'koishi' {
     interface Tables {
         blacklisted_guild: BlacklistedGuild;
         command_frequency_record: CommandFrequencyRecord;
         group_bot_status: GroupBotStatus;
+        small_group_whitelist: SmallGroupWhitelist;
     }
 }
 export declare const name = "group-control-database";
@@ -38,3 +43,7 @@ export declare function getCommandFrequencyRecord(ctx: Context, platform: string
 export declare function updateCommandFrequencyRecord(ctx: Context, platform: string, guildId: string, data: Partial<CommandFrequencyRecord>): Promise<void>;
 export declare function getGroupBotStatus(ctx: Context, platform: string, guildId: string): Promise<GroupBotStatus | null>;
 export declare function setGroupBotStatus(ctx: Context, platform: string, guildId: string, botEnabled: boolean): Promise<void>;
+export declare function isInSmallGroupWhitelist(ctx: Context, guildId: string): Promise<boolean>;
+export declare function addToSmallGroupWhitelist(ctx: Context, guildId: string): Promise<void>;
+export declare function removeFromSmallGroupWhitelist(ctx: Context, guildId: string): Promise<void>;
+export declare function getAllSmallGroupWhitelist(ctx: Context): Promise<SmallGroupWhitelist[]>;

package/lib/index.js

@@ -30,15 +30,19 @@ module.exports = __toCommonJS(src_exports);
 var database_exports = {};
 __export(database_exports, {
   BLACKLIST_PLATFORM: () => BLACKLIST_PLATFORM,
+  addToSmallGroupWhitelist: () => addToSmallGroupWhitelist,
   apply: () => apply,
   clearBlacklistedGuilds: () => clearBlacklistedGuilds,
   createBlacklistedGuild: () => createBlacklistedGuild,
   getAllBlacklistedGuilds: () => getAllBlacklistedGuilds,
+  getAllSmallGroupWhitelist: () => getAllSmallGroupWhitelist,
   getBlacklistedGuild: () => getBlacklistedGuild,
   getCommandFrequencyRecord: () => getCommandFrequencyRecord,
   getGroupBotStatus: () => getGroupBotStatus,
+  isInSmallGroupWhitelist: () => isInSmallGroupWhitelist,
   name: () => name,
   removeBlacklistedGuild: () => removeBlacklistedGuild,
+  removeFromSmallGroupWhitelist: () => removeFromSmallGroupWhitelist,
   setGroupBotStatus: () => setGroupBotStatus,
   updateCommandFrequencyRecord: () => updateCommandFrequencyRecord
 });
@@ -64,6 +68,10 @@ function apply(ctx) {
     guildId: "string",
     botEnabled: "boolean"
   }, { primary: ["platform", "guildId"] });
+  ctx.model.extend("small_group_whitelist", {
+    platform: "string",
+    guildId: "string"
+  }, { primary: ["platform", "guildId"] });
 }
 __name(apply, "apply");
 var BLACKLIST_PLATFORM = "onebot";
@@ -114,6 +122,23 @@ async function setGroupBotStatus(ctx, platform, guildId, botEnabled) {
   await ctx.model.upsert("group_bot_status", [{ platform, guildId, botEnabled }]);
 }
 __name(setGroupBotStatus, "setGroupBotStatus");
+async function isInSmallGroupWhitelist(ctx, guildId) {
+  const records = await ctx.model.get("small_group_whitelist", { platform: BLACKLIST_PLATFORM, guildId });
+  return records.length > 0;
+}
+__name(isInSmallGroupWhitelist, "isInSmallGroupWhitelist");
+async function addToSmallGroupWhitelist(ctx, guildId) {
+  await ctx.model.upsert("small_group_whitelist", [{ platform: BLACKLIST_PLATFORM, guildId }]);
+}
+__name(addToSmallGroupWhitelist, "addToSmallGroupWhitelist");
+async function removeFromSmallGroupWhitelist(ctx, guildId) {
+  await ctx.model.remove("small_group_whitelist", { platform: BLACKLIST_PLATFORM, guildId });
+}
+__name(removeFromSmallGroupWhitelist, "removeFromSmallGroupWhitelist");
+async function getAllSmallGroupWhitelist(ctx) {
+  return await ctx.model.get("small_group_whitelist", { platform: BLACKLIST_PLATFORM });
+}
+__name(getAllSmallGroupWhitelist, "getAllSmallGroupWhitelist");
 
 // src/modules/basic.ts
 var basic_exports = {};
@@ -208,7 +233,10 @@ var ADMIN_COMMANDS = /* @__PURE__ */ new Set([
   "clear-blacklist",
   "approve",
   "reject",
-  "pending-invites"
+  "pending-invites",
+  "allow-small-group",
+  "disallow-small-group",
+  "view-small-group-whitelist"
 ]);
 
 // src/state.ts
@@ -263,8 +291,10 @@ function apply2(ctx, config) {
       }
     }
     if (config.basic.smallGroupAutoQuit) {
-      if (approvedGroups.has(guildId)) {
-        approvedGroups.delete(guildId);
+      const inWhitelist = await isInSmallGroupWhitelist(ctx, guildId);
+      const wasApproved = approvedGroups.has(guildId);
+      if (wasApproved) approvedGroups.delete(guildId);
+      if (inWhitelist || wasApproved) {
       } else {
         const delay = config.basic.smallGroupCheckDelay || 3e3;
         setTimeout(async () => {
@@ -684,6 +714,27 @@ function apply5(ctx, config) {
   }
   __name(clearBlacklist, "clearBlacklist");
   ctx.command("clear-blacklist", "清空黑名单", { authority: 4 }).action(clearBlacklist);
+  ctx.command("allow-small-group <groupId:text>", "解除指定群聊的小群人数限制", { authority: 4 }).action(async ({}, input) => {
+    const guildId = parseGuildId(input);
+    if (!guildId) return "输入格式错误，请输入群号。";
+    const exists = await isInSmallGroupWhitelist(ctx, guildId);
+    if (exists) return `群聊 ${guildId} 已在小群白名单中。`;
+    await addToSmallGroupWhitelist(ctx, guildId);
+    return `已将群聊 ${guildId} 加入小群白名单，该群不再受小群人数限制。`;
+  });
+  ctx.command("disallow-small-group <groupId:text>", "恢复指定群聊的小群人数限制", { authority: 4 }).action(async ({}, input) => {
+    const guildId = parseGuildId(input);
+    if (!guildId) return "输入格式错误，请输入群号。";
+    const exists = await isInSmallGroupWhitelist(ctx, guildId);
+    if (!exists) return `群聊 ${guildId} 不在小群白名单中。`;
+    await removeFromSmallGroupWhitelist(ctx, guildId);
+    return `已将群聊 ${guildId} 从小群白名单移除，该群将恢复小群人数限制。`;
+  });
+  ctx.command("view-small-group-whitelist", "查看小群白名单", { authority: 4 }).action(async () => {
+    const records = await getAllSmallGroupWhitelist(ctx);
+    if (records.length === 0) return "小群白名单为空。";
+    return "小群白名单列表（以下群不受小群人数限制）：\n" + records.map((r) => `- ${r.guildId}`).join("\n");
+  });
 }
 __name(apply5, "apply");
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-group-control",
   "description": "Koishi 插件，一个多功能的群聊自管理工具。支持被踢出自动拉黑、刷屏自动屏蔽、开关控制等功能。(仅支持 OneBot 适配器)",
-  "version": "0.2.10",
+  "version": "0.2.11",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [