@cocorograph/hub-agent 0.7.13 → 0.7.15

package/package.json

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

package/src/chat-signals.mjs

@@ -35,6 +35,17 @@ const _byCwd = new Map()
 // 一次対策で、本 TTL はその取りこぼし時のバックストップ。
 const CHAT_SIGNAL_STALE_MS = 90 * 1000
 
+// RC4 (症状④): stream_event (拡張思考の thinking_delta / 長時間ツール中の partial delta)
+// 受信を status=processing の tick として記録する際の cwd 単位スロットル間隔。stream_event は
+// トークン delta 毎に高頻度で流れるため、無制限に recordChatActivity を呼ぶと _byCwd への書き込み
+// 洪水になる。この間隔で間引いても、>90s の拡張思考バースト / 単発ツールの間 statusAt・updatedAtMs
+// を確実に前進させ、CHAT_SIGNAL_STALE_MS による idle 誤降格 (純 SDK チャットで三点リーダーが消える
+// 事故) を防げる。
+const STREAM_TICK_THROTTLE_MS = 3 * 1000
+
+// cwd → 最後に stream tick を記録した時刻 (スロットル判定用)。
+const _lastStreamTickAt = new Map()
+
 const VALID_STATUS = new Set(["processing", "waiting", "idle"])
 
 function _now() {
@@ -86,6 +97,33 @@ export function recordChatActivity(cwd, patch = {}, now = _now()) {
   _byCwd.set(cwd, next)
 }
 
+/**
+ * RC4 (症状④): stream_event (拡張思考 thinking_delta / 長時間ツール中の partial delta) 受信を
+ * status=processing の「生きている」tick として記録する。assistant / result が流れない長い無音
+ * 区間 (>90s) でも statusAt・updatedAtMs を前進させ、getChatSignal が CHAT_SIGNAL_STALE_MS で
+ * null を返して三点リーダーが消える事故を防ぐ。
+ *
+ * 設計上の注意:
+ *  - cwd 単位で STREAM_TICK_THROTTLE_MS に間引く (delta 毎の Map 書き込み洪水を回避)。
+ *  - delta 本文は一切読まない (タイムスタンプ前進のみ)。拡張思考の本文は ephemeral で、ここで
+ *    保持・転送する必要は無く、プライバシー上も読むべきでない。
+ *  - SDK のイベント順序上 stream_event はターン中の assistant ブロックに属し result より前に
+ *    流れる。abort 時は SDK query が即停止して onEvent が止まるため、result/turnsettled で
+ *    waiting に落ちた後に trailing stream_event が processing を蘇生させる経路は無い。
+ *
+ * @param {string} cwd
+ * @param {number} [now]
+ */
+export function recordChatStreamTick(cwd, now = _now()) {
+  if (!cwd || typeof cwd !== "string") return
+  // 初回 (未 tick) は必ず記録する。`|| 0` で last を 0 既定にすると「時刻 0 で tick 済み」と
+  // 区別できず、初回 tick が now < throttle のとき誤って間引かれる。has で未 tick を判定する。
+  const last = _lastStreamTickAt.get(cwd)
+  if (last !== undefined && now - last < STREAM_TICK_THROTTLE_MS) return
+  _lastStreamTickAt.set(cwd, now)
+  recordChatActivity(cwd, { status: "processing" }, now)
+}
+
 /**
  * cwd に対応する生きているチャット信号を返す。stale / 不在なら null。
  *
@@ -116,11 +154,13 @@ export function getChatSignal(cwd, now = _now()) {
 export function clearChatSignal(cwd) {
   if (!cwd || typeof cwd !== "string") return
   _byCwd.delete(cwd)
+  _lastStreamTickAt.delete(cwd)
 }
 
 /** テスト用: ストアを空にする。 */
 export function _resetChatSignals() {
   _byCwd.clear()
+  _lastStreamTickAt.clear()
 }
 
-export { CHAT_SIGNAL_STALE_MS }
+export { CHAT_SIGNAL_STALE_MS, STREAM_TICK_THROTTLE_MS }

package/src/main.mjs

@@ -92,7 +92,12 @@ import {
   getUsage,
   recordChatRateLimit,
 } from "./usage.mjs"
-import { clearChatSignal, getChatSignal, recordChatActivity } from "./chat-signals.mjs"
+import {
+  clearChatSignal,
+  getChatSignal,
+  recordChatActivity,
+  recordChatStreamTick,
+} from "./chat-signals.mjs"
 import {
   ReadinessTracker,
   busyChildCount,
@@ -491,6 +496,17 @@ export async function startDaemon({ version, ptyModule, claudeSdk } = {}) {
         } catch {
           /* ignore */
         }
+      } else if (event?.type === "stream_event") {
+        // RC4 (症状④): 拡張思考 (thinking_delta) や長時間ツール中の partial delta では
+        // assistant / result が流れず statusAt / updatedAtMs が前進しないため、>90s の無音で
+        // getChatSignal が CHAT_SIGNAL_STALE_MS により null を返し、capture-pane が idle な
+        // 純 SDK チャットで三点リーダーが消える。stream_event 受信を throttle 付き processing
+        // tick として記録し statusAt を前進させる (本文は読まずタイムスタンプ前進のみ)。
+        try {
+          recordChatStreamTick(cwd)
+        } catch {
+          /* ignore */
+        }
       }
       client.send({ type: "claude.event", stream_id, session_id, event })
     })
