dsh-tap 0.20.0

package/scripts/probe-trae-live.mjs

@@ -0,0 +1,147 @@
+#!/usr/bin/env node
+/**
+ * Live integration probe for the Trae channel (REAL network, own account).
+ *
+ * The offline suite (verify:trae-provider) locks translation logic against
+ * mocks; this script is the one manual calibration step that needs a real
+ * login — the request envelope / SSE grammar of the Trae cloud gateway is
+ * extracted from binary strings and must be confirmed against the live wire.
+ *
+ *   node scripts/probe-trae-live.mjs --login
+ *     Real OAuth device flow against api.trae.cn: prints the authorization
+ *     URL (open it, log in), waits for the 127.0.0.1 callback, then
+ *     immediately exercises a DeviceProof refresh to prove the self-held
+ *     keypair works online. Tokens never print (masked only).
+ *
+ *   node scripts/probe-trae-live.mjs --chat "你好"
+ *     One-shot real chat via the gateway's request builder; dumps the RAW
+ *     response (status/headers/body) to docs/probes/trae-chat-live-*.json
+ *     for schema correction (no credentials in the dump).
+ *
+ *   node scripts/probe-trae-live.mjs --sig raw   # flip DeviceProof encoding
+ *
+ * Uses the SAME auth store as the plugin (~/.dsh/trae-plugin-auth.json,
+ * DSH_HOME respected) — one login serves both.
+ */
+
+import { writeFileSync, mkdirSync } from 'node:fs'
+import { homedir } from 'node:os'
+import { join } from 'node:path'
+
+import { createTraeOAuth } from '../providers/trae/oauth.js'
+import { buildChatRequest, TRAE_APP_ID, TRAE_IDE_VERSION, TRAE_IDE_VERSION_CODE } from '../providers/trae/gateway.js'
+
+const DSH_HOME = process.env.DSH_HOME ?? join(homedir(), '.dsh')
+const AUTH_PATH = join(DSH_HOME, 'trae-plugin-auth.json')
+
+const SETTINGS = {
+  traeAuthBaseURL: process.env.TRAE_AUTH_BASE ?? 'https://api.trae.cn',
+  traeChatBaseURL: process.env.TRAE_CHAT_BASE ?? 'https://trae-api-cn.mchost.guru',
+  traeLoginHost: process.env.TRAE_LOGIN_HOST ?? 'https://www.trae.cn',
+}
+
+function mask(t) {
+  return typeof t === 'string' && t.length > 12 ? `${t.slice(0, 6)}…${t.slice(-4)}（${t.length} 字符）` : '(none)'
+}
+
+const { readJson, writeJson } = await import('../core/json-store.js')
+const oauth = createTraeOAuth({ readAuth: () => readJson(AUTH_PATH), writeAuth: (v) => writeJson(AUTH_PATH, v) })
+
+const args = process.argv.slice(2)
+const mode = args.includes('--login') ? 'login' : args.includes('--chat') ? 'chat' : args.includes('--sig') ? 'sig' : null
+if (!mode) {
+  console.log('用法：--login（真实登录+刷新自证）｜ --chat "文本"（真实对话，证据落盘）｜ --sig der|raw（切换签名编码）')
+  process.exit(1)
+}
+
+if (mode === 'sig') {
+  const fmt = args[args.indexOf('--sig') + 1]
+  console.log('signatureFormat →', oauth.setSignatureFormat(fmt))
+  process.exit(0)
+}
+
+if (mode === 'login') {
+  const status0 = oauth.oauthStatus()
+  if (status0.signedIn) {
+    console.log('已有登录态：', status0.account ?? '', '令牌至', new Date(status0.accessTokenExpiresAt ?? 0).toISOString())
+  }
+  const { authUrl } = await oauth.startOAuth(SETTINGS)
+  console.log('\n在浏览器打开并完成登录（10 分钟内有效）：\n')
+  console.log(authUrl, '\n')
+  process.stdout.write('等待回调')
+  for (let i = 0; i < 600; i++) {
+    await new Promise((r) => setTimeout(r, 1000))
+    process.stdout.write('.')
+    const s = oauth.oauthStatus()
+    if (s.signedIn) {
+      console.log('\n\n登录成功 ✓  账号：', s.account ?? '(GetUserInfo 未返回昵称)')
+      console.log('accessToken：', mask(readJson(AUTH_PATH).auth?.accessToken))
+      console.log('refreshToken：', mask(readJson(AUTH_PATH).auth?.refreshToken))
+      // 立即自证刷新链路（DeviceProof 线上验签）。
+      const store = readJson(AUTH_PATH)
+      store.auth.expiresAt = Date.now() - 1000 // 强制走刷新
+      writeJson(AUTH_PATH, store)
+      const cred = await oauth.resolveTraeCredential(SETTINGS)
+      if (cred && readJson(AUTH_PATH).auth?.expiresAt > Date.now() + 60_000) {
+        console.log('DeviceProof 刷新成功 ✓（自持设备密钥线上可用）新令牌：', mask(readJson(AUTH_PATH).auth.accessToken))
+      } else {
+        console.log('DeviceProof 刷新失败 ✗ —— 尝试 `node scripts/probe-trae-live.mjs --sig raw` 后重新 --login')
+      }
+      process.exit(0)
+    }
+    if (!s.pending && s.error) {
+      console.log('\n\n登录失败：', s.error)
+      process.exit(1)
+    }
+  }
+  console.log('\n\n超时（10 分钟未完成回调）')
+  process.exit(1)
+}
+
+// mode === 'chat'
+const text = args[args.indexOf('--chat') + 1] ?? '你好'
+const cred = await oauth.resolveTraeCredential(SETTINGS)
+if (!cred) {
+  console.error('未登录（先跑 --login）。')
+  process.exit(1)
+}
+const payload = { model: process.env.TRAE_MODEL ?? 'glm-5.3', stream: true, messages: [{ role: 'user', content: text }] }
+const { body: reqBody, requestId } = buildChatRequest(payload, 'probe-live')
+const token = String(cred.authorization).replace(/^Cloud-IDE-JWT\s+/, '')
+const authStore = readJson(AUTH_PATH)
+const res = await fetch(`${SETTINGS.traeChatBaseURL}/api/agent/v3/llm_utils_chat`, {
+  method: 'POST',
+  headers: {
+    'Content-Type': 'application/json',
+    Accept: 'text/event-stream',
+    Authorization: `Cloud-IDE-JWT ${token}`,
+    'X-Cloudide-Token': token,
+    'x-ide-token': token,
+    'x-app-id': TRAE_APP_ID,
+    'x-ide-version': TRAE_IDE_VERSION,
+    'x-ide-version-code': TRAE_IDE_VERSION_CODE,
+    'x-request-id': requestId,
+    ...(authStore.account?.uid ? { 'x-uid': String(authStore.account.uid) } : {}),
+    ...(authStore.device?.deviceId ? { 'x-device-id': authStore.device.deviceId } : {}),
+    ...(authStore.device?.machineId ? { 'x-machine-id': authStore.device.machineId } : {}),
+  },
+  body: JSON.stringify(reqBody),
+})
+const raw = await res.text()
+const outPath = `docs/probes/trae-chat-live-${Date.now()}.json`
+mkdirSync('docs/probes', { recursive: true })
+writeFileSync(outPath, JSON.stringify({
+  at: new Date().toISOString(),
+  request: { url: `${SETTINGS.traeChatBaseURL}/api/agent/v3/llm_utils_chat`, body: reqBody },
+  response: {
+    status: res.status,
+    contentType: res.headers.get('content-type'),
+    headers: Object.fromEntries([...res.headers].filter(([k]) => !k.includes('token') && !k.includes('authorization'))),
+    bodyHead: raw.slice(0, 8000),
+    bodyBytes: raw.length,
+  },
+}, null, 2))
+console.log(`HTTP ${res.status}（${res.headers.get('content-type')}），body ${raw.length} 字节`)
+console.log('证据落盘 →', outPath)
+console.log('\n--- body 前 1200 字节 ---')
+console.log(raw.slice(0, 1200))

