koishi-plugin-smmcat-gptimage2 0.0.1

package/lib/chatActive.d.ts

@@ -0,0 +1,10 @@
+import { Context } from "koishi";
+import { Config } from ".";
+/** 内容发送模块 */
+export declare const ChatOfActive: {
+    config: Config;
+    ctx: Context;
+    init(config: Config, ctx: Context): Promise<void>;
+    /** 发送消息 */
+    send(guildId: string, content: string): Promise<void>;
+};

package/lib/chatPassive.d.ts

@@ -0,0 +1,17 @@
+import { Context, Session } from "koishi";
+import { Config } from ".";
+/** 内容发送模块 */
+export declare const ChatOfPassive: {
+    config: Config;
+    ctx: Context;
+    chache: {};
+    messageIdTemp: {};
+    init(config: Config, ctx: Context): Promise<void>;
+    /** 发送消息 */
+    send(session: Session, content: string): Promise<void>;
+    isUseMd(session: Session): boolean;
+    /** 清除一条 msgid 记录 */
+    clearChache(): void;
+    /** 清除一条 messageId 缓存 */
+    clearMessageIdTemp(): void;
+};

package/lib/index.d.ts

@@ -0,0 +1,10 @@
+import { Context, Schema } from 'koishi';
+export declare const name = "smmcat-gptimage2";
+export interface Config {
+    deBug: boolean;
+    useMd: boolean;
+    token: string;
+    onlyUse: string[];
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -0,0 +1,121 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __export = (target, all) => {
+  for (var name2 in all)
+    __defProp(target, name2, { get: all[name2], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var src_exports = {};
+__export(src_exports, {
+  Config: () => Config,
+  apply: () => apply,
+  name: () => name
+});
+module.exports = __toCommonJS(src_exports);
+var import_koishi = require("koishi");
+var name = "smmcat-gptimage2";
+var Config = import_koishi.Schema.object({
+  deBug: import_koishi.Schema.boolean().default(false).description("日志查看更多信息"),
+  useMd: import_koishi.Schema.boolean().default(false).description("使用qqbot的md发送（未实装）"),
+  token: import_koishi.Schema.string().default("").description("[万象灵算](https://wxls.bbroot.com/sign-up?aff=8FJR) 提供的 API 密钥"),
+  onlyUse: import_koishi.Schema.array(String).role("table").default([]).description("仅指定userId用户可用（为空则所有）")
+});
+function apply(ctx, config) {
+  function sleep(ms = 1e3) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+  }
+  __name(sleep, "sleep");
+  const ImgGpt = {
+    userCache: {},
+    async startGetImage(session, content, img = null, fn) {
+      const { userId } = session;
+      if (!ImgGpt.userCache[userId]) {
+        ImgGpt.userCache[userId] = {
+          progress: null,
+          lastResult: null
+        };
+      }
+      try {
+        const body = {
+          model: "gpt-image-2",
+          prompt: content,
+          aspect_ratio: "16:9",
+          response_format: "url",
+          n: 1,
+          images: img || void 0
+        };
+        config.deBug && console.log(body);
+        const respose = await ctx.http.post("https://wxls.bbroot.com/v1/images/generations", body, {
+          headers: { "Authorization": config.token }
+        });
+        let time = 0;
+        let progress = 0;
+        let url = null;
+        let prompt = null;
+        while (true) {
+          const data = await ctx.http.post(`https://tools.mgtv100.com/external/v1/pear/imageTaskShow`, { task_id: respose.id });
+          config.deBug && console.log(data);
+          progress = data.data?.progress || 0;
+          prompt = data.data?.input?.prompt;
+          if (data.data?.api_file_url) {
+            if (Array.isArray(data.data?.api_file_url)) {
+              url = data.data?.api_file_url[0];
+            } else {
+              url = data.data?.api_file_url;
+            }
+          }
+          if (progress == 100) {
+            ImgGpt.userCache[userId].progress = null;
+            ImgGpt.userCache[userId].lastResult = url;
+            break;
+          }
+          if (time % 6 == 0) {
+            fn && await fn({ progress, url });
+          }
+          await sleep(1e4);
+          time++;
+        }
+        return { progress, url, prompt };
+      } catch (error) {
+        console.log(error.message);
+        return null;
+      }
+    }
+  };
+  ctx.command("获取标识").action(async ({ session }) => {
+    return session.userId;
+  });
+  ctx.command("生图 <prompt:text>").action(async ({ session }, prompt) => {
+    if (!config.onlyUse.includes(session.userId)) return `该功能暂不对外支持`;
+    const text = import_koishi.h.select(prompt, "text")[0]?.attrs.content || null;
+    const img = import_koishi.h.select(prompt, "img").map((item) => item.attrs.src) || null;
+    if (!text) return `请携带需求发送，例如 /生图 生成一只猫娘`;
+    await session.send("稍等，正在给你生...");
+    const result = await ImgGpt.startGetImage(session, text, img, async (val) => {
+      await session.send(`当前进度：${val.progress}%`);
+    });
+    console.log(result);
+    if (!result) return `生成失败！`;
+    await session.send(import_koishi.h.image(result.url) + result.prompt);
+  });
+}
+__name(apply, "apply");
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  Config,
+  apply,
+  name
+});

package/package.json

@@ -0,0 +1,20 @@
+{
+  "name": "koishi-plugin-smmcat-gptimage2",
+  "description": "简单去使用云接口 GPT-Image2",
+  "version": "0.0.1",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "dist"
+  ],
+  "license": "MIT",
+  "keywords": [
+    "chatbot",
+    "koishi",
+    "plugin"
+  ],
+  "peerDependencies": {
+    "koishi": "^4.18.7"
+  }
+}

package/readme.md

@@ -0,0 +1,5 @@
+# koishi-plugin-smmcat-gptimage2
+
+[![npm](https://img.shields.io/npm/v/koishi-plugin-smmcat-gptimage2?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-smmcat-gptimage2)
+
+简单去使用 GPT-Image2