koishi-plugin-imgdraw-selfuse 0.0.6 → 0.0.7

package/lib/index.js

@@ -121,8 +121,19 @@ async function apply(ctx, cfg) {
         apiIdx.val++;
         return api;
     }
+    // ==================== 修复：增强 HTML/XML 清理 ====================
     function cleanHtmlTags(str) {
-        return str.replace(/<<[^>]+>/g, '').trim();
+        if (!str)
+            return '';
+        // 1. 清理标准 HTML 标签
+        let cleaned = str.replace(/<<[^>]+>/g, ' ');
+        // 2. 清理 QQ XML 图片标签（如 <img src="..." file="..."/>）
+        cleaned = cleaned.replace(/<<img\s+[^>]+\/>/gi, ' ');
+        // 3. 清理其他 XML 标签
+        cleaned = cleaned.replace(/<<[^>]+>/g, ' ');
+        // 4. 清理多余空格和换行
+        cleaned = cleaned.replace(/\s+/g, ' ').trim();
+        return cleaned;
     }
     // 增强图片提取函数
     function getImageUrlFromContent(text) {
@@ -471,6 +482,18 @@ async function apply(ctx, cfg) {
         await safeSend(session, cfg.messages.generating);
         await generateWithMultipleImages(session, finalPrompt, finalUrls, cfg.img2imgModel || cfg.model);
     }
+    // ==================== 新增：从 session.elements 提取纯文本 ====================
+    function extractTextFromElements(elements) {
+        if (!elements || !Array.isArray(elements))
+            return '';
+        const textParts = [];
+        for (const el of elements) {
+            if (el.type === 'text') {
+                textParts.push(el.attrs?.content || el.attrs?.text || '');
+            }
+        }
+        return cleanHtmlTags(textParts.join(' '));
+    }
     // ==================== 命令注册 ====================
     const cmd = ctx.command(`${cfg.command} <raw:text>`, 'draw');
     cfg.aliases.forEach((alias) => cmd.alias(alias));
@@ -507,13 +530,24 @@ async function apply(ctx, cfg) {
                 return safeSend(session, cfg.messages.blacklisted);
             if (!cfg.enableImg2Img)
                 return safeSend(session, cfg.messages.img2imgDisabled);
-            const prompt = cleanHtmlTags(raw || '');
+            // ==================== 修复：从 elements 提取纯文本，避免 XML 污染 ====================
+            let prompt = '';
+            if (session.elements && session.elements.length > 0) {
+                // 有 elements，从中提取纯文本（排除图片等）
+                prompt = extractTextFromElements(session.elements);
+                if (debug)
+                    logger.info('从 elements 提取的 prompt:', prompt);
+            }
+            // 如果 elements 没有文本，回退到 raw
+            if (!prompt && raw) {
+                prompt = cleanHtmlTags(raw);
+            }
             // 检查消息中是否包含图片（QQ 合并消息）
             const messageImages = session.elements ? koishi_1.h.select(session.elements, 'img') : [];
             if (messageImages.length > 0) {
                 // 合并消息：同时包含文字和图片，直接处理
                 if (debug)
-                    logger.info(`检测到合并消息，包含 ${messageImages.length} 张图片`);
+                    logger.info(`检测到合并消息，包含 ${messageImages.length} 张图片，prompt: "${prompt}"`);
                 if (!prompt) {
                     // 如果没有文字 prompt，使用默认提示
                     await processImg2ImgWithImages(session, '请根据图片内容进行编辑', messageImages);

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-imgdraw-selfuse",
-  "description": "修改自ai-image的画图插件,支持openai兼容api,增加base64转换以适配GPTimg2，支持合并消息图生图",
-  "version": "0.0.6",
+  "description": "修改自ai-image的画图插件,支持openai兼容api,增加base64转换以适配GPTimg2，支持发送合并消息图生图",
+  "version": "0.0.7",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [