koishi-plugin-smmcat-nai-diffusion 0.0.1

package/lib/chatSend.d.ts

@@ -0,0 +1,17 @@
+import { Context, Session } from "koishi";
+import { Config } from ".";
+/** 内容发送模块 */
+export declare const Chat: {
+    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,11 @@
+import { Context, Schema } from 'koishi';
+export declare const name = "smmcat-nai-diffusion";
+export interface Config {
+    token: string;
+    deBug: boolean;
+    useMd: boolean;
+    useGuildId: string[];
+    useGuildUser: string[];
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -0,0 +1,244 @@
+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");
+
+// src/chatSend.ts
+var Chat = {
+  config: {},
+  ctx: {},
+  chache: {},
+  messageIdTemp: {},
+  async init(config, ctx) {
+    Chat.config = config;
+    Chat.ctx = ctx;
+    ctx.on("message", (session) => {
+      if (!Chat.messageIdTemp[session.userId]) {
+        Chat.messageIdTemp[session.userId] = {};
+      }
+      Chat.messageIdTemp[session.userId] = session.messageId;
+      Chat.clearMessageIdTemp();
+    }, true);
+  },
+  /** 发送消息 */
+  async send(session, content) {
+    if (session.platform !== "qq") {
+      await session.send(content);
+      return;
+    }
+    if (Chat.config.useMd) {
+      const messageId = session.messageId;
+      const userId = session.userId;
+      if (Chat.chache[messageId] == void 0) {
+        Chat.chache[messageId] = 0;
+      }
+      const mdContent = {
+        msg_id: session.messageId,
+        msg_type: 2,
+        msg_seq: ++Chat.chache[messageId] == 1 ? 0 : Chat.chache[messageId],
+        markdown: {
+          content
+        }
+      };
+      if (Chat.chache[messageId] > 1) {
+        Chat.config.deBug && console.log(`检测到MD连续发送，尝试递增 msg_seq 再进行再次发送；第${Chat.chache[messageId] + 1}次。`);
+        if (Chat.chache[messageId] > 5) {
+          if (Chat.messageIdTemp[userId] && mdContent.msg_id != Chat.messageIdTemp[userId]) {
+            Chat.config.deBug && console.log("middleware 存在新 mse_id 数据，使用更新数据");
+            mdContent.msg_id = Chat.messageIdTemp[userId];
+            mdContent.msg_seq = 0;
+          } else {
+            await session.send(content);
+          }
+        }
+      }
+      try {
+        if (session.guildId) {
+          await session.bot.internal.sendMessage(session.channelId, mdContent);
+        } else {
+          mdContent.markdown.content = mdContent.markdown.content.replace(/<qqbot-at-user[^>]*\/>/g, "");
+          await session.bot.internal.sendPrivateMessage(session.userId, mdContent);
+        }
+      } catch (error) {
+        console.log(error);
+      }
+      Chat.clearChache();
+    } else {
+      await session.send(content);
+    }
+  },
+  // 当前 session 是否可以使用MD
+  isUseMd(session) {
+    if (session.platform !== "qq") return false;
+    if (Chat.chache[session.messageId] == void 0) return true;
+    return Chat.chache[session.messageId] <= 5;
+  },
+  /** 清除一条 msgid 记录 */
+  clearChache() {
+    const chache = Object.keys(Chat.chache);
+    if (chache.length >= 40) {
+      delete Chat.chache[chache[0]];
+    }
+  },
+  /** 清除一条 messageId 缓存 */
+  clearMessageIdTemp() {
+    const messageIdTempList = Object.keys(Chat.messageIdTemp);
+    if (messageIdTempList.length > 150) {
+      delete Chat.messageIdTemp[messageIdTempList[0]];
+    }
+  }
+};
+
+// src/index.ts
+var name = "smmcat-nai-diffusion";
+var Config = import_koishi.Schema.object({
+  deBug: import_koishi.Schema.boolean().default(false).description("获取更多日志消息"),
+  useMd: import_koishi.Schema.boolean().default(false).description("使用md发送消息"),
+  token: import_koishi.Schema.string().default("").description("[tuercha](https://api.tuercha.com/register) 绘图站提供的令牌"),
+  useGuildId: import_koishi.Schema.array(String).role("table").default([]).description("指定可用群"),
+  useGuildUser: import_koishi.Schema.array(String).role("table").default([]).description("指定可用人（无视可用群规则）")
+});
+function apply(ctx, config) {
+  ctx.on("ready", () => {
+    Chat.init(config, ctx);
+  });
+  const size = [[832, 1216], [1216, 832], [1024, 1024], [512, 768], [768, 512]];
+  const naiDiffusion = {
+    /** 通过词条获取图片 */
+    async getDrawByEntry(text) {
+      try {
+        const body = {
+          model: "nai-diffusion-4-5-full",
+          messages: [
+            {
+              role: "user",
+              content: JSON.stringify({
+                prompt: text,
+                negative_prompt: "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, ugly, deformed, poorly drawn hands, poorly drawn face, mutation, deformed, bad proportions, gross proportions, long neck, missing arms, missing legs, extra arms, extra legs, fused fingers, too many fingers, unclear eyes, bad body, out of focus, chromatic aberration",
+                size: size[Math.floor(Math.random() * size.length)],
+                steps: 23
+              })
+            }
+          ],
+          stream: false,
+          max_tokens: 1e5
+        };
+        const res = await ctx.http.post("https://api.tuercha.com/v1/chat/completions", body, {
+          headers: {
+            Authorization: "Bearer " + config.token
+          }
+        });
+        const imgList = [];
+        if (Array.isArray(res.choices)) {
+          for (const data of res.choices) {
+            const base64 = naiDiffusion.extractBase64(data.message.content || "");
+            if (base64) {
+              const imgUrl = await naiDiffusion.base64ToWebUrl(base64);
+              imgUrl && imgList.push(imgUrl);
+            }
+          }
+        }
+        return imgList;
+      } catch (error) {
+        console.log(error);
+        return null;
+      }
+    },
+    /** 输入的内容是否存在中文 */
+    containsForbiddenChars(str) {
+      return /[\p{Script=Han}\p{Script=Hiragana}\p{Script=Katakana}\p{Script=Hangul}\uff00-\uffef\u3000\u3001-\u303f]/u.test(str);
+    },
+    /** 过滤筛选 base64 内容 */
+    extractBase64(raw) {
+      const match = raw.match(/data:image\/\w+;base64,[A-Za-z0-9+/=]+/);
+      return match ? match[0] : null;
+    },
+    /** 将 base64 图片转为 网络图片地址 */
+    async base64ToWebUrl(base64) {
+      try {
+        const res = await ctx.http.post("https://img.smmcat.top/proxy/image/base64", { base64 });
+        if (res.code !== 0) {
+          return null;
+        }
+        return res.localUrl;
+      } catch (error) {
+        console.log(error);
+        return null;
+      }
+    }
+  };
+  const use = {
+    userId: {},
+    isUse(session) {
+      return use.userId[session.userId];
+    },
+    start(session) {
+      use.userId[session.userId] = +/* @__PURE__ */ new Date();
+    },
+    clear(session) {
+      delete use.userId[session.userId];
+    }
+  };
+  ctx.command("测试画 <prompt:text>").action(async ({ session }, prompt) => {
+    const { userId, guildId } = session;
+    if (!config.useGuildUser.includes(userId)) {
+      if (!(guildId && config.useGuildId.includes(guildId))) {
+        return `该群暂时不提供该功能...`;
+      }
+    }
+    if (!prompt) return "请传词条！";
+    if (use.isUse(session)) return `正在进行上一个绘画，请等待...`;
+    if (naiDiffusion.containsForbiddenChars(prompt)) return `输入的词条包含问题，请检查！`;
+    try {
+      use.start(session);
+      await Chat.send(session, `<qqbot-at-user id="${userId}" /> 稍等，正在给你画...`);
+      const imgs = await naiDiffusion.getDrawByEntry(prompt);
+      if (config.useMd) {
+        const md = `<qqbot-at-user id="${userId}" /> 画完了，请点击下方查看！
+
+${imgs.map((item) => {
+          return `[查看图片](${item})`;
+        }).join("\n")}`;
+        await Chat.send(session, md);
+      } else {
+        await Chat.send(session, imgs.map((item) => {
+          return import_koishi.h.image(item);
+        }).join("\n"));
+      }
+      use.clear(session);
+    } catch (error) {
+      use.clear(session);
+    }
+  });
+}
+__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-nai-diffusion",
+  "description": "来自 tuercha.com 公益站的绘图服务",
+  "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-nai-diffusion
+
+[![npm](https://img.shields.io/npm/v/koishi-plugin-smmcat-nai-diffusion?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-smmcat-nai-diffusion)
+
+来自 tuercha.com 公益站的绘图服务