@cocorograph/hub-agent 0.8.19 → 0.8.20

package/package.json

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

package/src/main.mjs

@@ -108,6 +108,7 @@ import {
 } from "./tmux.mjs"
 import { SiblingInjectBridge } from "./sibling-inject.mjs"
 import { isSessionWarm } from "./live-sessions.mjs"
+import { applyPointerContexts, resolvePointerContexts } from "./pointer-context.mjs"
 import {
   boundSessionId,
   codexContextPercentFromTokenUsage,
@@ -5067,10 +5068,32 @@ async function dispatch(msg, ctx) {
             }
             return { sessions: enriched, worktree_stubs: worktreeStubs }
           })
+          // 行ごとの「次に開く会話」の context% (任意)。宛先 (context_targets) は要求ごとに
+          // 違うため、共有される snapshot キャッシュの**外**で解決し、触る行だけ複製して
+          // 載せる (他クライアントの応答を汚さない)。context_targets が無ければ従来どおりの
+          // 応答 (後方互換)。
+          let sessions = snapshot.sessions
+          if (Array.isArray(msg.context_targets) && msg.context_targets.length > 0) {
+            try {
+              const pointerByName = await resolvePointerContexts(
+                msg.context_targets,
+                sessions,
+                { logger: ctx.logger },
+              )
+              sessions = applyPointerContexts(sessions, pointerByName)
+            } catch (err) {
+              // pointer context はあくまで付加情報。失敗してもセッション一覧は返す。
+              ctx.logger?.warn?.(
+                { err: err?.message },
+                "resolvePointerContexts failed",
+              )
+            }
+          }
           ctx.client.send({
             type: "tmux.sessions",
             request_id: msg.request_id,
             ...snapshot,
+            sessions,
           })
         } catch (err) {
           ctx.client.send({

package/src/pointer-context.mjs

@@ -0,0 +1,196 @@
+/**
+ * 「その行が次に開く会話」の context% (= pointer context) を解決する。
+ *
+ * ## 背景 (WHY)
+ * Cockpit のサイドバーは、コンテキスト使用率を **その行が次に開く会話に帰属させて**表示する
+ * ように直した (Hub 側 PR #4534)。しかし供給側 (agent) は「いま動いている会話」の値しか
+ * 出しておらず、一度も開いていない行・閉じた行では使用率が出ない。
+ *
+ * 従来は「その cwd で mtime が一番新しい jsonl」に落として埋めていたが、それが
+ * **サブエージェント / headless `claude -p` / 過去セッションの数字が混ざる**原因だったので
+ * 撤去した (cross-activity contamination)。よってここでは **frontend が行ごとに指定した
+ * 会話 ID (`session_id`) でしか値を引かない**。ID が無い行は推測せず null を返す。
+ *
+ * ## 不変条件 (ここを崩すと汚染が戻る)
+ *   1. `session_id` が渡されていない行では `contextForCwd` を **呼ばない** (mtime 最新への
+ *      フォールバックが走るため)。
+ *   2. `session_id` が渡された行でも、戻り値の `uuid` が要求 ID と**一致するときだけ**採用する。
+ *      `jsonlContextForCwd` は「その ID の jsonl が存在しない」とき mtime 最新へフォールバック
+ *      する仕様のため、件数ではなく**実体 (uuid) の同一性**で確認する。
+ *   3. 鮮度 (mtime) による足切りをしない。止まっている会話の使用率は古くならない
+ *      (次に開けばその値から再開する) ので、statusLine 経路の `SESSION_STALE_MS` は適用しない。
+ *
+ * ## N+1 対策
+ * サイドバーは 8 秒周期でポーリングし、複数タブが位相をずらして同時に叩く。行数分の jsonl
+ * 読み取りをそのまま毎回走らせないよう、`(cwd, session_id)` をキーにした短期 TTL +
+ * single-flight キャッシュを噛ませる (同時要求は 1 本の I/O に畳む)。
+ */
+
+import { cliKindForSessionName } from "./session-names.mjs"
+import { jsonlContextForCwd } from "./usage.mjs"
+
+/** pointer context のキャッシュ TTL (ms)。tmux スナップショットキャッシュと同じ 5 秒。 */
+const POINTER_CONTEXT_TTL_MS = Number(
+  process.env.HUB_AGENT_POINTER_CONTEXT_TTL_MS ?? 5_000,
+)
+
+/** キャッシュ保持数の上限 (会話 ID は増え続けるため LRU で頭打ちにする)。 */
+const POINTER_CONTEXT_MAX_ENTRIES = 512
+
+/**
+ * キー付きの短期 TTL + single-flight キャッシュ。
+ *
+ * `AsyncTtlCache` (単一値版) と役割は同じだが、こちらは `(cwd, session_id)` ごとに独立した
+ * エントリを持つ。build 失敗はキャッシュしない (次の要求で再試行できる)。
+ */
+export class KeyedAsyncTtlCache {
+  constructor({ ttlMs, now = () => Date.now(), maxEntries = POINTER_CONTEXT_MAX_ENTRIES } = {}) {
+    this.ttlMs = Math.max(0, Number(ttlMs) || 0)
+    this.now = now
+    this.maxEntries = Math.max(1, Number(maxEntries) || 1)
+    this.entries = new Map() // key → {value, expiresAt, inFlight}
+  }
+
+  async get(key, build) {
+    const hit = this.entries.get(key)
+    if (hit) {
+      if (hit.inFlight) return hit.inFlight // 同時要求は 1 本の I/O に畳む
+      if (this.now() < hit.expiresAt) return hit.value
+    }
+    const pending = Promise.resolve().then(build)
+    const entry = { value: undefined, expiresAt: 0, inFlight: pending }
+    // delete → set で挿入順 = 最近使った順にし、prune を LRU として使えるようにする。
+    this.entries.delete(key)
+    this.entries.set(key, entry)
+    try {
+      const value = await pending
+      entry.value = value
+      entry.expiresAt = this.now() + this.ttlMs
+      entry.inFlight = null
+      this.prune()
+      return value
+    } catch (err) {
+      if (this.entries.get(key) === entry) this.entries.delete(key)
+      throw err
+    }
+  }
+
+  /** 期限切れエントリを落とし、上限を超えていれば古い順に捨てる。 */
+  prune() {
+    const now = this.now()
+    for (const [key, e] of this.entries) {
+      if (!e.inFlight && now >= e.expiresAt) this.entries.delete(key)
+    }
+    while (this.entries.size > this.maxEntries) {
+      const oldest = this.entries.keys().next()
+      if (oldest.done) break
+      this.entries.delete(oldest.value)
+    }
+  }
+
+  clear() {
+    this.entries.clear()
+  }
+}
+
+/** 既定のプロセス共有キャッシュ (tmux.list_sessions ハンドラ用)。 */
+const defaultPointerContextCache = new KeyedAsyncTtlCache({ ttlMs: POINTER_CONTEXT_TTL_MS })
+
+/** テスト用: 既定キャッシュを空にする。 */
+export function _resetPointerContextCache() {
+  defaultPointerContextCache.clear()
+}
+
+/** cwd と会話 ID の連結。区切りは両者に現れ得ない制御文字 (エスケープ表記で書く)。 */
+function cacheKey(cwd, sessionId) {
+  return `${cwd}\u0000${sessionId}`
+}
+
+/**
+ * `context_targets` (行ごとの「次に開く会話」) を解決し、`session_name → context% | null` の
+ * Map を返す。
+ *
+ * 値が確定しなかった行にも **null を明示的に入れる** (「見に行った上で不明」を表す)。
+ * `context_targets` に含まれない行は Map に載らない = 応答にフィールドを付けない。
+ *
+ * @param {Array<{session_name?: string, session_id?: string|null}>|undefined} targets
+ * @param {Array<{name?: string, cwd?: string|null}>} rows tmux.list_sessions の行 (cwd 解決用)
+ * @param {{contextForCwd?: Function, cache?: KeyedAsyncTtlCache, logger?: object}} [opts]
+ * @returns {Promise<Map<string, number|null>>}
+ */
+export async function resolvePointerContexts(targets, rows, opts = {}) {
+  const {
+    contextForCwd = jsonlContextForCwd,
+    cache = defaultPointerContextCache,
+    logger,
+  } = opts
+  const out = new Map()
+  if (!Array.isArray(targets) || targets.length === 0) return out
+
+  const cwdByName = new Map()
+  for (const row of Array.isArray(rows) ? rows : []) {
+    if (row && typeof row.name === "string") cwdByName.set(row.name, row.cwd || null)
+  }
+
+  const jobs = []
+  for (const target of targets) {
+    const name = typeof target?.session_name === "string" ? target.session_name : ""
+    if (!name || out.has(name)) continue // 同名の重複指定は先勝ち
+    out.set(name, null)
+
+    const sessionId = typeof target?.session_id === "string" ? target.session_id.trim() : ""
+    // 不変条件 1: ID が無ければ読みに行かない (mtime 最新フォールバックを踏まない)。
+    if (!sessionId) continue
+    const cwd = cwdByName.get(name)
+    if (!cwd) continue
+    // codex 行は claude の jsonl を持たない。読んでも uuid 不一致で null になるだけなので、
+    // 無駄な readdir/stat を避けるためここで落とす (結果は同じ)。
+    if (cliKindForSessionName(name) !== "claude") continue
+    jobs.push({ name, cwd, sessionId })
+  }
+
+  await Promise.all(
+    jobs.map(async ({ name, cwd, sessionId }) => {
+      try {
+        const pct = await cache.get(cacheKey(cwd, sessionId), async () => {
+          const res = await contextForCwd(cwd, sessionId)
+          // 不変条件 2: 実体 (uuid) が一致したときだけ採用する。
+          if (!res || res.uuid !== sessionId) return null
+          return typeof res.percent === "number" ? res.percent : null
+        })
+        if (typeof pct === "number") out.set(name, pct)
+      } catch (err) {
+        logger?.warn?.(
+          { err: err?.message, session: name },
+          "pointer context resolve failed",
+        )
+      }
+    }),
+  )
+  return out
+}
+
+/**
+ * 解決済みの pointer context を session 行へ載せる (元配列・元オブジェクトは変更しない)。
+ *
+ * `tmux.list_sessions` の応答行は TTL スナップショットキャッシュで**複数の要求に共有される**
+ * ため、要求ごとに違う値を直接書き込むと他クライアントの応答を汚す。触る行だけ複製する。
+ *
+ * 既存の `context_pct` (いま動いている会話の値) とは**別フィールド**にする。混ぜると
+ * 「どちらの会話の値か」が曖昧になるため。
+ *
+ * @param {Array<object>} sessions
+ * @param {Map<string, number|null>} pointerByName
+ * @returns {Array<object>}
+ */
+export function applyPointerContexts(sessions, pointerByName) {
+  if (!Array.isArray(sessions)) return []
+  if (!(pointerByName instanceof Map) || pointerByName.size === 0) return sessions
+  return sessions.map((s) =>
+    s && pointerByName.has(s.name)
+      ? { ...s, pointer_context_pct: pointerByName.get(s.name) }
+      : s,
+  )
+}
+
+export default { KeyedAsyncTtlCache, resolvePointerContexts, applyPointerContexts }