@@ -1656,9 +1672,13 @@ export async function handleTuiInterrupt(msg, ctx) {
   const sendEsc = ctx.sendInterruptKey || sendInterruptKey
   const detect = ctx.detectSessionState || detectSessionState
   const delay = ctx.delay || ((ms) => new Promise((r) => setTimeout(r, ms)))
-  const settleMs = ctx.interruptSettleMs ?? 350
-  const pollMs = ctx.interruptPollMs ?? 350
-  const maxPolls = ctx.interruptMaxPolls ?? 7 // ~2.5s deadline
+  // 症状⑤: 初回チェックまでの settle と poll 間隔を 350→150ms に詰め、maxPolls を 7→16 に
+  // 増やす。総 deadline は ~2.4s で従来(~2.5s)とほぼ同じだが、実際に停止した瞬間の観測が
+  // 約2倍速くなり「停止押下→画面が止まる」体感ラグを縮める。capture-pane スクレイプ回数は
+  // 増えるが 150ms 間隔なら許容範囲。Ctrl+C 非エスカレーション方針は不変。
+  const settleMs = ctx.interruptSettleMs ?? 150
+  const pollMs = ctx.interruptPollMs ?? 150
+  const maxPolls = ctx.interruptMaxPolls ?? 16 // ~2.4s deadline (150 + 150*15)
   const reply = (stopped, attempts) =>
     ctx.client.send({
       type: "claude.tui.interrupt.result",

package/src/ws-client.mjs

@@ -3,7 +3,7 @@
  *
  * - 接続: `Authorization: Bearer <agent_id>:<agent_token>`
  * - 起動時に `hello`、30s おきに `heartbeat` を送信
- * - 切断時は exponential backoff (1s, 2s, 4s, ..., max 30s) で再接続
+ * - 切断時は exponential backoff (1s, 2s, 4s, 8s max) で再接続
  * - サーバから受け取った JSON は `onMessage` callback に渡す
  * - `bundleVersionProvider` + `bundleManifestPath` を渡すと、manifest.json の
  *   変更を fs.watch で検知して即時 heartbeat を送信し Cockpit UI に最新版を
@@ -20,7 +20,10 @@ import WebSocket from "ws"
 
 const HEARTBEAT_INTERVAL_MS = 30_000
 const MIN_BACKOFF_MS = 1_000
-const MAX_BACKOFF_MS = 30_000
+// 症状①(接続復帰の遅さ)対策: 旧 30s。本番ログで再接続 p50=1.9s に対し、フラッピング時に
+// backoff が指数増加して復帰レイテンシのテール(p90=9s, 最大35s)を作っていた。実測の切断は
+// ほぼ瞬断(CF/NW)で 8s も待てば回復に十分。jitter(±20%)で thundering herd は引き続き回避。
+const MAX_BACKOFF_MS = 8_000
 const BUNDLE_WATCH_DEBOUNCE_MS = 500
 
 // zombie WS (half-open TCP) 検知のしきい値。backend は heartbeat (30s) のたびに
@@ -196,15 +199,18 @@ export class WsClient extends EventEmitter {
    * のクロージャだと spy しづらい)。
    */
   _onOpen() {
-    // backoff の即時リセットはしない。接続が STABLE_CONNECTION_MS 維持できてから
-    // _armStableReset() でリセットする (open→即 close フラッピング時の再接続
-    // ストロボ対策)。すぐ切れた接続では backoff がリセットされず指数バックオフが
-    // 効き続けるため、1〜2 秒間隔の再接続ループに陥らない。
-    this._armStableReset()
     // open 自体が生存の証拠。直前接続の古い lastRecvAt で開幕直後に zombie 誤検知
-    // しないようリセットする。
+    // しないようリセットする。⚠️ _armStableReset() の armedAt 捕捉より「前」に設定すること。
+    // ack ベース判定 (lastRecvAt > armedAt) を初期状態で偽にするため (順序を逆にすると
+    // armedAt < lastRecvAt となり「inbound 受信済み」と誤判定し、half-open でも backoff を
+    // リセットしてしまう)。
     this.lastRecvAt = Date.now()
     this.lastTickAt = Date.now()
+    // backoff の即時リセットはしない。接続が STABLE_CONNECTION_MS 維持 + inbound 受信できてから
+    // _armStableReset() でリセットする (open→即 close フラッピング時の再接続ストロボ対策)。
+    // すぐ切れた / half-open の接続では backoff がリセットされず指数バックオフが効き続けるため、
+    // 1〜2 秒間隔の再接続ループに陥らない。
+    this._armStableReset()
     this.logger?.info("ws open")
     // 切断中に積んだ pty.data を flush。hello より先に送ると backend で stream
     // 未登録のため unknown_stream error になる可能性があるが、hello の応答待ち
@@ -583,11 +589,27 @@ export class WsClient extends EventEmitter {
    */
   _armStableReset() {
     this._clearStableReset()
+    // 症状①: backoff リセットを「ack ベース」にする。armedAt(≈open 時刻) を捕捉し、
+    // STABLE_CONNECTION_MS 後に「open 以降に inbound を受信したか(lastRecvAt > armedAt)」で
+    // 判定する。受信あり = 双方向に健全とみなし backoff/5xx をリセット。受信ゼロ =
+    // half-open(send は kernel buffer に書けて成功扱いだが TCP は死)とみなし、75s の zombie
+    // 検知を待たずに即 forceReconnect する(再接続後の half-open を 10s で回収し復帰を早める)。
+    // _onOpen は lastRecvAt=now を armedAt より前に設定するため、初期状態では
+    // lastRecvAt <= armedAt となり「未受信」と正しく判定される。
+    const armedAt = Date.now()
     this.stableResetTimer = setTimeout(() => {
       this.stableResetTimer = null
-      this.backoff = MIN_BACKOFF_MS
-      this.lastCloseWas5xx = false
-      this.logger?.debug("ws connection stable, backoff reset")
+      if (this.lastRecvAt > armedAt) {
+        this.backoff = MIN_BACKOFF_MS
+        this.lastCloseWas5xx = false
+        this.logger?.debug("ws connection stable (inbound received), backoff reset")
+      } else {
+        this.logger?.warn(
+          { sinceArmedMs: Date.now() - armedAt },
+          "ws stable window elapsed with no inbound (half-open suspected), forcing reconnect",
+        )
+        this._forceReconnect()
+      }
     }, STABLE_CONNECTION_MS)
     this.stableResetTimer.unref?.()
   }