koishi-plugin-starfx-bot 0.4.2 → 0.5.0

package/lib/index.d.ts

@@ -5,6 +5,7 @@ export declare let assetsDir: string;
 export interface Config {
     openLock: boolean;
     openSold: boolean;
+    bangdreamBorder: boolean;
     atNotSay: boolean;
     atNotSayOther: boolean;
     atNotSayProperty: number;
@@ -15,6 +16,7 @@ export interface Config {
 export declare const Config: Schema<Schemastery.ObjectS<{
     openLock: Schema<boolean, boolean>;
     openSold: Schema<boolean, boolean>;
+    bangdreamBorder: Schema<boolean, boolean>;
     atNotSay: Schema<boolean, boolean>;
     atNotSayProperty: Schema<number, number>;
     atNotSayOther: Schema<boolean, boolean>;
@@ -24,6 +26,7 @@ export declare const Config: Schema<Schemastery.ObjectS<{
 }>, {
     openLock: boolean;
     openSold: boolean;
+    bangdreamBorder: boolean;
     atNotSay: boolean;
     atNotSayProperty: number;
     atNotSayOther: boolean;
@@ -33,11 +36,18 @@ 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<any>;
-export declare function drawLock(ctx: Context, baseImage: string): Promise<h | "发生错误" | "输入无效">;
+export declare function getImageSrc(session: Session, param: string): Promise<string>;
+export declare function handleBanGDreamConfig(ctx: Context, options: any): Promise<{
+    color: string;
+    band: string;
+    starType: string;
+    starNum: number;
+    border: string;
+}>;
+export declare function drawLock(ctx: Context, baseImage: string): Promise<"发生错误" | "输入无效" | h>;
 /**
  * "卖掉了"绘图函数
  * @param ctx
  * @param baseImage
  */
-export declare function drawSold(ctx: Context, baseImage: string): Promise<h | "发生错误" | "输入无效">;
+export declare function drawSold(ctx: Context, baseImage: string): Promise<"发生错误" | "输入无效" | h>;

package/lib/index.js

@@ -47,6 +47,7 @@ __export(src_exports, {
   drawLock: () => drawLock,
   drawSold: () => drawSold,
   getImageSrc: () => getImageSrc,
+  handleBanGDreamConfig: () => handleBanGDreamConfig,
   name: () => name,
   usage: () => usage
 });
@@ -54,6 +55,7 @@ module.exports = __toCommonJS(src_exports);
 var import_koishi = require("koishi");
 var import_sharp = __toESM(require("sharp"));
 var fs = __toESM(require("fs"));
+var import_jimp = require("jimp");
 var name = "starfx-bot";
 var baseDir;
 var assetsDir;
@@ -61,6 +63,7 @@ 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!边框功能"),
     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("开启‘艾特他又不说话’功能"),
@@ -75,70 +78,47 @@ function apply(ctx, cfg) {
   ctx.i18n.define("zh-CN", require_zh_CN());
   baseDir = ctx.baseDir;
   assetsDir = `${ctx.baseDir}/data/starfx-bot/assets`;
-  initAssets("lock.png");
-  initAssets("sold.png");
+  initAssets();
   if (cfg.openLock) {
     ctx.command("封印 [param]").action(async ({ session }, param) => {
-      console.log("elements");
-      console.log(session.elements);
-      console.log("quote");
-      console.log(session.quote);
-      console.log("param");
-      console.log(param);
       return await drawLock(ctx, await getImageSrc(session, param));
     });
   }
   if (cfg.openSold) {
     ctx.command("卖掉了 [param]").action(async ({ session }, param) => {
-      console.log("elements");
-      console.log(session.elements);
-      console.log("quote");
-      console.log(session.quote);
-      console.log("param");
-      console.log(param);
       return await drawSold(ctx, await getImageSrc(session, param));
     });
   }
+  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) => {
+      session.send("图片处理中请稍等...");
+      console.log(param);
+      const drawConfig = await handleBanGDreamConfig(ctx, options);
+      const imageBase64 = await drawBanGDream(await getImageSrc(session, param), drawConfig);
+      return import_koishi.h.image(imageBase64);
+    });
+  }
   ctx.middleware(async (session, next) => {
     const elements = session.elements;
-    console.log(elements);
-    if (cfg.atNotSay || cfg.atNotSayOther) {
-      if (elements.length === 1 && elements[0].type === "at") {
-        if (elements[0].attrs.id === session.selfId) {
-          if (cfg.atNotSay) {
-            if (import_koishi.Random.bool(cfg.atNotSayProperty)) {
-              await session.send(session.text("middleware.messages.atNotReply"));
-            }
-          }
-        } else {
-          if (cfg.atNotSayOther) {
-            if (import_koishi.Random.bool(cfg.atNotSayOtherProperty)) {
-              await session.send(session.text("middleware.messages.atNotReplyOther"));
-            }
-          }
-        }
-      }
-    }
-    if (cfg.replyBot !== "关闭") {
-      const bots = ["bot", "机器人", "Bot", "BOT", "机器人！", "机器人!", "人机"];
-      const texts = elements?.filter((e) => e.type === "text").map((e) => e?.attrs?.content?.trim());
-      const ats = elements?.filter((e) => e.type === "at").map((e) => e?.attrs?.id);
-      const mentionedBot = texts?.some((t) => bots.includes(t));
-      const atMe = ats?.includes(session.selfId);
-      if (elements?.length === 1 && mentionedBot && cfg.replyBot === "无需at" || elements?.length === 2 && mentionedBot && atMe) {
-        await session.send(session.text("middleware.messages.notBot"));
-      }
-    }
-    if (cfg.iLoveYou) {
-      if (elements?.length === 2 && elements.some((e) => e.type === "at" && e?.attrs?.id === session.selfId) && elements.some((e) => e.type === "text" && e?.attrs?.content?.trim() === session.text("middleware.messages.loveMessage"))) {
-        await session.send(session.text("middleware.messages.iLoveU", {
-          at: import_koishi.h.at(session.userId),
-          quote: import_koishi.h.quote(session.messageId)
-        }));
-      }
-    }
+    await atNotSayReply(cfg, session, elements);
+    await replyBot(cfg, session, elements);
+    await iLoveYou(cfg, session, elements);
     return next();
   });
+  function initAssets() {
+    const fromUrl = `${__dirname}/../assets`;
+    assetsDir = `${ctx.baseDir}/data/starfx-bot/assets`;
+    if (!fs.existsSync(fromUrl)) return;
+    if (!fs.readdirSync(fromUrl)?.length) return;
+    if (!fs.existsSync(assetsDir)) {
+      fs.mkdirSync(assetsDir, { recursive: true });
+    }
+    fs.cpSync(fromUrl, assetsDir, { recursive: true, force: true });
+    if (process.env.NODE_ENV !== "development") {
+      fs.rmSync(fromUrl, { recursive: true });
+    }
+  }
+  __name(initAssets, "initAssets");
 }
 __name(apply, "apply");
 async function getImageSrc(session, param) {
@@ -168,14 +148,62 @@ async function getImageSrc(session, param) {
   return "";
 }
 __name(getImageSrc, "getImageSrc");
-function initAssets(fileName) {
-  if (fs.existsSync(`${__dirname}/../assets/${fileName}`)) {
-    fs.mkdirSync(`${assetsDir}/`, { recursive: true });
-    fs.copyFileSync(`${__dirname}/../assets/${fileName}`, `${assetsDir}/${fileName}`);
-    fs.rmSync(`${__dirname}/../assets/${fileName}`);
+async function handleBanGDreamConfig(ctx, options) {
+  const types = {
+    cool: ["cool", "blue", "蓝", "蓝色"],
+    powerful: ["powerful", "red", "红", "红色"],
+    peer: ["peer", "green", "绿", "绿色"],
+    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", "我去！！！！！", "我去!!!!!", "我去", "卖狗"]
+  };
+  const trains = {
+    "color_star": ["花后", "1", "彩", "true"],
+    "normal_star": ["花前", "0", "false"]
+  };
+  const drawConfig = {
+    color: "",
+    band: "",
+    starType: "",
+    starNum: 0,
+    border: ""
+  };
+  if (options.color) {
+    for (const [type, aliases] of Object.entries(types)) {
+      if (aliases.includes(options.color.toLowerCase())) {
+        drawConfig.color = type;
+        break;
+      }
+    }
   }
+  if (options.band) {
+    for (const [band, aliases] of Object.entries(bands)) {
+      if (aliases.includes(options.band.toLowerCase())) {
+        drawConfig.band = band;
+        break;
+      }
+    }
+  }
+  drawConfig.starNum = options.starNum ? parseInt(options.starNum) : 0;
+  if (drawConfig.starType) {
+    for (const [train, aliases] of Object.entries(trains)) {
+      if (aliases.includes(options.color.toLowerCase())) {
+        drawConfig.starType = train;
+        break;
+      }
+    }
+  }
+  return drawConfig;
 }
-__name(initAssets, "initAssets");
+__name(handleBanGDreamConfig, "handleBanGDreamConfig");
 async function drawLock(ctx, baseImage) {
   const image = await getImageFromUrl(ctx, baseImage);
   if (image === -1) {
@@ -245,6 +273,88 @@ async function getImageFromUrl(ctx, url) {
   return image;
 }
 __name(getImageFromUrl, "getImageFromUrl");
+async function atNotSayReply(cfg, session, elements) {
+  if ((cfg.atNotSay || cfg.atNotSayOther) && elements.length === 1 && elements[0].type === "at") {
+    const isAtSelf = elements[0].attrs.id === session.selfId;
+    if (isAtSelf && cfg.atNotSay && import_koishi.Random.bool(cfg.atNotSayProperty)) {
+      await session.send(session.text("middleware.messages.atNotReply"));
+    } else if (!isAtSelf && cfg.atNotSayOther && import_koishi.Random.bool(cfg.atNotSayOtherProperty)) {
+      await session.send(session.text("middleware.messages.atNotReplyOther"));
+    }
+  }
+}
+__name(atNotSayReply, "atNotSayReply");
+async function replyBot(cfg, session, elements) {
+  if (cfg.replyBot !== "关闭") {
+    const bots = ["bot", "机器人", "Bot", "BOT", "机器人！", "机器人!", "人机"];
+    const texts = elements?.filter((e) => e.type === "text").map((e) => e?.attrs?.content?.trim());
+    const ats = elements?.filter((e) => e.type === "at").map((e) => e?.attrs?.id);
+    const mentionedBot = texts?.some((t) => bots.includes(t));
+    const atMe = ats?.includes(session.selfId);
+    if (elements?.length === 1 && mentionedBot && cfg.replyBot === "无需at" || elements?.length === 2 && mentionedBot && atMe) {
+      await session.send(session.text("middleware.messages.notBot"));
+    }
+  }
+}
+__name(replyBot, "replyBot");
+async function iLoveYou(cfg, session, elements) {
+  if (cfg.iLoveYou) {
+    const hasAtMe = elements?.some((e) => e.type === "at" && e?.attrs?.id === session.selfId);
+    const hasLoveMessage = elements?.some(
+      (e) => e.type === "text" && e?.attrs?.content?.trim() === session.text("middleware.messages.loveMessage")
+    );
+    if (elements?.length === 2 && hasAtMe && hasLoveMessage) {
+      await session.send(session.text("middleware.messages.iLoveU", {
+        at: import_koishi.h.at(session.userId),
+        quote: import_koishi.h.quote(session.messageId)
+      }));
+    }
+  }
+}
+__name(iLoveYou, "iLoveYou");
+async function drawBanGDream(avatar, inputOptions) {
+  if (!avatar) {
+    return "";
+  }
+  const colors = ["cool", "pure", "happy", "powerful"];
+  const bands = ["ppp", "ag", "pp", "r", "hhw", "ras", "mnk", "go"];
+  const starTypes = ["normal_star", "color_star"];
+  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]),
+    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}` : ""}`;
+  const [image, colorImage, bandImage, starImage, borderImage] = await Promise.all([
+    import_jimp.Jimp.read(avatar),
+    import_jimp.Jimp.read(`${assetsDir}/bangborder/${options.color}.png`),
+    import_jimp.Jimp.read(`${assetsDir}/bangborder/${options.band}.png`),
+    import_jimp.Jimp.read(`${assetsDir}/bangborder/${options.starType}.png`),
+    import_jimp.Jimp.read(`${assetsDir}/bangborder/${options.border}.png`)
+  ]);
+  const zoom = 2;
+  image.cover({ w: 500 * zoom, h: 500 * zoom });
+  borderImage.cover({ w: 500 * zoom, h: 500 * zoom });
+  image.composite(borderImage);
+  colorImage.cover({ w: 130 * zoom, h: 130 * zoom });
+  image.composite(colorImage, image.width - colorImage.width - 3 * zoom, 5.5);
+  bandImage.width > bandImage.height ? bandImage.resize({ w: 120 * zoom }) : bandImage.resize({ h: 120 * zoom });
+  image.composite(bandImage, 15 * zoom, 15 * zoom);
+  starImage.resize({ w: 90 * zoom });
+  const step = 60 * zoom;
+  let hei = 410 * zoom;
+  let times = options.starNum;
+  while (times > 0) {
+    image.composite(starImage, 10 * zoom, hei);
+    hei -= step;
+    times--;
+  }
+  return `data:image/png;base64,${(await image.getBuffer("image/jpeg")).toString("base64")}`;
+}
+__name(drawBanGDream, "drawBanGDream");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   Config,
@@ -254,6 +364,7 @@ __name(getImageFromUrl, "getImageFromUrl");
   drawLock,
   drawSold,
   getImageSrc,
+  handleBanGDreamConfig,
   name,
   usage
 });

package/package.json

@@ -4,7 +4,7 @@
   "contributors": [
     "StarFreedomX <starfreedomx@outlook.com>"
   ],
-  "version": "0.4.2",
+  "version": "0.5.0",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [
@@ -36,6 +36,7 @@
     "koishi": "^4.18.7"
   },
   "dependencies": {
+    "jimp": "^1.6.0",
     "sharp": "^0.34.1"
   }
 }

package/readme.md

@@ -25,3 +25,4 @@ StarFreedomX机器人的小功能，自用
 | `0.3.6` | 添加"艾特我/他又不说话"的概率配置       |
 | `0.4.0` | 更改处理库为sharp              |
 | `0.4.1` | 修复sharp处理w>h的图报错的bug     |
+| `0.5.0` | 加入BanG Dream!边框功能        |