wechaty-puppet-matrix 0.0.54 → 0.0.55

package/src/puppet-matrix.ts

@@ -12,6 +12,7 @@ import Client from './matrix/service/request.js'
 import { getUnixTime } from 'date-fns'
 
 import { mergeWithFallback } from './matrix/utils/index.js'
+import { detectImageExt, fetchRemoteHead, replaceExt } from './matrix/utils/file-type.js'
 import path from 'path'
 // 参考
 import { CacheManager, RoomMemberMap } from './matrix/cache-manager.js'
@@ -1894,7 +1895,25 @@ class PuppetMatrix extends PUPPET.Puppet {
       log.error(`Can not get file for message: ${messageId}`)
     }
     const url = imageInfo.is_ww ? await this._client?.downloadWWMedia(imageInfo, 1) : await this._client?.downloadImage(imageInfo)
-    return this._fileBoxFromUrl(url, messageId)
+    return this._fileBoxFromUrl(url, messageId, url ? await this._imageNameByHead(url) : undefined)
+  }
+
+  /**
+   * 下载接口固定返回 .png 文件名，实际多为 JPEG。读取文件头修正扩展名，识别失败时保持原名
+   */
+  private async _imageNameByHead (url: string): Promise<string | undefined> {
+    const head = await fetchRemoteHead(url)
+    const ext = head && detectImageExt(head)
+    if (!ext) {
+      return undefined
+    }
+    let rawName = 'image'
+    try {
+      rawName = decodeURIComponent(new URL(url).pathname.split('/').pop() || rawName)
+    } catch {
+      // url 解析失败时用默认名
+    }
+    return replaceExt(rawName, ext)
   }
 
   private async _getMessageAudioFileBox (messageId: string, messagePayload: MessagePayload): Promise<FileBoxInterface> {