koishi-plugin-chatluna-toolbox 0.0.5 → 0.0.6

package/lib/index.js

@@ -31,6 +31,7 @@ __export(index_exports, {
   apply: () => apply,
   callOneBotAPI: () => callOneBotAPI,
   createBotInfoProvider: () => createBotInfoProvider,
+  createCharacterTempXmlRuntime: () => createCharacterTempXmlRuntime,
   createDeleteMessageTool: () => createDeleteMessageTool,
   createGroupInfoProvider: () => createGroupInfoProvider,
   createLogger: () => createLogger,
@@ -40,12 +41,12 @@ __export(index_exports, {
   createSetMsgEmojiTool: () => createSetMsgEmojiTool,
   createSetProfileTool: () => createSetProfileTool,
   createUserInfoProvider: () => createUserInfoProvider,
-  createXmlInterceptor: () => createXmlInterceptor,
+  createXmlProcessor: () => createXmlProcessor,
   ensureOneBotSession: () => ensureOneBotSession,
   getSession: () => getSession,
   inject: () => inject,
   name: () => name,
-  parseXmlActions: () => parseXmlActions,
+  parseSelfClosingXmlTags: () => parseSelfClosingXmlTags,
   registerNativeTools: () => registerNativeTools,
   registerVariables: () => registerVariables,
   registerXmlTools: () => registerXmlTools,
@@ -64,22 +65,22 @@ var import_koishi = require("koishi");
 var NativeToolsSchema = import_koishi.Schema.object({
   enableNapCatProtocol: import_koishi.Schema.boolean().default(true).description("\u542F\u7528 NapCat OneBot \u534F\u8BAE\uFF08\u4E0E LLBot \u4E8C\u9009\u4E00\uFF09"),
   enableLlbotProtocol: import_koishi.Schema.boolean().default(false).description("\u542F\u7528 LLBot OneBot \u534F\u8BAE\uFF08\u4E0E NapCat \u4E8C\u9009\u4E00\uFF09"),
-  enablePokeTool: import_koishi.Schema.boolean().default(false).description("\u6CE8\u518C ChatLuna \u5DE5\u5177\uFF1A\u6233\u4E00\u6233\uFF08\u5EFA\u8BAE\u4E0E XML\u5DE5\u5177 \u4E8C\u9009\u4E00\uFF09"),
+  enablePokeTool: import_koishi.Schema.boolean().default(false).description("\u6CE8\u518C ChatLuna \u5DE5\u5177\uFF1A\u6233\u4E00\u6233\uFF08\u4E0E XML\u5DE5\u5177 \u4E8C\u9009\u4E00\uFF09"),
   pokeToolName: import_koishi.Schema.string().default("poke_user").description("ChatLuna \u5DE5\u5177\u540D\u79F0\uFF1A\u6233\u4E00\u6233"),
   enableSetSelfProfileTool: import_koishi.Schema.boolean().default(false).description("\u6CE8\u518C ChatLuna \u5DE5\u5177\uFF1A\u4FEE\u6539\u81EA\u8EAB\u8D26\u6237\u4FE1\u606F"),
   setSelfProfileToolName: import_koishi.Schema.string().default("set_self_profile").description("ChatLuna \u5DE5\u5177\u540D\u79F0\uFF1A\u4FEE\u6539\u81EA\u8EAB\u8D26\u6237\u4FE1\u606F\uFF08\u652F\u6301\u6635\u79F0/\u7B7E\u540D/\u6027\u522B\uFF09"),
   enableSetGroupCardTool: import_koishi.Schema.boolean().default(false).description("\u6CE8\u518C ChatLuna \u5DE5\u5177\uFF1A\u4FEE\u6539\u7FA4\u6210\u5458\u6635\u79F0"),
   setGroupCardToolName: import_koishi.Schema.string().default("set_group_card").description("ChatLuna \u5DE5\u5177\u540D\u79F0\uFF1A\u4FEE\u6539\u7FA4\u6210\u5458\u6635\u79F0"),
-  enableSetMsgEmojiTool: import_koishi.Schema.boolean().default(false).description("\u6CE8\u518C ChatLuna \u5DE5\u5177\uFF1A\u7ED9\u6D88\u606F\u6DFB\u52A0\u8868\u60C5\uFF08\u9700 chatluna-character \u5F00\u542F enableMessageId\uFF0C\u5EFA\u8BAE\u4E0E XML\u5DE5\u5177 \u4E8C\u9009\u4E00\uFF09"),
+  enableSetMsgEmojiTool: import_koishi.Schema.boolean().default(false).description("\u6CE8\u518C ChatLuna \u5DE5\u5177\uFF1A\u7ED9\u6D88\u606F\u6DFB\u52A0\u8868\u60C5\uFF08\u9700 chatluna-character \u5F00\u542F enableMessageId\uFF0C\u4E0E XML\u5DE5\u5177 \u4E8C\u9009\u4E00\uFF0C\u8868\u60C5\u5BF9\u7167\u8868\uFF1Ahttps://bot.q.qq.com/wiki/develop/pythonsdk/model/emoji.html \uFF09"),
   setMsgEmojiToolName: import_koishi.Schema.string().default("set_msg_emoji").description("ChatLuna \u5DE5\u5177\u540D\u79F0\uFF1A\u7ED9\u6D88\u606F\u6DFB\u52A0\u8868\u60C5"),
-  enableDeleteMessageTool: import_koishi.Schema.boolean().default(false).description("\u6CE8\u518C ChatLuna \u5DE5\u5177\uFF1A\u64A4\u56DE\u6D88\u606F\uFF08\u9700 chatluna-character \u5F00\u542F enableMessageId\uFF0C\u5EFA\u8BAE\u4E0E XML\u5DE5\u5177 \u4E8C\u9009\u4E00\uFF09"),
+  enableDeleteMessageTool: import_koishi.Schema.boolean().default(false).description("\u6CE8\u518C ChatLuna \u5DE5\u5177\uFF1A\u64A4\u56DE\u6D88\u606F\uFF08\u9700 chatluna-character \u5F00\u542F enableMessageId\uFF0C\u4E0E XML\u5DE5\u5177 \u4E8C\u9009\u4E00\uFF09"),
   deleteMessageToolName: import_koishi.Schema.string().default("delete_msg").description("ChatLuna \u5DE5\u5177\u540D\u79F0\uFF1A\u64A4\u56DE\u6D88\u606F")
 }).description("\u539F\u751F\u5DE5\u5177");
 
 // src/schema/xml-tools.ts
 var import_koishi2 = require("koishi");
 var DEFAULT_XML_REFERENCE_PROMPT = `## \u52A8\u4F5C\u6307\u4EE4
-\u4F60\u53EF\u4EE5\u6839\u636E\u9700\u8981\u521B\u5EFA\u4E00\u4E2A\u72EC\u7ACB\u7684 <actions> \u5143\u7D20\u3002\u5B83\u7528\u4E8E\u6267\u884C\u975E\u8BED\u8A00\u7684\u7CFB\u7EDF\u6307\u4EE4\u3002\u5982\u679C\u4E0D\u9700\u8981\u6267\u884C\u4EFB\u4F55\u52A8\u4F5C\uFF0C\u8BF7\u7701\u7565\u6B64\u5143\u7D20\u3002
+\u4F60\u53EF\u4EE5\u6839\u636E\u9700\u8981\u5728\u6A21\u578B\u56DE\u590D\u4E2D\u8F93\u51FA\u4E00\u4E2A\u72EC\u7ACB\u7684 <actions> \u5143\u7D20\u3002\u5B83\u7528\u4E8E\u6267\u884C\u975E\u8BED\u8A00\u7684\u7CFB\u7EDF\u6307\u4EE4\u3002\u5982\u679C\u4E0D\u9700\u8981\u6267\u884C\u4EFB\u4F55\u52A8\u4F5C\uFF0C\u8BF7\u7701\u7565\u6B64\u5143\u7D20\u3002
 1. \u6233\u4E00\u6233: <poke id=""/>
   - id: user_id
   - \u9002\u7528\u573A\u666F:
@@ -111,10 +112,12 @@ var DEFAULT_XML_REFERENCE_PROMPT = `## \u52A8\u4F5C\u6307\u4EE4
   </actions>
 \`\`\``;
 var XmlToolsSchema = import_koishi2.Schema.object({
-  enablePokeXmlTool: import_koishi2.Schema.boolean().default(false).description("\u542F\u7528 XML \u5F62\u5F0F\u7684\u6233\u4E00\u6233\u8C03\u7528"),
-  enableEmojiXmlTool: import_koishi2.Schema.boolean().default(false).description("\u542F\u7528 XML \u5F62\u5F0F\u7684\u6D88\u606F\u8868\u60C5\u8C03\u7528"),
-  enableDeleteXmlTool: import_koishi2.Schema.boolean().default(false).description("\u542F\u7528 XML \u5F62\u5F0F\u7684\u6D88\u606F\u64A4\u56DE\u8C03\u7528"),
-  referencePrompt: import_koishi2.Schema.string().role("textarea").default(DEFAULT_XML_REFERENCE_PROMPT).description("\u53C2\u8003\u63D0\u793A\u8BCD")
+  enablePokeXmlTool: import_koishi2.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_koishi2.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"
+  ),
+  enableDeleteXmlTool: import_koishi2.Schema.boolean().default(false).description("\u542F\u7528 XML \u5F62\u5F0F\u7684\u6D88\u606F\u64A4\u56DE\u8C03\u7528\uFF08\u9700 chatluna-character \u5F00\u542F enableMessageId\uFF0C\u4E0E \u539F\u751F\u5DE5\u5177 \u4E8C\u9009\u4E00\uFF09"),
+  referencePrompt: import_koishi2.Schema.string().role("textarea").default(DEFAULT_XML_REFERENCE_PROMPT).description("\u6A21\u578B\u56DE\u590D XML \u53C2\u8003\u63D0\u793A\u8BCD\uFF0C\u81EA\u884C\u5199\u5165\u63D0\u793A\u8BCD\u4E2D\uFF0C\u4E0D\u4F1A\u81EA\u52A8\u6CE8\u5165")
 }).description("XML \u5DE5\u5177");
 
 // src/schema/variables.ts
@@ -600,311 +603,315 @@ function registerNativeTools(deps) {
   }
 }
 
-// src/features/xml-tools/interceptor.ts
-function createXmlInterceptor(deps) {
-  const { ctx, config, log, onResponse } = deps;
-  const sessionMap = /* @__PURE__ */ new Map();
-  let currentGuildId = null;
-  let lastSession = null;
-  const pendingSessions = [];
-  let responseSeq = 0;
-  const processedResponses = /* @__PURE__ */ new Set();
-  let monitorHandle = null;
-  let fastRetryHandle = null;
-  let startupHandle = null;
-  const responseRetryHandles = /* @__PURE__ */ new Map();
-  let activeService = null;
-  let activeLogger = null;
-  let originalDebug = null;
-  let collectorBound = false;
-  const restore = () => {
-    if (activeLogger && originalDebug) {
-      activeLogger.debug = originalDebug;
-    }
-    activeLogger = null;
-    originalDebug = null;
-  };
-  const isActive = () => {
-    const characterService = ctx.chatluna_character;
-    const debugFn = characterService?.logger?.debug;
-    return Boolean(debugFn?.[RAW_INTERCEPTOR_TAG]);
-  };
-  const resolveSession = () => {
-    while (pendingSessions.length > 100) pendingSessions.shift();
-    const now = Date.now();
-    while (pendingSessions.length > 0 && now - pendingSessions[0].timestamp > 12e3) {
-      pendingSessions.shift();
-    }
-    if (pendingSessions.length === 0) {
-      return (currentGuildId ? sessionMap.get(currentGuildId) : null) || lastSession || null;
-    }
-    const contextKeys = /* @__PURE__ */ new Set();
-    for (let i = 0; i < pendingSessions.length; i++) {
-      contextKeys.add(pendingSessions[i].key || "__null__");
+// src/features/xml-tools/parser.ts
+function parseSelfClosingXmlTags(text, tagName) {
+  const tags = Array.from(
+    text.matchAll(new RegExp(`<${tagName}\\b([^>]*)\\/>`, "gi"))
+  );
+  if (!tags.length) return [];
+  return tags.map((tag) => {
+    const attrText = String(tag[1] || "");
+    const attrs = {};
+    for (const pair of attrText.matchAll(/([a-zA-Z_][\w-]*)="([^"]*)"/g)) {
+      attrs[pair[1]] = pair[2];
     }
-    let resolved = pendingSessions[pendingSessions.length - 1];
-    if (currentGuildId) {
-      for (let i = pendingSessions.length - 1; i >= 0; i--) {
-        if (pendingSessions[i].key === currentGuildId) {
-          resolved = pendingSessions[i];
-          break;
+    return attrs;
+  });
+}
+
+// src/features/xml-tools/processor.ts
+function createXmlProcessor(deps) {
+  const {
+    config,
+    protocol,
+    log,
+    sendPoke: runPoke = sendPoke,
+    sendMsgEmoji: runEmoji = sendMsgEmoji,
+    sendDeleteMessage: runDelete = sendDeleteMessage
+  } = deps;
+  return async ({ response, session }) => {
+    const content = String(response || "").trim();
+    if (!content) return false;
+    const pokeTags = parseSelfClosingXmlTags(content, "poke");
+    const emojiTags = parseSelfClosingXmlTags(content, "emoji");
+    const deleteTags = parseSelfClosingXmlTags(content, "delete");
+    let handled = false;
+    if (config.enablePokeXmlTool && pokeTags.length > 0) {
+      if (!session) {
+        log?.("warn", "\u68C0\u6D4B\u5230\u6233\u4E00\u6233\u6807\u8BB0\u4F46\u7F3A\u5C11\u4F1A\u8BDD\u4E0A\u4E0B\u6587");
+      } else {
+        const userIds = pokeTags.map((attrs) => String(attrs.id || "").trim()).filter(Boolean);
+        if (userIds.length > 0) handled = true;
+        for (const userId of userIds) {
+          try {
+            await runPoke({ session, userId, protocol, log });
+          } catch (error) {
+            log?.("warn", "XML \u89E6\u53D1 poke \u5931\u8D25", error);
+          }
         }
       }
     }
-    pendingSessions.length = 0;
-    if (contextKeys.size > 1 && config.debugLogging) {
-      log?.("warn", "\u68C0\u6D4B\u5230\u591A\u4E2A\u4F1A\u8BDD\u4E0A\u4E0B\u6587\uFF0C\u5DF2\u4F7F\u7528\u6700\u8FD1\u4F1A\u8BDD\u7EE7\u7EED\u6267\u884C XML", {
-        contextCount: contextKeys.size,
-        preferredGuildId: currentGuildId,
-        resolvedGuildId: resolved.key
-      });
-    }
-    return resolved.key ? sessionMap.get(resolved.key) || resolved.session : resolved.session;
-  };
-  const markProcessed = (responseId) => {
-    processedResponses.add(responseId);
-    if (processedResponses.size > 100) {
-      const first = processedResponses.values().next().value;
-      if (first) processedResponses.delete(first);
-    }
-  };
-  const runResponse = (responseId, response, session) => {
-    const handled = onResponse(response, session);
-    if (handled) {
-      markProcessed(responseId);
-      const retryHandle = responseRetryHandles.get(responseId);
-      if (retryHandle) {
-        retryHandle();
-        responseRetryHandles.delete(responseId);
+    if (config.enableEmojiXmlTool && emojiTags.length > 0) {
+      if (!session) {
+        log?.("warn", "\u68C0\u6D4B\u5230\u8868\u60C5\u6807\u8BB0\u4F46\u7F3A\u5C11\u4F1A\u8BDD\u4E0A\u4E0B\u6587");
+      } else {
+        const items = emojiTags.map((attrs) => ({
+          messageId: String(attrs.message_id || "").trim(),
+          emojiId: String(attrs.emoji_id || "").trim()
+        })).filter((item) => item.messageId && item.emojiId);
+        if (items.length > 0) handled = true;
+        for (const item of items) {
+          try {
+            await runEmoji({
+              session,
+              messageId: item.messageId,
+              emojiId: item.emojiId,
+              protocol,
+              log
+            });
+          } catch (error) {
+            log?.("warn", "XML \u89E6\u53D1\u8868\u60C5\u5931\u8D25", error);
+          }
+        }
       }
     }
-    return handled;
-  };
-  const retryResponse = (responseId, response) => {
-    if (responseRetryHandles.has(responseId) || processedResponses.has(responseId))
-      return;
-    const handle = ctx.setTimeout(() => {
-      responseRetryHandles.delete(responseId);
-      if (processedResponses.has(responseId)) return;
-      const session = resolveSession();
+    if (config.enableDeleteXmlTool && deleteTags.length > 0) {
       if (!session) {
-        if (config.debugLogging) {
-          log?.("warn", "\u62E6\u622A\u5230\u539F\u59CB\u8F93\u51FA\u4F46\u7F3A\u5C11\u4F1A\u8BDD\u4E0A\u4E0B\u6587\uFF0CXML \u5DE5\u5177\u4E0D\u4F1A\u6267\u884C", {
-            length: response.length
-          });
+        log?.("warn", "\u68C0\u6D4B\u5230\u64A4\u56DE\u6807\u8BB0\u4F46\u7F3A\u5C11\u4F1A\u8BDD\u4E0A\u4E0B\u6587");
+      } else {
+        const messageIds = deleteTags.map((attrs) => String(attrs.message_id || "").trim()).filter(Boolean);
+        if (messageIds.length > 0) handled = true;
+        for (const messageId of messageIds) {
+          try {
+            await runDelete({ session, messageId, log });
+          } catch (error) {
+            log?.("warn", "XML \u89E6\u53D1\u64A4\u56DE\u5931\u8D25", error);
+          }
         }
-        runResponse(responseId, response, null);
-        return;
       }
-      runResponse(responseId, response, session);
-    }, 200);
-    responseRetryHandles.set(responseId, handle);
-  };
-  const attach = () => {
-    const characterService = ctx.chatluna_character;
-    if (!characterService) return false;
-    if (activeService !== characterService) {
-      activeService = characterService;
-      collectorBound = false;
-    }
-    if (!collectorBound && typeof characterService.collect === "function") {
-      characterService.collect(async (session) => {
-        const guildId = session?.guildId || session?.channelId || session?.userId || null;
-        currentGuildId = guildId;
-        if (guildId) sessionMap.set(guildId, session);
-        lastSession = session;
-        pendingSessions.push({ key: guildId, session, timestamp: Date.now() });
-      });
-      collectorBound = true;
     }
-    const loggerService = characterService.logger;
-    if (!loggerService || typeof loggerService.debug !== "function")
-      return false;
-    const tagged = loggerService.debug;
-    if (!tagged[RAW_INTERCEPTOR_TAG]) {
-      restore();
-      const raw = loggerService.debug.bind(loggerService);
-      const wrapped = (...args) => {
-        raw(...args);
-        const message = args[0];
-        if (typeof message !== "string") return;
-        const prefix = ["model response: ", "model response:\n"].find(
-          (item) => message.startsWith(item)
-        );
-        if (!prefix) return;
-        const response = message.substring(prefix.length);
-        if (!response) return;
-        const responseId = `${Date.now()}:${responseSeq++}`;
-        const session = resolveSession();
-        if (!session) {
-          retryResponse(responseId, response);
-          return;
+    return handled;
+  };
+}
+
+// src/features/xml-tools/temp-runtime.ts
+var GET_TEMP_TAG = /* @__PURE__ */ Symbol.for("chatlunaToolboxXmlGetTempRuntime");
+var GET_TEMP_ORIGINAL = /* @__PURE__ */ Symbol.for("chatlunaToolboxXmlOriginalGetTemp");
+var GET_TEMP_LISTENERS = /* @__PURE__ */ Symbol.for("chatlunaToolboxXmlGetTempListeners");
+var PUSH_DISPATCHER = /* @__PURE__ */ Symbol.for("chatlunaToolboxXmlPushDispatcher");
+function getMessageType(message) {
+  if (!message) return "";
+  if (typeof message._getType === "function") {
+    return String(message._getType() || "").trim().toLowerCase();
+  }
+  return String(message.type || message.role || "").trim().toLowerCase();
+}
+function isAssistantMessage(message) {
+  const type = getMessageType(message);
+  return type === "ai" || type === "assistant";
+}
+function extractText(value) {
+  if (typeof value === "string") return value;
+  if (value == null) return "";
+  if (Array.isArray(value)) {
+    return value.map((item) => extractText(item)).join("");
+  }
+  if (typeof value !== "object") return "";
+  const record = value;
+  if (typeof record.text === "string") return record.text;
+  if (record.content !== void 0 && record.content !== value) {
+    return extractText(record.content);
+  }
+  if (Array.isArray(record.children)) return extractText(record.children);
+  if (typeof record.attrs === "object" && record.attrs) {
+    const attrs = record.attrs;
+    if (typeof attrs.content === "string") return attrs.content;
+    if (typeof attrs.text === "string") return attrs.text;
+  }
+  return "";
+}
+function getResponseText(message) {
+  if (!isAssistantMessage(message)) return "";
+  return extractText(message?.content ?? message?.text).trim();
+}
+function getDispatcher(messages) {
+  return messages[PUSH_DISPATCHER] ?? null;
+}
+function setDispatcher(messages, dispatcher) {
+  const record = messages;
+  if (!dispatcher) {
+    delete record[PUSH_DISPATCHER];
+    return;
+  }
+  Object.defineProperty(record, PUSH_DISPATCHER, {
+    value: dispatcher,
+    configurable: true,
+    enumerable: false,
+    writable: true
+  });
+}
+function attachDispatcher(messages, log) {
+  const existing = getDispatcher(messages);
+  if (existing) return existing;
+  const listeners = /* @__PURE__ */ new Set();
+  const processedMessages = /* @__PURE__ */ new WeakSet();
+  const originalPush = messages.push;
+  const patchedPush = function patchedPush2(...items) {
+    const result = originalPush.apply(this, items);
+    for (const item of items) {
+      if (!item || typeof item !== "object") continue;
+      const message = item;
+      if (!isAssistantMessage(message)) continue;
+      if (processedMessages.has(item)) continue;
+      const response = getResponseText(message);
+      if (!response) continue;
+      processedMessages.add(item);
+      for (const listener of Array.from(listeners)) {
+        try {
+          listener(message);
+        } catch (error) {
+          log?.("warn", "\u5904\u7406 XML completionMessages \u76D1\u542C\u5668\u5931\u8D25", error);
         }
-        runResponse(responseId, response, session);
-      };
-      wrapped[RAW_INTERCEPTOR_TAG] = true;
-      loggerService.debug = wrapped;
-      activeLogger = loggerService;
-      originalDebug = raw;
+      }
     }
-    return true;
+    return result;
   };
-  const stopFastRetry = () => {
-    if (!fastRetryHandle) return;
-    fastRetryHandle();
-    fastRetryHandle = null;
+  const dispatcher = {
+    messages,
+    originalPush,
+    patchedPush,
+    listeners,
+    processedMessages
   };
-  const startFastRetry = () => {
-    if (fastRetryHandle) return;
-    fastRetryHandle = ctx.setInterval(() => {
-      if (isActive()) {
-        stopFastRetry();
-        return;
-      }
-      const ready = attach();
-      if (ready) {
-        log?.("info", "\u539F\u59CB\u8F93\u51FA\u62E6\u622A\u5DF2\u6062\u590D");
-        stopFastRetry();
+  messages.push = patchedPush;
+  setDispatcher(messages, dispatcher);
+  return dispatcher;
+}
+function restoreDispatcher(dispatcher) {
+  if (dispatcher.messages.push === dispatcher.patchedPush) {
+    dispatcher.messages.push = dispatcher.originalPush;
+  }
+  setDispatcher(dispatcher.messages, null);
+}
+function registerGetTempListener(service, listener) {
+  const getTemp = service.getTemp;
+  if (typeof getTemp !== "function") return null;
+  const serviceRecord = service;
+  let listeners = serviceRecord[GET_TEMP_LISTENERS];
+  if (!listeners) {
+    listeners = /* @__PURE__ */ new Set();
+    serviceRecord[GET_TEMP_LISTENERS] = listeners;
+  }
+  if (!serviceRecord[GET_TEMP_TAG]) {
+    Object.defineProperty(serviceRecord, GET_TEMP_ORIGINAL, {
+      value: getTemp,
+      configurable: true,
+      enumerable: false,
+      writable: true
+    });
+    service.getTemp = async (...args) => {
+      const originalGetTemp = serviceRecord[GET_TEMP_ORIGINAL];
+      const temp = await originalGetTemp?.apply(
+        service,
+        args
+      );
+      const activeListeners = serviceRecord[GET_TEMP_LISTENERS];
+      const session = args[0] && typeof args[0] === "object" ? args[0] : null;
+      if (temp && activeListeners?.size) {
+        for (const handler of Array.from(activeListeners)) {
+          handler(temp, session);
+        }
       }
-    }, 3e3);
+      return temp;
+    };
+    serviceRecord[GET_TEMP_TAG] = true;
+  }
+  listeners.add(listener);
+  return () => {
+    const currentListeners = serviceRecord[GET_TEMP_LISTENERS];
+    currentListeners?.delete(listener);
+    if (currentListeners?.size) return;
+    const originalGetTemp = serviceRecord[GET_TEMP_ORIGINAL];
+    if (originalGetTemp) service.getTemp = originalGetTemp;
+    delete serviceRecord[GET_TEMP_ORIGINAL];
+    delete serviceRecord[GET_TEMP_TAG];
+    delete serviceRecord[GET_TEMP_LISTENERS];
   };
-  const ensureActive = () => {
-    if (isActive()) {
-      stopFastRetry();
-      return;
-    }
-    const ready = attach();
-    if (!ready) {
-      startFastRetry();
-    }
+}
+function createCharacterTempXmlRuntime(params) {
+  const { getCharacterService, processModelResponse, log } = params;
+  let attachedDispatchers = [];
+  let stopListening = null;
+  const compactAttachedDispatchers = () => {
+    attachedDispatchers = attachedDispatchers.filter(
+      (entry) => Boolean(entry.ref.deref())
+    );
+  };
+  const hasAttachedListener = (dispatcher) => {
+    compactAttachedDispatchers();
+    return attachedDispatchers.some(
+      (entry) => entry.ref.deref() === dispatcher
+    );
   };
-  const startMonitor = () => {
-    if (monitorHandle) return;
-    monitorHandle = ctx.setInterval(() => {
-      ensureActive();
-    }, 5e3);
+  const detachDispatcherListener = (entry) => {
+    const dispatcher = entry.ref.deref();
+    if (!dispatcher) return;
+    dispatcher.listeners.delete(entry.listener);
+    if (dispatcher.listeners.size === 0) restoreDispatcher(dispatcher);
   };
-  const stopMonitor = () => {
-    if (!monitorHandle) return;
-    monitorHandle();
-    monitorHandle = null;
+  const handleTemp = (temp, session) => {
+    const messages = temp.completionMessages;
+    if (!Array.isArray(messages) || typeof messages.push !== "function") return;
+    const dispatcher = attachDispatcher(messages, log);
+    if (hasAttachedListener(dispatcher)) return;
+    const listener = (message) => {
+      const response = getResponseText(message);
+      if (!response) return;
+      void processModelResponse({ response, session }).catch((error) => {
+        log?.("warn", "\u5904\u7406 XML \u6A21\u578B\u54CD\u5E94\u5931\u8D25", error);
+      });
+    };
+    dispatcher.listeners.add(listener);
+    attachedDispatchers.push({ ref: new WeakRef(dispatcher), listener });
   };
   return {
     start: () => {
-      const delay = 3e3;
-      if (startupHandle) return;
-      startupHandle = ctx.setTimeout(() => {
-        startupHandle = null;
-        const ready = attach();
-        if (ready) {
-          log?.("info", "\u5DF2\u542F\u7528\u539F\u59CB\u8F93\u51FA\u62E6\u622A\u6A21\u5F0F");
-        } else {
-          log?.("warn", "chatluna_character \u670D\u52A1\u4E0D\u53EF\u7528\uFF0C\u5C06\u6BCF3\u79D2\u91CD\u8BD5\u4E00\u6B21");
-          startFastRetry();
+      if (stopListening) return true;
+      const service = getCharacterService();
+      if (!service) return false;
+      const detach = registerGetTempListener(service, handleTemp);
+      if (!detach) return false;
+      stopListening = () => {
+        compactAttachedDispatchers();
+        for (const entry of attachedDispatchers) {
+          detachDispatcherListener(entry);
         }
-        startMonitor();
-      }, delay);
+        attachedDispatchers = [];
+        detach();
+        stopListening = null;
+      };
+      return true;
     },
     stop: () => {
-      if (startupHandle) {
-        startupHandle();
-        startupHandle = null;
-      }
-      stopFastRetry();
-      stopMonitor();
-      for (const handle of responseRetryHandles.values()) handle();
-      responseRetryHandles.clear();
-      processedResponses.clear();
-      restore();
-      sessionMap.clear();
-      pendingSessions.length = 0;
-      currentGuildId = null;
-      lastSession = null;
-      activeService = null;
-    }
-  };
-}
-
-// src/features/xml-tools/parser.ts
-function unique(values) {
-  return Array.from(new Set(values.map((item) => item.trim()).filter(Boolean)));
-}
-function parseXmlActions(response) {
-  const pokeMatches = Array.from(response.matchAll(/<poke\s+id="([^"]+)"\s*\/>/gi));
-  const emojiMatches = Array.from(
-    response.matchAll(/<emoji\s+message_id="([^"]+)"\s+emoji_id="([^"]+)"\s*\/>/gi)
-  );
-  const deleteMatches = Array.from(response.matchAll(/<delete\s+message_id="([^"]+)"\s*\/?>/gi));
-  return {
-    pokeUserIds: unique(pokeMatches.map((match) => String(match[1] || ""))),
-    emojis: emojiMatches.map((match) => ({
-      messageId: String(match[1] || "").trim(),
-      emojiId: String(match[2] || "").trim()
-    })).filter((item) => item.messageId && item.emojiId),
-    deleteMessageIds: unique(deleteMatches.map((match) => String(match[1] || "")))
+      stopListening?.();
+    },
+    isActive: () => Boolean(stopListening)
   };
 }
 
 // src/features/xml-tools/register.ts
 function registerXmlTools(deps) {
   const { ctx, config, protocol, log } = deps;
-  const interceptor = createXmlInterceptor({
-    ctx,
-    config,
-    log,
-    onResponse: (response, session) => {
-      const actions = parseXmlActions(response);
-      let handled = false;
-      if (config.enablePokeXmlTool && actions.pokeUserIds.length > 0) {
-        if (!session) {
-          log?.("warn", "\u68C0\u6D4B\u5230\u6233\u4E00\u6233\u6807\u8BB0\u4F46\u7F3A\u5C11\u4F1A\u8BDD\u4E0A\u4E0B\u6587");
-        } else {
-          handled = true;
-          for (const userId of actions.pokeUserIds) {
-            void sendPoke({ session, userId, protocol, log }).catch((error) => {
-              log?.("warn", "XML \u89E6\u53D1 poke \u5931\u8D25", error);
-            });
-          }
-        }
-      }
-      if (config.enableEmojiXmlTool && actions.emojis.length > 0) {
-        if (!session) {
-          log?.("warn", "\u68C0\u6D4B\u5230\u8868\u60C5\u6807\u8BB0\u4F46\u7F3A\u5C11\u4F1A\u8BDD\u4E0A\u4E0B\u6587");
-        } else {
-          handled = true;
-          for (const item of actions.emojis) {
-            void sendMsgEmoji({
-              session,
-              messageId: item.messageId,
-              emojiId: item.emojiId,
-              protocol,
-              log
-            }).catch((error) => {
-              log?.("warn", "XML \u89E6\u53D1\u8868\u60C5\u5931\u8D25", error);
-            });
-          }
-        }
-      }
-      if (config.enableDeleteXmlTool && actions.deleteMessageIds.length > 0) {
-        if (!session) {
-          log?.("warn", "\u68C0\u6D4B\u5230\u64A4\u56DE\u6807\u8BB0\u4F46\u7F3A\u5C11\u4F1A\u8BDD\u4E0A\u4E0B\u6587");
-        } else {
-          handled = true;
-          for (const messageId of actions.deleteMessageIds) {
-            void sendDeleteMessage({ session, messageId, log }).catch(
-              (error) => {
-                log?.("warn", "XML \u89E6\u53D1\u64A4\u56DE\u5931\u8D25", error);
-              }
-            );
-          }
-        }
-      }
-      return handled;
-    }
+  const processor = createXmlProcessor({ config, protocol, log });
+  const runtime = createCharacterTempXmlRuntime({
+    getCharacterService: () => ctx.chatluna_character,
+    processModelResponse: processor,
+    log
   });
   return {
-    start: () => interceptor.start(),
-    stop: () => interceptor.stop()
+    start: () => runtime.start(),
+    stop: () => runtime.stop(),
+    isActive: () => runtime.isActive()
   };
 }
 
@@ -1509,22 +1516,35 @@ function apply(ctx, config) {
   };
   const log = createLogger(ctx, config);
   let xmlRuntime = null;
+  let characterCtx = 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 });
-    if (config.enablePokeXmlTool || config.enableEmojiXmlTool || config.enableDeleteXmlTool) {
-      xmlRuntime = registerXmlTools({ ctx, config, protocol, log });
-      xmlRuntime.start();
+    xmlRuntime = registerXmlTools({ ctx, config, protocol, log });
+    if (characterCtx && xmlRuntime.start()) {
       log("info", "XML \u5DE5\u5177\u5DF2\u542F\u7528");
     }
     log("info", "toolbox \u521D\u59CB\u5316\u5B8C\u6210");
   };
   const dispose = () => {
+    characterCtx = null;
     xmlRuntime?.stop();
     xmlRuntime = null;
   };
+  if (config.enablePokeXmlTool || config.enableEmojiXmlTool || config.enableDeleteXmlTool) {
+    ctx.inject(["chatluna_character"], (innerCtx) => {
+      characterCtx = innerCtx;
+      if (!xmlRuntime) return;
+      const started = xmlRuntime.start();
+      if (started) log("info", "XML \u5DE5\u5177\u5DF2\u542F\u7528");
+      innerCtx.on("dispose", () => {
+        if (characterCtx === innerCtx) characterCtx = null;
+        xmlRuntime?.stop();
+      });
+    });
+  }
   ctx.on("dispose", dispose);
   if (ctx.root.lifecycle.isActive) {
     void initializeServices();
@@ -1547,6 +1567,7 @@ function apply(ctx, config) {
   apply,
   callOneBotAPI,
   createBotInfoProvider,
+  createCharacterTempXmlRuntime,
   createDeleteMessageTool,
   createGroupInfoProvider,
   createLogger,
@@ -1556,12 +1577,12 @@ function apply(ctx, config) {
   createSetMsgEmojiTool,
   createSetProfileTool,
   createUserInfoProvider,
-  createXmlInterceptor,
+  createXmlProcessor,
   ensureOneBotSession,
   getSession,
   inject,
   name,
-  parseXmlActions,
+  parseSelfClosingXmlTags,
   registerNativeTools,
   registerVariables,
   registerXmlTools,

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chatluna-toolbox",
   "description": "为 ChatLuna 提供更多原生工具、XML 工具与变量。",
-  "version": "0.0.5",
+  "version": "0.0.6",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [