koishi-plugin-hupohuhu-tupian 1.0.1 → 1.0.6

package/lib/index.d.ts

@@ -3,7 +3,6 @@ export declare const name = "hupohuhu-tupian";
 export interface Config {
     endpoint: string;
     kohaku: string;
-    maxBatchSize: number;
 }
 export declare const Config: Schema<Config>;
 export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -29,19 +29,68 @@ var import_koishi = require("koishi");
 var name = "hupohuhu-tupian";
 var Config = import_koishi.Schema.object({
   endpoint: import_koishi.Schema.string().default(""),
-  kohaku: import_koishi.Schema.string().role("secret").default(""),
-  maxBatchSize: import_koishi.Schema.number().default(4)
+  kohaku: import_koishi.Schema.string().role("secret").default("")
 });
 function apply(ctx, config) {
-  ctx.command("draw <prompt:text>").alias("生图").option("width", "-W <w:number>").option("height", "-H <h:number>").option("steps", "-t <t:number>").option("scale", "-g <g:number>").option("seed", "-s <s:number>", { fallback: -1 }).option("batch", "-n <n:number>", { fallback: 1 }).option("negative", "-u <u:text>").option("enhance", "-e", { fallback: true }).action(async ({ options }, prompt) => {
-    if (!prompt) return "请输入提示词";
-    const { data } = await ctx.http("POST", config.endpoint, {
-      data: { ...options, prompt, kohaku: config.kohaku, batch_size: options.batch || 1 },
-      responseType: "arraybuffer"
-    });
-    if ((options.batch || 1) === 1) return import_koishi.h.image(data, "image/png");
-    const { images } = JSON.parse(new TextDecoder().decode(data));
-    return images.map((img) => import_koishi.h.image(img.base64));
+  ctx.command("hupo <prompt>").option("negative", "-n <val:string>").option("negative", "-N <val:string>").option("width", "-b <val:number>").option("width", "-B <val:number>").option("height", "-a <val:number>").option("height", "-A <val:number>").option("steps", "-i <val:number>").option("steps", "-I <val:number>").option("scale", "-f <val:number>").option("scale", "-F <val:number>").option("seed", "-s <val:number>").option("seed", "-S <val:number>").option("enhance", "-e").option("enhance", "-E").option("batch", "-x <val:number>").option("batch", "-X <val:number>").action(async ({ session, options }, prompt) => {
+    const cleanPrompt = prompt ? prompt.replace(/^["'“](.*)["'”]$/, "$1") : "";
+    if (!cleanPrompt) return "❌ 没词生个鬼，爬";
+    const check8 = /* @__PURE__ */ __name((val, name2) => {
+      if (val && val % 8 !== 0) {
+        const cut = Math.floor(val / 8) * 8;
+        throw new Error(`❌ ${name2}不是8的倍. 帮你一刀砍成 ${cut} 了。重新提交吧。`);
+      }
+      return val;
+    }, "check8");
+    try {
+      const finalPayload = {
+        prompt: cleanPrompt,
+        // P
+        negative_prompt: options.negative || "",
+        // N
+        width: check8(options.width, "宽度") || 1024,
+        // B
+        height: check8(options.height, "高度") || 1024,
+        // A
+        steps: options.steps || 20,
+        // I
+        guidance_scale: options.scale || 9,
+        // F
+        seed: options.seed || -1,
+        // S
+        enhancePrompt: !!options.enhance || session.content.includes("琥珀私照"),
+        // E
+        batch_size: options.batch || 1,
+        // X
+        kohaku: config.kohaku
+      };
+      const data = await ctx.http.post(config.endpoint, finalPayload, {
+        responseType: "arraybuffer"
+      });
+      const res = [
+        (0, import_koishi.h)("quote", { id: session.messageId }),
+        `✨ 已经生成参数为：
+P (提示词): ${finalPayload.prompt}
+N (负面): ${finalPayload.negative_prompt || "无"}
+B (宽度): ${finalPayload.width}
+A (高度): ${finalPayload.height}
+I (步数): ${finalPayload.steps}
+F (系数): ${finalPayload.guidance_scale}
+S (种子): ${finalPayload.seed}
+E (增强): ${finalPayload.enhancePrompt ? "ON" : "OFF"}
+的图片【${finalPayload.batch_size}】张`
+      ];
+      const view = new Uint8Array(data);
+      if (view[0] === 137) {
+        res.push(import_koishi.h.image(data, "image/png"));
+        return res;
+      }
+      const { images } = JSON.parse(Buffer.from(data).toString());
+      images.forEach((img) => res.push(import_koishi.h.image(img.base64)));
+      return res;
+    } catch (e) {
+      return e.message;
+    }
   });
 }
 __name(apply, "apply");

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-hupohuhu-tupian",
   "description": "kon!This is a customed image generate frontend",
-  "version": "1.0.1",
+  "version": "1.0.6",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [
@@ -11,12 +11,30 @@
   "license": "MIT",
   "scripts": {},
   "keywords": [
-    "chatbot",
+    "example",
     "koishi",
     "plugin"
   ],
+  "private": false,
   "devDependencies": {},
   "peerDependencies": {
-    "koishi": "^4.18.7"
+    "koishi": "^4.0.0"
+  },
+  "koishi": {
+    "description": {
+      "en": "this is a customed image generator ",
+      "zh": "这是一个高度定制化的生图前端"
+    },
+    "service": {
+      "required": [
+        "database"
+      ],
+      "optional": [
+        "assets"
+      ],
+      "implements": [
+        "dialogue"
+      ]
+    }
   }
 }