@cocorograph/hub-agent 0.8.5 → 0.8.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cocorograph/hub-agent",
-  "version": "0.8.5",
+  "version": "0.8.7",
   "description": "Hub Hosted Cockpit のローカル常駐 agent。Hub と outbound WSS で接続し、ローカルの tmux/pty を中継する。",
   "type": "module",
   "license": "UNLICENSED",

package/src/async-ttl-cache.mjs

@@ -0,0 +1,51 @@
+/**
+ * 読み取り専用スナップショット向けの短期 TTL + single-flight キャッシュ。
+ *
+ * - TTL 内は直前の成功値を即返す
+ * - TTL 切れ後に複数 caller が重なった場合も build は 1 本だけ実行する
+ * - build 失敗はキャッシュせず、次回 caller が再試行できる
+ * - clear() は進行中 build の呼び出し元も新世代の build へ合流し直す
+ */
+export class AsyncTtlCache {
+  constructor({ ttlMs, now = () => Date.now() }) {
+    this.ttlMs = Math.max(0, Number(ttlMs) || 0)
+    this.now = now
+    this.hasValue = false
+    this.value = undefined
+    this.expiresAt = 0
+    this.inFlight = null
+    this.generation = 0
+  }
+
+  async get(build) {
+    if (this.hasValue && this.now() < this.expiresAt) return this.value
+    if (this.inFlight) return this.inFlight
+
+    const generation = this.generation
+    const pending = Promise.resolve().then(build)
+    this.inFlight = pending
+    try {
+      const value = await pending
+      // build 中に mutation / state push が走った場合、旧値を最初の caller にも返さない。
+      // clear() が inFlight を外しているため、新世代の build を開始するか既存の新 build に
+      // 合流する。read-only snapshot 専用なので再実行は安全。
+      if (this.generation !== generation) return this.get(build)
+      this.value = value
+      this.hasValue = true
+      this.expiresAt = this.now() + this.ttlMs
+      return value
+    } finally {
+      if (this.inFlight === pending) this.inFlight = null
+    }
+  }
+
+  clear() {
+    this.generation += 1
+    this.hasValue = false
+    this.value = undefined
+    this.expiresAt = 0
+    // 旧 build 自体は安全に完走させるが、新 caller はそこへ合流させない。旧 build の
+    // finally は identity 比較により、新世代の inFlight を誤って消さない。
+    this.inFlight = null
+  }
+}

package/src/codex-stream-bridge.mjs

@@ -43,6 +43,7 @@
 import { EventEmitter } from "node:events"
 import { randomUUID } from "node:crypto"
 
