koishi-plugin-p-draw 1.3.0 → 1.3.1

package/index.js

@@ -150,6 +150,7 @@ const {
   animaOotdI2IWorkflow, buildI2IWorkflow, customWorkflow,
 } = require('./lib/workflows')
 const { outputImages, waitComfyResult } = require('./lib/comfy')
+const { materializeImageSource } = require('./lib/media')
 const {
   MULTI_PERSON_NEGATIVE_TAGS, buildMultiPersonPlanPrompt, parseMultiPersonPlan,
   renderMultiPersonCharacter, multiPersonAutoSize,
@@ -484,9 +485,12 @@ exports.apply = async function apply(ctx, cfg) {
     }
   }
 
-  // 发图：引用用户触发指令的原消息，失败回退普通发送
-  async function sendImagesWithQuote(session, outputs) {
-    const imageElements = outputs.map(src => h.image(src))
+  // 发图：引用用户触发指令的原消息，失败回退普通发送
+  async function sendImagesWithQuote(session, outputs) {
+    const imageElements = await Promise.all(outputs.map(async (src) => {
+      const materialized = await materializeImageSource(src)
+      return Buffer.isBuffer(materialized) ? h.image(materialized) : h.image(src)
+    }))
     try {
       await session.send(h.quote(session.messageId) + imageElements.join(''))
     } catch (e) {

package/lib/media.js

@@ -0,0 +1,22 @@
+// 消息媒体来源处理。
+// OneBot 不一定能读取机器人进程本地的 file:// 路径，因此发送前转换为 Buffer。
+
+const fsp = require('fs/promises')
+const { fileURLToPath } = require('url')
+
+async function materializeImageSource(source, readFile = fsp.readFile) {
+  if (typeof source !== 'string' || !/^file:\/\//i.test(source)) return source
+  let filePath
+  try {
+    filePath = fileURLToPath(source)
+  } catch (e) {
+    return source
+  }
+  try {
+    return await readFile(filePath)
+  } catch (e) {
+    return source
+  }
+}
+
+module.exports = { materializeImageSource }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-p-draw",
   "description": "p点-绘图，连接本地 ComfyUI 生图。支持自然语言优化、尺寸选择、画师组、固定角色、生成队列，需要配合 p-qiandao 使用",
-  "version": "1.3.0",
+  "version": "1.3.1",
   "main": "index.js",
   "files": [
     "index.js",