koishi-plugin-starfx-bot 0.8.0 → 0.10.0

package/lib/index.d.ts

@@ -8,48 +8,61 @@ export interface Config {
     openLock: boolean;
     openSold: boolean;
     bangdreamBorder: boolean;
+    record: boolean;
+    roll: boolean;
     atNotSay: boolean;
-    atNotSayOther: boolean;
     atNotSayProperty: number;
+    atNotSayOther: boolean;
     atNotSayOtherProperty: number;
-    record: boolean;
+    iLoveYou: boolean;
+    replyBot: string;
     openRepeat: boolean;
     minRepeatTimes: number;
     repeatPossibility: number;
-    saveArchive: boolean;
-    replyBot: string;
-    iLoveYou: boolean;
+    featureControl: string;
 }
 export declare const Config: Schema<Schemastery.ObjectS<{
     openLock: Schema<boolean, boolean>;
     openSold: Schema<boolean, boolean>;
     bangdreamBorder: Schema<boolean, boolean>;
+}> | Schemastery.ObjectS<{
     record: Schema<boolean, boolean>;
-    saveArchive: Schema<boolean, boolean>;
+}> | Schemastery.ObjectS<{
+    roll: Schema<boolean, boolean>;
+}> | Schemastery.ObjectS<{
     atNotSay: Schema<boolean, boolean>;
     atNotSayProperty: Schema<number, number>;
     atNotSayOther: Schema<boolean, boolean>;
     atNotSayOtherProperty: Schema<number, number>;
+    iLoveYou: Schema<boolean, boolean>;
+    replyBot: Schema<"关闭" | "无需at" | "必须at", "关闭" | "无需at" | "必须at">;
+}> | Schemastery.ObjectS<{
     openRepeat: Schema<boolean, boolean>;
     minRepeatTimes: Schema<number, number>;
     repeatPossibility: Schema<number, number>;
-    iLoveYou: Schema<boolean, boolean>;
-    replyBot: Schema<"关闭" | "无需at" | "必须at", "关闭" | "无需at" | "必须at">;
+}> | Schemastery.ObjectS<{
+    featureControl: Schema<string, string>;
 }>, {
     openLock: boolean;
     openSold: boolean;
     bangdreamBorder: boolean;
+} & import("cosmokit").Dict & {
     record: boolean;
-    saveArchive: boolean;
+} & {
+    roll: boolean;
+} & {
     atNotSay: boolean;
     atNotSayProperty: number;
     atNotSayOther: boolean;
     atNotSayOtherProperty: number;
+    iLoveYou: boolean;
+    replyBot: "关闭" | "无需at" | "必须at";
+} & {
     openRepeat: boolean;
     minRepeatTimes: number;
     repeatPossibility: number;
-    iLoveYou: boolean;
-    replyBot: "关闭" | "无需at" | "必须at";
-} & import("cosmokit").Dict>;
-export declare const usage = "<h5>StarFreedomX\u7684\u81EA\u7528\u63D2\u4EF6 \u653E\u4E86\u4E00\u4E9B\u5C0F\u529F\u80FD</h5>\n  ";
+} & {
+    featureControl: string;
+}>;
+export declare const usage = "<h2>StarFreedomX\u7684\u81EA\u7528\u63D2\u4EF6 \u653E\u4E86\u4E00\u4E9B\u5C0F\u529F\u80FD</h2>\n  ";
 export declare function apply(ctx: Context, cfg: Config): void;

package/lib/index.js

@@ -33,7 +33,7 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/locales/zh-CN.yml
 var require_zh_CN = __commonJS({
   "src/locales/zh-CN.yml"(exports2, module2) {
-    module2.exports = { commands: null, middleware: { messages: { atNotReply: "艾特我又不说话", atNotReplyOther: "艾特他又不说话", notBot: "我才不是机器人！", loveMessage: "我喜欢你", iLoveU: "我也喜欢你" } } };
+    module2.exports = { commands: { roll: { messages: { noParam: "你问啥?", invalid: "你要不看看你说的是啥", "too-many": "太、太多了啦!", noodles: "扔出了{num}个{noodles}面的骰子，点数如下:{points}", possibility: "{param}是{possibility}%", choose: "当然是{option}啦!" } } }, middleware: { messages: { atNotReply: "艾特我又不说话", atNotReplyOther: "艾特他又不说话", notBot: "我才不是机器人！", loveMessage: "我喜欢你", iLoveU: "我也喜欢你" } } };
   }
 });
 
