@cocorograph/hub-agent 0.6.64 → 0.6.66

package/package.json

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

package/src/main.mjs

@@ -1044,6 +1044,53 @@ export function handleStreamsSyncResponse(msg, ctx) {
   }
 }
 
+/**
+ * end-to-end 配達確認 (tracked pty.data) の冪等化レジストリ。
+ *
+ * browser はチャット送信 (paste / 確定 CR) に input_id を付け、agent からの
+ * pty.input.ack が来ない限り再送する (zombie WS / 切断窓 / backend group_send の
+ * 無言喪失への対策)。「実際は届いていたが ack 側が消えた」再送で同じ本文が
+ * 二重 paste されないよう、処理済み input_id を記憶して 2 回目以降は write せず
+ * ack だけ返す。Set は挿入順 iterate できるので最古から間引く。
+ */
+const SEEN_INPUT_IDS_MAX = 1_000
+const seenInputIds = new Set()
+function rememberInputId(input_id) {
+  seenInputIds.add(input_id)
+  while (seenInputIds.size > SEEN_INPUT_IDS_MAX) {
+    seenInputIds.delete(seenInputIds.values().next().value)
+  }
+}
+
+/**
+ * input_id 付き pty.data (tracked 入力) の処理。write 成功で pty.input.ack を返し、
+ * stream 不在なら pty.error を返して browser の再 attach → 再送に繋げる。
+ * テストから直接呼べるよう dispatch から切り出して export する。
+ */
+export function handleTrackedPtyData(msg, ctx) {
+  const { stream_id, input_id } = msg
+  if (seenInputIds.has(input_id)) {
+    // 再送 (ack 喪失 or 再接続 flush)。書き込み済みなので ack だけ返す。
+    ctx.logger?.info({ stream_id, input_id }, "tracked pty.data dedup (ack only)")
+    ctx.client.send({ type: "pty.input.ack", stream_id, input_id })
+    return
+  }
+  const ok = ctx.ptyBridge.write({ stream_id, data: msg.data })
+  if (ok) {
+    rememberInputId(input_id)
+    ctx.client.send({ type: "pty.input.ack", stream_id, input_id })
+  } else {
+    // stream 不在 (reap 済み等)。browser は pty.error で再 attach し、未 ack の
+    // tracked 入力は ack ウォッチドッグが再送する。
+    ctx.client.send({
+      type: "pty.error",
+      stream_id,
+      error: "stream_missing",
+      input_id,
+    })
+  }
+}
+
 async function dispatch(msg, ctx) {
   const t = msg?.type || ""
   try {
@@ -1073,6 +1120,10 @@ async function dispatch(msg, ctx) {
             stream_id,
             session_name: msg.session_name || "",
             plugin: info.plugin,
+            // end-to-end 配達確認 (input_id 付き pty.data → pty.input.ack) に対応
+            // していることを browser へ通知する capability フラグ。旧 agent には
+            // このフラグが無く、browser 側は tracked 送信を自動で無効化する。
+            supports_input_ack: true,
           })
         } catch (err) {
           ctx.client.send({
@@ -1085,6 +1136,12 @@ async function dispatch(msg, ctx) {
         return
       }
       case "pty.data":
+        // input_id 付き = tracked 入力 (チャット送信)。冪等化 + ack 返却で
+        // end-to-end 配達確認に応える。無印は従来どおり fire-and-forget。
+        if (typeof msg.input_id === "string" && msg.input_id) {
+          handleTrackedPtyData(msg, ctx)
+          return
+        }
         ctx.ptyBridge.write({ stream_id: msg.stream_id, data: msg.data })
         return
       case "pty.resize":
@@ -1400,15 +1457,19 @@ async function dispatch(msg, ctx) {
         // model="" は「デフォルト」= `/model default`。frontend へはそのまま空で返し、
         // 解決後の実 id は次ターンの jsonl 由来 (message.model) に委ねる。
         const model = typeof msg.model === "string" ? msg.model : ""
+        // 先に broadcast してフロントの fallback (raw `/model` paste) を即キャンセルさせる。
+        // setTuiModel は「Switch model?」確認ダイアログの描画検知で数秒かかり得るため、その
+        // 完了を待つとフロントの MODEL_SYNC_FALLBACK_MS が先に発火して二重 paste になる。
+        // バッジは要求モデルを即表示し、確定値は次ターンの jsonl (message.model) で補正される。
+        ctx.client.send({
+          type: "claude.tui.model",
+          cwd: cwd || undefined,
+          session_name: sessionName,
+          model,
+        })
         ;(async () => {
           try {
             await setTuiModel(sessionName, model || "default", { logger })
-            ctx.client.send({
-              type: "claude.tui.model",
-              cwd: cwd || undefined,
-              session_name: sessionName,
-              model,
-            })
             logger.info(
               { session: sessionName, cwd, model: model || "(default)" },
               "tui model switched → notified browser",
@@ -1433,15 +1494,19 @@ async function dispatch(msg, ctx) {
         if (!sessionName) return
         // effort="" は「auto」= `/effort auto`。frontend へはそのまま空で返し、バッジは auto 表示。
         const effort = typeof msg.effort === "string" ? msg.effort : ""
+        // setModel と同様、先に broadcast してフロントの fallback (raw `/effort` paste) を即
+        // キャンセルさせる。setTuiEffort も「Change effort level?」確認ダイアログの描画検知で
+        // 数秒かかり得るため、完了待ちだと EFFORT_SYNC_FALLBACK_MS が先に発火して二重 paste
+        // になる。effort は jsonl に記録されないので、この即時 broadcast が表示の根拠になる。
+        ctx.client.send({
+          type: "claude.tui.effort",
+          cwd: cwd || undefined,
+          session_name: sessionName,
+          effort,
+        })
         ;(async () => {
           try {
             await setTuiEffort(sessionName, effort || "auto", { logger })
-            ctx.client.send({
-              type: "claude.tui.effort",
-              cwd: cwd || undefined,
-              session_name: sessionName,
-              effort,
-            })
             logger.info(
               { session: sessionName, cwd, effort: effort || "(auto)" },
               "tui effort switched → notified browser",

package/src/tmux.mjs

@@ -785,6 +785,74 @@ export function buildFreshCmd(opts = {}) {
 
 const _delay = (ms) => new Promise((r) => setTimeout(r, ms))
 
+/**
+ * `/model` / `/effort` 切替時に claude TUI が出す確認ダイアログを検知する正規表現。
+ *
+ * claude v2.1.170 実機確認 (2026-06-10): 会話に出力済みの履歴 (prior output) があると、
+ * モデル/effort 切替は即時反映されず確認ダイアログを挟む。ダイアログは:
+ *   - タイトル: "Switch model?" / "Change effort level?"
+ *   - 本文: "This conversation is cached for the current model. Switching to … means the
+ *     full history gets re-read on your next message."
+ *   - 選択肢: "Yes, switch to …" (デフォルトハイライト) / "No, go back"
+ * デフォルトが確定側なので Enter 1 回で承認できる。複数マーカーのいずれかで判定する
+ * (文言の小変更に多少強くする)。
+ */
+const SWITCH_CONFIRM_DIALOG_RE =
+  /Switch model\?|Change effort level\?|Yes, switch to|No, go back|full history gets re-read/i
+
+/**
+ * `/model` / `/effort` の確認ダイアログを「描画を検知してから」Enter で畳む。
+ *
+ * 旧実装は本文確定 Enter の一定時間後に盲打ちで追い Enter を 1 回送っていたが、確認
+ * ダイアログのレンダはコマンド確定から遅延することがあり (TUI の全画面再描画ラグ)、
+ * 固定待ちの Enter がダイアログ描画前に着弾して取りこぼされ、ダイアログが残ったまま
+ * 次の入力の Enter を吸収する事故になっていた。代わりにペインを短周期で読み、ダイアログ
+ * が実際に描画されたのを確認してから Enter を送り、消えるまで数回リトライする。ダイアログ
+ * が出ない場合 (prior output 無し = 即時反映) は余計な Enter を送らない (空 Enter が
+ * 別の場面で誤確定するのを避ける)。ベストエフォート。
+ *
+ * タイミングは opts で上書きできる (テストで短縮するため):
+ *   - confirmAppearMs: ダイアログ出現を待つ上限 (既定 2400ms)
+ *   - confirmPollMs:   出現待ちのポーリング間隔 (既定 150ms)
+ *   - confirmDismissMs: 承認後にダイアログ消失を待つ上限 (既定 2000ms)
+ *
+ * @param {string} bin tmux バイナリパス
+ * @param {string} name tmux セッション名
+ * @param {{logger?:object,tmuxBin?:string,confirmAppearMs?:number,confirmPollMs?:number,confirmDismissMs?:number}} opts
+ * @returns {Promise<boolean>} 確認ダイアログを承認したら true
+ */
+async function _confirmSwitchDialog(bin, name, opts) {
+  const appearMs = opts.confirmAppearMs ?? 2400
+  const pollMs = opts.confirmPollMs ?? 150
+  const dismissMs = opts.confirmDismissMs ?? 2000
+  // ダイアログ出現待ち: コマンド確定から最大 appearMs 観測する (レンダ遅延の上限想定)。
+  const APPEAR_DEADLINE = Date.now() + appearMs
+  let appeared = false
+  while (Date.now() < APPEAR_DEADLINE) {
+    const text = await capturePane(name, { ...opts, noCache: true })
+    if (SWITCH_CONFIRM_DIALOG_RE.test(text)) {
+      appeared = true
+      break
+    }
+    await _delay(pollMs)
+  }
+  if (!appeared) return false // prior output 無し = 即時反映済み。追い Enter 不要。
+
+  // ダイアログが消える (= 承認反映) まで Enter を送ってリトライする。
+  const DISMISS_DEADLINE = Date.now() + dismissMs
+  while (Date.now() < DISMISS_DEADLINE) {
+    await execFileP(bin, ["send-keys", "-t", name, "Enter"])
+    await _delay(180)
+    const after = await capturePane(name, { ...opts, noCache: true })
+    if (!SWITCH_CONFIRM_DIALOG_RE.test(after)) return true
+  }
+  opts.logger?.warn(
+    { session: name },
+    "switch confirm dialog still visible after retries",
+  )
+  return true
+}
+
 /**
  * 遅延回答 resume: 承認/質問カードへの回答がフックの生存中に間に合わなかったとき、
  * 回答を「新しいユーザーメッセージ」として同一 tmux セッションの対話 claude へ届ける。
@@ -921,9 +989,9 @@ export async function setTuiModel(name, modelArg, opts = {}) {
     await _delay(120)
     // Enter で本文確定 (send-keys の離散イベントなので paste 吸収は起きにくい)。
     await execFileP(bin, ["send-keys", "-t", name, "Enter"])
-    // prior output ありの場合に出るモデル切替の確認プロンプトを Enter で畳む。
-    await _delay(450)
-    await execFileP(bin, ["send-keys", "-t", name, "Enter"])
+    // prior output ありの場合に出る「Switch model?」確認ダイアログを、描画を検知してから
+    // Enter で畳む (固定待ちの盲打ち Enter はレンダ遅延と競合して取りこぼすため)。
+    await _confirmSwitchDialog(bin, name, opts)
     return { ok: true }
   } catch (err) {
     opts.logger?.warn(
@@ -941,9 +1009,11 @@ export async function setTuiModel(name, modelArg, opts = {}) {
  * 同設計で、frontend は raw pty.data ではなく claude.tui.setEffort を送り、agent 側で本関数を
  * 実行 → 全ブラウザへ claude.tui.effort を broadcast して全端末を実 effort に揃える。
  *
- * `/effort` は引数 (low/medium/high/xhigh/max/auto) を受理し即時反映する (確認プロンプトを
- * 挟まない)。effort 非対応レベルを与えても CLI がモデルの対応上限へ自動フォールバックする。
- * `auto` はモデル既定へリセット。copy-mode に入っているとキーが奪われるので先に抜ける。
+ * `/effort` は引数 (low/medium/high/xhigh/max/auto) を受理する。effort 非対応レベルを
+ * 与えても CLI がモデルの対応上限へ自動フォールバックする。`auto` はモデル既定へリセット。
+ * claude v2.1.170 以降は会話に出力済みの履歴があると「Change effort level?」確認ダイアログ
+ * を挟むようになった (モデル切替と同じ S1_ コンポーネント) ため、setTuiModel と同様に描画を
+ * 検知してから Enter で畳む。copy-mode に入っているとキーが奪われるので先に抜ける。
  * ベストエフォート。
  *
  * @param {string} name tmux セッション名
@@ -977,6 +1047,9 @@ export async function setTuiEffort(name, effortArg, opts = {}) {
     await _delay(120)
     // Enter で本文確定 (send-keys の離散イベントなので paste 吸収は起きにくい)。
     await execFileP(bin, ["send-keys", "-t", name, "Enter"])
+    // prior output ありの場合に出る「Change effort level?」確認ダイアログを、描画を検知して
+    // から Enter で畳む (setTuiModel と同じ堅牢化)。
+    await _confirmSwitchDialog(bin, name, opts)
     return { ok: true }
   } catch (err) {
     opts.logger?.warn(

package/src/ws-client.mjs

@@ -23,6 +23,19 @@ const MIN_BACKOFF_MS = 1_000
 const MAX_BACKOFF_MS = 30_000
 const BUNDLE_WATCH_DEBOUNCE_MS = 500
 
+// zombie WS (half-open TCP) 検知のしきい値。backend は heartbeat (30s) のたびに
+// ack を返すため、健全な接続なら inbound frame が最低 30s+RTT に 1 回は届く。
+// これを超えて沈黙した接続は「readyState=OPEN だが TCP は死んでいる」zombie の
+// 可能性が高い (send は kernel buffer に書けて成功扱いになるため throw しない)。
+// heartbeat 2.5 周期 = 75s を超えたら強制再接続する。browser 側 CockpitTerminal の
+// LIVENESS_TIMEOUT_MS (PR#2342) と同じパターンの agent 版。
+const LIVENESS_TIMEOUT_MS = HEARTBEAT_INTERVAL_MS * 2.5
+// heartbeat tick 間隔の実測ジャンプ検知 (Mac スリープ復帰対策)。スリープ中は
+// プロセスごと止まるため setInterval の実発火間隔が大きく開く。復帰直後の TCP は
+// ほぼ確実に死んでいる (half-open) のに send は成功扱いになるので、ジャンプを
+// 検知したら鮮度判定を待たず即座に再接続する。2 周期 (60s) 超のギャップで発火。
+const TICK_JUMP_MS = HEARTBEAT_INTERVAL_MS * 2
+
 // 接続が安定したとみなすまでの猶予。open 直後に即 backoff をリセットすると、
 // open→即 close のフラッピング時に毎回 backoff が最小 (1s) へ戻り、1〜2 秒間隔の
 // 再接続ストロボに陥る (CF/backend 側の瞬断時に hub-agent↔backend WS で観測)。
@@ -85,6 +98,11 @@ export class WsClient extends EventEmitter {
     // 接続が STABLE_CONNECTION_MS 維持できたら backoff/5xx フラグをリセットする
     // タイマー。close (_clearStableReset) でキャンセルされる。
     this.stableResetTimer = null
+    // zombie WS 検知: 最後に inbound frame (heartbeat ack / agent.relay 等の何か) を
+    // 受信した時刻。健全なら heartbeat ack で 30s ごとに必ず更新される。
+    this.lastRecvAt = 0
+    // heartbeat tick の直前実発火時刻。スリープ復帰のジャンプ検知に使う。
+    this.lastTickAt = 0
   }
 
   /** WSS 接続を開始する。`stop()` まで自動で reconnect 続行。 */
@@ -102,6 +120,9 @@ export class WsClient extends EventEmitter {
     ws.on("open", () => this._onOpen())
 
     ws.on("message", (data) => {
+      // 何かを受信した = TCP は生きている。zombie 検知の鮮度を更新する
+      // (parse 失敗でも「届いた」事実は生存の証拠なので先に記録する)。
+      this.lastRecvAt = Date.now()
       let msg
       try {
         msg = JSON.parse(data.toString("utf-8"))
@@ -151,6 +172,10 @@ export class WsClient extends EventEmitter {
     // ストロボ対策)。すぐ切れた接続では backoff がリセットされず指数バックオフが
     // 効き続けるため、1〜2 秒間隔の再接続ループに陥らない。
     this._armStableReset()
+    // open 自体が生存の証拠。直前接続の古い lastRecvAt で開幕直後に zombie 誤検知
+    // しないようリセットする。
+    this.lastRecvAt = Date.now()
+    this.lastTickAt = Date.now()
     this.logger?.info("ws open")
     // 切断中に積んだ pty.data を flush。hello より先に送ると backend で stream
     // 未登録のため unknown_stream error になる可能性があるが、hello の応答待ち
@@ -284,7 +309,35 @@ export class WsClient extends EventEmitter {
 
   _startHeartbeat() {
     this._stopHeartbeat()
+    this.lastTickAt = Date.now()
     this.heartbeatTimer = setInterval(() => {
+      const now = Date.now()
+      // スリープ復帰検知: スリープ中はプロセスごと止まり、setInterval の実発火間隔が
+      // 大きく開く。復帰直後の TCP は half-open でほぼ確実に死んでいるのに send は
+      // 成功扱いになるため、鮮度判定 (受信側) を待たず即座に再接続へ倒す。
+      const tickGap = this.lastTickAt > 0 ? now - this.lastTickAt : 0
+      this.lastTickAt = now
+      if (tickGap > TICK_JUMP_MS) {
+        this.logger?.warn(
+          { tickGapMs: tickGap },
+          "heartbeat tick jump detected (likely sleep wake), forcing reconnect",
+        )
+        this._forceReconnect()
+        return
+      }
+      // zombie WS 検知 (inbound 鮮度): backend は heartbeat のたびに ack を返すため、
+      // 健全なら lastRecvAt は 30s+RTT 以内に必ず更新される。LIVENESS_TIMEOUT_MS を
+      // 超えて何も届いていない接続は half-open とみなし強制再接続する。send の
+      // throw 判定 (下) では half-open を検知できない (kernel buffer に書けて
+      // 成功扱いになる) ことへの対策。browser 側 PR#2342 と同じパターン。
+      if (this.lastRecvAt > 0 && now - this.lastRecvAt > LIVENESS_TIMEOUT_MS) {
+        this.logger?.warn(
+          { sinceLastRecvMs: now - this.lastRecvAt },
+          "ws liveness timeout (zombie suspected), forcing reconnect",
+        )
+        this._forceReconnect()
+        return
+      }
       // heartbeat 都度 provider 経由で bundle version を最新化する。
       // fs.watch を取り逃した変更 (atomic rename / 別マウント越し等) への保険。
       this._refreshBundleVersion()