dsh-qq-tools 0.8.0 → 0.8.2

package/README.md

@@ -99,7 +99,7 @@ tools/qq-send-poke/
 
 私聊和群聊在程序里是分开的两条路径：群聊靠轮询 @，私聊靠 WebSocket 的 `message_type=private` 事件。私聊没有观察窗，每收到一条私聊消息都会直接唤醒目标会话。
 
-- 唤醒提示里会带发送者昵称、QQ 号、`message_id` 和文字；如果消息里有图片，也会把图片一起给模型看（GIF 会拆成首/中/尾最多 3 帧，和群聊记录的处理一致）。
+- 唤醒提示里会带发送者昵称、QQ 号、`message_id` 和文字；如果消息里有图片，也会把图片一起给模型看（有 sharp 时 GIF 会拆成首/中/尾最多 3 帧；没有 sharp 时交给 DSH 显示首帧，和群聊记录的处理一致）。
 - 模型用 `qq_get_private_history` 看和对方的最近记录，用 `qq_send_private_message` 回文字，用 `qq_send_face` 回表情。
 - `qq_list_friends` 列好友，可以按昵称、备注或 QQ 号搜；`privateUsers` 非空时只列白名单里的好友。
 - `qq_get_private_history` 的消息带 `#message_id`，可以传给 `qq_send_private_message` / `qq_send_face` 的 `reply_to` 做引用回复。
@@ -128,7 +128,7 @@ dsh plugin --profile <profile> add dsh-qq-tools
 需要固定版本时：
 
 ```sh
-dsh plugin --profile <profile> add dsh-qq-tools@0.8.0
+dsh plugin --profile <profile> add dsh-qq-tools@0.8.2
 ```
 
 ### 本地开发 / 自己改代码
@@ -141,7 +141,7 @@ dsh plugin --profile <profile> add link:/path/to/dsh-qq-tools
 
 ```sh
 pnpm pack
-dsh plugin --profile <profile> add ./dsh-qq-tools-0.8.0.tgz
+dsh plugin --profile <profile> add ./dsh-qq-tools-0.8.2.tgz
 ```
 
 `dsh plugin` 会在安装后自动把这个包追加进 profile 的 `dsh.profile.bundles`，不用手改 `package.json`。卸载用：
@@ -155,8 +155,8 @@ dsh plugin --profile <profile> remove dsh-qq-tools
 - DSH 0.2.x（cordis 插件体系）
 - Node.js 22+（内置 `fetch` / `WebSocket`）
 - SnowLuma 提供 OneBot 11 HTTP 接口和 WebSocket 事件（收藏表情需要 SnowLuma；系统表情 NapCat 也能发）
-- 可选：`sharp`（GIF 帧拆分）。会自动使用插件自己装的、`SHARP_PATH` 指定的，或 DSH 宿主自带的 sharp；都找不到时 GIF 按静图处理
+- 可选：`sharp`（GIF 拆帧、尺寸被拒时的兜底重编码）。会从 `SHARP_PATH`、插件自己的 `node_modules`、DSH 宿主 runtimeDir（桌面版通过 `process.argv[2]` 传入）等位置自动查找；找不到时 GIF 交给 DSH 显示首帧，大图仍由 DSH 自动归一化（默认单张源图上限 20MiB），图片不会因此丢失。
 
 ---
 
-版本：0.8.0
+版本：0.8.2

package/lib/shared.js

@@ -34,7 +34,7 @@
 //
 // 工具本身的开关在 DSH 插件页的组件列表里：关掉某个组件，它就不再注册对应工具。
 
-import { existsSync, readFileSync, writeFileSync, mkdirSync, appendFileSync, renameSync } from 'node:fs'
+import { existsSync, readFileSync, writeFileSync, mkdirSync, appendFileSync, renameSync, statSync } from 'node:fs'
 import { join } from 'node:path'
 import { homedir } from 'node:os'
 import { randomUUID } from 'node:crypto'
@@ -47,34 +47,82 @@ const DIAG_PATH = join(DSH_HOME, 'qq-tools-diag.log')
 
 
 const requireFromHere = createRequire(import.meta.url)
+let sharpSource = ''
 