package/scripts/probe-trae-max-effort.mjs

@@ -0,0 +1,179 @@
+/**
+ * 探测 Trae remote 通道是否接受 reasoning_effort=max 拼写。
+ *
+ * 目录只声明了 light/high/extra_high，但上游（trae-api-cn.mchost.guru）可能
+ * 也接受 max（CodeBuddy 侧的拼写）。本脚本用本地存储的 Trae OAuth token，
+ * 向 remote 会话端点发 reasoning_effort=max 的创建请求，看是否 200 + 正常事件流。
+ *
+ * 用法：node scripts/probe-trae-max-effort.mjs
+ *   --model deepseek-v4.1-flash  （默认）
+ *   --base https://trae-api-cn.mchost.guru
+ *   --effort max  （可重复，默认 max + extra_high 对照）
+ */
+import { readFileSync } from 'node:fs'
+import { homedir } from 'node:os'
+import { join } from 'node:path'
+
+const args = Object.fromEntries(process.argv.slice(2).reduce((acc, a, i, arr) => {
+  if (a.startsWith('--')) {
+    const k = a.slice(2)
+    const v = arr[i + 1]
+    if (v && !v.startsWith('--')) { acc.push([k, v]); arr.splice(i + 1, 1) }
+    else acc.push([k, 'true'])
+  }
+  return acc
+}, []))
+
+const MODEL = args.model || 'deepseek-v4.1-flash'
+const BASE = args.base || 'https://trae-api-cn.mchost.guru'
+const EFFORTS = args.effort
+  ? [args.effort].flat()
+  : ['max', 'extra_high'] // max = 待测；extra_high = 对照
+
+const authPath = join(homedir(), '.dsh', 'trae-plugin-auth.json')
+const authStore = JSON.parse(readFileSync(authPath, 'utf8'))
+const token = authStore.auth?.accessToken
+if (!token) { console.error('NO_TOKEN'); process.exit(1) }
+
+const WEB_UA = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/149.0.0.0 Safari/537.36'
+const headers = (stream) => ({
+  Authorization: `Cloud-IDE-JWT ${token}`,
+  'Content-Type': 'application/json',
+  'X-Trae-Client-Type': 'web',
+  'X-Preferenced-Language': 'zh-CN',
+  'x-user-region': 'CN',
+  Origin: 'https://solo.trae.cn',
+  Referer: 'https://solo.trae.cn/',
+  'User-Agent': WEB_UA,
+  Accept: stream ? 'text/event-stream' : 'application/json',
+})
+
+const commonParams = JSON.stringify({
+  language: 'zh-cn', app_language: 'zh-CN', quality: 'stable',
+  app_version: '1.0.0.1229', web_id: '', user_identity: 'Free',
+  is_freshman: '0', biz_user_id: '', user_unique_id: '',
+  scope: 'marscode-cn', tenant: 'marscode', region: 'cn', aiRegion: 'cn',
+  is_privacy_mode: 0, privacy_mode: 'off', solo_chat_mode: 'code',
+})
+
+function buildBody(model, effort) {
+  const im = {
+    chat_session_id: '', content: [],
+    query: JSON.stringify([{ type: 'text', data: { content: '一个笼子里有鸡和兔共 10 只，脚共 28 只。鸡兔各几只？只给最终数字。' } }]),
+    model_name: model, agent_type: 'solo_agent_remote',
+    model_selection_strategy: 'manual', common_params: commonParams,
+  }
+  if (effort) {
+    im.custom_model = {
+      model_name: model, config_name: model, config_source: 1,
+      is_preset: true, use_remote_service: true, multimodal: false,
+      reasoning_effort: effort,
+    }
+  }
+  return { mode: 'code', environment_id: 'default', initial_message: im, env: 'remote', auto_create_project: false, origin: 'web' }
+}
+
+async function probe(effort) {
+  const label = effort || 'none(omit)'
+  const t0 = Date.now()
+  try {
+    // 1. create session
+    const createResp = await fetch(`${BASE}/api/remote/v1/chat_sessions`, {
+      method: 'POST', headers: headers(false),
+      body: JSON.stringify(buildBody(MODEL, effort)),
+      signal: AbortSignal.timeout(20_000),
+    })
+    const createText = await createResp.text()
+    if (createResp.status >= 400) {
+      return { effort: label, createHttp: createResp.status, createBody: createText.slice(0, 300), ms: Date.now() - t0 }
+    }
+    let data
+    try { data = JSON.parse(createText) } catch {
+      return { effort: label, createHttp: createResp.status, error: 'non-json create', ms: Date.now() - t0 }
+    }
+    if (data.code != null && data.code !== 0 && !data.data) {
+      return { effort: label, createHttp: createResp.status, bizCode: data.code, bizMsg: data.message, ms: Date.now() - t0 }
+    }
+    const payload = data.data ?? data
+    const sessionId = String(payload.chat_session_id ?? '')
+    const messageId = String(payload.message_id ?? '')
+    if (!sessionId || !messageId) {
+      return { effort: label, createHttp: createResp.status, error: 'missing session/message id', ms: Date.now() - t0 }
+    }
+
+    // 2. open events, read entire stream to EOF (like probe-trae-thinking-scene)
+    const eventsResp = await fetch(
+      `${BASE}/api/remote/v1/chat_sessions/${sessionId}/events?reply_to_message_id=${messageId}`,
+      { headers: headers(true) },
+    )
+    if (eventsResp.status >= 400) {
+      const evText = await eventsResp.text().catch(() => '')
+      return { effort: label, createHttp: 200, eventsHttp: eventsResp.status, eventsBody: evText.slice(0, 300), ms: Date.now() - t0 }
+    }
+    const raw = await eventsResp.text()
+    let reasoningChars = 0, textChars = 0, actualModel = null, finish = null, bytes = raw.length
+    const eventTypes = {}
+    let eventSamples = null
+    let lastEventName = null
+    // also capture raw SSE lines for debugging
+    let rawLines = []
+    for (const line of raw.split('\n')) {
+      const s = line.trim()
+      if (!s) { lastEventName = null; continue }
+      if (s.startsWith('event:')) { lastEventName = s.slice(6).trim(); continue }
+      if (s.startsWith('id:') || s.startsWith(':')) continue
+      if (!s.startsWith('data:')) continue
+      const dataStr = s.slice(5).trim()
+      if (!dataStr) continue
+      if (dataStr === '[DONE]') { finish = finish || 'done'; continue }
+      let ev
+      try { ev = JSON.parse(dataStr) } catch { continue }
+      // use .event field (production parser way) or SSE event: line
+      const eventName = (typeof ev.event === 'string' && ev.event) ? ev.event : lastEventName
+      const et = eventName || (ev.model_config ? 'model_config'
+        : ev.plan_item ? 'plan_item'
+        : ev.done ? 'done'
+        : ev.error ? 'error'
+        : 'other')
+      eventTypes[et] = (eventTypes[et] || 0) + 1
+      if (!eventSamples) eventSamples = []
+      if (eventSamples.length < 8) eventSamples.push({ event: eventName || null, data: dataStr.slice(0, 250) })
+      if (rawLines.length < 30) rawLines.push(s.slice(0, 200))
+      if (eventName === 'model_config' || ev.model_config) {
+        if (typeof ev.config_name === 'string') actualModel = ev.config_name
+      }
+      if (eventName === 'plan_item' || ev.plan_item) {
+        const pi = ev.plan_item || ev
+        if (typeof pi.reasoning_content === 'string') reasoningChars += pi.reasoning_content.length
+        if (typeof pi.thought === 'string') textChars += pi.thought.length
+        if (pi.tool_call_info?.name === 'finish') {
+          finish = 'finish'
+          const sp = pi.tool_call_info?.params?.summary
+          if (typeof sp === 'string') textChars += sp.length
+        }
+      }
+      if (eventName === 'done' || ev.done) finish = finish || 'done'
+      if (eventName === 'error' || ev.error) { return { effort: label, createHttp: 200, eventsHttp: 200, streamError: ev.error || ev, eventTypes, eventSamples, rawLines, ms: Date.now() - t0 } }
+    }
+    return { effort: label, createHttp: 200, eventsHttp: 200, reasoningChars, textChars, actualModel, finish, bytes, eventTypes, eventSamples, rawLines, ms: Date.now() - t0 }
+  } catch (e) {
+    return { effort: label, error: e?.message ?? String(e), ms: Date.now() - t0 }
+  }
+}
+
+console.log(`Trae max-effort probe: model=${MODEL} base=${BASE} efforts=${EFFORTS.join(',')}`)
+const results = []
+for (let i = 0; i < EFFORTS.length; i++) {
+  if (i > 0) { console.log('  (waiting 10s for session slot...)'); await new Promise(r => setTimeout(r, 10_000)) }
+  const r = await probe(EFFORTS[i])
+  results.push(r)
+  console.log(JSON.stringify(r))
+}
+console.log('\n--- summary ---')
+for (const r of results) {
+  const ok = r.createHttp === 200 && r.eventsHttp === 200 && !r.streamError && !r.error
+  const verdict = ok
+    ? (r.reasoningChars > 0 ? `ACCEPTED (reasoning ${r.reasoningChars} chars)` : 'ACCEPTED (no reasoning)')
+    : `REJECTED (${r.error || r.bizCode || r.createHttp})`
+  console.log(`  ${r.effort}: ${verdict}`)
+}