koishi-plugin-imgdraw-selfuse 0.0.5 → 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) {
@@ -440,6 +451,49 @@ async function apply(ctx, cfg) {
             deleteAllCachedFiles(imageUrls);
         }
     }
+    // ==================== 新增：处理带图片的合并消息 ====================
+    async function processImg2ImgWithImages(session, prompt, images) {
+        const assets = ctx.assets;
+        if (!assets) {
+            await safeSend(session, cfg.messages.needAssets);
+            return;
+        }
+        // 上传所有图片
+        const uploadResults = await Promise.allSettled(images.map(img => assets.upload(img.attrs.src, 'ref_image.jpg')));
+        const imageUrls = [];
+        for (const res of uploadResults) {
+            if (res.status === 'fulfilled' && /^https?:\/\//.test(res.value)) {
+                imageUrls.push(res.value);
+            }
+        }
+        if (imageUrls.length === 0) {
+            await safeSend(session, cfg.messages.needAssets);
+            return;
+        }
+        // 限制图片数量
+        const finalUrls = imageUrls.slice(0, cfg.maxImages);
+        if (finalUrls.length < imageUrls.length) {
+            logger.warn(`图片数量超过限制，已截取前 ${cfg.maxImages} 张`);
+        }
+        // 构建 prompt
+        const finalPrompt = cfg.img2imgPrompt
+            .replace('{url}', finalUrls.join(', '))
+            .replace('{prompt}', prompt);
+        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));
@@ -464,8 +518,10 @@ async function apply(ctx, cfg) {
             await safeSend(session, cfg.messages.fail);
         }
     });
+    // ==================== 修改：图生图命令支持合并消息 ====================
     const imgCmd = ctx.command(`${cfg.img2imgCommand} <raw:text>`, 'imgdraw');
     cfg.img2imgAliases.forEach((alias) => imgCmd.alias(alias));
+    // 修改 action 以支持检测消息中的图片
     imgCmd.action(async ({ session }, raw) => {
         try {
             if (!session)
@@ -474,12 +530,39 @@ async function apply(ctx, cfg) {
                 return safeSend(session, cfg.messages.blacklisted);
             if (!cfg.enableImg2Img)
                 return safeSend(session, cfg.messages.img2imgDisabled);
+            // ==================== 修复：从 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} 张图片，prompt: "${prompt}"`);
+                if (!prompt) {
+                    // 如果没有文字 prompt，使用默认提示
+                    await processImg2ImgWithImages(session, '请根据图片内容进行编辑', messageImages);
+                }
+                else {
+                    await processImg2ImgWithImages(session, prompt, messageImages);
+                }
+                return;
+            }
+            // 传统模式：只发送了文字，进入等待图片状态
+            if (!prompt)
+                return safeSend(session, cfg.messages.empty);
             const assets = ctx.assets;
             if (!assets)
                 return safeSend(session, cfg.messages.needAssets);
-            const prompt = cleanHtmlTags(raw || '');
-            if (!prompt)
-                return safeSend(session, cfg.messages.empty);
             const key = `${session.guildId || 'private'}-${session.userId}`;
             if (waitingMap.has(key)) {
                 return safeSend(session, cfg.messages.alreadyWaiting);
@@ -505,7 +588,7 @@ async function apply(ctx, cfg) {
             await safeSend(session, cfg.messages.fail);
         }
     });
-    // 消息监听
+    // ==================== 修改：消息监听支持追加图片 ====================
     ctx.on('message', async (session) => {
         try {
             if (!session.elements)
@@ -514,6 +597,7 @@ async function apply(ctx, cfg) {
                 return;
             const key = `${session.guildId || 'private'}-${session.userId}`;
             const task = waitingMap.get(key);
+            // 如果没有等待任务，检查是否是新的合并消息（已在上面的 imgCmd action 中处理）
             if (!task)
                 return;
             const imgs = koishi_1.h.select(session.elements, 'img');

package/package.json

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