koishi-plugin-p-draw 1.5.3 → 1.5.5

package/index.js

@@ -2,6 +2,7 @@ const { Schema, h } = require('koishi')
 const fs = require('fs')
 const fsp = require('fs/promises')
 const path = require('path')
+const os = require('os')
 const crypto = require('crypto')
 const { pathToFileURL } = require('url')
 
@@ -173,10 +174,18 @@ exports.apply = async function apply(ctx, cfg) {
   // 用户自选模型偏好（userid -> unet 文件名），持久化在 p_draw_config.user_models
   if (!cfg.userModels || typeof cfg.userModels !== 'object') cfg.userModels = {}
 
-  const tempDir = path.join(__dirname, 'temp')
-  if (!fs.existsSync(tempDir)) {
-    try { fs.mkdirSync(tempDir, { recursive: true }) } catch (e) { logger.warn('无法创建临时目录：' + e.message) }
-  }
+  const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'pdraw-'))
+  // 启动时清理上次运行遗留的临时文件
+  try {
+    const leftovers = fs.readdirSync(os.tmpdir()).filter((n) => n.startsWith('pdraw-'))
+    for (const name of leftovers) {
+      const dir = path.join(os.tmpdir(), name)
+      try {
+        for (const file of fs.readdirSync(dir)) fs.unlinkSync(path.join(dir, file))
+        fs.rmdirSync(dir)
+      } catch (e) { /* ignore */ }
+    }
+  } catch (e) { /* ignore */ }
 
   const baseUrl = () => normalizeBaseUrl(cfg.comfyuiBaseUrl)
 
@@ -462,9 +471,11 @@ exports.apply = async function apply(ctx, cfg) {
     return { notices }
   }
 
-  function buildChargeNotice({ isAdmin, totalPrice, unetName, seeds }) {
-    if (isAdmin || !seeds.length) return ''
-    return `已扣除 ${totalPrice} P 点，当前模型：${unetName}，--seed=${seeds.join(',')}`
+  function buildChargeNotice({ isAdmin, totalPrice, unetName, seeds }) {
+    if (!seeds.length) return ''
+    const modelSeed = `当前模型：${unetName}，--seed=${seeds.join(',')}`
+    if (isAdmin) return modelSeed
+    return `已扣除 ${totalPrice} P 点，${modelSeed}`
   }
 
   function buildGenerationReply(session, { successCount, count, failures, notes = [], chargeNotice = '' }) {
@@ -2150,6 +2161,7 @@ exports.apply = async function apply(ctx, cfg) {
       for (const file of files) {
         try { fs.unlinkSync(path.join(tempDir, file)) } catch (e) { /* ignore */ }
       }
+      try { fs.rmdirSync(tempDir) } catch (e) { /* ignore */ }
     } catch (e) { /* ignore */ }
   })
 

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-p-draw",
   "description": "p点-绘图，连接本地 ComfyUI 生图。支持自然语言优化、尺寸选择、画师组、固定角色、生成队列，需要配合 p-qiandao 使用",
-  "version": "1.5.3",
+  "version": "1.5.5",
   "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`）。管理员/免扣费用户不显示扣费金额，但仍显示模型和 seed。
 
 ## 安装