dsh-account-pool 0.1.0

package/lib/headers.js

@@ -0,0 +1,76 @@
+/**
+ * 上游 HTTP 请求头：所有出站请求共用一套。
+ *
+ * 为什么单独成模块：原先 accounts.js 与 upstream.js 各有一份 commonHeaders，
+ * 改一处漏一处。出站头是上游风控的判据，必须只有一处定义。
+ *
+ * 头的来源与理由（对齐 workbuddy2api 网关的 headers.go，那是它验证过的形态）：
+ *   - X-CodeBuddy-Request: 1  官方客户端的风控闸门头，所有 API 请求必带
+ *   - X-Machine-ID/X-Session-ID  按 uid 稳定派生的设备标识，跨重启不变、
+ *                                账号间互异；uid 为空时不发（上游不要求）
+ *   - Accept-Language  按区域切：国内 zh-CN、国际 en-US，与官方客户端一致
+ *   - Origin/Referer   按区域切，送错域会被 openresty 直接拒
+ */
+
+import { createHash } from 'node:crypto'
+
+/** CLI 形态 User-Agent——上游按它选通道。 */
+export const CLIENT_UA = 'CLI/2.63.2 CodeBuddy/2.63.2'
+
+/** 桌面端 User-Agent：拉国际版模型目录时必须用它才能拿到完整清单。 */
+export const DESKTOP_UA = 'WorkBuddy/5.5.2'
+
+/** 官方 IDE 的 User-Agent：/v3/config 用它才返回完整目录。 */
+export const IDE_UA = 'CodeBuddyIDE/4.12.0 CodeBuddy/4.12.0'
+
+/**
+ * 按 uid 派生一个稳定的设备标识。
+ *
+ * 与网关同算法（sha256 取前 36 个 hex 字符）：同一账号每次得到同一个值，
+ * 不同账号互不相同。上游把这些当作设备指纹，随机生成会被判为异常设备。
+ *
+ * @param {string} uid 账号 uid
+ * @param {'machine'|'session'} purpose 用途
+ * @returns {string|undefined} uid 为空时返回 undefined（调用方据此不发该头）
+ */
+export function deriveDeviceId(uid, purpose) {
+  if (typeof uid !== 'string' || uid === '') return undefined
+  return createHash('sha256').update(`wb2a:${purpose}:${uid}`).digest('hex').slice(0, 36)
+}
+
+/** 按区域返回 Accept-Language。 */
+export function acceptLanguageOf(region) {
+  return region === 'global' ? 'en-US' : 'zh-CN'
+}
+
+/**
+ * 构造所有上游请求共用的头。
+ *
+ * @param {object} options
+ * @param {string} options.origin Origin/Referer 基地址（按区域算好传进来）
+ * @param {'cn'|'global'} [options.region] 区域，决定 Accept-Language
+ * @param {string} [options.uid] 账号 uid；给了就注入设备头
+ * @param {string} [options.userAgent] 覆盖 User-Agent（拉目录时要换）
+ * @returns {Record<string,string>}
+ */
+export function commonHeaders({ origin, region = 'cn', uid, userAgent }) {
+  const headers = {
+    'content-type': 'application/json',
+    accept: 'application/json',
+    'x-requested-with': 'XMLHttpRequest',
+    origin,
+    referer: `${origin}/`,
+    'user-agent': userAgent ?? CLIENT_UA,
+    // 官方客户端风控闸门头：所有 API 请求必带，缺了可能被当机器人。
+    'x-codebuddy-request': '1',
+    'accept-language': acceptLanguageOf(region),
+  }
+
+  // 设备头：按账号稳定派生，账号间互异。uid 缺失时不发。
+  const machineId = deriveDeviceId(uid, 'machine')
+  const sessionId = deriveDeviceId(uid, 'session')
+  if (machineId !== undefined) headers['x-machine-id'] = machineId
+  if (sessionId !== undefined) headers['x-session-id'] = sessionId
+
+  return headers
+}