koishi-plugin-imgdraw-selfuse 0.0.4 → 0.0.6

package/lib/index.js

@@ -440,6 +440,37 @@ 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);
+    }
     // ==================== 命令注册 ====================
     const cmd = ctx.command(`${cfg.command} <raw:text>`, 'draw');
     cfg.aliases.forEach((alias) => cmd.alias(alias));
@@ -464,8 +495,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 +507,28 @@ async function apply(ctx, cfg) {
                 return safeSend(session, cfg.messages.blacklisted);
             if (!cfg.enableImg2Img)
                 return safeSend(session, cfg.messages.img2imgDisabled);
-            const assets = ctx.assets;
-            if (!assets)
-                return safeSend(session, cfg.messages.needAssets);
             const 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} 张图片`);
+                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 key = `${session.guildId || 'private'}-${session.userId}`;
             if (waitingMap.has(key)) {
                 return safeSend(session, cfg.messages.alreadyWaiting);
@@ -505,7 +554,7 @@ async function apply(ctx, cfg) {
             await safeSend(session, cfg.messages.fail);
         }
     });
-    // 消息监听
+    // ==================== 修改：消息监听支持追加图片 ====================
     ctx.on('message', async (session) => {
         try {
             if (!session.elements)
@@ -514,6 +563,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": "画图",
-  "version": "0.0.4",
+  "description": "修改自ai-image的画图插件,支持openai兼容api,增加base64转换以适配GPTimg2，支持合并消息图生图",
+  "version": "0.0.6",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [