dsh-account-pool 0.1.0

package/lib/trae-bridge.js

@@ -0,0 +1,266 @@
+/**
+ * Trae SSE → OpenAI chat-completions SSE 协议转换。
+ *
+ * ## 为什么必须有这一层
+ *
+ * shim 对 pi-ai 承诺的是 OpenAI 格式的流（`data: {"choices":[{"delta":...}]}`）。
+ * WorkBuddy 的上游本来就是 OpenAI 格式，直通即可；**Trae 的上游是自己的
+ * 命名事件格式**（`event:metadata` / `event:output` / `event:done`…）。
+ *
+ * 没有这一层时 shim 把 Trae 原生事件原样转发，pi-ai 认不出任何一个，
+ * 等不到 `delta`，撑到流空闲超时判失败——表现为「很慢、几乎都失败」，
+ * 而直连上游实测 1.5 秒就正常出字。协议转换是 Trae 接入的必要环节，
+ * 参照实现（dsh-connect-trae 的 solo-bridge）也是这么做的。
+ *
+ * ## 事件映射
+ *
+ * ```text
+ * Trae                          →  OpenAI chunk
+ * ─────────────────────────────────────────────────────────
+ * output {response,             →  delta.content
+ *         reasoning_content,       delta.reasoning_content
+ *         tool_calls[]}            delta.tool_calls
+ * token_usage {prompt_tokens…}  →  （累积，附在末尾 chunk 上）
+ * error {code,message}          →  流错误（上游失败要暴露，不能装成功）
+ * done {finish_reason}          →  finish_reason chunk + data: [DONE]
+ * 其余（metadata/timing…）      →  忽略
+ * ```
+ */
+
+import { randomUUID } from 'node:crypto'
+
+/** 上游错误，转换层抛出、shim 捕获后按失败处置（换号或回错）。 */
+export class TraeUpstreamError extends Error {
+  constructor(message, code) {
+    super(message)
+    this.name = 'TraeUpstreamError'
+    this.code = code
+  }
+}
+
+/** 容错 JSON 解析：Trae 的 data 偶有非 JSON 行，不能让它断流。 */
+function parseJson(text) {
+  try {
+    return JSON.parse(text)
+  } catch {
+    return undefined
+  }
+}
+
+/**
+ * 把上游的原始响应体转换成 OpenAI SSE 流。
+ *
+ * @param {object} response 上游 fetch 的 Response（body 为 Trae 命名事件流）
+ * @param {string} model DSH 请求的模型名（原样回填进 chunk，便于 pi-ai 对账）
+ * @returns {object} 带 OpenAI 格式 body 的新 Response
+ */
+export function bridgeTraeSseToOpenAI(response, model) {
+  const source = response.body
+  if (source === null) {
+    return new Response(null, { status: 502 })
+  }
+
+  const id = `chatcmpl-${randomUUID().replaceAll('-', '').slice(0, 24)}`
+  const created = Math.floor(Date.now() / 1000)
+  const decoder = new TextDecoder()
+  const encoder = new TextEncoder()
+
+  /** 工具调用是否出现过：决定 finish_reason 用 tool_calls 还是 stop。 */
+  let sawToolCalls = false
+  /** 是否已发过 finish chunk：done 与流 EOF 可能都触发，只发一次。 */
+  let emittedFinish = false
+  /** 上游是否已明确结束（done 事件）。 */
+  let upstreamEnded = false
+  /** 上游业务错误：error 事件或 code>=4000 都算，不能伪装成成功。 */
+  let upstreamError
+  /** token_usage 累积：OpenAI 惯例附在最后一个 chunk 上。 */
+  let usage
+
+  /** 组一个 OpenAI chunk 的字节序列。 */
+  const chunk = (delta, finishReason = null) => encoder.encode(
+    `data: ${JSON.stringify({
+      id,
+      object: 'chat.completion.chunk',
+      created,
+      model,
+      choices: [{ index: 0, delta, finish_reason: finishReason }],
+      ...(usage === undefined ? {} : { usage }),
+    })}\n\n`,
+  )
+
+  /**
+   * 处理一条 Trae 事件。
+   *
+   * Trae 事件的 data 有两种形态：裸字符串（progress_notice）和 JSON 对象，
+   * 都要容错。
+   */
+  const consume = (event) => {
+    if (event.data === '[DONE]') {
+      // Trae 偶尔也发 [DONE]；真正收尾在流 EOF 处统一处理
+      return
+    }
+    const payload = parseJson(event.data)
+
+    // —— 错误：error 事件，或任何带 code>=4000 的载荷 ——
+    const code = typeof payload?.code === 'number' ? payload.code : undefined
+    if (event.event === 'error' || (code !== undefined && code >= 4000)) {
+      upstreamError = new TraeUpstreamError(
+        typeof payload?.message === 'string' && payload.message !== ''
+          ? payload.message
+          : `Trae 上游错误（code ${code ?? '?'}）`,
+        code,
+      )
+      return
+    }
+
+    // —— 用量 ——
+    if (event.event === 'token_usage') {
+      usage = {
+        ...(typeof payload?.prompt_tokens === 'number' ? { prompt_tokens: payload.prompt_tokens } : {}),
+        ...(typeof payload?.completion_tokens === 'number' ? { completion_tokens: payload.completion_tokens } : {}),
+        ...(typeof payload?.total_tokens === 'number' ? { total_tokens: payload.total_tokens } : {}),
+      }
+      return
+    }
+
+    // —— 结束 ——
+    if (event.event === 'done' || typeof payload?.finish_reason === 'string') {
+      upstreamEnded = true
+      if (!emittedFinish) {
+        emittedFinish = true
+        if (upstreamError !== undefined) {
+          controller.error(upstreamError)
+          return
+        }
+        // 工具调用过的流 finish_reason 要用 tool_calls，pi-ai 靠它走工具循环
+        controller.enqueue(chunk({}, sawToolCalls ? 'tool_calls' : (payload?.finish_reason ?? 'stop')))
+      }
+      return
+    }
+
+    // —— 增量内容 ——
+    if (
+      event.event === 'output'
+      || payload?.response !== undefined
+      || payload?.reasoning_content !== undefined
+    ) {
+      const delta = {}
+      const text = typeof payload?.response === 'string' ? payload.response : ''
+      const reasoning = typeof payload?.reasoning_content === 'string' ? payload.reasoning_content : ''
+      if (text !== '') delta.content = text
+      if (reasoning !== '') delta.reasoning_content = reasoning
+      // tool_calls：Trae 用 function_call{name,arguments}，转回 OpenAI 的 function 形态
+      const calls = normalizeToolCalls(payload?.tool_calls)
+      if (calls.length > 0) {
+        sawToolCalls = true
+        delta.tool_calls = calls
+      }
+      // 全空的 delta 不发：pi-ai 对空 chunk 宽容，但没必要制造噪音
+      if (Object.keys(delta).length > 0) controller.enqueue(chunk(delta))
+    }
+    // metadata / timing_cost / progress_notice 等与内容无关，忽略
+  }
+
+  let controller
+  const stream = new ReadableStream({
+    start(c) {
+      controller = c
+    },
+    async pull() {
+      // 读取由 start 里的异步循环驱动；pull 不使用
+    },
+    cancel(reason) {
+      return source.cancel(reason)
+    },
+  })
+
+  // 驱动循环：读上游 → 解析 SSE → 转换 → 推给消费者
+  void (async () => {
+    const reader = source.getReader()
+    let buffer = ''
+    let currentEvent
+    const lines = []
+
+    /** 逐行喂 SSE；空行 = 一条事件结束。 */
+    const feed = (line) => {
+      if (line === '') {
+        if (lines.length > 0) {
+          const dataLines = []
+          for (const l of lines) {
+            if (l.startsWith('data:')) dataLines.push(l.slice(5).trimStart())
+          }
+          consume({ event: currentEvent, data: dataLines.join('\n') })
+          lines.length = 0
+          currentEvent = undefined
+        }
+        return
+      }
+      if (line.startsWith('event:')) currentEvent = line.slice(6).trim()
+      lines.push(line)
+    }
+
+    try {
+      while (true) {
+        const next = await reader.read()
+        if (next.done) break
+        buffer += decoder.decode(next.value, { stream: true })
+        let idx
+        while ((idx = buffer.indexOf('\n')) !== -1) {
+          const line = buffer.slice(0, idx).replace(/\r$/, '')
+          buffer = buffer.slice(idx + 1)
+          feed(line)
+        }
+      }
+      if (buffer !== '') feed(buffer)
+
+      // 上游 EOF：即使没有显式 done，也要合成 finish（Trae 有时省略）
+      if (!upstreamEnded) {
+        emittedFinish = true
+        if (upstreamError !== undefined) {
+          controller.error(upstreamError)
+          return
+        }
+        controller.enqueue(chunk({}, sawToolCalls ? 'tool_calls' : 'stop'))
+      }
+      if (upstreamError !== undefined) {
+        // done 已到但错误在先：仍要以失败收场
+        controller.error(upstreamError)
+        return
+      }
+      controller.enqueue(encoder.encode('data: [DONE]\n\n'))
+      controller.close()
+    } catch (error) {
+      controller.error(error)
+    } finally {
+      reader.releaseLock()
+    }
+  })()
+
+  return new Response(stream, { status: 200, headers: { 'Content-Type': 'text/event-stream' } })
+
+  /** tool_calls 归一化：Trae 的 function_call 形态 → OpenAI 形态。 */
+  function normalizeToolCalls(value) {
+    if (!Array.isArray(value)) return []
+    const calls = []
+    for (const raw of value) {
+      if (typeof raw !== 'object' || raw === null) continue
+      const fn = typeof raw.function_call === 'object' && raw.function_call !== null
+        ? raw.function_call
+        : typeof raw.function === 'object' && raw.function !== null
+          ? raw.function
+          : {}
+      const name = typeof fn.name === 'string' ? fn.name : ''
+      if (name === '') continue // 上游要求 Name 必填，空的不转
+      calls.push({
+        index: typeof raw.index === 'number' ? raw.index : calls.length,
+        ...(typeof raw.id === 'string' ? { id: raw.id } : {}),
+        type: 'function',
+        function: {
+          name,
+          ...(typeof fn.arguments === 'string' ? { arguments: fn.arguments } : {}),
+        },
+      })
+    }
+    return calls
+  }
+}

package/lib/trae-storage.js

@@ -0,0 +1,215 @@
+/**
+ * 解析 Trae 桌面端的 storage.json 凭证。
+ *
+ * ## 为什么走这条路
+ *
+ * Trae 授权页是把 token **fetch 投递到 `127.0.0.1:18080`** 的（不是导航跳转）。
+ * 当 DSH 跑在 NAS、浏览器在另一台电脑时，那个 127.0.0.1 指的是**用户自己的
+ * 电脑**，请求被拒 → 页面报「网络错误」，而且因为是 fetch 而非导航，
+ * 地址栏不会出现带 token 的链接，**连手动复制都拿不到**。
+ *
+ * 所以改为「从桌面端已登录的凭证里读」：
+ * Trae 桌面端把登录态写在 `storage.json` 的 `iCubeAuthInfo://icube.cloudide`
+ * 字段里（AES-CBC 加密）。用户把这个文件拷过来即可，不需要在 NAS 上
+ * 复现整个 OAuth 流程。
+ *
+ * ## 关键结论：解密不依赖本机
+ *
+ * 密钥完全由「硬编码盐值 + 密文自带的随机数」派生：
+ *
+ * ```text
+ * salt    = SALT_A xor SALT_B          （或 aes-private 用 SALT_C xor SALT_D）
+ * first   = sha512(random)             （random 取自密文第 7..38 字节）
+ * derived = sha512(first + salt)
+ * key     = derived[0..16]   iv = derived[16..32]
+ * ```
+ *
+ * 没有机器码、没有 DPAPI、没有系统密钥——**任何机器上都能解**。
+ * 明文前 64 字节是 sha512 校验值，用来验证解密正确。
+ */
+
+import { createDecipheriv, createHash } from 'node:crypto'
+
+/** storage.json 里存放鉴权信息的键。 */
+export const AUTH_STORAGE_KEY = 'iCubeAuthInfo://icube.cloudide'
+
+/** 设备中心 id 的键前缀，用于取 deviceId。 */
+const DEVICE_CENTER_PREFIX = 'iCubeAuthInfo://icube-dc:'
+
+// 硬编码盐值。这些是 Trae 客户端自身的常量，不是可配置项。
+const SALT_A = Uint8Array.from([82, 9, 106, 213, 48, 54, 165, 56, 191, 64, 163, 158, 129, 243, 215, 251, 124, 227, 57, 130, 155, 47, 255, 135, 52, 142, 67, 68, 196, 222, 233, 203, 84, 123, 148, 50, 166, 194, 35, 61, 238, 76, 149, 11, 66, 250, 195, 78, 8, 46, 161, 102, 40, 217, 36, 178, 118, 91, 162, 73, 109, 139, 209, 37])
+const SALT_B = Uint8Array.from([31, 221, 168, 51, 136, 7, 199, 49, 177, 18, 16, 89, 39, 128, 236, 95, 96, 81, 127, 169, 25, 181, 74, 13, 45, 229, 122, 159, 147, 201, 156, 239, 160, 224, 59, 77, 174, 42, 245, 176, 200, 235, 187, 60, 131, 83, 153, 97, 23, 43, 4, 126, 186, 119, 214, 38, 225, 105, 20, 99, 85, 33, 12, 125])
+const SALT_C = Uint8Array.from([191, 192, 216, 250, 122, 246, 220, 97, 31, 254, 98, 27, 8, 72, 71, 176, 135, 99, 96, 18, 127, 101, 203, 104, 211, 102, 191, 125, 37, 72, 150, 156, 51, 229, 121, 35, 17, 153, 141, 177, 110, 131, 150, 128, 172, 255, 254, 6, 18, 140, 55, 62, 236, 249, 135, 64, 135, 12, 117, 4, 89, 149, 168, 209])
+const SALT_D = Uint8Array.from([246, 204, 26, 232, 232, 70, 129, 109, 223, 146, 169, 242, 23, 241, 105, 145, 50, 196, 165, 42, 254, 120, 3, 54, 244, 207, 209, 85, 53, 6, 138, 106, 175, 148, 31, 204, 186, 186, 165, 182, 87, 142, 49, 10, 39, 110, 26, 154, 86, 56, 173, 125, 18, 64, 198, 225, 99, 99, 83, 82, 191, 134, 76, 170])
+
+/** 两种密文头，对应两套盐值。 */
+const HEADER_AES = Buffer.from([0x74, 0x63, 0x05, 0x10, 0x00, 0x00])
+const HEADER_AES_PRIVATE = Buffer.from([18, 57, 32, 32, 2, 3])
+
+/** 逐字节异或。 */
+function xor(a, b) {
+  return Buffer.from(a.map((value, index) => value ^ (b[index] ?? 0)))
+}
+
+function saltFor(header) {
+  if (header.equals(HEADER_AES)) return xor(SALT_A, SALT_B)
+  if (header.equals(HEADER_AES_PRIVATE)) return xor(SALT_C, SALT_D)
+  throw new Error('无法识别的凭证加密头（可能不是 Trae 的 storage.json）')
+}
+
+/**
+ * 解密 `iCubeAuthInfo://icube.cloudide` 字段的值。
+ *
+ * @param {string} encoded base64 密文
+ * @returns {string} 明文 JSON 字符串
+ */
+export function decryptAuthValue(encoded) {
+  const buffer = Buffer.from(String(encoded ?? '').trim(), 'base64')
+  if (buffer.length <= 102) throw new Error('凭证密文太短，可能复制不完整')
+
+  const salt = saltFor(buffer.subarray(0, 6))
+  const random = buffer.subarray(6, 38)
+  const encrypted = buffer.subarray(38)
+
+  const first = createHash('sha512').update(random).digest()
+  const derived = createHash('sha512').update(Buffer.concat([first, salt])).digest()
+  const decipher = createDecipheriv('aes-128-cbc', derived.subarray(0, 16), derived.subarray(16, 32))
+  const decrypted = Buffer.concat([decipher.update(encrypted), decipher.final()])
+
+  if (decrypted.length < 64) throw new Error('凭证明文太短')
+  // 前 64 字节是明文自身的 sha512，用来验证解密是否成功
+  const expected = decrypted.subarray(0, 64)
+  const plaintext = decrypted.subarray(64)
+  const actual = createHash('sha512').update(plaintext).digest()
+  if (!expected.equals(actual)) {
+    throw new Error('凭证完整性校验失败（文件可能被改动或复制不完整）')
+  }
+  return plaintext.toString('utf8')
+}
+
+/** 值是明文 JSON 就直接用，是密文就解。 */
+function readAuthJson(value) {
+  const text = String(value ?? '').trim()
+  if (text === '') throw new Error('凭证字段为空')
+  const plaintext = text.startsWith('{') ? text : decryptAuthValue(text)
+  let parsed
+  try {
+    parsed = JSON.parse(plaintext)
+  } catch {
+    throw new Error('凭证解密后不是合法 JSON')
+  }
+  if (parsed === null || typeof parsed !== 'object' || Array.isArray(parsed)) {
+    throw new Error('凭证结构异常')
+  }
+  return parsed
+}
+
+/** 取第一个非空字符串。 */
+function firstString(...values) {
+  for (const value of values) {
+    if (typeof value === 'string' && value.trim() !== '') return value.trim()
+  }
+  return undefined
+}
+
+/** 时间字段可能是秒或毫秒，统一成毫秒。 */
+function toMs(value) {
+  // 数字：秒（<1e12）或毫秒
+  if (typeof value === 'number' && Number.isFinite(value) && value > 0) {
+    return value > 1e12 ? value : value * 1000
+  }
+  // 字符串：可能是纯数字时间戳，也可能是 ISO 日期。
+  // Trae 各版本字段类型不统一——只处理数字会丢掉 ISO 形式，
+  // 导致 expiresAtMs 落成 0、token 被误判为已过期而立刻触发刷新。
+  if (typeof value === 'string' && value.trim() !== '') {
+    const numeric = Number(value)
+    if (Number.isFinite(numeric) && numeric > 0) {
+      return numeric > 1e12 ? numeric : numeric * 1000
+    }
+    const parsed = Date.parse(value)
+    return Number.isFinite(parsed) ? parsed : undefined
+  }
+  return undefined
+}
+
+/** 从 storage.json 文本里取设备标识。 */
+function readDeviceId(storage) {
+  const keys = Object.keys(storage).filter(key => key.startsWith(DEVICE_CENTER_PREFIX))
+  // 只有一个候选时才用，避免多个设备 id 时猜错
+  if (keys.length !== 1) return { deviceId: '', machineId: '' }
+  const deviceId = keys[0].slice(DEVICE_CENTER_PREFIX.length)
+  const machineId = firstString(storage['telemetry.machineId']) ?? ''
+  return { deviceId, machineId }
+}
+
+/**
+ * 解析 Trae 桌面端的 storage.json。
+ *
+ * 支持两种输入：
+ *   1. 完整的 storage.json 内容（含 iCubeAuthInfo://icube.cloudide 键）
+ *   2. 直接就是那个字段的值（密文或明文 JSON）
+ *
+ * @param {string} text 文件内容或字段值
+ * @returns {object} 归一化后的凭证
+ */
+export function parseTraeStorage(text) {
+  const raw = String(text ?? '').trim()
+  if (raw === '') throw new Error('内容为空')
+
+  let authValue
+  let device = { deviceId: '', machineId: '' }
+
+  if (raw.startsWith('{')) {
+    let document
+    try {
+      document = JSON.parse(raw)
+    } catch {
+      throw new Error('不是合法的 JSON')
+    }
+    if (document[AUTH_STORAGE_KEY] !== undefined) {
+      // 情况 1：完整 storage.json
+      authValue = document[AUTH_STORAGE_KEY]
+      device = readDeviceId(document)
+    } else if (document.token !== undefined || document.accessToken !== undefined) {
+      // 情况 2：已经是鉴权对象本身
+      authValue = raw
+    } else {
+      throw new Error(
+        `这份 JSON 里没有 ${AUTH_STORAGE_KEY} 字段，`
+        + '可能不是 Trae 的 storage.json（或选错了 edition 的文件）',
+      )
+    }
+  } else {
+    // 密文
+    authValue = raw
+  }
+
+  const auth = readAuthJson(authValue)
+  const accessToken = firstString(auth.token, auth.accessToken)
+  if (accessToken === undefined) throw new Error('凭证里没有 token/accessToken')
+
+  const account = typeof auth.account === 'object' && auth.account !== null ? auth.account : undefined
+  const refreshToken = firstString(auth.refreshToken) ?? ''
+  const userId = firstString(auth.userId, auth.uid) ?? ''
+
+  // 区域：优先凭证自带的 userRegion（大小写都可能），据它决定上游主机
+  const regionRaw = firstString(
+    typeof auth.userRegion === 'string' ? auth.userRegion : undefined,
+    auth.userRegion?.region,
+    auth.region,
+  )
+  const region = regionRaw?.toLowerCase() === 'sg' ? 'sg' : 'cn'
+
+  return {
+    accessToken,
+    refreshToken,
+    userId,
+    accountName: firstString(account?.username) ?? '',
+    host: firstString(auth.host) ?? '',
+    region,
+    expiresAtMs: toMs(auth.expiredAt ?? auth.expiresAt) ?? 0,
+    refreshExpiresAtMs: toMs(auth.refreshExpiredAt ?? auth.refreshExpiresAt),
+    deviceId: device.deviceId,
+    machineId: device.machineId,
+  }
+}