+import { AsyncTtlCache } from "./async-ttl-cache.mjs"
 import { CodexAppServerClient } from "./codex-appserver-client.mjs"
 import {
   snapshotCodexDeliveryAudit,
@@ -323,6 +324,7 @@ export class CodexStreamBridge extends EventEmitter {
     listThreadsFromDisk,
     historyMaxItems,
     historyMaxBytes,
+    listThreadsCacheTtlMs,
   } = {}) {
     super()
     this.codexCmd = codexCmd
@@ -357,10 +359,12 @@ export class CodexStreamBridge extends EventEmitter {
     /** @type {Map<string, object>} stream_id -> session record */
     this.sessions = new Map()
     this._accountRateLimitsByProfile = new Map()
-    // Phase 3 (Read-plane): listThreads の single-flight。key(`cwd::limit`) -> in-flight Promise。
-    // codex.sessions.request を直列キューから外したため、同一 cwd への同時リクエストが
-    // app-server の同時 spawn (storm) を起こさないよう 1 本に畳む。
-    this._listThreadsInFlight = new Map()
+    // 複数 Cockpit タブ + pre-attach/ready の連続 request が、同じ cwd で rollout
+    // header 最大 800 件を短時間に何度も走査するのを防ぐ。thread picker の読み取り専用
+    // データなので数秒の鮮度差は許容し、実 turn/attach の正本には使わない。各 key の
+    // AsyncTtlCache が single-flight と、thread/start 中の世代切替をまとめて担う。
+    this._listThreadsCacheTtlMs = listThreadsCacheTtlMs ?? 5_000
+    this._listThreadsCache = new Map()
   }
 
   /**
@@ -576,6 +580,9 @@ export class CodexStreamBridge extends EventEmitter {
         ? await client.request("thread/resume", { ...startParams, threadId: resumeThreadId })
         : await client.request("thread/start", startParams)
       session.threadId = result?.thread?.id
+      // thread/start は一覧そのものを変更するため、pre-attach で得た短期キャッシュを破棄する。
+      // resume/reattach は既存 thread の再閲覧なのでキャッシュを維持し、ready 後の重複走査を防ぐ。
+      if (!resuming) this._invalidateListThreadsCacheForCwd(cwd)
       this.logger?.info?.(
         {
           stream_id,
@@ -836,13 +843,33 @@ export class CodexStreamBridge extends EventEmitter {
     // sessionsDir 未指定 (旧経路) のときだけ env 由来の値へフォールバックする。
     const readRoot = sessionsDir || process.env.CODEX_HOME || ""
     const key = `${readRoot}::${cwd || ""}::${limit ?? ""}`
-    const existing = this._listThreadsInFlight.get(key)
-    if (existing) return existing
-    const p = this._doListThreads({ cwd, limit, sessionsDir }).finally(() => {
-      this._listThreadsInFlight.delete(key)
-    })
-    this._listThreadsInFlight.set(key, p)
-    return p
+    let entry = this._listThreadsCache.get(key)
+    if (!entry) {
+      entry = {
+        cwd: cwd || "",
+        cache: new AsyncTtlCache({ ttlMs: this._listThreadsCacheTtlMs }),
+      }
+      this._listThreadsCache.set(key, entry)
+      this._pruneListThreadsCache()
+    }
+    return entry.cache.get(() => this._doListThreads({ cwd, limit, sessionsDir }))
+  }
+
+  _invalidateListThreadsCacheForCwd(cwd) {
+    const normalizedCwd = cwd || ""
+    for (const entry of this._listThreadsCache.values()) {
+      if (entry.cwd === normalizedCwd) entry.cache.clear()
+    }
+  }
+
+  _pruneListThreadsCache() {
+    if (this._listThreadsCache.size <= 256) return
+    for (const [key, entry] of this._listThreadsCache) {
+      if (this._listThreadsCache.size <= 256) break
+      // 走査中の entry を map から外すと、同じ key の次 caller が別走査を開始して
+      // single-flight を破る。完了済みの古い entry だけを先に落とす。
+      if (!entry.cache.inFlight) this._listThreadsCache.delete(key)
+    }
   }
 
   async _doListThreads({ cwd, limit, sessionsDir } = {}) {

package/src/main.mjs

@@ -48,6 +48,7 @@ import {
 import { loadPlugins, runHookBroadcast, runHookChain } from "./plugin-loader.mjs"
 import { WsClient } from "./ws-client.mjs"
 import { DispatchScheduler } from "./dispatch-scheduler.mjs"
+import { AsyncTtlCache } from "./async-ttl-cache.mjs"
 import { PtyBridge } from "./pty-bridge.mjs"
 import { ClaudeStreamBridge } from "./claude-stream-bridge.mjs"
 import { CodexMcpControl } from "./codex-mcp-control.mjs"
@@ -92,6 +93,7 @@ import {
 } from "./claude-credentials.mjs"
 import {
   buildClaudeCmd,
+  buildWorktreeIndex,
   createSession as createTmuxSession,
   createWorktreeDir,
   execTmux,
@@ -1717,6 +1719,11 @@ const lastTurnTextsCache = new Map()
  */
 const lastTurnTextsInFlight = new Map()
 
+// 複数 Cockpit タブは各自 8 秒 poll を持つため、位相がずれると agent は同じ
+// tmux/worktree/jsonl スナップショットを数秒おきに再構築してしまう。5 秒だけ成功値を
+// 共有し、同時要求は single-flight に畳む。create/kill/worktree 変更時は明示 clear する。
+const tmuxSessionsSnapshotCache = new AsyncTtlCache({ ttlMs: 5_000 })
+
 /** cold cache 時に全セッションを一斉に読まないための同時実行上限。 */
 const LAST_TEXTS_CONCURRENCY = 4
 
@@ -1779,12 +1786,21 @@ function isSafeSessionIdForPath(id) {
  *
  * @param {Array} sessions listTmuxSessions の素の行 (name / cwd を使う)
  * @param {((name: string) => object | null) | undefined} getSnap state loop の getSnapshot
+ * @param {object} [options] I/O と session ID 解決の注入点
+ * @param {((session: object) => string | null) | undefined} [options.resolveFallbackSessionId]
+ *        snapshot の context_session_id が再起動後の `unbound:*` のとき、TUI bind / live
+ *        Claude bridge から同じ cwd の会話 ID を復元する。戻り値にも containment 検査を行う。
  * @returns {Promise<Map<string, {user: string, assistant: string}>>}
  */
 export async function readAllSessionLastTexts(
   sessions,
   getSnap,
-  { extract = extractLastTurnTexts, statFn = stat, resolvePath = jsonlPath } = {},
+  {
+    extract = extractLastTurnTexts,
+    statFn = stat,
+    resolvePath = jsonlPath,
+    resolveFallbackSessionId,
+  } = {},
 ) {
   const out = new Map()
   if (!getSnap) return out
@@ -1794,7 +1810,17 @@ export async function readAllSessionLastTexts(
     try {
       const snap = getSnap(s.name)
       if (!snap || snap.cli_kind !== "claude") return
-      const sessionId = snap.context_session_id
+      let sessionId = snap.context_session_id
+      // hub-agent 再起動で in-memory の TUI bind が消えると、state loop の snapshot は
+      // `unbound:<session>` へ落ちる。一方、Claude bridge の warm/live session は cwd
+      // から同じ会話 ID を復元できる。ここでフォールバックしないと、再起動後に全行の
+      // 「最後の依頼 / 応答」だけが消え、次に chat signal が fresh になるまで戻らない。
+      //
+      // snapshot が安全な実 ID を持つ場合は必ずそちらを優先し、fallback は placeholder /
+      // 危険値の場合だけ使う。fallback も同じ containment 検査を通してから path 化する。
+      if (!isSafeSessionIdForPath(sessionId) && resolveFallbackSessionId) {
+        sessionId = resolveFallbackSessionId(s)
+      }
       // パス要素として安全な形だけを通す (外部由来の値なので containment を局所保証)。
       if (!isSafeSessionIdForPath(sessionId)) return
       if (!s.cwd) return
@@ -1950,6 +1976,9 @@ async function startSessionEventWatcher({ client, logger, readinessTracker }) {
           "session.event push",
         )
       }
+      // 直前に作った一覧 snapshot がこの event より古い場合、次の poll で last_event を
+      // 逆戻りさせない。build 中なら AsyncTtlCache が新世代へ合流し直す。
+      tmuxSessionsSnapshotCache.clear()
       client.send({
         type: "session.event",
         session_name: sessionName,
@@ -2274,6 +2303,7 @@ export async function pushImmediateChatState({
     for (const s of states) {
       if (s.cwd !== cwd) continue
       if (cliKindForSessionName(s.session_name) !== cliKind) continue
+      tmuxSessionsSnapshotCache.clear()
       client.send({
         type: "session.state",
         session_name: s.session_name,
@@ -2380,6 +2410,7 @@ export function startStateLoop({ client, plugins, logger, intervalMs, claudeBrid
   // これで再接続直後の次 tick (≤intervalMs) に短縮する。初回接続時は空 Map の clear で no-op。
   const onReopen = () => {
     lastByName.clear()
+    tmuxSessionsSnapshotCache.clear()
   }
   client.on("open", onReopen)
 
@@ -2678,6 +2709,9 @@ export function startStateLoop({ client, plugins, logger, intervalMs, claudeBrid
             "session.state push",
           )
           }
+          // state push より古い一覧を後続 poll が返すと sidebar 行の status/context が
+          // 逆戻りするため、送信前に snapshot 世代を切り替える。
+          tmuxSessionsSnapshotCache.clear()
           client.send({
             type: "session.state",
             session_name: s.session_name,
@@ -2715,6 +2749,7 @@ export function startStateLoop({ client, plugins, logger, intervalMs, claudeBrid
             // の消滅はここが唯一の検知点。frontend が states/lastEvent を忘れてドットを
             // 'down' に倒せるよう、ローカル Map を forget する前に通知する。
             try {
+              tmuxSessionsSnapshotCache.clear()
               client.send({ type: "session.gone", session_name: name })
             } catch {
               /* ignore */
@@ -4434,6 +4469,7 @@ async function dispatch(msg, ctx) {
           ctx.claudeBridge?.shutdown()
           // tmux TUI セッションも旧アカウントのまま残さない: kill → 同名・同 cwd で再作成。
           const tmuxResult = await restartAllTmuxSessionsForProfileSwitch(ctx)
+          tmuxSessionsSnapshotCache.clear()
           // 切替先の config ディレクトリに認証情報が無ければ要ログイン (バッジ/案内用)。
           const hasCreds = isKeychainPlatform()
             ? await profileHasKeychainCredentials(target.configDir)
@@ -4573,6 +4609,7 @@ async function dispatch(msg, ctx) {
           // Codex セッションだけ撤去 (Claude セッションは残す = 独立切替)。
           const before = ctx.codexBridge?.sessions?.size ?? 0
           ctx.codexBridge?.shutdown()
+          tmuxSessionsSnapshotCache.clear()
           const hasCreds = await codexProfiles.hasCodexCredentials(target.configDir)
           const needsLogin = !hasCreds
           const isDefaultTarget = target.id === codexProfiles.DEFAULT_CODEX_PROFILE_ID
@@ -4706,48 +4743,50 @@ async function dispatch(msg, ctx) {
       }
       case "tmux.list_sessions": {
         try {
-          // boundSessions: 各行の context% (ドーナツ) を bound session の jsonl から引く
-          // (cwd dir 内の別セッション jsonl を最新と取り違える=メタ欠陥#1 の根治)。
-          // claudeBridge: TUI bind が無い SDK チャット cwd 向けの同根治フォールバック
-          // (2026-07 追加。tmux.mjs listSessions 参照)。
-          const sessions = await listTmuxSessions({
-            plugins: ctx.plugins,
-            logger: ctx.logger,
-            boundSessions: ctx.tuiReboundSessions,
-            claudeBridge: ctx.claudeBridge,
+          const snapshot = await tmuxSessionsSnapshotCache.get(async () => {
+            // worktree index は listSessions の親判定と stub 列挙で共用する。従来は同じ
+            // 全 workspace + git branch 走査を 1 応答内で 2 回直列実行していた。
+            const worktreeIndex = await buildWorktreeIndex()
+            // boundSessions: 各行の context% (ドーナツ) を bound session の jsonl から引く
+            // (cwd dir 内の別セッション jsonl を最新と取り違える=メタ欠陥#1 の根治)。
+            const sessions = await listTmuxSessions({
+              plugins: ctx.plugins,
+              logger: ctx.logger,
+              boundSessions: ctx.tuiReboundSessions,
+              claudeBridge: ctx.claudeBridge,
+              worktreeIndex,
+            })
+            const [lastEventByName, lastTextsByName] = await Promise.all([
+              readAllSessionEvents(),
+              readAllSessionLastTexts(sessions, ctx.getSessionStateSnapshot, {
+                resolveFallbackSessionId: (session) =>
+                  resolveStateLoopBoundSessionId(
+                    ctx.tuiReboundSessions,
+                    session.name,
+                    session.cwd,
+                    ctx.claudeBridge,
+                  ),
+              }),
+            ])
+            const enriched = enrichTmuxSessionsForListResponse(
+              sessions,
+              lastEventByName,
+              ctx.getSessionStateSnapshot,
+              lastTextsByName,
+            )
+            const liveNames = new Set(enriched.map((s) => s.name))
+            let worktreeStubs = []
+            try {
+              worktreeStubs = await listWorktreeStubs(liveNames, { worktreeIndex })
+            } catch (err) {
+              ctx.logger?.warn?.({ err: err?.message }, "listWorktreeStubs failed")
+            }
+            return { sessions: enriched, worktree_stubs: worktreeStubs }
           })
-          // 各 session の最新 hook event を /tmp/cockpit_session_events/ から読んで
-          // レスポンスに含める (Phase 4: ハードリロード時の lastEvent 復元用)。
-          // fs.watch 由来の session.event push は揮発性のため、frontend が新規
-          // マウントすると過去 event を取れず全グレー表示になる事象を解消する。
-          const lastEventByName = await readAllSessionEvents()
-          // サイドバー行に「何を依頼したか」を出すため、各セッションの jsonl 末尾から
-          // 最新の依頼 / 応答テキストを読む (mtime キャッシュ付き)。失敗は空に落ちる。
-          const lastTextsByName = await readAllSessionLastTexts(
-            sessions,
-            ctx.getSessionStateSnapshot,
-          )
-          const enriched = enrichTmuxSessionsForListResponse(
-            sessions,
-            lastEventByName,
-            ctx.getSessionStateSnapshot,
-            lastTextsByName,
-          )
-          // cockpit (PR 1719) で未起動 worktree をサイドバーに可視化するために
-          // filesystem 上は存在するが tmux session が無い worktree dir のリストを
-          // 同梱する。古い cockpit は worktree_stubs を無視するので互換 OK。
-          const liveNames = new Set(enriched.map((s) => s.name))
-          let worktreeStubs = []
-          try {
-            worktreeStubs = await listWorktreeStubs(liveNames)
-          } catch (err) {
-            ctx.logger?.warn?.({ err: err?.message }, "listWorktreeStubs failed")
-          }
           ctx.client.send({
             type: "tmux.sessions",
             request_id: msg.request_id,
-            sessions: enriched,
-            worktree_stubs: worktreeStubs,
+            ...snapshot,
           })
         } catch (err) {
           ctx.client.send({
@@ -4797,6 +4836,7 @@ async function dispatch(msg, ctx) {
               typeof msg.initial_prompt === "string" ? msg.initial_prompt : undefined,
             logger: ctx.logger,
           })
+          tmuxSessionsSnapshotCache.clear()
           ctx.client.send({
             type: "tmux.create_session.result",
             request_id: msg.request_id,
@@ -4867,6 +4907,7 @@ async function dispatch(msg, ctx) {
               typeof msg.initial_prompt === "string" ? msg.initial_prompt : undefined,
             logger: ctx.logger,
           })
+          tmuxSessionsSnapshotCache.clear()
           ctx.client.send({
             type: "worktree.create.result",
             request_id: msg.request_id,
@@ -4944,6 +4985,7 @@ async function dispatch(msg, ctx) {
         const force = msg.force === true
         try {
           const result = await removeWorktreeDir(name, { force })
+          if (result.dirty !== true) tmuxSessionsSnapshotCache.clear()
           ctx.client.send({
             type: "worktree.remove.result",
             request_id: msg.request_id,
@@ -4966,6 +5008,7 @@ async function dispatch(msg, ctx) {
       }
       case "tmux.kill_session": {
         await handleKillSession(msg, ctx)
+        tmuxSessionsSnapshotCache.clear()
         return
       }
       case "skills.request": {

package/src/tmux.mjs

@@ -174,7 +174,7 @@ export async function createWorktreeDir(parentDir, branch) {
  *
  * 移植元: D00000_cockpit/webapp/lib/workspaces.ts (findWorktrees)
  */
-async function buildWorktreeIndex() {
+export async function buildWorktreeIndex() {
   const out = new Map()
   const projectsBase = hubProjectsBase()
   let projects
@@ -240,8 +240,11 @@ async function buildWorktreeParentMap() {
  * worktree (`<wt>__codex`) が「未起動 stub」としても同時に列挙され、サイドバーに live 行と
  * stub 行が二重表示される事故を防ぐ。
  */
-export async function listWorktreeStubs(liveSessionNames) {
-  const idx = await buildWorktreeIndex()
+export async function listWorktreeStubs(liveSessionNames, opts = {}) {
+  const idx =
+    opts.worktreeIndex instanceof Map
+      ? opts.worktreeIndex
+      : await buildWorktreeIndex()
   const liveBaseNames = new Set()
   for (const n of liveSessionNames) liveBaseNames.add(baseSessionName(n))
   const stubs = []
@@ -649,7 +652,16 @@ export async function listSessions(opts = {}) {
   // pane 解析 (state.detectSessionState) はフォーマット変動に脆いので、
   // statusLine cache がある cwd では優先採用する。
   const [wtParentMap, sessionUsages] = await Promise.all([
-    buildWorktreeParentMap(),
+    opts.worktreeIndex instanceof Map
+      ? Promise.resolve(
+          new Map(
+            [...opts.worktreeIndex].map(([name, info]) => [
+              name,
+              info.parent_session_name,
+            ]),
+          ),
+        )
+      : buildWorktreeParentMap(),
     getSessionUsages().catch(() => []),
   ])
   // cwd → context_pct の lookup (同一 cwd で複数 session_id ある場合は最新 mtime 優先)