dsh-workspace-api-key 0.2.2 → 0.2.3

package/README.md

@@ -72,7 +72,7 @@ Then add `"dsh-workspace-api-key"` to `dsh.profile.bundles` in that profile's `p
 
 ```sh
 node test/host.test.mjs     # 125 assertions
-node test/client.test.mjs   # 121 assertions
+node test/client.test.mjs   # 124 assertions
 ```
 
 Neither suite needs a running DSH process. The host suite drives `apply()` against a mock context (fake `credentials`, `workspaceRegistry`, `agents`, `sessionController`, `sessions`, `webServer`, stubbed `fetch`) and covers ref derivation, the session → workspace → default cascade, workspace matching, redirecting a second adapter's `apiKeyEnv`, clearing back to the default, 401/402/429/5xx classification, the loopback and origin checks, state persistence, and the session probe including the `{items}` envelope and the live-session fallback. The client suite loads `lib/client.js` through a minimal `window.__ModuleLoader__` plus a small React runtime, renders the panel and the overlay, and asserts slot registration, the first render, session expand/collapse, per-session saving and testing, sessions recovered from the browser-side store when the host has none (with `displayTitle` winning and subagent sessions hidden), the ungrouped-directory card, the stale and mismatch prompts, the "go configure" hand-off, graceful degradation without a session list, and back-navigation when `ctx.get('layout')` only becomes available after the plugin was applied.

package/README.zh.md

@@ -72,7 +72,7 @@ pnpm add "dsh-workspace-api-key@github:COH2357/dsh-workspace-api-key"
 
 ```sh
 node test/host.test.mjs     # 125 条断言
-node test/client.test.mjs   # 121 条断言
+node test/client.test.mjs   # 124 条断言
 ```
 
 两个套件都不需要启动 DSH。宿主套件用一个假 ctx（假的 `credentials`、`workspaceRegistry`、`agents`、`sessionController`、`sessions`、`webServer`，以及打桩的 `fetch`）驱动 `apply()`，覆盖 ref 派生、会话→工作区→默认的级联、工作区匹配、把第二个适配器的 `apiKeyEnv` 一起重定向、清回默认、401/402/429/5xx 的判定、回环与来源校验、状态落盘，以及会话探针（含 `{items}` 信封与 live 兜底）。浏览器套件通过一个极小的 `window.__ModuleLoader__` 与自制 React 运行时加载 `lib/client.js`，渲染出面板与覆盖层，断言槽注册、首屏渲染、会话折叠/展开、会话级保存与测试、宿主列不出会话时从浏览器端存储恢复会话（标题取 `displayTitle`、隐藏子代理会话）、未分组目录卡片、失效与不匹配提醒、「去配置」的交接、没有会话列表时的降级，以及 `ctx.get('layout')` 在插件 apply 之后才可用时返回键仍然有效。

package/lib/client.js