@@ -350,6 +350,77 @@ async function drawBanGDream(avatar, inputOptions) {
   return `data:image/png;base64,${(await image.getBuffer("image/jpeg")).toString("base64")}`;
 }
 __name(drawBanGDream, "drawBanGDream");
+function parseJsonControl(text) {
+  try {
+    return JSON.parse(text);
+  } catch (e) {
+    starfxLogger.warn("[功能控制] JSON 解析失败");
+    return null;
+  }
+}
+__name(parseJsonControl, "parseJsonControl");
+function detectControl(controlJson, guildId, funName) {
+  const rule = controlJson?.[funName];
+  if (!rule || rule.whitelist === void 0 || !Array.isArray(rule.groups)) {
+    return true;
+  }
+  const inList = rule.groups.includes(Number(guildId));
+  return rule.whitelist ? inList : !inList;
+}
+__name(detectControl, "detectControl");
+function handleRoll(session) {
+  const elements = session.elements;
+  let parts = [];
+  for (const element of elements) {
+    if (element?.type === "text") {
+      parts.push(...element.attrs.content.split(/(?:\s+)+/).filter(Boolean));
+    } else {
+      parts.push(element);
+    }
+  }
+  const commandLength = parts[0].length;
+  console.log(parts);
+  parts.shift();
+  if (!parts) return session.text(".noParam");
+  const last = session.elements[session.elements.length - 1];
+  if (last?.type === "text" && last?.attrs?.content?.endsWith("的概率") && last?.attrs?.content?.length > 3) {
+    return session.text(".possibility", {
+      param: parts,
+      possibility: Math.floor(Math.random() * 1e4 + 1) / 100
+    });
+  }
+  const items = parts.join(" ").split("r");
+  if (items.length === 2) {
+    const [num, noodles] = items.map(Number);
+    return getPoints(session, num, noodles);
+  }
+  const newParts = [];
+  parts.forEach((element) => {
+    if (typeof element === "string") {
+      newParts.push(...element.split(/(?:、|还是|，|,)+/).filter(Boolean));
+    } else {
+      newParts.push(element);
+    }
+  });
+  if (newParts.length > 1) {
+    return session.text(".choose", {
+      option: import_koishi.Random.pick(newParts)
+    });
+  }
+  return session.text(".noParam");
+}
+__name(handleRoll, "handleRoll");
+function getPoints(session, num, noodles) {
+  if (isNaN(num) || isNaN(noodles)) return session.text(".invalid");
+  if (num > 20 || noodles > 1e8) return session.text(".too-many");
+  const points = Array(num).fill(0).map(() => Math.floor(Math.random() * noodles + 1));
+  return session.text(".noodles", {
+    num,
+    noodles,
+    points: points.join(", ")
+  });
+}
+__name(getPoints, "getPoints");
 
 // src/index.ts
 var name = "starfx-bot";
