dsh-workspace-api-key 0.2.16 → 0.3.0

package/lib/index.js

@@ -32,6 +32,7 @@
 
 import { mkdirSync, writeFileSync, readFileSync, existsSync } from 'node:fs'
 import { createHash } from 'node:crypto'
+import { createRequire } from 'node:module'
 import { homedir } from 'node:os'
 import { join, dirname, basename, resolve as pathResolve } from 'node:path'
 
@@ -116,6 +117,11 @@ function isSearchRefFor(ctx, ref) {
   if (typeof ref !== 'string' || ref === '') return false
   const web = currentWeb(ctx)
   if (web.apiChannel !== true) return false
+  // 新版计费模型：搜索只认 `searchApiKeyEnv` 指向的自有 ref；聊天 ref（DEEPSEEK_API_KEY）
+  // 绝不能被当成搜索处理，否则「统一搜索 key」会把聊天也一起换掉。
+  if (web.mode === 'billing') {
+    return ref === SEARCH_DEFAULT_REF || (web.ownSearchRef !== undefined && ref === web.ownSearchRef)
+  }
   // 官方页还没被接管时，搜索用的是聊天那把 ref：这时分辨不了，按聊天处理（安全）。
   if (web.searchKeyEnv === DEFAULT_API_KEY_REF) return ref === SEARCH_DEFAULT_REF
   return ref === SEARCH_DEFAULT_REF || ref === web.searchKeyEnv || ref === web.legacyKeyEnv
@@ -142,6 +148,39 @@ const PLUGIN_VERSION = (() => {
   }
 })()
 
+/**
+ * 宿主运行时版本（尽力而为，只用于界面展示与日志）。
+ *
+ * 行为判断**不**依赖它：桌面端宿主代码打在 app.asar 里，`@deepseek-ai/dsh`
+ * 不一定能从插件目录解析到；而新版搜索配置的 schema 特征（searchBilling /
+ * fetchBilling）与行为一一对应，所以 gating 用特性检测，版本号只是辅助信息。
+ */
+const HOST_VERSION = (() => {
+  try {
+    const req = createRequire(import.meta.url)
+    let main
+    try {
+      main = req.resolve('@deepseek-ai/dsh')
+    } catch {
+      main = undefined
+    }
+    if (main === undefined) return undefined
+    // 从入口文件往上找最近的 package.json。
+    let dir = dirname(main)
+    for (let depth = 0; depth < 5; depth += 1) {
+      const candidate = join(dir, 'package.json')
+      if (existsSync(candidate)) {
+        const pkg = JSON.parse(readFileSync(candidate, 'utf8'))
+        if (typeof pkg?.version === 'string') return pkg.version
+      }
+      dir = dirname(dir)
+    }
+  } catch {
+    /* ignore */
+  }
+  return undefined
+})()
+
 // ─────────────────────────── 路径 / 工作区键 ───────────────────────────
 
 /** 统一的路径比较用规范形：绝对化 + 分隔符统一 + Windows 折大小写。 */
@@ -243,11 +282,11 @@ function workspaceIndex(ctx) {
 }
 
 /** cwd → 工作区描述；没有匹配工作区时回退成「以该目录为单位」。 */
