weflow-cli 1.16.0 → 1.17.0

package/src/services/wechatMedia.ts

@@ -1,166 +1,166 @@
-/**
- * 官方通道（ilink）**入站媒体**的落地：把 CDN 上的密文拉下来、解密、写到本地磁盘。
- *
- * 为什么值得做：入站图片/语音/文件/视频此前只解析出"有这一项"，`filePath` 是空串，于是助手拿到消息
- * 也不知道图里是什么（那四项的落地当时是显式延后的）。下载与解密的本事本来就有
- * （`WechatClient.downloadMedia`：`<cdn>/download?encrypted_query_param=…` + AES-128-ECB + 去 PKCS#7），
- * 缺的只是把它接上入站这条路。
- *
- * 字段名与优先级照**厂商自己的实现**（`third-party/WeKnora/internal/im/wechat/longpoll.go`）：
- * 图片优先用 `image_item.aeskey`（十六进制字符串，**不走 base64**），退回 `media.aes_key`（base64）；
- * 语音/文件/视频只有 `media.aes_key`。两种编码都由 `WechatClient.parseMediaAesKey` 认。
- *
- * 三条边界写在这里，不指望调用方记得：
- * 1. **它不是默认行为** —— 打开 `wechatMediaDownload` 才发生网络访问；
- * 2. **文件名永远由本地决定** —— 服务端给的名字只取 basename 并过滤字符，`../` 之类必须在这里被吃掉；
- * 3. **超过上限就不写**（`WECHAT_MEDIA_MAX_BYTES`），失败不留半个文件 —— 半个文件比没有更难查。
- */
-import { basename, join } from 'path'
-import { mkdir, writeFile } from 'fs/promises'
-
-/** 单件上限。够大（手机拍的视频也能过），又不至于一次 CDN 事故把磁盘写满 */
-export const WECHAT_MEDIA_MAX_BYTES = 25 * 1024 * 1024
-
-/** 落地目录：默认仓库相对 `output/wechat-media/`（`output/` 已在 .gitignore 里），可用环境变量改（测试用临时目录） */
-export function wechatMediaDir(env: NodeJS.ProcessEnv = process.env): string {
-  const fromEnv = env.WEFLOW_WECHAT_MEDIA_DIR
-  return fromEnv && fromEnv.trim() ? fromEnv : join('output', 'wechat-media')
-}
-
-export type InboundMediaType = 'image' | 'record' | 'file' | 'video'
-
-/** item.type 的数字编码（与服务端 item_list 一致） */
-export function mediaTypeOf(itemType: number): InboundMediaType | null {
-  if (itemType === 2) return 'image'
-  if (itemType === 3) return 'record'
-  if (itemType === 4) return 'file'
-  if (itemType === 5) return 'video'
-  return null
-}
-
-/** 服务端给的名字只当作"一个字符串"用：取 basename、去掉控制字符与路径字符、去掉前导点、限长 */
-export function sanitizeFileName(name: string): string {
-  const base = basename(String(name || '').replace(/\\/g, '/')).trim()
-  return base
-    .replace(/[\u0000-\u001f<>:"|?*]/g, '_')
-    .replace(/^\.+/, '')
-    .trim()
-    .slice(0, 80)
-}
-
-/** 按魔数认扩展名。认不出就返回空串（调用方退回按类型给的后缀），不猜 */
-export function extFromMagic(buf: Buffer): string {
-  const b = buf
-  if (b.length >= 3 && b[0] === 0xff && b[1] === 0xd8 && b[2] === 0xff) return '.jpg'
-  if (b.length >= 8 && b[0] === 0x89 && b[1] === 0x50 && b[2] === 0x4e && b[3] === 0x47) return '.png'
-  if (b.length >= 6 && b.subarray(0, 6).toString('latin1').startsWith('GIF8')) return '.gif'
-  if (b.length >= 12 && b.subarray(0, 4).toString('latin1') === 'RIFF'
-    && b.subarray(8, 12).toString('latin1') === 'WEBP') return '.webp'
-  if (b.length >= 2 && b[0] === 0x42 && b[1] === 0x4d) return '.bmp'
-  if (b.length >= 12 && b.subarray(4, 8).toString('latin1') === 'ftyp') return '.mp4'
-  if (b.length >= 4 && b[0] === 0x1a && b[1] === 0x45 && b[2] === 0xdf && b[3] === 0xa3) return '.webm'
-  if (b.length >= 5 && b.subarray(0, 5).toString('latin1') === '#!AMR') return '.amr'
-  if (b.length >= 4 && b.subarray(0, 4).toString('latin1') === 'OggS') return '.ogg'
-  if (b.length >= 2 && b[0] === 0x49 && b[1] === 0x44 && b[2] === 0x33) return '.mp3'
-  if (b.length >= 12 && b.subarray(0, 4).toString('latin1') === 'RIFF'
-    && b.subarray(8, 12).toString('latin1') === 'WAVE') return '.wav'
-  return ''
-}
-
-/** 该媒体类型在服务端字段树里的节点名 */
-function mediaNode(item: any, mediaType: InboundMediaType): any {
-  if (mediaType === 'image') return item?.image_item
-  if (mediaType === 'record') return item?.voice_item
-  if (mediaType === 'file') return item?.file_item
-  return item?.video_item
-}
-
-/** 下载参数（`encrypt_query_param`）。取不到就没法下载——**这时候不猜、不留空文件** */
-export function pickEncryptParam(item: any, mediaType: InboundMediaType): string {
-  const node = mediaNode(item, mediaType)
-  return String(node?.media?.encrypt_query_param || '')
-}
-
-/**
- * AES key 的取值优先级。**图片这一支必须照厂商实现**：`image_item.aeskey` 是十六进制字符串，
- * 比 `media.aes_key`（base64）更靠前；漏了这个优先级，图片会解出一堆乱码却不报错。
- */
-export function pickAesKeyValue(item: any, mediaType: InboundMediaType): string {
-  const node = mediaNode(item, mediaType)
-  if (mediaType === 'image' && node?.aeskey) return String(node.aeskey)
-  return String(node?.media?.aes_key || '')
-}
-
-export interface SaveMediaOptions {
-  item: any
-  mediaType: InboundMediaType
-  /** 同一条消息里的第几项（一条消息可以带多张图） */
-  index: number
-  /** 名字前缀：优先用服务端给的 client_id */
-  messageId: string
-  dir?: string
-  /** 注入的下载器（默认由调用方传 `client.downloadMedia`），测试里换成假的 */
-  download: (encryptedQueryParam: string, aesKeyValue: string) => Promise<Buffer>
-  /** 文件类消息里服务端给的文件名 */
-  serverFileName?: string
-  maxBytes?: number
-}
-
-export interface SaveMediaResult {
-  /** 成功是磁盘路径；失败是空串（**绝不返回"看起来成功"的半成品**） */
-  path: string
-  /** 失败原因，进日志用 */
-  reason: string
-}
-
-/**
- * 下载一件入站媒体并落盘。失败一律返回 `{ path: '', reason }` 且**不留下文件**：
- * 助手与导出拿到空 path 时知道"没有这玩意"，比拿到一个 0 字节文件强。
- */
-export async function saveInboundMedia(opts: SaveMediaOptions): Promise<SaveMediaResult> {
-  const { item, mediaType, index, messageId, download } = opts
-  const cap = opts.maxBytes ?? WECHAT_MEDIA_MAX_BYTES
-
-  const encryptParam = pickEncryptParam(item, mediaType)
-  if (!encryptParam) return { path: '', reason: '服务端没给 encrypt_query_param' }
-
-  const aesKeyValue = pickAesKeyValue(item, mediaType)
-  if (!aesKeyValue) return { path: '', reason: '服务端没给 aes key' }
-
-  let bytes: Buffer
-  try {
-    bytes = await download(encryptParam, aesKeyValue)
-  } catch (error: any) {
-    return { path: '', reason: `下载或解密失败：${error?.message || error}` }
-  }
-  if (!bytes || bytes.length === 0) return { path: '', reason: '下载结果为空' }
-  if (bytes.length > cap) {
-    return { path: '', reason: `超过上限（${bytes.length} > ${cap} 字节）` }
-  }
-
-  const byMagic = extFromMagic(bytes)
-  const fallbackExt = mediaType === 'image' ? '.jpg' : ''
-  const prefix = sanitizeFileName(messageId) || 'msg'
-  let name: string
-  if (mediaType === 'file' && opts.serverFileName) {
-    const safe = sanitizeFileName(opts.serverFileName)
-    if (safe) {
-      const hasExt = /\.[A-Za-z0-9]{1,8}$/.test(safe)
-      name = `${prefix}-${index}-${hasExt || !byMagic ? safe : safe + byMagic}`
-    } else {
-      name = `${prefix}-${index}${byMagic}`
-    }
-  } else {
-    name = `${prefix}-${index}${byMagic || fallbackExt}`
-  }
-
-  const dir = opts.dir || wechatMediaDir()
-  const full = join(dir, name)
-  try {
-    await mkdir(dir, { recursive: true })
-    await writeFile(full, bytes)
-  } catch (error: any) {
-    return { path: '', reason: `写盘失败：${error?.message || error}` }
-  }
-  return { path: full, reason: '' }
-}
+/**
+ * 官方通道（ilink）**入站媒体**的落地：把 CDN 上的密文拉下来、解密、写到本地磁盘。
+ *
+ * 为什么值得做：入站图片/语音/文件/视频此前只解析出"有这一项"，`filePath` 是空串，于是助手拿到消息
+ * 也不知道图里是什么（那四项的落地当时是显式延后的）。下载与解密的本事本来就有
+ * （`WechatClient.downloadMedia`：`<cdn>/download?encrypted_query_param=…` + AES-128-ECB + 去 PKCS#7），
+ * 缺的只是把它接上入站这条路。
+ *
+ * 字段名与优先级照**厂商自己的实现**（`third-party/WeKnora/internal/im/wechat/longpoll.go`）：
+ * 图片优先用 `image_item.aeskey`（十六进制字符串，**不走 base64**），退回 `media.aes_key`（base64）；
+ * 语音/文件/视频只有 `media.aes_key`。两种编码都由 `WechatClient.parseMediaAesKey` 认。
+ *
+ * 三条边界写在这里，不指望调用方记得：
+ * 1. **它不是默认行为** —— 打开 `wechatMediaDownload` 才发生网络访问；
+ * 2. **文件名永远由本地决定** —— 服务端给的名字只取 basename 并过滤字符，`../` 之类必须在这里被吃掉；
+ * 3. **超过上限就不写**（`WECHAT_MEDIA_MAX_BYTES`），失败不留半个文件 —— 半个文件比没有更难查。
+ */
+import { basename, join } from 'path'
+import { mkdir, writeFile } from 'fs/promises'
+
+/** 单件上限。够大（手机拍的视频也能过），又不至于一次 CDN 事故把磁盘写满 */
+export const WECHAT_MEDIA_MAX_BYTES = 25 * 1024 * 1024
+
+/** 落地目录：默认仓库相对 `output/wechat-media/`（`output/` 已在 .gitignore 里），可用环境变量改（测试用临时目录） */
+export function wechatMediaDir(env: NodeJS.ProcessEnv = process.env): string {
+  const fromEnv = env.WEFLOW_WECHAT_MEDIA_DIR
+  return fromEnv && fromEnv.trim() ? fromEnv : join('output', 'wechat-media')
+}
+
+export type InboundMediaType = 'image' | 'record' | 'file' | 'video'
+
+/** item.type 的数字编码（与服务端 item_list 一致） */
+export function mediaTypeOf(itemType: number): InboundMediaType | null {
+  if (itemType === 2) return 'image'
+  if (itemType === 3) return 'record'
+  if (itemType === 4) return 'file'
+  if (itemType === 5) return 'video'
+  return null
+}
+
+/** 服务端给的名字只当作"一个字符串"用：取 basename、去掉控制字符与路径字符、去掉前导点、限长 */
+export function sanitizeFileName(name: string): string {
+  const base = basename(String(name || '').replace(/\\/g, '/')).trim()
+  return base
+    .replace(/[\u0000-\u001f<>:"|?*]/g, '_')
+    .replace(/^\.+/, '')
+    .trim()
+    .slice(0, 80)
+}
+
+/** 按魔数认扩展名。认不出就返回空串（调用方退回按类型给的后缀），不猜 */
+export function extFromMagic(buf: Buffer): string {
+  const b = buf
+  if (b.length >= 3 && b[0] === 0xff && b[1] === 0xd8 && b[2] === 0xff) return '.jpg'
+  if (b.length >= 8 && b[0] === 0x89 && b[1] === 0x50 && b[2] === 0x4e && b[3] === 0x47) return '.png'
+  if (b.length >= 6 && b.subarray(0, 6).toString('latin1').startsWith('GIF8')) return '.gif'
+  if (b.length >= 12 && b.subarray(0, 4).toString('latin1') === 'RIFF'
+    && b.subarray(8, 12).toString('latin1') === 'WEBP') return '.webp'
+  if (b.length >= 2 && b[0] === 0x42 && b[1] === 0x4d) return '.bmp'
+  if (b.length >= 12 && b.subarray(4, 8).toString('latin1') === 'ftyp') return '.mp4'
+  if (b.length >= 4 && b[0] === 0x1a && b[1] === 0x45 && b[2] === 0xdf && b[3] === 0xa3) return '.webm'
+  if (b.length >= 5 && b.subarray(0, 5).toString('latin1') === '#!AMR') return '.amr'
+  if (b.length >= 4 && b.subarray(0, 4).toString('latin1') === 'OggS') return '.ogg'
+  if (b.length >= 2 && b[0] === 0x49 && b[1] === 0x44 && b[2] === 0x33) return '.mp3'
+  if (b.length >= 12 && b.subarray(0, 4).toString('latin1') === 'RIFF'
+    && b.subarray(8, 12).toString('latin1') === 'WAVE') return '.wav'
+  return ''
+}
+
+/** 该媒体类型在服务端字段树里的节点名 */
+function mediaNode(item: any, mediaType: InboundMediaType): any {
+  if (mediaType === 'image') return item?.image_item
+  if (mediaType === 'record') return item?.voice_item
+  if (mediaType === 'file') return item?.file_item
+  return item?.video_item
+}
+
+/** 下载参数（`encrypt_query_param`）。取不到就没法下载——**这时候不猜、不留空文件** */
+export function pickEncryptParam(item: any, mediaType: InboundMediaType): string {
+  const node = mediaNode(item, mediaType)
+  return String(node?.media?.encrypt_query_param || '')
+}
+
+/**
+ * AES key 的取值优先级。**图片这一支必须照厂商实现**：`image_item.aeskey` 是十六进制字符串，
+ * 比 `media.aes_key`（base64）更靠前；漏了这个优先级，图片会解出一堆乱码却不报错。
+ */
+export function pickAesKeyValue(item: any, mediaType: InboundMediaType): string {
+  const node = mediaNode(item, mediaType)
+  if (mediaType === 'image' && node?.aeskey) return String(node.aeskey)
+  return String(node?.media?.aes_key || '')
+}
+
+export interface SaveMediaOptions {
+  item: any
+  mediaType: InboundMediaType
+  /** 同一条消息里的第几项（一条消息可以带多张图） */
+  index: number
+  /** 名字前缀：优先用服务端给的 client_id */
+  messageId: string
+  dir?: string
+  /** 注入的下载器（默认由调用方传 `client.downloadMedia`），测试里换成假的 */
+  download: (encryptedQueryParam: string, aesKeyValue: string) => Promise<Buffer>
+  /** 文件类消息里服务端给的文件名 */
+  serverFileName?: string
+  maxBytes?: number
+}
+
+export interface SaveMediaResult {
+  /** 成功是磁盘路径；失败是空串（**绝不返回"看起来成功"的半成品**） */
+  path: string
+  /** 失败原因，进日志用 */
+  reason: string
+}
+
+/**
+ * 下载一件入站媒体并落盘。失败一律返回 `{ path: '', reason }` 且**不留下文件**：
+ * 助手与导出拿到空 path 时知道"没有这玩意"，比拿到一个 0 字节文件强。
+ */
+export async function saveInboundMedia(opts: SaveMediaOptions): Promise<SaveMediaResult> {
+  const { item, mediaType, index, messageId, download } = opts
+  const cap = opts.maxBytes ?? WECHAT_MEDIA_MAX_BYTES
+
+  const encryptParam = pickEncryptParam(item, mediaType)
+  if (!encryptParam) return { path: '', reason: '服务端没给 encrypt_query_param' }
+
+  const aesKeyValue = pickAesKeyValue(item, mediaType)
+  if (!aesKeyValue) return { path: '', reason: '服务端没给 aes key' }
+
+  let bytes: Buffer
+  try {
+    bytes = await download(encryptParam, aesKeyValue)
+  } catch (error: any) {
+    return { path: '', reason: `下载或解密失败：${error?.message || error}` }
+  }
+  if (!bytes || bytes.length === 0) return { path: '', reason: '下载结果为空' }
+  if (bytes.length > cap) {
+    return { path: '', reason: `超过上限（${bytes.length} > ${cap} 字节）` }
+  }
+
+  const byMagic = extFromMagic(bytes)
+  const fallbackExt = mediaType === 'image' ? '.jpg' : ''
+  const prefix = sanitizeFileName(messageId) || 'msg'
+  let name: string
+  if (mediaType === 'file' && opts.serverFileName) {
+    const safe = sanitizeFileName(opts.serverFileName)
+    if (safe) {
+      const hasExt = /\.[A-Za-z0-9]{1,8}$/.test(safe)
+      name = `${prefix}-${index}-${hasExt || !byMagic ? safe : safe + byMagic}`
+    } else {
+      name = `${prefix}-${index}${byMagic}`
+    }
+  } else {
+    name = `${prefix}-${index}${byMagic || fallbackExt}`
+  }
+
+  const dir = opts.dir || wechatMediaDir()
+  const full = join(dir, name)
+  try {
+    await mkdir(dir, { recursive: true })
+    await writeFile(full, bytes)
+  } catch (error: any) {
+    return { path: '', reason: `写盘失败：${error?.message || error}` }
+  }
+  return { path: full, reason: '' }
+}

package/src/utils/python.ts

@@ -1,6 +1,7 @@
 import { execFileSync } from 'child_process'
 import { delimiter, join } from 'path'
-import { accessSync, constants } from 'fs'
+import { accessSync, constants, existsSync } from 'fs'
+import { homedir } from 'os'
 
 let cached: string | null = null
 
@@ -17,6 +18,26 @@ function isExecutable(path: string): boolean {
   }
 }
 
+/** `weflow-cli setup` 建的自管 venv 的解释器路径；没建过就是 null。 */
+export function managedVenvPython(): string | null {
+  const exe = process.platform === 'win32'
+    ? join(homedir(), '.weflow-cli', 'venv', 'Scripts', 'python.exe')
+    : join(homedir(), '.weflow-cli', 'venv', 'bin', 'python')
+  return existsSync(exe) ? exe : null
+}
+
+/**
+ * 候选序：自管 venv 永远排最前，其余按调用方给的序。纯函数，供测试钉住
+ * "建了 venv 就该优先用它"这件事——它坏了的表现是 setup 装好了包、
+ * 运行时却仍在用 PATH 上的解释器，两边各自"正常"。
+ */
+export function orderCandidates(
+  managed: string | null,
+  rest: string[],
+): string[] {
+  return managed ? [managed, ...rest.filter(c => c !== managed)] : [...rest]
+}
+
 /** 按顺序返回 PATH 中所有可用的 Python 解释器（含命令名候选） */
 function candidates(): string[] {
   const found: string[] = []
@@ -55,13 +76,16 @@ function hasSqlcipher(cmd: string): boolean {
 }
 
 function detect(): string {
+  // 自管 venv（`weflow-cli setup`）优先：它就是为了"这台机器有一个能
+  // import sqlcipher3 的 Python"而存在的。venv 坏了（装完又被卸了包）会
+  // 自然落到下面的 PATH 候选——行为退回 setup 之前，而不是更糟。
+  const ordered = orderCandidates(managedVenvPython(), candidates())
   // 优先选择能 import sqlcipher3 的解释器（NT 解密核心依赖），
   // 避免命中 PATH 中仅有基础库的裸 Python 环境（如被 IDE 注入的 VM Python 抢占）。
-  const list = candidates()
-  for (const cmd of list) {
+  for (const cmd of ordered) {
     if (canRun(cmd) && hasSqlcipher(cmd)) return cmd
   }
-  for (const cmd of list) {
+  for (const cmd of ordered) {
     if (canRun(cmd)) return cmd
   }
   return 'python'
@@ -71,3 +95,8 @@ export function getPythonCommand(): string {
   cached ??= detect()
   return cached
 }
+
+/** setup 装完后清缓存，让同一进程里后续的 getPythonCommand 看见新 venv。 */
+export function clearPythonCommandCache(): void {
+  cached = null
+}