koishi-plugin-starfx-bot 0.5.0 → 0.6.0

package/lib/index.d.ts

@@ -1,7 +1,8 @@
-import { Context, h, Schema, Session } from 'koishi';
+import { Context, h, Logger, Schema, Session } from 'koishi';
 export declare const name = "starfx-bot";
 export declare let baseDir: string;
 export declare let assetsDir: string;
+export declare const starfxLogger: Logger;
 export interface Config {
     openLock: boolean;
     openSold: boolean;
@@ -10,6 +11,7 @@ export interface Config {
     atNotSayOther: boolean;
     atNotSayProperty: number;
     atNotSayOtherProperty: number;
+    record: boolean;
     replyBot: string;
     iLoveYou: boolean;
 }
@@ -17,6 +19,7 @@ export declare const Config: Schema<Schemastery.ObjectS<{
     openLock: Schema<boolean, boolean>;
     openSold: Schema<boolean, boolean>;
     bangdreamBorder: Schema<boolean, boolean>;
+    record: Schema<boolean, boolean>;
     atNotSay: Schema<boolean, boolean>;
     atNotSayProperty: Schema<number, number>;
     atNotSayOther: Schema<boolean, boolean>;
@@ -27,6 +30,7 @@ export declare const Config: Schema<Schemastery.ObjectS<{
     openLock: boolean;
     openSold: boolean;
     bangdreamBorder: boolean;
+    record: boolean;
     atNotSay: boolean;
     atNotSayProperty: number;
     atNotSayOther: boolean;
@@ -36,7 +40,15 @@ export declare const Config: Schema<Schemastery.ObjectS<{
 } & import("cosmokit").Dict>;
 export declare const usage = "<h5>StarFreedomX\u7684\u81EA\u7528\u63D2\u4EF6 \u653E\u4E86\u4E00\u4E9B\u5C0F\u529F\u80FD</h5>\n  ";
 export declare function apply(ctx: Context, cfg: Config): void;
-export declare function getImageSrc(session: Session, param: string): Promise<string>;
+export declare function addRecord(ctx: Context, gid: string, avatarUrl: string): Promise<string>;
+export declare function getRecord(gid: string): Promise<string>;
+export declare function getImageSrc(session: Session, param: string, option?: {
+    number?: boolean;
+    img?: boolean;
+    at?: boolean;
+    noParam?: boolean;
+    quote?: boolean;
+}): Promise<string>;
 export declare function handleBanGDreamConfig(ctx: Context, options: any): Promise<{
     color: string;
     band: string;

package/lib/index.js

@@ -41,14 +41,17 @@ var require_zh_CN = __commonJS({
 var src_exports = {};
 __export(src_exports, {
   Config: () => Config,
+  addRecord: () => addRecord,
   apply: () => apply,
   assetsDir: () => assetsDir,
   baseDir: () => baseDir,
   drawLock: () => drawLock,
   drawSold: () => drawSold,
   getImageSrc: () => getImageSrc,
+  getRecord: () => getRecord,
   handleBanGDreamConfig: () => handleBanGDreamConfig,
   name: () => name,
+  starfxLogger: () => starfxLogger,
   usage: () => usage
 });
 module.exports = __toCommonJS(src_exports);
@@ -56,14 +59,17 @@ var import_koishi = require("koishi");
 var import_sharp = __toESM(require("sharp"));
 var fs = __toESM(require("fs"));
 var import_jimp = require("jimp");
+var import_node_path = __toESM(require("node:path"));
 var name = "starfx-bot";
 var baseDir;
 var assetsDir;
+var starfxLogger = new import_koishi.Logger("starfx-bot");
 var Config = import_koishi.Schema.intersect([
   import_koishi.Schema.object({
     openLock: import_koishi.Schema.boolean().default(true).description("开启明日方舟封印功能"),
     openSold: import_koishi.Schema.boolean().default(true).description('开启闲鱼"卖掉了"功能'),
     bangdreamBorder: import_koishi.Schema.boolean().default(true).description("开启BanG Dream!边框功能"),
+    record: import_koishi.Schema.boolean().default(true).description("开启群语录功能"),
     atNotSay: import_koishi.Schema.boolean().default(true).description("开启‘艾特我又不说话’功能"),
     atNotSayProperty: import_koishi.Schema.number().role("slider").min(0).max(1).step(0.01).default(0.5).description("'艾特我又不说话'回复概率"),
     atNotSayOther: import_koishi.Schema.boolean().default(true).description("开启‘艾特他又不说话’功能"),
@@ -98,11 +104,36 @@ function apply(ctx, cfg) {
       return import_koishi.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 (!imageSrc) {
+        return "请发送带图片的指令消息或引用图片消息进行投稿";
+      }
+      return await addRecord(ctx, session.gid.replace(":", "_"), imageSrc);
+    });
+  }
   ctx.middleware(async (session, next) => {
     const elements = session.elements;
     await atNotSayReply(cfg, session, elements);
     await replyBot(cfg, session, elements);
     await iLoveYou(cfg, session, elements);
+    if (cfg.record && session.content?.startsWith("语录")) {
+      const filepath = await getRecord(session.gid.replace(":", "_"));
+      starfxLogger.info(`send record: ${filepath}`);
+      if (!filepath) return "暂无语录呢";
+      return import_koishi.h.image(filepath);
+    }
     return next();
   });
   function initAssets() {
@@ -121,27 +152,80 @@ function apply(ctx, cfg) {
   __name(initAssets, "initAssets");
 }
 __name(apply, "apply");
-async function getImageSrc(session, param) {
-  if (param?.length && param?.length === String(Number(param))?.length) {
+async function addRecord(ctx, gid, avatarUrl) {
+  const recordDir = `${assetsDir}/record/${gid}`;
+  const avatarBuffer = await ctx.http.get(avatarUrl, { responseType: "arraybuffer" });
+  saveImage(avatarBuffer, recordDir);
+  return "添加成功!";
+}
+__name(addRecord, "addRecord");
+async function getRecord(gid) {
+  const recordDir = `${assetsDir}/record/${gid}`;
+  const files = fs.readdirSync(recordDir);
+  return files?.length ? import_node_path.default.join(recordDir, import_koishi.Random.pick(files)) : null;
+}
+__name(getRecord, "getRecord");
+function getTodayPrefix() {
+  const now = /* @__PURE__ */ new Date();
+  const yyyy = now.getFullYear();
+  const mm = String(now.getMonth() + 1).padStart(2, "0");
+  const dd = String(now.getDate()).padStart(2, "0");
+  return `${yyyy}${mm}${dd}`;
+}
+__name(getTodayPrefix, "getTodayPrefix");
+function getNextSequenceNumber(directory, prefix) {
+  const files = fs.existsSync(directory) ? fs.readdirSync(directory) : [];
+  const regex = new RegExp(`^${prefix}-(\\d+)\\.jpg$`);
+  let maxNum = 0;
+  files.forEach((file) => {
+    const match = file.match(regex);
+    if (match) {
+      const num = parseInt(match[1], 10);
+      if (num > maxNum) {
+        maxNum = num;
+      }
+    }
+  });
+  return maxNum + 1;
+}
+__name(getNextSequenceNumber, "getNextSequenceNumber");
+function saveImage(arrayBuffer, directory) {
+  const prefix = getTodayPrefix();
+  const seq = getNextSequenceNumber(directory, prefix);
+  const filename = `${prefix}-${seq}.jpg`;
+  const filepath = import_node_path.default.join(directory, filename);
+  if (!fs.existsSync(directory)) {
+    fs.mkdirSync(directory, { recursive: true });
+  }
+  const buffer = Buffer.from(arrayBuffer);
+  fs.writeFileSync(filepath, buffer);
+  console.log(`已保存图片：${filepath}`);
+}
+__name(saveImage, "saveImage");
+async function getImageSrc(session, param, option) {
+  const number = option?.number ?? true, img = option?.img ?? true, at = option?.at ?? true, noParam = option?.noParam ?? true, quote = option?.quote ?? true;
+  if (number && param?.length && param?.length === String(Number(param))?.length) {
     return `https://q1.qlogo.cn/g?b=qq&nk=${param}&s=640`;
-  } else if (!param?.length) {
+  } else if (noParam && !param?.length) {
     return `https://q1.qlogo.cn/g?b=qq&nk=${session.userId}&s=640`;
   }
   const elementArray = session.elements;
   for (const element of elementArray) {
-    if (element?.type === "img") {
+    if (img && element?.type === "img") {
       return element?.attrs?.src;
-    } else if (element?.type === "at" && element?.attrs?.id) {
+    } else if (at && element?.type === "at" && element?.attrs?.id) {
       return `https://q1.qlogo.cn/g?b=qq&nk=${element.attrs.id}&s=640`;
     }
   }
-  const quoteElementArray = session?.quote?.elements;
-  if (quoteElementArray?.length) {
-    for (const element of quoteElementArray) {
-      if (element?.type === "img") {
-        return element?.attrs?.src;
-      } else if (element?.type === "at" && element?.attrs?.id && element.attrs.id !== session.selfId) {
-        return `https://q1.qlogo.cn/g?b=qq&nk=${element?.attrs?.id}&s=640`;
+  if (quote) {
+    const quoteElementArray = session?.quote?.elements;
+    if (quoteElementArray?.length) {
+      for (const element of quoteElementArray) {
+        if (img && element?.type === "img") {
+          return element?.attrs?.src;
+        } else if (at && element?.type === "at" && element?.attrs?.id && element.attrs.id !== session.selfId) {
+          return `https://q1.qlogo.cn/g?b=qq&nk=${element?.attrs?.id}&s=640`;
+        }
       }
     }
   }
@@ -156,14 +240,14 @@ async function handleBanGDreamConfig(ctx, options) {
     happy: ["happy", "orange", "橙", "橙色"]
   };
   const bands = {
-    ppp: ["ppp", "Poppin'Party", "破琵琶", "步品破茶", "poppin", "popipa", "poppinparty", "ポピパ"],
-    ag: ["ag", "Afterglow", "夕阳红", "悪蓋愚狼"],
-    pp: ["pp", "Pastel＊Palettes", "Pastel*Palettes", "PastelPalettes", "怕死怕累", "pastel", "palettes", "pasupare", "パスパレ", "破巣照破烈斗"],
-    hhw: ["hhw", "ハロー、ハッピーワールド！", "Hello,HappyWorld!", "helloHappyWorld", "ハロハピ", "Hello，HappyWorld！", "harohapi", "破狼法被威悪怒", "儿歌团", "好好玩"],
-    r: ["r", "Roselia", "露世里恶", "萝", "露世裏悪", "ロゼリア", "r组", "相声团", "相声组"],
-    ras: ["ras", "RaiseASuilen", "raise", "suilen", "ラス", "零図悪酔恋", "睡莲", "麗厨唖睡蓮", "睡蓮"],
-    mnk: ["mnk", "モニカ", "蝶团", "蝶", "Morfonica", "毛二力", "monika", "monica"],
-    go: ["go", "MyGO!!!!!", "MyGO！！！！！", "mygo", "我去！！！！！", "我去!!!!!", "我去", "卖狗"]
+    ppp: ["ppp", "poppin'Party", "破琵琶", "步品破茶", "poppin", "popipa", "poppinparty", "ポピパ"],
+    ag: ["ag", "afterglow", "夕阳红", "悪蓋愚狼"],
+    pp: ["pp", "pastel＊palettes", "pastel*palettes", "pastelPalettes", "怕死怕累", "pastel", "palettes", "pasupare", "パスパレ", "破巣照破烈斗"],
+    hhw: ["hhw", "ハロー、ハッピーワールド！", "hello,happyworld!", "hellohappyworld", "ハロハピ", "hello，happyworld！", "harohapi", "破狼法被威悪怒", "儿歌团", "好好玩"],
+    r: ["r", "roselia", "露世里恶", "萝", "露世裏悪", "ロゼリア", "r组", "相声团", "相声组"],
+    ras: ["ras", "raiseasuilen", "raise", "suilen", "ラス", "零図悪酔恋", "睡莲", "麗厨唖睡蓮", "睡蓮"],
+    mnk: ["mnk", "モニカ", "蝶团", "蝶", "morfonica", "毛二力", "monika", "monica"],
+    go: ["go", "mygo!!!!!", "mygo！！！！！", "mygo", "我去！！！！！", "我去!!!!!", "我去", "卖狗"]
   };
   const trains = {
     "color_star": ["花后", "1", "彩", "true"],
@@ -192,10 +276,11 @@ async function handleBanGDreamConfig(ctx, options) {
       }
     }
   }
-  drawConfig.starNum = options.starNum ? parseInt(options.starNum) : 0;
-  if (drawConfig.starType) {
+  const starNum = options.starNum ? parseInt(options.starNum) : 0;
+  drawConfig.starNum = starNum > 0 && starNum < 10 ? starNum : 0;
+  if (options.train) {
     for (const [train, aliases] of Object.entries(trains)) {
-      if (aliases.includes(options.color.toLowerCase())) {
+      if (aliases.includes(options.train.toLowerCase())) {
         drawConfig.starType = train;
         break;
       }
@@ -319,15 +404,16 @@ async function drawBanGDream(avatar, inputOptions) {
   const colors = ["cool", "pure", "happy", "powerful"];
   const bands = ["ppp", "ag", "pp", "r", "hhw", "ras", "mnk", "go"];
   const starTypes = ["normal_star", "color_star"];
+  const starNums = [1, 2, 3, 4, 5];
   const options = {
     color: inputOptions.color || import_koishi.Random.pick(colors),
     band: inputOptions.band || import_koishi.Random.pick(bands),
-    starNum: inputOptions.starNum || import_koishi.Random.pick([1, 2, 3, 4, 5]),
+    starNum: inputOptions.starNum || import_koishi.Random.pick(starNums),
     starType: inputOptions.starType || "",
     border: inputOptions.border || ""
   };
   options.starType ||= options.starNum < 3 ? starTypes[0] : import_koishi.Random.pick(starTypes);
-  options.border ||= `card-${options.starNum}${options.starNum == 1 ? `-${options.color}` : ""}`;
+  options.border ||= `card-${starNums.includes(options.starNum) ? options.starNum : 5}${options.starNum == 1 ? `-${options.color}` : ""}`;
   const [image, colorImage, bandImage, starImage, borderImage] = await Promise.all([
     import_jimp.Jimp.read(avatar),
     import_jimp.Jimp.read(`${assetsDir}/bangborder/${options.color}.png`),
@@ -358,13 +444,16 @@ __name(drawBanGDream, "drawBanGDream");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   Config,
+  addRecord,
   apply,
   assetsDir,
   baseDir,
   drawLock,
   drawSold,
   getImageSrc,
+  getRecord,
   handleBanGDreamConfig,
   name,
+  starfxLogger,
   usage
 });

package/package.json

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

package/readme.md

@@ -13,16 +13,18 @@ StarFreedomX机器人的小功能，自用
 * 我也喜欢你！
 
 # 更新日志
-| 版本      | 更新日志                     |
-|---------|--------------------------|
-| `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.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 |