@@ -361,93 +432,104 @@ var Config2 = import_koishi2.Schema.intersect([
   import_koishi2.Schema.object({
     openLock: import_koishi2.Schema.boolean().default(true).description("开启明日方舟封印功能"),
     openSold: import_koishi2.Schema.boolean().default(true).description('开启闲鱼"卖掉了"功能'),
-    bangdreamBorder: import_koishi2.Schema.boolean().default(true).description("开启BanG Dream!边框功能"),
-    record: import_koishi2.Schema.boolean().default(true).description("开启群语录功能"),
-    saveArchive: import_koishi2.Schema.boolean().default(false).description("开启入典功能").hidden(),
+    bangdreamBorder: import_koishi2.Schema.boolean().default(true).description("开启BanG Dream!边框功能")
+  }).description("绘图功能"),
+  import_koishi2.Schema.object({
+    record: import_koishi2.Schema.boolean().default(true).description("开启群语录功能")
+    //saveArchive: Schema.boolean().default(false).description('开启入典功能').hidden(),
+  }).description("语录记录功能"),
+  import_koishi2.Schema.object({
+    roll: import_koishi2.Schema.boolean().default(true).description("开启roll随机数功能")
+  }).description("指令小功能"),
+  import_koishi2.Schema.object({
     atNotSay: import_koishi2.Schema.boolean().default(true).description("开启‘艾特我又不说话’功能"),
     atNotSayProperty: import_koishi2.Schema.number().role("slider").min(0).max(1).step(0.01).default(0.5).description("'艾特我又不说话'回复概率"),
     atNotSayOther: import_koishi2.Schema.boolean().default(true).description("开启‘艾特他又不说话’功能"),
     atNotSayOtherProperty: import_koishi2.Schema.number().role("slider").min(0).max(1).step(0.01).default(0.5).description("'艾特他又不说话'回复概率"),
-    openRepeat: import_koishi2.Schema.boolean().default(true).description("开启复读功能"),
-    minRepeatTimes: import_koishi2.Schema.number().default(2).description("最少重复次数"),
-    repeatPossibility: import_koishi2.Schema.number().role("slider").min(0).max(1).step(0.01).default(0.3).description("复读发生概率"),
     iLoveYou: import_koishi2.Schema.boolean().default(true).description("开启‘我喜欢你’功能"),
     replyBot: import_koishi2.Schema.union(["关闭", "无需at", "必须at"]).default("无需at").description("回复‘我才不是机器人！’功能")
-  })
+  }).description("特定回应功能"),
+  import_koishi2.Schema.object({
+    openRepeat: import_koishi2.Schema.boolean().default(true).description("开启复读功能"),
+    minRepeatTimes: import_koishi2.Schema.number().default(2).description("最少重复次数"),
+    repeatPossibility: import_koishi2.Schema.number().role("slider").min(0).max(1).step(0.01).default(0.3).description("复读发生概率")
+  }).description("复读功能"),
+  import_koishi2.Schema.object({
+    featureControl: import_koishi2.Schema.string().role("textarea", { rows: [15] }).default("{\n\n}").description(`黑/白名单配置，语法为JSON格式(可以不缩进)，<br>
+可配置功能键及语法详见 [项目地址](https://github.com/StarFreedomX/starfx-bot)或[npm发布页](https://www.npmjs.com/package/koishi-plugin-bangdream-ccg)`)
+  }).description("高级配置")
 ]);
