dsh-qq-tools 0.8.0 → 0.8.1

package/README.md

@@ -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.1
 ```
 
 ### 本地开发 / 自己改代码
@@ -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.1.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 帧拆分、超过 4MiB 的图片缩放、其他格式转码）。会自动使用插件自己装的、`SHARP_PATH` 指定的，或 DSH 宿主自带的 sharp；都找不到时 GIF 按静图处理，大图直接交给 DSH 归一化（DSH 默认单张源图上限 20MiB）。
 
 ---
 
-版本：0.8.0
+版本：0.8.1

package/lib/shared.js

@@ -354,12 +354,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 +467,80 @@ 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 提示都可能撒谎，所以只信字节。DSH 默认单张源图上限 20MiB，
+// 归一化目标 4MiB；源图超过 4MiB 时本地先缩一下，避免白白走一遍宿主重编码。
+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 +613,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 +725,21 @@ export function admitImagesFor(ctx) {
           diag('GIF 帧拆分不可用：没有加载到 sharp，动图会按静图处理')
         }
       }
+      // 以前这里直接丢弃 >5MiB 的图；DSH 实际能收 20MiB 并自动归一化，所以改成先缩放。
+      if (bytes.length > IMAGE_SOFT_LIMIT) {
+        const shrunk = await shrinkImage(bytes)
+        if (shrunk) {
+          bytes = shrunk.bytes
+          mediaType = shrunk.mediaType
+        }
+      }
       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.1",
   "description": "给大肥鱼会话用的 QQ 群/私聊工具：列群、看群记录、发群消息、发表情、收藏表情（SnowLuma 收藏夹）、好友/私聊记录、发私聊消息、私聊唤醒、群成员、戳一戳感知/戳回去、挂载会话选择。8 个工具加 1 个设置组件，可在插件页逐个开关。",
   "type": "module",
   "main": "lib/index.js",