-// sharp 是可选的，按这个顺序找：SHARP_PATH → 插件自己装的 → 宿主进程能解析到的。
-// DSH 桌面版把 sharp 打在 resources/app.asar/dsh/node_modules 下，CLI 版跟着
-// @deepseek-ai/dsh 的依赖走，所以从宿主的启动入口解析最稳，不用写死安装路径。
+function sharpFromDir(dir) {
+  try {
+    const s = createRequire(join(String(dir), 'noop.js'))('sharp')
+    return s || null
+  } catch (error) {
+    return null
+  }
+}
+
+function sharpFromCache() {
+  try {
+    const cache = requireFromHere.cache || {}
+    for (const key of Object.keys(cache)) {
+      if (!/[\\/]node_modules[\\/]sharp[\\/]/.test(key)) continue
+      const mod = cache[key]
+      if (mod && mod.exports) return mod.exports
+    }
+  } catch (error) {}
+  return null
+}
+
+// sharp 是可选的，按顺序找：SHARP_PATH → 插件自己装的 → 宿主 runtime → 常见宿主目录。
+// 关键的一条是 process.argv[2]：DSH 桌面版的 dsh-desktop-host 用 argv[2] 接 runtimeDir，
+// 插件和宿主在同一个进程里，所以 runtimeDir/node_modules/sharp 就是宿主实际加载的那份。
+// 不写死安装路径；找不到时核心图片逻辑照常工作，只是不拆 GIF 帧，
+// 尺寸被宿主拒绝时也没法兜底重编码。
 function loadSharp() {
   const probes = []
-  if (process.env.SHARP_PATH) probes.push(() => requireFromHere(process.env.SHARP_PATH))
-  probes.push(() => requireFromHere('sharp'))
-  if (process.argv[1]) probes.push(() => createRequire(process.argv[1])('sharp'))
-  if (process.resourcesPath) {
-    probes.push(() => createRequire(join(process.resourcesPath, 'app.asar', 'dsh', 'package.json'))('sharp'))
+  const add = (label, fn) => probes.push({ label, fn })
+
+  if (process.env.SHARP_PATH) add('SHARP_PATH', () => requireFromHere(process.env.SHARP_PATH))
+  add('插件 node_modules', () => requireFromHere('sharp'))
+  if (process.argv[1]) add('宿主入口', () => createRequire(process.argv[1])('sharp'))
+  if (process.argv[2]) add('宿主 runtimeDir', () => sharpFromDir(process.argv[2]))
+
+  // 不同启动方式下 runtimeDir 的位置可能变；把 argv 里看起来像目录的路径都试一遍。
+  for (let i = 1; i < process.argv.length; i++) {
+    const raw = String(process.argv[i] || '')
+    if (!raw) continue
+    try {
+      if (existsSync(raw) && statSync(raw).isDirectory()) add('argv 目录 ' + raw, () => sharpFromDir(raw))
+    } catch (error) {}
   }
-  for (const p of [
-    '/usr/local/lib/node_modules/@deepseek-ai/dsh/node_modules/sharp',
-    '/usr/lib/node_modules/@deepseek-ai/dsh/node_modules/sharp',
-  ]) {
-    probes.push(() => requireFromHere(p))
+
+  if (process.cwd()) add('当前目录', () => sharpFromDir(process.cwd()))
+  if (process.resourcesPath) {
+    for (const root of [
+      join(process.resourcesPath, 'app.asar', 'dsh'),
+      join(process.resourcesPath, 'app.asar.unpacked', 'dsh'),
+      join(process.resourcesPath, 'app.asar'),
+      join(process.resourcesPath, 'app.asar.unpacked'),
+      join(process.resourcesPath, 'runtime'),
+    ]) {
+      add('resources ' + root, () => sharpFromDir(root))
+    }
   }
+  add('已加载模块缓存', sharpFromCache)
+
   for (const probe of probes) {
     try {
-      const s = probe()
-      if (s) return s
+      const s = probe.fn()
+      if (s) {
+        sharpSource = probe.label
+        return s
+      }
     } catch (error) {}
   }
   return null
 }
 
 const sharp = loadSharp()
+diag('sharp：' + (sharp
+  ? '已加载（' + sharpSource + '）'
+  : '没找到；GIF 只取首帧，大图交给 DSH 归一化，图片不会因此丢失'))
 let gifWarned = false
 
 const DEFAULT_GROUPS = []
@@ -354,12 +402,14 @@ function segText(segs) {
     .join(''))
 }
 
-// mface（商城表情/表情包）基本都是动图；普通 image 段只能从名字和摘要里猜。
+// 这里只做「优先展示动图」的排序提示，不决定 mediaType。
+// QQ 的 [动画表情] 摘要经常贴在静态 PNG/JPEG 上，所以别再信 summary 里的「动画」；
+// 真正是不是 GIF 由后面按字节嗅探，只有 mface 和 .gif 扩展名算动图候选。
 function animatedSeg(seg) {
   if (!seg || !seg.data) return false
+  if (seg.type === 'mface') return true
   const d = seg.data
-  return seg.type === 'mface' ||
-    /gif|动画/i.test(String(d.summary || '') + ' ' + String(d.file || '') + ' ' + String(d.url || ''))
+  return /\.gif(?:[?#]|$)/i.test(String(d.file || '')) || /\.gif(?:[?#]|$)/i.test(String(d.url || ''))
 }
 
 function fmtTime(sec) {
@@ -465,13 +515,81 @@ export function makeListTool() {
 }
 
 
-function mediaTypeOf(name) {
-  const s = String(name || '').toLowerCase()
-  if (s.endsWith('.png')) return 'image/png'
-  if (s.endsWith('.gif')) return 'image/gif'
-  if (s.endsWith('.webp')) return 'image/webp'
-  if (s.endsWith('.bmp')) return 'image/bmp'
-  return 'image/jpeg'
+// ---------- 图片类型与尺寸 ----------
+// DSH 附件层按真实字节复核 mediaType（IMAGE_TYPE_MISMATCH），文件名和 QQ 的
+// animated 提示都可能撒谎，所以只信字节。大图不依赖 sharp：DSH 默认单张源图
+// 上限 20MiB、归一化目标 4MiB，直接交给它处理；sharp 只用于 GIF 拆帧和
+// admit 被尺寸拒绝后的兜底重编码。
+const ADMIT_IMAGE_LIMIT = 6
+const IMAGE_SOFT_LIMIT = 4 * 1024 * 1024
+const IMAGE_MEDIA_TYPES = {
+  png: 'image/png',
+  jpeg: 'image/jpeg',
+  webp: 'image/webp',
+  gif: 'image/gif',
+}
+
+function sniffImageType(bytes) {
+  if (!bytes || bytes.length < 12) return ''
+  if (bytes[0] === 0x89 && bytes[1] === 0x50 && bytes[2] === 0x4e && bytes[3] === 0x47 &&
+      bytes[4] === 0x0d && bytes[5] === 0x0a && bytes[6] === 0x1a && bytes[7] === 0x0a) return 'image/png'
+  if (bytes[0] === 0xff && bytes[1] === 0xd8 && bytes[2] === 0xff) return 'image/jpeg'
+  if (bytes[0] === 0x47 && bytes[1] === 0x49 && bytes[2] === 0x46 && bytes[3] === 0x38 &&
+      (bytes[4] === 0x37 || bytes[4] === 0x39) && bytes[5] === 0x61) return 'image/gif'
+  if (bytes.toString('latin1', 0, 4) === 'RIFF' && bytes.toString('latin1', 8, 12) === 'WEBP') return 'image/webp'
+  return ''
+}
+
+async function sharpImageType(bytes) {
+  if (!sharp) return ''
+  try {
+    const meta = await sharp(bytes, { animated: true }).metadata()
+    return IMAGE_MEDIA_TYPES[String((meta && meta.format) || '').toLowerCase()] || ''
+  } catch (error) {
+    return ''
+  }
+}
+
+// DSH 只收 png/jpeg/webp/gif；sharp 能解的其他格式（TIFF 等）先转成它认识的格式。
+async function convertImage(bytes) {
+  if (!sharp) return null
+  try {
+    const meta = await sharp(bytes).metadata()
+    const hasAlpha = !!meta.hasAlpha
+    const out = hasAlpha
+      ? await sharp(bytes).webp({ quality: 82 }).toBuffer()
+      : await sharp(bytes).jpeg({ quality: 82 }).toBuffer()
+    return { bytes: out, mediaType: hasAlpha ? 'image/webp' : 'image/jpeg' }
+  } catch (error) {
+    return null
+  }
+}
+
+// 大图不丢：超过 DSH 归一化目标就先缩到 2048 长边并重编码，通常能压到 1MiB 以内。
+async function shrinkImage(bytes) {
+  if (!sharp || !bytes || !bytes.length) return null
+  try {
+    const meta = await sharp(bytes, { animated: false }).metadata()
+    const width = Number(meta && meta.width) || 0
+    const height = Number(meta && meta.height) || 0
+    if (!width || !height) return null
+    const hasAlpha = !!meta.hasAlpha
+    const scale = Math.min(1, 2048 / Math.max(width, height))
+    const targetWidth = Math.max(1, Math.round(width * scale))
+    const targetHeight = Math.max(1, Math.round(height * scale))
+    for (const quality of [82, 70, 58]) {
+      const pipeline = sharp(bytes, { animated: false })
+        .rotate()
+        .resize({ width: targetWidth, height: targetHeight, fit: 'inside', withoutEnlargement: true })
+      const out = hasAlpha
+        ? await pipeline.webp({ quality }).toBuffer()
+        : await pipeline.jpeg({ quality }).toBuffer()
+      if (out.length <= IMAGE_SOFT_LIMIT) {
+        return { bytes: out, mediaType: hasAlpha ? 'image/webp' : 'image/jpeg' }
+      }
+    }
+  } catch (error) {}
+  return null
 }
 
 
@@ -544,29 +662,111 @@ async function imageBytes(data, cfg) {
   return null
 }
 
+function admitErrorText(error) {
+  if (!error) return '未知错误'
+  return (error.code ? String(error.code) + '：' : '') + String(error.message || error)
+}
+
+// 单张被拒时可以重编码再试；批量/存储层错误重试没有意义。
+const RETRYABLE_IMAGE_ERRORS = new Set([
+  'IMAGE_TYPE_MISMATCH',
+  'INVALID_IMAGE',
+  'UNSUPPORTED_IMAGE_TYPE',
+  'IMAGE_TOO_LARGE',
+  'IMAGE_TOO_MANY_PIXELS',
+  'IMAGE_DIMENSION_TOO_LARGE',
+])
+
+async function admitOneImage(attachments, block) {
+  try {
+    const out = await attachments.admitPromptContent([block])
+    const hit = (Array.isArray(out) ? out : []).find((x) => x && x.type === 'image')
+    return hit ? { block: hit } : { error: new Error('附件层没有返回图片') }
+  } catch (error) {
+    return { error }
+  }
+}
+
+async function retryImageBlock(block) {
+  let bytes
+  try {
+    bytes = Buffer.from(String((block && block.data) || ''), 'base64')
+  } catch (error) {
+    return null
+  }
+  if (!bytes || !bytes.length) return null
+  const type = sniffImageType(bytes)
+  const fixed = type ? await shrinkImage(bytes) : await convertImage(bytes)
+  if (!fixed) return null
+  return { ...block, data: fixed.bytes.toString('base64'), mediaType: fixed.mediaType }
+}
+
+// 一张坏图不能拖垮整批：先整批送，失败后退回逐张，逐张失败的再重编码试一次。
+async function admitPreparedImages(attachments, prepared) {
+  const blocks = prepared.map((x) => x.block)
+  try {
+    const out = await attachments.admitPromptContent(blocks)
+    const images = (Array.isArray(out) ? out : []).filter((x) => x && x.type === 'image')
+    if (images.length === blocks.length) {
+      return images.map((block, i) => ({ caption: prepared[i].caption, block }))
+    }
+    diag('图片 admit 返回数量不符（' + images.length + '/' + blocks.length + '），改为逐张处理')
+  } catch (error) {
+    diag('图片批量 admit 失败，改为逐张重试：' + admitErrorText(error))
+  }
+  const admitted = []
+  let firstError = ''
+  for (const item of prepared) {
+    let result = await admitOneImage(attachments, item.block)
+    if (result.error && RETRYABLE_IMAGE_ERRORS.has(String(result.error.code || ''))) {
+      const retried = await retryImageBlock(item.block)
+      if (retried) result = await admitOneImage(attachments, retried)
+    }
+    if (result.block) admitted.push({ caption: item.caption, block: result.block })
+    else if (!firstError) firstError = admitErrorText(result.error)
+  }
+  if (firstError && admitted.length < prepared.length) {
+    diag('有图片没能进来（已跳过，其余保留）：' + firstError)
+  }
+  return admitted
+}
+
 export function admitImagesFor(ctx) {
   return async function admitImages(list) {
     const attachments = ctx.attachments || (typeof ctx.get === 'function' ? ctx.get('attachments') : null)
     if (!attachments || typeof attachments.admitPromptContent !== 'function') return []
     const prepared = []
     for (const x of Array.isArray(list) ? list : []) {
-      const bytes = await imageBytes(x.data, x.cfg || loadCfg())
+      if (prepared.length >= ADMIT_IMAGE_LIMIT) break
+      let bytes = await imageBytes(x.data, x.cfg || loadCfg())
       if (!bytes || !bytes.length) continue
-      if (bytes.length > 5 * 1024 * 1024) continue
       const name = String(x.name || 'qq-image.jpg')
-      const isGif = x.animated || /\.gif$/i.test(name) || (bytes[0] === 0x47 && bytes[1] === 0x49 && bytes[2] === 0x46)
-      if (isGif) {
+      // animated 只是提示，type 一律按字节来；猜错会被 DSH 按 IMAGE_TYPE_MISMATCH 整批拒掉。
+      let mediaType = sniffImageType(bytes) || (await sharpImageType(bytes))
+      if (!mediaType) {
+        const converted = await convertImage(bytes)
+        if (!converted) {
+          diag('图片跳过：认不出格式（' + name + '）')
+          continue
+        }
+        bytes = converted.bytes
+        mediaType = converted.mediaType
+      }
+      if (mediaType === 'image/gif') {
         const frames = await gifFrames(bytes, 3)
         if (frames) {
           for (const f of frames) {
-            const frameName = name.replace(/\.[^.]+$/, '') + `-gif-${f.page}.png`
             prepared.push({
               caption: String(x.caption || '') + `（GIF 第 ${f.page}/${f.pages} 帧）`,
-              block: { type: 'image', data: f.bytes.toString('base64'), mediaType: 'image/png', name: frameName },
+              block: {
+                type: 'image',
+                data: f.bytes.toString('base64'),
+                mediaType: 'image/png',
+                name: name.replace(/\.[^.]+$/, '') + `-gif-${f.page}.png`,
+              },
             })
-            if (prepared.length >= 6) break
+            if (prepared.length >= ADMIT_IMAGE_LIMIT) break
           }
-          if (prepared.length >= 6) break
           continue
         }
         if (!sharp && !gifWarned) {
@@ -574,21 +774,15 @@ export function admitImagesFor(ctx) {
           diag('GIF 帧拆分不可用：没有加载到 sharp，动图会按静图处理')
         }
       }
+      // 大图不在这里预缩放：DSH 自己会把源图归一化到 4MiB，20MiB 以内都能收。
+      // 只有 admit 因尺寸被拒时才在 retryImageBlock 里用 sharp 兜底重编码。
       prepared.push({
         caption: String(x.caption || ''),
-        block: { type: 'image', data: bytes.toString('base64'), mediaType: isGif ? 'image/gif' : mediaTypeOf(name), name },
+        block: { type: 'image', data: bytes.toString('base64'), mediaType, name },
       })
-      if (prepared.length >= 6) break
     }
     if (!prepared.length) return []
-    try {
-      const out = await attachments.admitPromptContent(prepared.map((x) => x.block))
-      const blocks = (out || []).filter((x) => x && x.type === 'image')
-      return blocks.map((block, i) => ({ caption: prepared[i] ? prepared[i].caption : '', block }))
-    } catch (error) {
-      diag('图片 admit 失败：' + String((error && error.message) || error))
-      return []
-    }
+    return admitPreparedImages(attachments, prepared)
   }
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-qq-tools",
-  "version": "0.8.0",
+  "version": "0.8.2",
   "description": "给大肥鱼会话用的 QQ 群/私聊工具：列群、看群记录、发群消息、发表情、收藏表情（SnowLuma 收藏夹）、好友/私聊记录、发私聊消息、私聊唤醒、群成员、戳一戳感知/戳回去、挂载会话选择。8 个工具加 1 个设置组件，可在插件页逐个开关。",
   "type": "module",
   "main": "lib/index.js",