-var usage = `<h5>StarFreedomX的自用插件 放了一些小功能</h5>
+var usage = `<h2>StarFreedomX的自用插件 放了一些小功能</h2>
   `;
 function apply(ctx, cfg) {
   ctx.i18n.define("zh-CN", require_zh_CN());
   baseDir = ctx.baseDir;
   assetsDir = `${ctx.baseDir}/data/starfx-bot/assets`;
   initAssets();
+  const controlJson = parseJsonControl(cfg.featureControl);
   if (cfg.openLock) {
     ctx.command("封印 [param]").action(async ({ session }, param) => {
-      return await drawLock(ctx, await getImageSrc(session, param));
+      if (detectControl(controlJson, session.guildId, "lock"))
+        return await drawLock(ctx, await getImageSrc(session, param));
     });
   }
   if (cfg.openSold) {
     ctx.command("卖掉了 [param]").action(async ({ session }, param) => {
-      return await drawSold(ctx, await getImageSrc(session, param));
+      if (detectControl(controlJson, session.guildId, "sold"))
+        return await drawSold(ctx, await getImageSrc(session, param));
+    });
+  }
+  if (cfg.roll) {
+    ctx.command("roll").action(async ({ session }) => {
+      if (detectControl(controlJson, session.guildId, "roll")) {
+        return handleRoll(session);
+      }
     });
   }
   if (cfg.bangdreamBorder) {
-    ctx.command("bangdreamborder [param]").alias("bdbd").option("starNum", "-n <starNum: number>").option("color", "-c <color: string>").option("train", "-t <train: string>").option("band", "-b <band: string>").action(async ({ session, options }, param) => {
-      const p = session.send("图片处理中请稍等...");
-      console.log(param);
-      const drawConfig = await handleBanGDreamConfig(options);
-      const imageBase64 = await drawBanGDream(await getImageSrc(session, param), drawConfig);
-      await p;
-      return import_koishi2.h.image(imageBase64);
+    ctx.command("bdbd [param]").option("starNum", "-n <starNum: number>").option("color", "-c <color: string>").option("train", "-t <train: string>").option("band", "-b <band: string>").action(async ({ session, options }, param) => {
+      if (detectControl(controlJson, session.guildId, "bdbd")) {
+        const p = session.send("图片处理中请稍等...");
+        console.log(param);
+        const drawConfig = await handleBanGDreamConfig(options);
+        const imageBase64 = await drawBanGDream(await getImageSrc(session, param), drawConfig);
+        await p;
+        return import_koishi2.h.image(imageBase64);
+      }
     });
   }
   if (cfg.record) {
     ctx.command("投稿 [param]").action(async ({ session }, param) => {
-      const imageSrc = await getImageSrc(
-        session,
-        param,
-        {
-          img: true,
-          at: false,
-          quote: true,
-          noParam: false,
-          number: false
+      if (detectControl(controlJson, session.guildId, "record")) {
+        const imageSrc = await getImageSrc(
+          session,
+          param,
+          {
+            img: true,
+            at: false,
+            quote: true,
+            noParam: false,
+            number: false
+          }
+        );
+        if (!imageSrc) {
+          return "请发送带图片的指令消息或引用图片消息进行投稿";
         }
-      );
-      if (!imageSrc) {
-        return "请发送带图片的指令消息或引用图片消息进行投稿";
+        return await addRecord(ctx, session.gid.replace(":", "_"), imageSrc);
       }
-      return await addRecord(ctx, session.gid.replace(":", "_"), imageSrc);
     });
     ctx.command("语录").action(async ({ session }) => {
-      const filepath = await getRecord(session.gid.replace(":", "_"));
-      starfxLogger.info(`send record: ${filepath}`);
-      if (!filepath) return "暂无语录呢";
-      return import_koishi2.h.image(filepath);
-    });
-  }
-  if (cfg.saveArchive) {
-    ctx.command("入典 [param]").action(async ({ session }, param) => {
-      const imageSrc = await getImageSrc(
-        session,
-        param,
-        {
-          img: true,
-          at: false,
-          quote: true,
-          noParam: false,
-          number: false
-        }
-      );
-      if (!imageSrc) {
-        return "请发送带图片的指令消息或引用图片消息进行投稿";
+      if (detectControl(controlJson, session.guildId, "record")) {
+        const filepath = await getRecord(session.gid.replace(":", "_"));
+        starfxLogger.info(`send record: ${filepath}`);
+        if (!filepath) return "暂无语录呢";
+        return import_koishi2.h.image(filepath);
       }
     });
   }
   ctx.middleware(async (session, next) => {
     const elements = session.elements;
-    if (cfg.openRepeat) {
+    if (cfg.openRepeat && detectControl(controlJson, session.guildId, "repeat")) {
       const content = session.content;
       const ctxArr = repeatContextMap.get(session.gid);
       if (!ctxArr || ctxArr[0] !== content) {
@@ -459,9 +541,12 @@ function apply(ctx, cfg) {
         }
       }
     }
-    await atNotSayReply(cfg, session, elements);
-    await replyBot(cfg, session, elements);
-    await iLoveYou(cfg, session, elements);
+    if (detectControl(controlJson, session.guildId, "atNotSay"))
+      await atNotSayReply(cfg, session, elements);
+    if (detectControl(controlJson, session.guildId, "replyBot"))
+      await replyBot(cfg, session, elements);
+    if (detectControl(controlJson, session.guildId, "iLoveYou"))
+      await iLoveYou(cfg, session, elements);
     return next();
   });
   function initAssets() {

package/lib/utils.d.ts

@@ -1,6 +1,12 @@
 import { Context, h, Session } from "koishi";
 import sharp from "sharp";
 import { Config } from "./index";
+interface FeatureControl {
+    [feature: string]: {
+        whitelist: boolean;
+        groups: number[];
+    };
+}
 /**
  * 添加投稿
  * @param ctx Context
@@ -120,3 +126,7 @@ export declare function drawBanGDream(avatar: string, inputOptions?: {
     starNum: number;
     border: string;
 }): Promise<string>;
+export declare function parseJsonControl(text: string): FeatureControl | null;
+export declare function detectControl(controlJson: FeatureControl, guildId: string, funName: string): boolean;
+export declare function handleRoll(session: Session): string;
+export {};

package/package.json

@@ -4,7 +4,7 @@
   "contributors": [
     "StarFreedomX <starfreedomx@outlook.com>"
   ],
-  "version": "0.8.0",
+  "version": "0.10.0",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [

package/readme.md

@@ -4,7 +4,7 @@
 
 StarFreedomX机器人的小功能，自用
 
-# 已加入的功能列表：
+## 已加入的功能列表：
 
 * 明日方舟封印功能
 * 闲鱼“卖掉了”功能
@@ -15,27 +15,70 @@ StarFreedomX机器人的小功能，自用
 * 群语录
 * 复读
 
-# List to Do
+## List to Do
 
 * 为每种功能添加单独的群聊控制开关
 * 修改BanG Dream!边框的绘制为使用sharp，加快速度
 * 增加入典功能
 
-# 更新日志
-| 版本      | 更新日志                          |
-|---------|-------------------------------|
-| `0.0.1` | 加入封印功能                        |
-| `0.1.0` | 加入"卖掉了"功能                     |
-| `0.2.0` | 加入"艾特我又不说话"回复功能               |
-| `0.3.0` | 加入"我喜欢你"和"我才不是机器人！"回复功能       |
-| `0.3.3` | 修订添加"我才不是机器人"是否需要at           |
-| `0.3.4` | 分离"艾特我/他又不说话"的两个选项为单独的开关      |
-| `0.3.5` | 修复丢失的assets                   |
-| `0.3.6` | 添加"艾特我/他又不说话"的概率配置            |
-| `0.4.0` | 更改处理库为sharp                   |
-| `0.4.1` | 修复sharp处理w>h的图报错的bug          |
-| `0.5.0` | 加入BanG Dream!边框功能             |
-| `0.5.1` | 对边框的参数做了限定，修复了异常输入导致的报错       |
-| `0.6.0` | 加入群语录功能，修复一处bangdream绘制的bug   |
-| `0.7.0` | 加入复读功能                        |
-| `0.8.0` | 将各个工具函数封装在utils.ts中，更改语录的触发方式 |
+
+## 🔧 功能权限控制（可选）
+
+本插件支持为各个功能设置 **群聊白名单 / 黑名单**，用于控制不同功能在指定群聊中是否启用。
+
+配置格式为 JSON
+
+
+- `whitelist: true`：启用白名单模式，仅允许列出的群使用该功能
+- `whitelist: false`：启用黑名单模式，禁止列出的群使用该功能
+- `groups`：群号数组，必须为数字
+
+若未配置某功能项，则默认所有群均可使用。
+
+---
+
+### 📌 可配置功能键一览
+
+| 键名         | 功能说明                                   |
+|------------|----------------------------------------|
+| `lock`     | 明日方舟封印功能（对应 `openLock`）                |
+| `sold`     | 闲鱼“卖掉了”功能（对应 `openSold`）               |
+| `repeat`   | 群复读功能（对应 `openRepeat`）                 |
+| `record`   | 群语录功能（对应 `投稿、语录`）                      |
+| `atNotSay` | “艾特我/他又不说话”系列功能                        |
+| `replyBot` | “我才不是机器人！”系列功能                         |
+| `iLoveYou` | “我也喜欢你”系列功能                            |
+| `bdbd`     | BanG Dream! 边框功能（对应 `bangdreamBorder`） |
+| `roll`     | 随机数功能                                  |
+
+---
+
+### 🧪 示例：仅允许特定群使用 `roll`，禁止某群使用 `sold`
+
+```json
+{
+  "roll": {
+    "whitelist": true,
+    "groups": [123456789,114514191]
+  },
+  "sold": {
+    "whitelist": false,
+    "groups": [987654321]
+  }
+}
+```
+
+## 更新日志
+| 版本       | 更新日志                          |
+|----------|-------------------------------|
+| `0.0.1`  | 加入封印功能                        |
+| `0.1.0`  | 加入"卖掉了"功能                     |
+| `0.2.0`  | 加入"艾特我又不说话"回复功能               |
+| `0.3.0`  | 加入"我喜欢你"和"我才不是机器人！"回复功能       |
+| `0.4.0`  | 更改处理库为sharp                   |
+| `0.5.0`  | 加入BanG Dream!边框功能             |
+| `0.6.0`  | 加入群语录功能，修复一处bangdream绘制的bug   |
+| `0.7.0`  | 加入复读功能                        |
+| `0.8.0`  | 将各个工具函数封装在utils.ts中，更改语录的触发方式 |
+| `0.9.0`  | 添加白名单/黑名单模式，对每个功能做过滤          |
+| `0.10.0` | 添加随机数功能                       |