@cocorograph/hub-agent 0.6.86 → 0.6.87

package/package.json

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

package/src/main.mjs

@@ -1137,6 +1137,20 @@ export function handleTrackedPtyData(msg, ctx) {
   }
 }
 
+/**
+ * 無印 (input_id 無し / fire-and-forget) pty.data の処理。raw 打鍵 (term.onData) が
+ * これを通る。stream 不在 (reap 済み等) なら pty.error を返して browser を再 attach →
+ * 自己修復させる (T-6)。従来は write の戻り値を見ずに捨てていたため、reap 済み stream
+ * への打鍵が無言ドロップされていた。tracked 経路 (handleTrackedPtyData) と対称。
+ * テストから直接呼べるよう dispatch から切り出して export する。
+ */
+export function handleUntrackedPtyData(msg, ctx) {
+  const { stream_id } = msg
+  if (!ctx.ptyBridge.write({ stream_id, data: msg.data })) {
+    ctx.client.send({ type: "pty.error", stream_id, error: "stream_missing" })
+  }
+}
+
 async function dispatch(msg, ctx) {
   const t = msg?.type || ""
   try {
@@ -1203,7 +1217,7 @@ async function dispatch(msg, ctx) {
           handleTrackedPtyData(msg, ctx)
           return
         }
-        ctx.ptyBridge.write({ stream_id: msg.stream_id, data: msg.data })
+        handleUntrackedPtyData(msg, ctx)
         return
       case "pty.resize":
         ctx.ptyBridge.resize({

package/src/pty-bridge.mjs

@@ -86,9 +86,11 @@ export class PtyBridge extends EventEmitter {
      * posix_spawnp が `EAGAIN` で失敗する (= "posix_spawnp failed" の真因)。
      *
      * 対処: 各 stream に `lastSeenAt` を持たせ、`attach` / `write` / `resize`
-     * のたびに touch する。`gcIntervalMs` 周期で `gcStaleMs` を超えた stream を
-     * 自動 detach する (= orphan GC)。**アクティブな stream は browser からの
-     * pty.data / pty.resize で常に touch されるため絶対に kill されない**。
+     * および **pty 出力 (onData)** のたびに touch する。`gcIntervalMs` 周期で
+     * `gcStaleMs` を超えた stream を自動 detach する (= orphan GC)。**アクティブな
+     * stream は browser からの pty.data / pty.resize、または claude の出力で常に
+     * touch されるため絶対に kill されない** (生成中で出力は流れるが入力が間遠い
+     * 閲覧中の TUI チャットが reap される A-4 を出力 touch で塞ぐ)。
      * リロード時の短時間切断 (秒〜十数秒) も `gcStaleMs` (デフォルト 10 分)
      * 未満なので無傷。何時間も放置された孤児だけ掃除される。
      *
@@ -382,6 +384,16 @@ export class PtyBridge extends EventEmitter {
       this.coalesceState.set(stream_id, state)
       disposables.push(
         pty.onData((data) => {
+          // 出力でも lastSeenAt を touch する (A-4 修正)。orphan GC は lastSeenAt が
+          // gcStaleMs 古い stream を reap するが、従来は write/resize/attach でしか
+          // touch していなかったため「生成中で出力は流れているが入力が間遠い閲覧中の
+          // TUI チャット」が 10 分で reap され、以後の送信が stream_missing で落ちていた
+          // (本番 agent.log で reaped_total 多数 + 同 input_id の再送 dedup を観測)。
+          // 出力が流れている = まだ生きて使われている stream なので touch して延命する。
+          // 真にブラウザが去った stream は claude がターン完了後に redraw を止めて出力が
+          // 止まり、gcStaleMs 経過で reap される。万一の runaway 出力 orphan は maxStreams
+          // 上限の最古 reap が backstop になる。
+          this._touch(stream_id)
           state.buf += data
           if (!state.timer) {
             state.timer = setTimeout(() => {
@@ -397,6 +409,8 @@ export class PtyBridge extends EventEmitter {
     } else {
       disposables.push(
         pty.onData((data) => {
+          // 出力でも touch (A-4 修正)。理由は上の coalesce 経路のコメント参照。
+          this._touch(stream_id)
           this.emit("output", { stream_id, data })
         }),
       )

package/src/ws-client.mjs

@@ -58,6 +58,31 @@ const PTY_BUFFER_MAX_FRAMES = 500
 // 30 秒以上経過した pty.data は flush 時に破棄する。
 const PTY_BUFFER_MAX_AGE_MS = 30_000
 
+// agent→browser の制御/イベント系メッセージのうち、WS not OPEN 中に捨てると
+// Cockpit の挙動が壊れるものを reconnect 越しに保持する (機構③: 本番 agent.log で
+// `claude.jsonl.event ... "ws send skipped (not open)"` のドロップを観測)。
+//   - pty.input.ack: 喪失すると browser の pendingAcks が永久滞留し「送信待ち(未配信)」が
+//     消えない (S1/S3)。再接続 flush で遅れて届けば即解消する。
+//   - claude.jsonl.event / claude.event / session.event: 喪失すると新規セッションの
+//     楽観バブルが本物へ昇格できず「生成は始まったがバブルが出ない」(S5)。
+//   - pty.exit / pty.ready / pty.error: stream ライフサイクルの節目。落とすと browser の
+//     再 attach/復帰判断が狂う。
+// pty.data とは別バッファにするのは、高頻度な pty.data の洪水で低頻度な ack/event が
+// リング evict されるのを防ぐため (pty.data は冪等で最新フレームだけ届けば良いが、
+// ack/event は 1 件ずつ意味を持つ)。
+const BUFFERED_CTRL_TYPES = new Set([
+  "pty.input.ack",
+  "pty.exit",
+  "pty.ready",
+  "pty.error",
+  "claude.jsonl.event",
+  "claude.event",
+  "session.event",
+])
+// 制御/イベントバッファの上限。低頻度なので frames は小さめ、age は pty と揃える。
+const CTRL_BUFFER_MAX_FRAMES = 200
+const CTRL_BUFFER_MAX_AGE_MS = 30_000
+
 export class WsClient extends EventEmitter {
   /**
    * @param {{ hub_url: string, agent_id: string, agent_token: string }} config
@@ -94,6 +119,9 @@ export class WsClient extends EventEmitter {
     // pty.data の outbound buffer (WS not OPEN 中に積み、open 後 flush)。
     // entry: { obj, ts }
     this.ptyOutboundBuffer = []
+    // 制御/イベント系 (ack / jsonl.event 等) の outbound buffer。pty.data とは別管理で、
+    // 高頻度 pty.data の洪水に evict されないようにする (機構③)。entry: { obj, ts }
+    this.ctrlOutboundBuffer = []
     // 直前の close 原因が CF/origin の 5xx か。次回 backoff の下限を引き上げる。
     this.lastCloseWas5xx = false
     // 接続が STABLE_CONNECTION_MS 維持できたら backoff/5xx フラグをリセットする
@@ -203,23 +231,30 @@ export class WsClient extends EventEmitter {
       request_id: randomUUID(),
     })
     this._flushPtyBuffer()
+    this._flushCtrlBuffer()
     this._startHeartbeat()
     this._startBundleWatcher()
     this.emit("open")
   }
 
-  /** メッセージを送る。未接続時は pty.data だけ buffer に積み、reconnect 後に flush。
+  /** メッセージを送る。未接続時は pty.data と制御/イベント系 (ack / jsonl.event 等) を
+   * buffer に積み、reconnect 後に flush する。
    *
-   * heartbeat / hello / agent.streams.sync.* など制御系は古くなると意味が無いので
-   * buffer しない (warn のみ)。
+   * heartbeat / hello / agent.streams.sync.* など「古くなると意味が無い」制御系は
+   * buffer しない (warn のみ)。ack / event 系は 1 件ずつ意味を持つので別バッファに保持する。
    */
   send(obj) {
     if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
-      if (obj?.type === "pty.data") {
+      const type = obj?.type
+      if (type === "pty.data") {
         this._bufferPtyData(obj)
         return false
       }
-      this.logger?.warn({ type: obj?.type }, "ws send skipped (not open)")
+      if (BUFFERED_CTRL_TYPES.has(type)) {
+        this._bufferCtrl(obj)
+        return false
+      }
+      this.logger?.warn({ type }, "ws send skipped (not open)")
       return false
     }
     return this._sendJson(obj)
@@ -269,6 +304,48 @@ export class WsClient extends EventEmitter {
     )
   }
 
+  /** 制御/イベント系 (ack / jsonl.event 等) を outbound buffer に積む。リング (drop oldest)。 */
+  _bufferCtrl(obj) {
+    this.ctrlOutboundBuffer.push({ obj, ts: Date.now() })
+    if (this.ctrlOutboundBuffer.length > CTRL_BUFFER_MAX_FRAMES) {
+      const dropped = this.ctrlOutboundBuffer.length - CTRL_BUFFER_MAX_FRAMES
+      this.ctrlOutboundBuffer.splice(0, dropped)
+      this.logger?.warn(
+        { dropped, kept: this.ctrlOutboundBuffer.length },
+        "ctrl outbound buffer overflow (oldest dropped)"
+      )
+    }
+  }
+
+  /** open 直後に制御/イベントバッファを flush。古すぎる entry は破棄する。
+   * 構造は _flushPtyBuffer と同じ (退避 → 期限切れ skip → 送信失敗で残りを順序保持で戻す)。
+   */
+  _flushCtrlBuffer() {
+    if (this.ctrlOutboundBuffer.length === 0) return
+    const now = Date.now()
+    const buf = this.ctrlOutboundBuffer
+    this.ctrlOutboundBuffer = []
+    let sent = 0
+    let expired = 0
+    for (let i = 0; i < buf.length; i++) {
+      const entry = buf[i]
+      if (now - entry.ts > CTRL_BUFFER_MAX_AGE_MS) {
+        expired += 1
+        continue
+      }
+      const ok = this._sendJson(entry.obj)
+      if (!ok) {
+        this.ctrlOutboundBuffer = buf.slice(i).concat(this.ctrlOutboundBuffer)
+        break
+      }
+      sent += 1
+    }
+    this.logger?.info(
+      { sent, expired, total: buf.length, requeued: this.ctrlOutboundBuffer.length },
+      "ctrl outbound buffer flushed"
+    )
+  }
+
   /** Reconnect を止めて切断する。 */
   stop() {
     this.stopped = true