@cocorograph/hub-agent 0.7.98 → 0.7.99

package/package.json

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

package/src/main.mjs

@@ -2861,9 +2861,12 @@ const STREAMS_SYNC_GRACE_MS = 45_000
  * Plan ε: `agent.streams.sync.response` を受けて backend ↔ agent の現存
  * stream_id を能動同期する。
  *
- * - local にあって backend にない → ptyBridge.detach で即時 kill (孤児)
- * - backend にあって local にない → backend に pty.error を通知 (browser に
- *   "pty 死んでます" を表示するため)
+ * - PTY: local にあって backend にない → ptyBridge.detach、逆差分 → pty.error
+ * - Claude/Codex: mode 別 snapshot から外れた古い subscriber を grace 付きで掃除
+ *
+ * active_stream_ids だけを全件 PTY とみなすと、正常な Claude/Codex に
+ * pty.error(stream_not_found_on_agent) を返して再 attach storm を誘発するため、
+ * 新 backend の active_streams を優先する。
  *
  * dispatch から呼ぶ用のヘルパーだが、テストから直接叩けるよう export している。
  */
@@ -2871,6 +2874,32 @@ export function handleStreamsSyncResponse(msg, ctx) {
   const activeIds = new Set(
     Array.isArray(msg?.active_stream_ids) ? msg.active_stream_ids : [],
   )
+  const activeByMode = { pty: new Set(), claude: new Set(), codex: new Set() }
+  const classifiedIds = new Set()
+  if (Array.isArray(msg?.active_streams)) {
+    for (const item of msg.active_streams) {
+      const streamId = item?.stream_id
+      const mode = item?.mode
+      const modeSet = activeByMode[mode]
+      if (typeof streamId !== "string" || !streamId || !(modeSet instanceof Set)) continue
+      modeSet.add(streamId)
+      classifiedIds.add(streamId)
+    }
+  }
+  // Blue-Green ローリング中に旧 BrowserConsumer の mode 無し応答が混ざった場合は
+  // typed snapshot を authoritative とみなさない。既知 chat stream だけ除外して
+  // legacy 互換を維持し、正常な Claude/Codex に ghost PTY error を返さない。
+  const haveCompleteModeSnapshot =
+    Array.isArray(msg?.active_streams)
+    && classifiedIds.size === activeIds.size
+    && Array.from(classifiedIds).every((id) => activeIds.has(id))
+  const knownChatIds = new Set([
+    ...Array.from(ctx.claudeBridge?.sessions?.keys?.() ?? []),
+    ...Array.from(ctx.codexBridge?.sessions?.keys?.() ?? []),
+  ])
+  const activePtyIds = haveCompleteModeSnapshot
+    ? activeByMode.pty
+    : new Set(Array.from(activeIds).filter((id) => !knownChatIds.has(id)))
   const localIds = new Set(ctx.ptyBridge.list())
   // churn 退化スナップショット保護 (ack_lost 根治 / 2026-06-17 根因 rank3)。
   // backend は browser 群へ _STREAMS_SYNC_TIMEOUT_S(0.5s) だけ問い合わせて active set を
@@ -2881,7 +2910,7 @@ export function handleStreamsSyncResponse(msg, ctx) {
   // 空 set は信用せず kill を一切行わない (真の孤児は 10min GC + maxStreams 上限が回収する)。
   const haveAuthoritativeActiveSet = activeIds.size > 0
   for (const sid of localIds) {
-    if (activeIds.has(sid)) continue
+    if (activePtyIds.has(sid)) continue
     if (!haveAuthoritativeActiveSet) continue
     // grace guard: 直近 STREAMS_SYNC_GRACE_MS 以内に touch (attach/write/resize/pty 出力) された
     // stream は「browser が再接続中で 0.5s 集計に間に合わなかっただけ」の可能性が高い。猶予内は
@@ -2901,7 +2930,7 @@ export function handleStreamsSyncResponse(msg, ctx) {
     )
     ctx.ptyBridge.detach({ stream_id: sid })
   }
-  for (const sid of activeIds) {
+  for (const sid of activePtyIds) {
     if (!localIds.has(sid)) {
       ctx.logger?.warn?.(
         { stream_id: sid, request_id: msg?.request_id },
@@ -2914,6 +2943,51 @@ export function handleStreamsSyncResponse(msg, ctx) {
       })
     }
   }
+
+  if (haveCompleteModeSnapshot) {
+    // reconnect 集計は browser ごとの復帰順で一時的に部分集合になり得る。
+    // 同じ mode の現役購読を1件も確認できない段階では、その mode 全体を掃除しない。
+    if (activeByMode.claude.size > 0) {
+      pruneInactiveChatSubscribers(
+        ctx.claudeBridge,
+        activeByMode.claude,
+        "claude",
+        ctx.logger,
+      )
+    }
+    if (activeByMode.codex.size > 0) {
+      pruneInactiveChatSubscribers(
+        ctx.codexBridge,
+        activeByMode.codex,
+        "codex",
+        ctx.logger,
+      )
+    }
+  }
+}
+
+/** backend の mode-aware snapshot に存在しない死 browser subscriber を除去する。
+ * 再接続が 0.5s 集計に間に合わなかった場合に備え、直近活性が grace 内の stream は保護する。 */
+export function pruneInactiveChatSubscribers(bridge, activeIds, mode, logger, now = Date.now()) {
+  if (!bridge?.sessions || typeof bridge.detach !== "function") return 0
+  let removed = 0
+  for (const session of new Set(bridge.sessions.values())) {
+    for (const streamId of Array.from(session?.streamIds ?? [])) {
+      if (activeIds.has(streamId)) continue
+      const lastSeen = session?.lastActivityByStream?.get?.(streamId) ?? 0
+      if (lastSeen > 0 && now - lastSeen < STREAMS_SYNC_GRACE_MS) continue
+      bridge.detach({ stream_id: streamId })
+      // Codex は live session の索引を最低1件残す不変条件があり、最後の1件では
+      // detachRequested を立てるだけで streamIds からは消さない。実際に消えた件だけ数える。
+      if (session.streamIds?.has?.(streamId)) continue
+      removed += 1
+      logger?.info?.(
+        { stream_id: streamId, mode, idle_ms: lastSeen > 0 ? now - lastSeen : null },
+        "inactive chat subscriber pruned after streams sync",
+      )
+    }
+  }
+  return removed
 }
 
 /**
@@ -3522,6 +3596,9 @@ async function dispatch(msg, ctx) {
         // 5s タイムアウトのような明示的な timer は持たず、再接続時に再度 request
         // が飛ぶので無害。
         handleStreamsSyncResponse(msg, ctx)
+        // buffered pty/control event は browser の購読集合が確定してから再送する。
+        // request_id は WsClient 側で現 reconnect 世代と照合され、古い応答は無視される。
+        ctx.client.completeStreamsSync?.(msg)
         return
       case "claude.attach": {
         // Sprint G: stream-json モードで Claude Code を起動。

package/src/ws-client.mjs

@@ -66,8 +66,9 @@ const PTY_BUFFER_MAX_AGE_MS = 30_000
 // `claude.jsonl.event ... "ws send skipped (not open)"` のドロップを観測)。
 //   - pty.input.ack: 喪失すると browser の pendingAcks が永久滞留し「送信待ち(未配信)」が
 //     消えない (S1/S3)。再接続 flush で遅れて届けば即解消する。
-//   - claude.jsonl.event / claude.event / session.event: 喪失すると新規セッションの
-//     楽観バブルが本物へ昇格できず「生成は始まったがバブルが出ない」(S5)。
+//   - claude/codex の event / ready / exit / error / permission / input ack,nack:
+//     喪失すると新規セッションの楽観バブルが本物へ昇格できず、再 attach も固着する。
+//   - claude.jsonl.event / session.event: TUI と一覧の確定状態を復元する。
 //   - pty.exit / pty.ready / pty.error: stream ライフサイクルの節目。落とすと browser の
 //     再 attach/復帰判断が狂う。
 // pty.data とは別バッファにするのは、高頻度な pty.data の洪水で低頻度な ack/event が
@@ -80,11 +81,29 @@ const BUFFERED_CTRL_TYPES = new Set([
   "pty.error",
   "claude.jsonl.event",
   "claude.event",
+  "claude.ready",
+  "claude.exit",
+  "claude.error",
+  "claude.permission.request",
+  "claude.permission.cancel",
+  "claude.input.ack",
+  "claude.input.nack",
+  "codex.event",
+  "codex.ready",
+  "codex.exit",
+  "codex.error",
+  "codex.permission.request",
+  "codex.input.ack",
+  "codex.input.nack",
   "session.event",
 ])
 // 制御/イベントバッファの上限。低頻度なので frames は小さめ、age は pty と揃える。
 const CTRL_BUFFER_MAX_FRAMES = 200
 const CTRL_BUFFER_MAX_AGE_MS = 30_000
+// reconnect 後の buffered event は backend が browser の active stream 集合を
+// 回収し終えるまで流さない。通常は 0.5s で応答する。旧 backend 互換のため 2s で
+// barrier を解除するが、通常時の送受信経路には一切待ちを追加しない。
+const STREAMS_SYNC_BARRIER_TIMEOUT_MS = 2_000
 
 /** SDK-P2 (2026-07-08): claude.event が「ephemeral な生成中 delta」かどうか。
  *  stream_event の content_block_delta (text_delta / thinking_delta) は高頻度で流れる
@@ -97,6 +116,12 @@ function isEphemeralDeltaEvent(obj) {
   return ev.event?.type === "content_block_delta"
 }
 
+/** Codex App Server の item/agentMessage/delta 等は再水和可能な高頻度delta。 */
+function isEphemeralCodexDeltaEvent(obj) {
+  const method = obj?.event?.method
+  return typeof method === "string" && method.toLowerCase().endsWith("/delta")
+}
+
 export class WsClient extends EventEmitter {
   /**
    * @param {{ hub_url: string, agent_id: string, agent_token: string }} config
@@ -149,6 +174,10 @@ export class WsClient extends EventEmitter {
     // 制御/イベント系 (ack / jsonl.event 等) の outbound buffer。pty.data とは別管理で、
     // 高頻度 pty.data の洪水に evict されないようにする (機構③)。entry: { obj, ts }
     this.ctrlOutboundBuffer = []
+    // reconnect 世代ごとの stream sync barrier。古い request_id の遅延応答で
+    // 現接続の buffer を flush しないよう request_id を照合する。
+    this.pendingStreamsSyncRequestId = null
+    this.streamsSyncBarrierTimer = null
     // 直前の close 原因が CF/origin の 5xx か。次回 backoff の下限を引き上げる。
     this.lastCloseWas5xx = false
     // 接続が STABLE_CONNECTION_MS 維持できたら backoff/5xx フラグをリセットする
@@ -190,6 +219,7 @@ export class WsClient extends EventEmitter {
     })
 
     ws.on("close", (code, reason) => {
+      this._clearStreamsSyncBarrier()
       this._clearStableReset()
       this._stopHeartbeat()
       this._stopBundleWatcher()
@@ -236,11 +266,8 @@ export class WsClient extends EventEmitter {
     // 1〜2 秒間隔の再接続ループに陥らない。
     this._armStableReset()
     this.logger?.info("ws open")
-    // 切断中に積んだ pty.data を flush。hello より先に送ると backend で stream
-    // 未登録のため unknown_stream error になる可能性があるが、hello の応答待ち
-    // 同期は backend が承認するため hello 送信後の同期 send_json で問題ない。
-    // ただし flush は hello 直後ではなく streams.sync.request の後に置く方が
-    // 安全 (sync 完了で stream_id がサーバに復元される想定)。
+    // 切断中に積んだ frame は hello 後も即時 flush しない。streams.sync.response で
+    // browser の購読集合が復元されたことを確認してから、現役 stream 宛だけ再送する。
     // hello 前に最新の bundle version / CLI 別ステータスを読み直す (再接続時の鮮度確保)
     this._refreshBundleVersion()
     this._refreshBundleStatus()
@@ -260,12 +287,12 @@ export class WsClient extends EventEmitter {
     // ptyBridge.list() と差分を取って kill する。request_id は uuid で
     // 重複排除。backend が古い (= Plan ε 未対応) なら "unknown_type" error が
     // 返るだけで動作には影響しない。
+    const streamsSyncRequestId = randomUUID()
     this._sendJson({
       type: "agent.streams.sync.request",
-      request_id: randomUUID(),
+      request_id: streamsSyncRequestId,
     })
-    this._flushPtyBuffer()
-    this._flushCtrlBuffer()
+    this._armStreamsSyncBarrier(streamsSyncRequestId)
     this._startHeartbeat()
     this._startBundleWatcher()
     this.emit("open")
@@ -278,8 +305,13 @@ export class WsClient extends EventEmitter {
    * buffer しない (warn のみ)。ack / event 系は 1 件ずつ意味を持つので別バッファに保持する。
    */
   send(obj) {
-    if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
-      const type = obj?.type
+    const type = obj?.type
+    const wsOpen = this.ws?.readyState === WebSocket.OPEN
+    // sync barrier 中に新しく発生した frame も同じ buffer へ積み、切断中 frame より
+    // 先に送って時系列を逆転させない。heartbeat 等の非 buffer 制御系は通常どおり送る。
+    const holdForStreamsSync = Boolean(this.pendingStreamsSyncRequestId)
+      && (type === "pty.data" || BUFFERED_CTRL_TYPES.has(type))
+    if (!wsOpen || holdForStreamsSync) {
       if (type === "pty.data") {
         this._bufferPtyData(obj)
         return false
@@ -294,11 +326,16 @@ export class WsClient extends EventEmitter {
         if (type === "claude.event" && isEphemeralDeltaEvent(obj)) {
           return false
         }
+        if (type === "codex.event" && isEphemeralCodexDeltaEvent(obj)) {
+          return false
+        }
         this._bufferCtrl(obj)
         return false
       }
-      this.logger?.warn({ type }, "ws send skipped (not open)")
-      return false
+      if (!wsOpen) {
+        this.logger?.warn({ type }, "ws send skipped (not open)")
+        return false
+      }
     }
     return this._sendJson(obj)
   }
@@ -389,9 +426,55 @@ export class WsClient extends EventEmitter {
     )
   }
 
+  /** reconnect stream sync の完了通知。現接続の request_id と一致し、browser の
+   * active stream が1件以上復元されたときだけ buffered frame を flush する。
+   * @returns {boolean} 現在の barrier を解除した場合 true。 */
+  completeStreamsSync(msg) {
+    const requestId = msg?.request_id
+    if (!requestId || requestId !== this.pendingStreamsSyncRequestId) return false
+
+    const activeIds = new Set(
+      Array.isArray(msg?.active_stream_ids)
+        ? msg.active_stream_ids.filter((id) => typeof id === "string" && id)
+        : [],
+    )
+    // backend の 0.5s 集計窓では、browser がまだ再接続しておらず空 snapshot が
+    // 返ることがある。空を authoritative として flush/drop すると live stream の
+    // frame を失うため、2s compatibility timeout まで barrier を維持する。
+    if (activeIds.size === 0) return false
+    this._clearStreamsSyncBarrier()
+    this._flushPtyBuffer()
+    this._flushCtrlBuffer()
+    return true
+  }
+
+  _armStreamsSyncBarrier(requestId) {
+    this._clearStreamsSyncBarrier()
+    this.pendingStreamsSyncRequestId = requestId
+    this.streamsSyncBarrierTimer = setTimeout(() => {
+      if (this.pendingStreamsSyncRequestId !== requestId) return
+      this.pendingStreamsSyncRequestId = null
+      this.streamsSyncBarrierTimer = null
+      this.logger?.warn?.(
+        { request_id: requestId },
+        "streams sync barrier timed out; flushing buffered frames for legacy backend",
+      )
+      this._flushPtyBuffer()
+      this._flushCtrlBuffer()
+    }, STREAMS_SYNC_BARRIER_TIMEOUT_MS)
+    this.streamsSyncBarrierTimer.unref?.()
+  }
+
+  _clearStreamsSyncBarrier() {
+    if (this.streamsSyncBarrierTimer) clearTimeout(this.streamsSyncBarrierTimer)
+    this.streamsSyncBarrierTimer = null
+    this.pendingStreamsSyncRequestId = null
+  }
+
   /** Reconnect を止めて切断する。 */
   stop() {
     this.stopped = true
+    this._clearStreamsSyncBarrier()
     this._clearStableReset()
     this._stopHeartbeat()
     this._stopBundleWatcher()