@@ -212,6 +212,11 @@ window.__ModuleLoader__.load({
       return parts[parts.length - 1] ?? p
     }
 
+    /** 会话按最近使用排序。 */
+    function sortByUpdated(rows) {
+      return rows.slice().sort((a, b) => (b?.updatedAt ?? 0) - (a?.updatedAt ?? 0))
+    }
+
     /** 浏览器端 sessions 服务快照 → 顶层会话条目（子代理会话不算）。 */
     function clientSessionEntries(sessions) {
       const snap = sessions?.list?.getSnapshot?.()
@@ -237,9 +242,10 @@ window.__ModuleLoader__.load({
     }
 
     /**
-     * 用浏览器端的会话列表重建「工作区/目录 → 会话」的树。
-     * 宿主行（含 key 层级）按 sessionId 合并进来；宿主没给出的会话用
-     * /state 里的覆盖记录补出层级，保证按钮不会因为宿主列表为空而消失。
+     * 用浏览器端的会话列表补全「工作区/目录 → 会话」的树。
+     * 宿主行始终保留（不会被浏览器端列表挤掉），浏览器端额外只做两件事：
+     * 给同名会话补标题/cwd，以及把宿主漏掉的会话补进来（层级从 /state 的
+     * 覆盖记录推出）。这样宿主列表为空时面板照样有会话可列。
      */
     function mergeSessions(state, sessions) {
       const entries = clientSessionEntries(sessions)
@@ -261,18 +267,27 @@ window.__ModuleLoader__.load({
         if (key !== undefined) workspaceByPath.set(key, ws)
       }
 
-      const buckets = new Map()
-      const extras = new Map()
+      // 第一遍：宿主行能补什么就补什么，宿主没有的会话合成一行。
+      const enriched = new Map()
+      const additions = []
       for (const entry of entries) {
         const host = hostRows.get(entry.sessionId)
-        const cwd = typeof host?.cwd === 'string' && host.cwd !== '' ? host.cwd : entry.cwd
+        if (host !== undefined) {
+          const patch = {}
+          if ((host.title === undefined || host.title === '') && entry.title !== undefined) patch.title = entry.title
+          if ((host.cwd === undefined || host.cwd === '') && entry.cwd !== undefined) patch.cwd = entry.cwd
+          if (Object.keys(patch).length > 0) enriched.set(entry.sessionId, { ...host, ...patch })
+          continue
+        }
+        const cwd = entry.cwd
         const pathKey = normKey(cwd)
         const ws = pathKey !== undefined ? workspaceByPath.get(pathKey) : undefined
-        let row = host
-        if (row === undefined) {
-          const own = overrideBySession.get(entry.sessionId)
-          const level = own !== undefined ? 'session' : ws?.configured === true ? 'workspace' : 'default'
-          row = {
+        const own = overrideBySession.get(entry.sessionId)
+        const level = own !== undefined ? 'session' : ws?.configured === true ? 'workspace' : 'default'
+        additions.push({
+          pathKey,
+          workspace: ws,
+          row: {
             sessionId: entry.sessionId,
             title: entry.title,
             cwd,
@@ -297,45 +312,50 @@ window.__ModuleLoader__.load({
             effectiveTitle: undefined,
             invalid: (level === 'session' ? own?.invalid : level === 'workspace' ? ws?.invalid : null) ?? null,
             mismatch: null,
-          }
-        } else {
-          const patch = {}
-          if ((row.title === undefined || row.title === '') && entry.title !== undefined) patch.title = entry.title
-          if ((row.cwd === undefined || row.cwd === '') && cwd !== undefined) patch.cwd = cwd
-          if (Object.keys(patch).length > 0) row = { ...row, ...patch }
-        }
-        const bucketKey = pathKey ?? ''
-        const target = ws !== undefined ? buckets : extras
-        const list = target.get(ws !== undefined ? ws.path : bucketKey) ?? []
-        list.push(row)
-        target.set(ws !== undefined ? ws.path : bucketKey, list)
+          },
+        })
       }
 
-      const sorted = (rows) => rows.slice().sort((a, b) => (b.updatedAt ?? 0) - (a.updatedAt ?? 0))
-      const workspaces = (state.workspaces ?? []).map((ws) => (
-        { ...ws, sessions: sorted(buckets.get(ws.path) ?? []) }
-      ))
+      const fill = (rows) => (Array.isArray(rows) ? rows : [])
+        .map((row) => enriched.get(row?.sessionId) ?? row)
+
+      // 第二遍：宿主结构照旧，浏览器端多出来的会话按 cwd 归档。
+      const workspaces = (state.workspaces ?? []).map((ws) => {
+        const rows = fill(ws.sessions)
+        for (const item of additions) {
+          if (item.workspace?.path === ws.path) rows.push(item.row)
+        }
+        return { ...ws, sessions: sortByUpdated(rows) }
+      })
+
       const ungrouped = []
+      const extras = new Map()
+      for (const item of additions) {
+        if (item.workspace !== undefined) continue
+        const key = item.pathKey ?? ''
+        const list = extras.get(key) ?? []
+        list.push(item.row)
+        extras.set(key, list)
+      }
+      for (const group of (state.ungrouped ?? [])) {
+        const key = normKey(group?.path) ?? String(group?.path ?? '')
+        const rows = fill(group.sessions)
+        for (const row of (extras.get(key) ?? [])) rows.push(row)
+        extras.delete(key)
+        ungrouped.push({ ...group, sessions: sortByUpdated(rows) })
+      }
       for (const [key, rows] of extras) {
-        const known = (state.ungrouped ?? []).find((group) => normKey(group?.path) === key)
-        const path = known?.path ?? rows[0]?.cwd ?? undefined
+        const path = key !== '' ? key : undefined
         ungrouped.push({
-          ...(known ?? {
-            id: undefined,
-            path,
-            title: path !== undefined && path !== '' ? dirName(path) : '（没有目录）',
-            kind: 'directory',
-            configured: false,
-            exists: true,
-          }),
-          sessions: sorted(rows),
+          id: undefined,
+          path: rows[0]?.cwd ?? path,
+          title: path !== undefined && path !== '' ? dirName(path) : '（没有目录）',
+          kind: 'directory',
+          configured: false,
+          exists: true,
+          sessions: sortByUpdated(rows),
         })
       }
-      for (const group of (state.ungrouped ?? [])) {
-        if (!ungrouped.some((item) => item.path === group.path)) {
-          ungrouped.push({ ...group, sessions: sorted(buckets.get(group.path) ?? []) })
-        }
-      }
       return { ...state, workspaces, ungrouped, sessionSource: 'client' }
     }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-workspace-api-key",
-  "version": "0.2.2",
+  "version": "0.2.3",
   "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",