koishi-plugin-p-draw 1.5.1 → 1.5.3

package/index.js

@@ -467,21 +467,30 @@ exports.apply = async function apply(ctx, cfg) {
     return `已扣除 ${totalPrice} P 点，当前模型：${unetName}，--seed=${seeds.join(',')}`
   }
 
-  function buildGenerationReply(session, { successCount, count, failures, notes = [] }) {
+  function buildGenerationReply(session, { successCount, count, failures, notes = [], chargeNotice = '' }) {
     const reply = []
+    if (chargeNotice) reply.push(chargeNotice)
     if (notes.length) reply.push(notes.join('\n'))
     if (failures.length) reply.push(session.text('.batch-partial', [successCount, count, failures.length, failures.join('；')]))
     return reply.filter(Boolean).join('\n')
   }
 
-  async function sendNotices(session, notices, opts = {}) {
-    const content = notices.filter(Boolean).join('\n')
-    if (!content) return
-    try {
-      const message = opts.quote && session.messageId ? h.quote(session.messageId) + content : content
-      await session.send(message)
-    } catch (e) {
-      logger.warn(`发送反馈消息失败：${e.message}`)
+  async function sendNotices(session, notices, opts = {}) {
+    const content = notices.filter(Boolean).join('\n')
+    if (!content) return
+    try {
+      const quoteId = session.event?.message?.id || session.messageId
+      let message = content
+      if (opts.quote && quoteId) {
+        const quote = session.platform === 'onebot'
+          ? h('reply', { id: quoteId })
+          : h.quote(quoteId)
+        message = quote + content
+      }
+      if (cfg.outputLogs && opts.quote) logger.info(`[p-draw] 发送扣费引用通知 platform=${session.platform || 'unknown'} messageId=${quoteId || 'missing'}`)
+      await session.send(message)
+    } catch (e) {
+      logger.warn(`发送反馈消息失败：${e.message}`)
     }
   }
 
@@ -1462,9 +1471,7 @@ exports.apply = async function apply(ctx, cfg) {
       unetName: unet,
       seeds,
     })
-    await sendNotices(session, [chargeNotice], { quote: true })
-
-    if (!allOutputs.length) {
+    if (!allOutputs.length) {
       if (cfg.outputLogs) logger.warn(`多人生成全部失败（${USERID}），已按张退款`)
       return session.text('.generate-failed', ['全部失败（已按张退款）'])
     }
@@ -1474,7 +1481,7 @@ exports.apply = async function apply(ctx, cfg) {
     // 发图：合并转发，不引用原指令
     await sendImagesAsForward(session, forwardOutputs)
 
-    return buildGenerationReply(session, { successCount, count, failures, notes })
+    return buildGenerationReply(session, { successCount, count, failures, notes, chargeNotice })
   }
 
   // ---------------- 权限 ----------------
@@ -1992,9 +1999,7 @@ exports.apply = async function apply(ctx, cfg) {
       unetName: unet,
       seeds,
     })
-    await sendNotices(session, [chargeNotice], { quote: true })
-
-    if (!allOutputs.length) {
+    if (!allOutputs.length) {
       if (cfg.outputLogs) logger.warn(`生成全部失败（${USERID}），已按张退款`)
       return session.text('.generate-failed', ['全部失败（已按张退款）'])
     }
@@ -2004,7 +2009,7 @@ exports.apply = async function apply(ctx, cfg) {
     // 发图：合并转发，不引用原指令
     await sendImagesAsForward(session, forwardOutputs)
 
-    return buildGenerationReply(session, { successCount, count, failures })
+    return buildGenerationReply(session, { successCount, count, failures, chargeNotice })
   }
   // ---------------- 提示词优化券交互式确认 ----------------
   // 仅全局优化关闭 + 非管理员 + 已配置 LLM（tokenOpt 分支）时进入。

package/package.json

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

package/readme.md

@@ -20,7 +20,7 @@ p点-绘图插件，连接本地 ComfyUI 生图，参考 AstrBot 的 [anima 绘
 - 生成队列（逐张顺序执行，防止争抢 GPU）。
 - `p-draw 状态` 查看 ComfyUI 连接状态与模型可用性。
 - `p-draw help`（或 `帮助`）查看全部指令帮助，包含生成 / 多人 / 批量张数 / 尺寸 / 画师组 / 状态等。
-- 发图时会以合并转发发送，每张图片后附对应的 Positive / Negative 提示词；扣费提醒单独引用触发指令，并显示实际模型和 seed（如 `已扣除 750 P 点，当前模型：animagine-xl-3.1.safetensors，--seed=123456`）。
+- 发图时会以合并转发发送，每张图片后附对应的 Positive / Negative 提示词；生成完成消息会显示实际模型和 seed（如 `已扣除 750 P 点，当前模型：animagine-xl-3.1.safetensors，--seed=123456`）。
 
 ## 安装