-function locateWorkspace(ctx, cwd) {
+function locateWorkspace(ctx, cwd, index) {
   if (cwd === undefined) return undefined
   const norm = normPath(cwd)
   if (norm === undefined) return undefined
-  const hit = workspaceIndex(ctx).get(norm)
+  const hit = (index ?? workspaceIndex(ctx)).get(norm)
   if (hit !== undefined) return { ...hit, kind: 'workspace' }
   return { id: undefined, path: cwd, title: basename(cwd) || cwd, kind: 'directory' }
 }
@@ -256,7 +295,7 @@ function locateWorkspace(ctx, cwd) {
  * 给定会话/目录，算出当前生效的覆盖层级（纯函数，不读凭据）：
  * session → workspace → default。
  */
-function effectiveOverride(ctx, state, { sessionId, cwd }) {
+function effectiveOverride(ctx, state, { sessionId, cwd }, index) {
   if (sessionId !== undefined) {
     const sref = sessionRef(sessionId)
     if (sref !== undefined && state.overrides[sref] !== undefined) {
@@ -264,7 +303,9 @@ function effectiveOverride(ctx, state, { sessionId, cwd }) {
     }
   }
   if (cwd !== undefined) {
-    const ws = locateWorkspace(ctx, cwd)
+    // index 是可选的「工作区索引」快照：面板一次要算几十个会话，每行都去
+    // registry.list() 一遍会让整个事件循环卡住（0.2.16 实测 /state 要 50 秒）。
+    const ws = locateWorkspace(ctx, cwd, index)
     const wref = ws !== undefined ? derivedRef(ws.path) : undefined
     if (wref !== undefined && state.overrides[wref] !== undefined) {
       return { level: 'workspace', ref: wref, entry: state.overrides[wref], workspace: ws }
@@ -352,6 +393,32 @@ async function bindOfficialSearchRef(ctx, ref) {
   return { ok: true, changed }
 }
 
+/**
+ * 写新版搜索 API 的 schema 字段（实验性「统一搜索 key」用）。
+ *
+ * 只改 `searchApiKeyEnv` / `fetchApiKeyEnv` 这类声明过的字段，不碰任何 key 值；
+ * 空字符串表示「清掉自有 ref、回到官方计费方式」（官方 resolveTarget 用 nonEmpty 判断）。
+ */
+async function updateSearchApiSettings(ctx, patch) {
+  let settings
+  try {
+    settings = ctx.get('settings')
+  } catch {
+    settings = undefined
+  }
+  if (typeof settings?.update !== 'function' || typeof settings?.describe !== 'function') {
+    return { ok: false, reason: 'settings-unavailable' }
+  }
+  try {
+    const row = (settings.describe() ?? []).find((item) => item?.ns === SEARCH_API_NS)
+    if (row === undefined) return { ok: false, reason: 'search-settings-missing' }
+    await settings.update(SEARCH_API_NS, patch, row.revision)
+    return { ok: true }
+  } catch (error) {
+    return { ok: false, reason: error?.message ?? String(error) }
+  }
+}
+
 /**
  * 启动时把官方「网页搜索」页的 key 指向拉回一致。
  *
@@ -390,7 +457,7 @@ async function reconcileOfficialSearchRef(ctx) {
  *   旧 Messages 通道：搜索就在用聊天 key，固定返回搜索 ref。
  */
 async function searchRefFor(ctx, state, { sessionId, cwd, web, searchDefault }) {
-  const w = web ?? readWebSettings(ctx)
+  const w = web ?? currentWeb(ctx)
   if (!w.apiChannel) return w.searchKeyEnv
   const own = sessionOverrideRef(state, sessionId)
   if (own !== undefined) return own
@@ -803,8 +870,10 @@ function interceptableRefs(ctx, state) {
   // 搜索/抓取通道的 ref 也要拦：用户可能把「新版 Web 搜索 API」的 key 指到
   // 独立 ref（例如 DEEPSEEK_SEARCH_API_KEY）。不拦的话，勾了「同步搜索 API」
   // 也会静默不生效——面板显示已同步、实际还在用系统默认 key。
+  // 注意这里走 currentWeb（缓存）而不是 readWebSettings：本函数在 credentials.resolve
+  // 的热路径上被调用，每次都 settings.describe() 会把聊天通道一并拖慢。
   try {
-    const web = readWebSettings(ctx)
+    const web = currentWeb(ctx)
     for (const ref of [web.apiKeyEnv, web.legacyKeyEnv]) {
       if (typeof ref === 'string' && ref !== '') refs.add(ref)
     }
@@ -875,6 +944,14 @@ function webDefaults() {
     apiChannel: false,
     searchKeyEnv: DEFAULT_API_KEY_REF,
     searchLiteralKey: undefined,
+    // 搜索配置模型：'ref' = 旧版（apiKeyEnv 指针）；'billing' = 0.2.2-rc.2 起的计费方式模型。
+    mode: 'ref',
+    searchBilling: undefined,
+    fetchBilling: undefined,
+    ownSearchRef: undefined,
+    ownFetchRef: undefined,
+    searchBaseURL: undefined,
+    fetchBaseURL: undefined,
   }
 }
 
@@ -899,6 +976,27 @@ function webConfigOf(rows) {
     : FALLBACK_FETCH_PROVIDER
   const apiLiteralKey = typeof api.apiKey === 'string' && api.apiKey !== '' ? api.apiKey : undefined
   const legacyLiteralKey = typeof legacy.apiKey === 'string' && legacy.apiKey !== '' ? legacy.apiKey : undefined
+
+  // 0.2.2-rc.2 起，搜索改由「计费方式」决定凭据：searchBilling / fetchBilling +
+  // （可选的）自有端点/ref。旧字段 apiKeyEnv 还能加载，但已经不再生效。
+  // 用「行里有没有 *_Billing」做特性检测，而不是解析宿主版本号：版本号在桌面端
+  // 拿不到（宿主代码打在 asar 里），而这个 schema 特征与行为是一一对应的。
+  const searchBilling = typeof api.searchBilling === 'string' && api.searchBilling !== '' ? api.searchBilling : undefined
+  const fetchBilling = typeof api.fetchBilling === 'string' && api.fetchBilling !== '' ? api.fetchBilling : undefined
+  const mode = searchBilling !== undefined || fetchBilling !== undefined ? 'billing' : 'ref'
+  const ownSearchRef = typeof api.searchApiKeyEnv === 'string' && api.searchApiKeyEnv !== ''
+    ? api.searchApiKeyEnv
+    : undefined
+  const ownFetchRef = typeof api.fetchApiKeyEnv === 'string' && api.fetchApiKeyEnv !== ''
+    ? api.fetchApiKeyEnv
+    : undefined
+  const searchBaseURL = typeof api.searchBaseURL === 'string' && api.searchBaseURL !== ''
+    ? api.searchBaseURL
+    : FALLBACK_SEARCH_BASE_URL
+  const fetchBaseURL = typeof api.fetchBaseURL === 'string' && api.fetchBaseURL !== ''
+    ? api.fetchBaseURL
+    : FALLBACK_SEARCH_BASE_URL
+
   return {
     apiKeyEnv,
     legacyKeyEnv,
@@ -911,8 +1009,17 @@ function webConfigOf(rows) {
     /** 新版 Search/Fetch API 是否被选中（官方开关）= provider 选择被改成了 API 版。 */
     apiChannel: searchProvider === SEARCH_PROVIDER_ID || fetchProvider === SEARCH_FETCH_PROVIDER_ID,
     /** 搜索请求真正会用的那套配置的 key 引用。 */
-    searchKeyEnv: searchProvider === SEARCH_PROVIDER_ID ? apiKeyEnv : legacyKeyEnv,
+    searchKeyEnv: searchProvider === SEARCH_PROVIDER_ID
+      ? (mode === 'billing' ? ownSearchRef ?? DEFAULT_API_KEY_REF : apiKeyEnv)
+      : legacyKeyEnv,
     searchLiteralKey: searchProvider === SEARCH_PROVIDER_ID ? apiLiteralKey : legacyLiteralKey,
+    mode,
+    searchBilling,
+    fetchBilling,
+    ownSearchRef,
+    ownFetchRef,
+    searchBaseURL,
+    fetchBaseURL,
   }
 }
 
@@ -923,6 +1030,30 @@ function webConfigOf(rows) {
  */
 async function searchWarnings(ctx, state, web) {
   const out = []
+
+  // 新版计费模型：搜索不再读 apiKeyEnv 指针，旧提醒（同步/搜索默认 key）全部失去意义。
+  // 这里只保留两条与新版真正相关的提示，避免面板给出过时或错误的引导。
+  if (web.mode === 'billing') {
+    if (web.ownSearchRef === SEARCH_DEFAULT_REF) {
+      out.push({
+        code: 'search-unified',
+        level: 'info',
+        title: '实验性：所有会话的搜索共用同一把 key',
+        detail: `搜索/获取被指向插件自有的 ${SEARCH_DEFAULT_REF}，官方「计费方式」暂时不生效。`
+          + '要恢复按会话/工作区计费，到「设置 → API Key 分配」里关掉这项实验功能。',
+      })
+    } else if (web.searchBilling === 'account') {
+      out.push({
+        code: 'search-billing-account',
+        level: 'warn',
+        title: '搜索当前按「已登录的 DeepSeek 账号」计费',
+        detail: '计费方式设为 account 时，搜索只使用账号令牌，本插件配置的 API key 不会参与搜索。'
+          + '要让搜索跟随会话/工作区的 key，请到「设置 → 插件 → 网页搜索」把计费方式改成「DeepSeek API Key」。',
+      })
+    }
+    return out
+  }
+
   const apiKeyConfigured = web.apiKeyEnv === DEFAULT_API_KEY_REF
     ? (await describeKey(ctx, DEFAULT_API_KEY_REF)).configured === true
     : (await describeKey(ctx, web.apiKeyEnv)).configured === true
@@ -1317,7 +1448,9 @@ export function apply(ctx) {
   }
   installResolver(ctx, state)
   // 「直接禁用官方页面里的搜索 key 设置」：把官方 apiKeyEnv 指到插件自有 ref（异步、失败不影响启动）。
-  void reconcileOfficialSearchRef(ctx)
+  // 0.2.2-rc.2 起官方改用计费方式（searchBilling/fetchBilling），apiKeyEnv 已不再生效：
+  // 这时再写它只会制造「看起来配了、其实没用」的无效配置，所以只在旧模型下接管。
+  if (currentWeb(ctx).mode !== 'billing') void reconcileOfficialSearchRef(ctx)
 
   // ── 2. REST API（loopback 可信）──────────────────────────────────────
   let webServer
@@ -1400,6 +1533,15 @@ function installResolver(ctx, state) {
       if (isSearchRefFor(ctx, ref)) {
         const web = currentWeb(ctx)
         if (web.apiChannel) {
+          // 实验性「统一搜索 key」：搜索/获取固定用插件自有的那一把，优先于会话/工作区配置。
+          // 只在计费模型下生效：旧模型里 searchApiKeyEnv 就是「搜索默认 key」，语义不同。
+          if (web.mode === 'billing' && web.ownSearchRef === SEARCH_DEFAULT_REF) {
+            const forced = await resolveKey(ctx, SEARCH_DEFAULT_REF)
+            if (forced.value !== undefined && forced.value !== '') {
+              log(`search ref ${ref} → ${SEARCH_DEFAULT_REF} (unified)`)
+              return { value: forced.value, source: forced.source ?? 'plugin' }
+            }
+          }
           // ① 会话单独设置的 key 永远优先——哪怕官方搜索插件里配了别的 key。
           const sessionKey = sessionOverrideRef(state, sessionId)
           if (sessionKey !== undefined) {
@@ -1551,7 +1693,7 @@ async function readJson(req) {
 }
 
 /** 一个会话的展示 + 生效情况（供面板与会话进入提醒使用）。 */
-async function sessionRow(ctx, state, row, fallbackCwd) {
+async function sessionRow(ctx, state, row, fallbackCwd, index) {
   const sessionId = typeof row?.sessionId === 'string' && row.sessionId !== '' ? row.sessionId : undefined
   if (sessionId === undefined) return undefined
   const selection = modelSelectionOf(row)
@@ -1559,7 +1701,7 @@ async function sessionRow(ctx, state, row, fallbackCwd) {
   const ref = sessionRef(sessionId)
   const entry = ref !== undefined ? state.overrides[ref] : undefined
   const directHit = entry !== undefined ? await resolveKey(ctx, ref) : undefined
-  const eff = effectiveOverride(ctx, state, { sessionId, cwd })
+  const eff = effectiveOverride(ctx, state, { sessionId, cwd }, index)
   const effHit = eff.ref !== undefined ? await resolveKey(ctx, eff.ref) : undefined
   return {
     sessionId,
@@ -1591,14 +1733,19 @@ async function sessionRow(ctx, state, row, fallbackCwd) {
   }
 }
 
-async function workspaceRow(ctx, state, { id, path: dir, title, kind }, rows, claimed) {
+async function workspaceRow(ctx, state, { id, path: dir, title, kind }, rows, claimed, env) {
+  // 每个工作区行都要的一份上下文（工作区索引、官方搜索配置、搜索默认 key 是否已配）。
+  // 由 snapshot() 统一算一次传进来：panel 打开一次要渲染 20+ 行，每行各算一遍
+  // settings.describe() + registry.list() 会把事件循环卡住（0.2.16 实测 /state ≈ 50 s）。
+  const index = env?.index
+  const web = env?.web ?? currentWeb(ctx)
   const ref = derivedRef(dir)
   const configured = ref !== undefined && state.overrides[ref] !== undefined
   const desc = configured ? await describeKey(ctx, ref) : undefined
   const hit = configured ? await resolveKey(ctx, ref) : undefined
   const sessions = []
   for (const row of rows) {
-    const item = await sessionRow(ctx, state, row, dir)
+    const item = await sessionRow(ctx, state, row, dir, index)
     if (item !== undefined) sessions.push(item)
   }
   sessions.sort((left, right) => (right.updatedAt ?? 0) - (left.updatedAt ?? 0))
@@ -1606,9 +1753,9 @@ async function workspaceRow(ctx, state, { id, path: dir, title, kind }, rows, cl
   // 搜索通道：只有新版 Search/Fetch API 才有「同步 / 不同步」的区别（旧 Messages
   // 通道本来就用聊天 key）。这里给出「该工作区的搜索请求实际会用哪把 key」，
   // 让用户勾了复选框之后马上能看到效果，而不用等到搜索失败才发现。
-  const web = readWebSettings(ctx)
   const synced = state.searchSync?.[normPath(dir)] === true
-  const searchDefaultConfigured = web.apiChannel && (await describeKey(ctx, searchDefaultRefOf(web))).configured === true
+  const searchDefaultConfigured = env?.searchDefaultConfigured
+    ?? (web.apiChannel && (await describeKey(ctx, searchDefaultRefOf(web))).configured === true)
   const searchKeyRef = web.apiChannel
     ? synced
       ? configured
@@ -1655,13 +1802,21 @@ function hasVisibleSession(rows) {
 }
 
 async function snapshot(ctx, state) {
+  const startedAt = Date.now()
   // 官方「网页搜索」配置：每次面板刷新都重读一次（缓存只在请求入口更新，解析路径只读）。
-  refreshWebCache(ctx)
+  // 整个 snapshot 只读这一次：下面每行都用 currentWeb(ctx) 拿缓存，不再各自 describe。
+  const web = refreshWebCache(ctx)
+  // 工作区索引也一样：一次 snapshot 只调一次 registry.list()，行渲染复用同一份快照。
+  const index = workspaceIndex(ctx)
   const credentials = ctx.get('credentials')
   const defaultRefName = 'DEEPSEEK_API_KEY'
   const defaultInfo = await describeKey(ctx, defaultRefName)
   const defaultHit = await resolveKey(ctx, defaultRefName)
   const probe = await sessionProbe(ctx)
+  const probeMs = Date.now() - startedAt
+  const searchDefaultConfigured = web.apiChannel
+    && (await describeKey(ctx, searchDefaultRefOf(web))).configured === true
+  const rowEnv = { index, web, searchDefaultConfigured }
   const rows = probe.rows
   const byCwd = new Map()
   for (const row of rows) {
@@ -1685,7 +1840,7 @@ async function snapshot(ctx, state) {
     }
     if (typeof dir !== 'string' || dir === '') continue
     const norm = normPath(dir)
-    workspaces.push(await workspaceRow(ctx, state, { id, path: dir, title, kind: 'workspace' }, byCwd.get(norm) ?? [], claimed))
+    workspaces.push(await workspaceRow(ctx, state, { id, path: dir, title, kind: 'workspace' }, byCwd.get(norm) ?? [], claimed, rowEnv))
   }
 
   // 有覆盖记录、但对应工作区已不在注册表里的（被删掉的工作区）也列出来，方便清理。
@@ -1725,7 +1880,7 @@ async function snapshot(ctx, state) {
     const dir = typeof list[0]?.cwd === 'string' ? list[0].cwd : norm
     // 诊断用：这个目录里一个「真实会话」都没有（界面把它藏起来，但用户能看到原因）。
     if (!hasVisibleSession(list)) hiddenBlankDirs.push(dir)
-    ungrouped.push(await workspaceRow(ctx, state, { id: undefined, path: dir, title: basename(dir) || dir, kind: 'directory' }, list))
+    ungrouped.push(await workspaceRow(ctx, state, { id: undefined, path: dir, title: basename(dir) || dir, kind: 'directory' }, list, undefined, rowEnv))
   }
 
   // 会话级覆盖总览（含已不在列表里的会话，便于清理）。
@@ -1751,8 +1906,7 @@ async function snapshot(ctx, state) {
   }
 
   // ── 搜索 / 抓取通道总览 ──────────────────────────────────────────────
-  // 每次面板刷新都重读一次：开关改过之后，缓存必须跟着变。
-  const web = refreshWebCache(ctx)
+  // 上面的 refreshWebCache() 已经把官方配置读到 web 里了，这里直接复用。
   const searchKeyRef = web.searchKeyEnv
   const searchKeyInfo = await describeKey(ctx, searchKeyRef)
   const searchKeyHit = await resolveKey(ctx, searchKeyRef)
@@ -1763,10 +1917,14 @@ async function snapshot(ctx, state) {
   const syncPaths = Object.keys(state.searchSync ?? {}).filter((key) => state.searchSync[key] === true)
   const warnings = await searchWarnings(ctx, state, web)
 
+  const totalMs = Date.now() - startedAt
   return {
     ok: true,
     pluginVersion: PLUGIN_VERSION,
+    hostVersion: HOST_VERSION,
     storeVersion: STATE_VERSION,
+    // 面板卡顿排查用：会话探针（宿主 list 一次要扫所有会话）与整行的耗时。
+    timings: { probeMs, rowsMs: Math.max(0, totalMs - probeMs), totalMs },
     sessionProbe: {
       available: probe.available,
       count: probe.count,
@@ -1788,7 +1946,7 @@ async function snapshot(ctx, state) {
     },
     providers: providerList(ctx),
     interceptableRefs: [...interceptableRefs(ctx, state)],
-    workspaceCount: workspaceIndex(ctx).size,
+    workspaceCount: index.size,
     credentialsAvailable: credentials !== undefined,
     sessionListAvailable: probe.available,
     sessionControllerAvailable: sessionListAvailable(ctx),
@@ -1802,6 +1960,13 @@ async function snapshot(ctx, state) {
     // 搜索 / 抓取通道：官方「设置 → 插件 → 网页搜索」里的配置 + 本插件怎么参与。
     // 面板据此显示通道、生效 key 与提醒；不暴露任何明文。
     search: {
+      // 'billing' = 0.2.2-rc.2 起的计费方式模型；'ref' = 旧版 apiKeyEnv 指针模型。
+      mode: web.mode ?? 'ref',
+      billing: { search: web.searchBilling, fetch: web.fetchBilling },
+      ownSearchRef: web.ownSearchRef,
+      ownFetchRef: web.ownFetchRef,
+      // 实验性「统一搜索 key」：把官方 searchApiKeyEnv 指向插件自有 ref。
+      unified: web.mode === 'billing' && web.searchKeyEnv === SEARCH_DEFAULT_REF,
       channel: web.apiChannel ? 'api' : 'messages',
       apiChannel: web.apiChannel,
       searchProvider: web.searchProvider,
@@ -1941,7 +2106,8 @@ async function checkSession(ctx, state, params) {
 
 /** 进会话提醒用的搜索通道摘要（不含明文 key）。 */
 async function searchStateOf(ctx, state, { sessionId, cwd }) {
-  const web = readWebSettings(ctx)
+  // 面板/会话入口不是热路径，但也没必要每次 describe：缓存由请求入口与设置事件维护。
+  const web = currentWeb(ctx)
   const sync = workspaceSynced(state, cwd)
   if (!web.apiChannel) {
     const idle = web.apiLiteralKey !== undefined
@@ -2029,7 +2195,7 @@ async function handle(ctx, state, req, res) {
       // 于是面板和官方页改的是同一把 key，不会各存一份。官方还指向聊天那把（DEEPSEEK_API_KEY）时，
       // 先把两套搜索插件的 apiKeyEnv 改到插件自有 ref，再写值。
       if (body.scope === 'search-default') {
-        const web = readWebSettings(ctx)
+        const web = refreshWebCache(ctx)
         const official = officialSearchRef(web)
         const needsBind = official === DEFAULT_API_KEY_REF
         const target = needsBind ? SEARCH_DEFAULT_REF : official
@@ -2089,7 +2255,7 @@ async function handle(ctx, state, req, res) {
 
     if (pathname === `${API_BASE}/test` && method === 'POST') {
       const body = await readJson(req)
-      const web = readWebSettings(ctx)
+      const web = refreshWebCache(ctx)
       const channel = body.channel === 'search' ? 'search' : 'chat'
       let key = typeof body.key === 'string' && body.key.trim() !== '' ? body.key.trim() : undefined
       let sourceRef = typeof body.ref === 'string' && body.ref !== '' ? body.ref : undefined
@@ -2148,6 +2314,43 @@ async function handle(ctx, state, req, res) {
       return
     }
 
+    if (pathname === `${API_BASE}/search-unified` && method === 'POST') {
+      const body = await readJson(req)
+      const web = refreshWebCache(ctx)
+      if (web.mode !== 'billing') {
+        sendJson(res, 400, { ok: false, error: '当前宿主还是旧版搜索配置，不支持该实验功能' })
+        return
+      }
+      const enabled = body.enabled !== false
+      if (enabled) {
+        const key = typeof body.key === 'string' ? body.key.trim() : ''
+        if (key === '') {
+          sendJson(res, 400, { ok: false, error: '请先填写一把搜索 key' })
+          return
+        }
+        await storeKey(ctx, SEARCH_DEFAULT_REF, key)
+        const updated = await updateSearchApiSettings(ctx, {
+          searchApiKeyEnv: SEARCH_DEFAULT_REF,
+          fetchApiKeyEnv: SEARCH_DEFAULT_REF,
+        })
+        if (updated.ok !== true) {
+          sendJson(res, 500, { ok: false, error: `写入官方搜索配置失败：${updated.reason}` })
+          return
+        }
+      } else {
+        const updated = await updateSearchApiSettings(ctx, { searchApiKeyEnv: '', fetchApiKeyEnv: '' })
+        if (updated.ok !== true) {
+          sendJson(res, 500, { ok: false, error: `写入官方搜索配置失败：${updated.reason}` })
+          return
+        }
+      }
+      webCache = undefined
+      refreshWebCache(ctx)
+      log(`unified search key ${enabled ? 'on' : 'off'}`)
+      sendJson(res, 200, { ok: true, enabled, ...(await snapshot(ctx, state)) })
+      return
+    }
+
     if (pathname === `${API_BASE}/search-sync` && method === 'POST') {
       const body = await readJson(req)
       const dir = typeof body.path === 'string' ? body.path : undefined

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-workspace-api-key",
-  "version": "0.2.16",
+  "version": "0.3.0",
   "description": "A DeepSeek Harness plugin that binds a separate DeepSeek API key to each workspace, or to a single session, so the spend of one key stays with the sessions you choose.",
   "type": "module",
   "license": "MIT",
@@ -50,6 +50,6 @@
     "lib/client.js"
   ],
   "scripts": {
-    "test": "node test/host.test.mjs && node test/client.test.mjs"
+    "test": "node test/host.test.mjs && node test/client.test.mjs && node test/client-i18n.test.mjs && node test/client-billing.test.mjs"
   }
 }