koishi-plugin-chatluna-toolbox 0.0.10 → 0.0.11

package/lib/index.js

@@ -47,9 +47,11 @@ __export(index_exports, {
   createXmlProcessor: () => createXmlProcessor,
   ensureOneBotSession: () => ensureOneBotSession,
   getSession: () => getSession,
+  hasReplyToolsEnabled: () => hasReplyToolsEnabled,
   inject: () => inject,
   name: () => name,
   parseSelfClosingXmlTags: () => parseSelfClosingXmlTags,
+  registerCharacterReplyTools: () => registerCharacterReplyTools,
   registerNativeTools: () => registerNativeTools,
   registerVariables: () => registerVariables,
   registerXmlTools: () => registerXmlTools,
@@ -57,7 +59,9 @@ __export(index_exports, {
   sendDeleteMessage: () => sendDeleteMessage,
   sendGroupBan: () => sendGroupBan,
   sendMsgEmoji: () => sendMsgEmoji,
-  sendPoke: () => sendPoke
+  sendPoke: () => sendPoke,
+  sendSetGroupCard: () => sendSetGroupCard,
+  sendSetProfile: () => sendSetProfile
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -195,6 +199,7 @@ var DEFAULT_XML_REFERENCE_PROMPT = `## \u52A8\u4F5C\u6307\u4EE4
   </actions>
 \`\`\``;
 var XmlToolsSchema = import_koishi3.Schema.object({
+  injectXmlToolAsReplyTool: import_koishi3.Schema.boolean().default(false).description("\u5C06 XML \u5DE5\u5177\u6539\u4E3A\u6CE8\u5165\u5B9E\u9A8C\u6027\u201C\u5DE5\u5177\u8C03\u7528\u56DE\u590D\u201D\u7684\u53C2\u6570\u4E2D"),
   enablePokeXmlTool: import_koishi3.Schema.boolean().default(false).description("\u542F\u7528 XML \u5F62\u5F0F\u7684\u6233\u4E00\u6233\u8C03\u7528\uFF08\u4E0E \u539F\u751F\u5DE5\u5177 \u4E8C\u9009\u4E00\uFF09"),
   enableEmojiXmlTool: import_koishi3.Schema.boolean().default(false).description(
     "\u542F\u7528 XML \u5F62\u5F0F\u7684\u6D88\u606F\u8868\u60C5\u8C03\u7528\uFF0C\uFF08\u9700 chatluna-character \u5F00\u542F enableMessageId\uFF0C\u4E0E \u539F\u751F\u5DE5\u5177 \u4E8C\u9009\u4E00\uFF0Cemoji_id \u5BF9\u7167\u8868\uFF1Ahttps://bot.q.qq.com/wiki/develop/pythonsdk/model/emoji.html \uFF09"
@@ -566,6 +571,34 @@ function createSetGroupBanTool(deps) {
 // src/features/native-tools/tools/set-group-card.ts
 var import_zod4 = require("zod");
 var import_tools3 = require("@langchain/core/tools");
+async function sendSetGroupCard(params) {
+  try {
+    const { session, userId, card, groupId, log } = params;
+    if (!session) return "No session context available.";
+    const resolvedGroupId = groupId?.trim() || (session.guildId ? String(session.guildId).trim() : "") || (session.channelId ? String(session.channelId).trim() : "");
+    if (!resolvedGroupId) {
+      return "Missing groupId. Provide groupId explicitly or run inside a group session.";
+    }
+    const userIdRaw = userId.trim();
+    const cardRaw = card.trim();
+    if (!userIdRaw) return "userId is required.";
+    if (!cardRaw) return "card is required.";
+    const { error, internal } = ensureOneBotSession(session);
+    if (error) return error;
+    await callOneBotAPI(
+      internal,
+      "set_group_card",
+      { group_id: resolvedGroupId, user_id: userIdRaw, card: cardRaw },
+      ["setGroupCard"]
+    );
+    const message = `\u7FA4\u6635\u79F0\u5DF2\u66F4\u65B0\uFF1A${userIdRaw} -> ${cardRaw}`;
+    log?.("info", message);
+    return message;
+  } catch (error) {
+    params.log?.("warn", "set_group_card failed", error);
+    return `set_group_card failed: ${error.message}`;
+  }
+}
 function createSetGroupCardTool(deps) {
   const { toolName, description, log } = deps;
   return new class extends import_tools3.StructuredTool {
@@ -577,31 +610,13 @@ function createSetGroupCardTool(deps) {
       card: import_zod4.z.string().min(1, "card is required").describe("New group card for the member.")
     });
     async _call(input, _manager, runnable) {
-      try {
-        const session = getSession(runnable);
-        if (!session) return "No session context available.";
-        const groupId = input.groupId?.trim() || (session.guildId ? String(session.guildId).trim() : "") || (session.channelId ? String(session.channelId).trim() : "");
-        if (!groupId)
-          return "Missing groupId. Provide groupId explicitly or run inside a group session.";
-        const userId = input.userId.trim();
-        const card = input.card.trim();
-        if (!userId) return "userId is required.";
-        if (!card) return "card is required.";
-        const { error, internal } = ensureOneBotSession(session);
-        if (error) return error;
-        await callOneBotAPI(
-          internal,
-          "set_group_card",
-          { group_id: groupId, user_id: userId, card },
-          ["setGroupCard"]
-        );
-        const message = `\u7FA4\u6635\u79F0\u5DF2\u66F4\u65B0\uFF1A${userId} -> ${card}`;
-        log?.("info", message);
-        return message;
-      } catch (error) {
-        log?.("warn", "set_group_card failed", error);
-        return `set_group_card failed: ${error.message}`;
-      }
+      return sendSetGroupCard({
+        session: getSession(runnable),
+        groupId: input.groupId,
+        userId: input.userId,
+        card: input.card,
+        log
+      });
     }
   }();
 }
@@ -667,6 +682,25 @@ var genders = {
   male: "1",
   female: "2"
 };
+async function sendSetProfile(params) {
+  try {
+    const { session, nickname, signature, gender, protocol, log } = params;
+    const { error, internal } = ensureOneBotSession(session);
+    if (error) return error;
+    const payload = { nickname };
+    if (signature) payload.personal_note = signature;
+    if (gender && protocol !== "llbot") payload.sex = genders[gender];
+    await callOneBotAPI(internal, "set_qq_profile", payload, [
+      "setQQProfile"
+    ]);
+    const message = "\u673A\u5668\u4EBA\u8D44\u6599\u5DF2\u66F4\u65B0\u3002";
+    log?.("info", message);
+    return message;
+  } catch (error) {
+    params.log?.("warn", "\u4FEE\u6539\u673A\u5668\u4EBA\u8D26\u6237\u4FE1\u606F\u5931\u8D25", error);
+    return `\u4FEE\u6539\u673A\u5668\u4EBA\u8D26\u6237\u4FE1\u606F\u5931\u8D25\uFF1A${error.message}`;
+  }
+}
 function createSetProfileTool(deps) {
   const { toolName, description, log, protocol } = deps;
   return new class extends import_tools5.StructuredTool {
@@ -678,24 +712,14 @@ function createSetProfileTool(deps) {
       gender: import_zod6.z.enum(["unknown", "male", "female"]).optional().describe("Optional: the new gender.")
     });
     async _call(input, _manager, runnable) {
-      try {
-        const session = getSession(runnable);
-        const { error, internal } = ensureOneBotSession(session);
-        if (error) return error;
-        const payload = { nickname: input.nickname };
-        if (input.signature) payload.personal_note = input.signature;
-        if (input.gender && protocol !== "llbot")
-          payload.sex = genders[input.gender];
-        await callOneBotAPI(internal, "set_qq_profile", payload, [
-          "setQQProfile"
-        ]);
-        const message = "\u673A\u5668\u4EBA\u8D44\u6599\u5DF2\u66F4\u65B0\u3002";
-        log?.("info", message);
-        return message;
-      } catch (error) {
-        log?.("warn", "\u4FEE\u6539\u673A\u5668\u4EBA\u8D26\u6237\u4FE1\u606F\u5931\u8D25", error);
-        return `\u4FEE\u6539\u673A\u5668\u4EBA\u8D26\u6237\u4FE1\u606F\u5931\u8D25\uFF1A${error.message}`;
-      }
+      return sendSetProfile({
+        session: getSession(runnable),
+        nickname: input.nickname,
+        signature: input.signature,
+        gender: input.gender,
+        protocol,
+        log
+      });
     }
   }();
 }
@@ -816,6 +840,253 @@ function registerNativeTools(deps) {
   }
 }
 
+// src/features/reply-tools/register.ts
+function escapeAttr(value) {
+  return String(value ?? "").replaceAll("&", "&amp;").replaceAll("<", "&lt;").replaceAll(">", "&gt;").replaceAll('"', "&quot;");
+}
+function hasReplyToolsEnabled(config) {
+  return !!(config.injectXmlToolAsReplyTool && (config.enablePokeXmlTool || config.enableBanXmlTool || config.enableEmojiXmlTool || config.enableDeleteXmlTool));
+}
+function registerCharacterReplyTools(deps) {
+  const { ctx, config, protocol, log } = deps;
+  const service = ctx.chatluna_character;
+  if (!service?.registerReplyToolField) {
+    return () => {
+    };
+  }
+  const disposers = [];
+  if (config.injectXmlToolAsReplyTool && config.enablePokeXmlTool) {
+    disposers.push(
+      service.registerReplyToolField({
+        name: "toolbox_poke",
+        schema: {
+          type: "array",
+          description: "Poke one or more users after sending this reply. Each item is one poke action.",
+          items: {
+            type: "object",
+            properties: {
+              user_id: {
+                type: "string",
+                description: "Target user ID to poke."
+              },
+              group_id: {
+                type: "string",
+                description: "Optional target group ID. Defaults to the current session group."
+              }
+            },
+            required: ["user_id"]
+          }
+        },
+        async invoke(_, session, value) {
+          if (!Array.isArray(value)) return;
+          for (const item of value) {
+            if (!item || typeof item !== "object" || Array.isArray(item)) {
+              continue;
+            }
+            const action = item;
+            if (typeof action.user_id !== "string") continue;
+            await sendPoke({
+              session,
+              userId: action.user_id,
+              groupId: typeof action.group_id === "string" ? action.group_id : void 0,
+              protocol,
+              log
+            });
+          }
+        },
+        render(_, __, value) {
+          if (!Array.isArray(value)) return;
+          return value.flatMap((item) => {
+            if (!item || typeof item !== "object" || Array.isArray(item)) {
+              return [];
+            }
+            const action = item;
+            if (typeof action.user_id !== "string") return [];
+            const group = typeof action.group_id === "string" && action.group_id.trim() ? ` group_id="${escapeAttr(action.group_id)}"` : "";
+            return [`<poke id="${escapeAttr(action.user_id)}"${group} />`];
+          });
+        }
+      })
+    );
+  }
+  if (config.injectXmlToolAsReplyTool && config.enableBanXmlTool) {
+    disposers.push(
+      service.registerReplyToolField({
+        name: "toolbox_set_group_ban",
+        schema: {
+          type: "array",
+          description: "Mute or unmute one or more users after this reply. Each item is one group ban action.",
+          items: {
+            type: "object",
+            properties: {
+              user_id: {
+                type: "string",
+                description: "Target member user ID."
+              },
+              duration: {
+                type: "number",
+                description: "Mute duration in seconds. Use 0 to unmute the member."
+              },
+              group_id: {
+                type: "string",
+                description: "Optional target group ID. Defaults to the current session group."
+              }
+            },
+            required: ["user_id", "duration"]
+          }
+        },
+        async invoke(_, session, value) {
+          if (!Array.isArray(value)) return;
+          for (const item of value) {
+            if (!item || typeof item !== "object" || Array.isArray(item)) {
+              continue;
+            }
+            const action = item;
+            if (typeof action.user_id !== "string") continue;
+            await sendGroupBan({
+              session,
+              userId: action.user_id,
+              duration: typeof action.duration === "number" || typeof action.duration === "string" ? action.duration : "",
+              groupId: typeof action.group_id === "string" ? action.group_id : void 0,
+              protocol,
+              log
+            });
+          }
+        },
+        render(_, __, value) {
+          if (!Array.isArray(value)) return;
+          return value.flatMap((item) => {
+            if (!item || typeof item !== "object" || Array.isArray(item)) {
+              return [];
+            }
+            const action = item;
+            if (typeof action.user_id !== "string") return [];
+            const duration = typeof action.duration === "number" || typeof action.duration === "string" ? String(action.duration) : "";
+            if (!duration) return [];
+            const group = typeof action.group_id === "string" && action.group_id.trim() ? ` group_id="${escapeAttr(action.group_id)}"` : "";
+            return [
+              `<ban id="${escapeAttr(action.user_id)}" duration="${escapeAttr(duration)}"${group} />`
+            ];
+          });
+        }
+      })
+    );
+  }
+  if (config.injectXmlToolAsReplyTool && config.enableEmojiXmlTool) {
+    disposers.push(
+      service.registerReplyToolField({
+        name: "toolbox_set_msg_emoji",
+        schema: {
+          type: "array",
+          description: "Add emoji reactions to one or more messages after this reply. Each item is one emoji action.",
+          items: {
+            type: "object",
+            properties: {
+              message_id: {
+                type: "string",
+                description: "Target message ID."
+              },
+              emoji_id: {
+                type: "string",
+                description: "Emoji ID to send."
+              }
+            },
+            required: ["message_id", "emoji_id"]
+          }
+        },
+        async invoke(_, session, value) {
+          if (!Array.isArray(value)) return;
+          for (const item of value) {
+            if (!item || typeof item !== "object" || Array.isArray(item)) {
+              continue;
+            }
+            const action = item;
+            if (typeof action.message_id !== "string" || typeof action.emoji_id !== "string") {
+              continue;
+            }
+            await sendMsgEmoji({
+              session,
+              messageId: action.message_id,
+              emojiId: action.emoji_id,
+              protocol,
+              log
+            });
+          }
+        },
+        render(_, __, value) {
+          if (!Array.isArray(value)) return;
+          return value.flatMap((item) => {
+            if (!item || typeof item !== "object" || Array.isArray(item)) {
+              return [];
+            }
+            const action = item;
+            if (typeof action.message_id !== "string" || typeof action.emoji_id !== "string") {
+              return [];
+            }
+            return [
+              `<emoji message_id="${escapeAttr(action.message_id)}" emoji_id="${escapeAttr(action.emoji_id)}" />`
+            ];
+          });
+        }
+      })
+    );
+  }
+  if (config.injectXmlToolAsReplyTool && config.enableDeleteXmlTool) {
+    disposers.push(
+      service.registerReplyToolField({
+        name: "toolbox_delete_message",
+        schema: {
+          type: "array",
+          description: "Delete one or more messages after this reply. Each item is one delete action.",
+          items: {
+            type: "object",
+            properties: {
+              message_id: {
+                type: "string",
+                description: "Target message ID to delete."
+              }
+            },
+            required: ["message_id"]
+          }
+        },
+        async invoke(_, session, value) {
+          if (!Array.isArray(value)) return;
+          for (const item of value) {
+            if (!item || typeof item !== "object" || Array.isArray(item)) {
+              continue;
+            }
+            const action = item;
+            if (typeof action.message_id !== "string") continue;
+            await sendDeleteMessage({
+              session,
+              messageId: action.message_id,
+              log
+            });
+          }
+        },
+        render(_, __, value) {
+          if (!Array.isArray(value)) return;
+          return value.flatMap((item) => {
+            if (!item || typeof item !== "object" || Array.isArray(item)) {
+              return [];
+            }
+            const action = item;
+            if (typeof action.message_id !== "string") return [];
+            return [
+              `<delete message_id="${escapeAttr(action.message_id)}" />`
+            ];
+          });
+        }
+      })
+    );
+  }
+  return () => {
+    for (const dispose of disposers.reverse()) {
+      dispose();
+    }
+  };
+}
+
 // src/features/xml-tools/parser.ts
 function parseSelfClosingXmlTags(text, tagName) {
   const tags = Array.from(
@@ -1875,34 +2146,47 @@ function apply(ctx, config) {
     }
   };
   const log = createLogger(ctx, config);
+  const enableXmlRuntime = !config.injectXmlToolAsReplyTool && (config.enablePokeXmlTool || config.enableEmojiXmlTool || config.enableDeleteXmlTool || config.enableBanXmlTool);
   let xmlRuntime = null;
   let characterCtx = null;
+  let replyToolsDispose = null;
   const initializeServices = async () => {
     log("info", "toolbox \u521D\u59CB\u5316\u5F00\u59CB");
     registerVariables({ ctx, config, log });
     const protocol = resolveOneBotProtocol(config, log);
     registerNativeTools({ ctx, config, plugin, protocol, log });
-    xmlRuntime = registerXmlTools({ ctx, config, protocol, log });
-    if (characterCtx && xmlRuntime.start()) {
+    xmlRuntime = enableXmlRuntime ? registerXmlTools({ ctx, config, protocol, log }) : null;
+    if (characterCtx && xmlRuntime?.start()) {
       log("info", "XML \u5DE5\u5177\u5DF2\u542F\u7528");
     }
     log("info", "toolbox \u521D\u59CB\u5316\u5B8C\u6210");
   };
   const dispose = () => {
     characterCtx = null;
+    replyToolsDispose?.();
+    replyToolsDispose = null;
     xmlRuntime?.stop();
     xmlRuntime = null;
   };
-  if (config.enablePokeXmlTool || config.enableEmojiXmlTool || config.enableDeleteXmlTool || config.enableBanXmlTool) {
+  if (hasReplyToolsEnabled(config) || enableXmlRuntime) {
     ctx.inject(["chatluna_character"], (innerCtx) => {
       characterCtx = innerCtx;
-      if (!xmlRuntime) return;
-      const started = xmlRuntime.start();
-      if (started) log("info", "XML \u5DE5\u5177\u5DF2\u542F\u7528");
+      replyToolsDispose?.();
+      replyToolsDispose = hasReplyToolsEnabled(config) ? registerCharacterReplyTools({
+        ctx: innerCtx,
+        config,
+        protocol: resolveOneBotProtocol(config, log),
+        log
+      }) : null;
       innerCtx.on("dispose", () => {
         if (characterCtx === innerCtx) characterCtx = null;
+        replyToolsDispose?.();
+        replyToolsDispose = null;
         xmlRuntime?.stop();
       });
+      if (!xmlRuntime) return;
+      const started = xmlRuntime.start();
+      if (started) log("info", "XML \u5DE5\u5177\u5DF2\u542F\u7528");
     });
   }
   ctx.on("dispose", dispose);
@@ -1943,9 +2227,11 @@ function apply(ctx, config) {
   createXmlProcessor,
   ensureOneBotSession,
   getSession,
+  hasReplyToolsEnabled,
   inject,
   name,
   parseSelfClosingXmlTags,
+  registerCharacterReplyTools,
   registerNativeTools,
   registerVariables,
   registerXmlTools,
@@ -1953,5 +2239,7 @@ function apply(ctx, config) {
   sendDeleteMessage,
   sendGroupBan,
   sendMsgEmoji,
-  sendPoke
+  sendPoke,
+  sendSetGroupCard,
+  sendSetProfile
 });

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chatluna-toolbox",
   "description": "为 ChatLuna 提供更多原生工具、XML 工具与变量，仅支持 onebot 平台。",
-  "version": "0.0.10",
+  "version": "0.0.11",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",
   "files": [
@@ -42,7 +42,7 @@
   "scripts": {
     "build": "tsup",
     "watch": "tsup --watch",
-    "typecheck": "tsc --noEmit",
+    "typecheck": "NODE_OPTIONS=--max-old-space-size=8192 tsc --noEmit",
     "test": "vitest run"
   }
 }