@cocorograph/hub-agent 0.7.9 → 0.7.10

package/package.json

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

package/src/main.mjs

@@ -76,6 +76,7 @@ import {
   removeWorktree as removeWorktreeDir,
   resumeWithMessage,
   pasteToSessionByName,
+  sendInterruptKey,
   setSessionMouse,
   setTmuxGlobalEnv,
   setTuiEffort,
@@ -1519,6 +1520,156 @@ export function handleUntrackedPtyData(msg, ctx) {
 const _seenFlushIds = new Set()
 const _SEEN_FLUSH_CAP = 2000
 
+/**
+ * claude.tui.queue.flush の処理 (離席中キュー自動投入の agent 側受け口)。
+ * ブラウザの常駐ドレイン (useCockpitTuiQueueDrainAll) が、表示していないセッションの送信待ち
+ * キュー先頭 1 件をここへ送る。tmux send-keys (PTY 非依存) で投入するので、そのセッションの
+ * ターミナルを誰も開いていなくても届く。結果を flush.ack で返し、ブラウザは ack を受けて
+ * はじめて localStorage から 1 件 dequeue する。
+ *
+ * 冪等性は 2 段:
+ *   1. flush_id 単位 (_seenFlushIds): WS 再送・ack 取りこぼしによる同一 flush の二重投入を防ぐ。
+ *   2. 論理 item id 単位 (cross-path): browser の active 経路 (tracked PTY paste) は同一 item を
+ *      input_id `qd:<itemId>:p` / `:r` で送る。ここと単一レジストリ (seenInputIds) を共有し、同一
+ *      item がどちらの経路で配送されても agent は at-most-once でしか paste しない (二重送信の根治)。
+ *
+ * テストから直接呼べるよう dispatch から切り出して export する。paste 実体は ctx で差し替え可能。
+ */
+export async function handleQueueFlush(msg, ctx) {
+  const paste = ctx.pasteToSessionByName || pasteToSessionByName
+  const sessionName =
+    typeof msg.session_name === "string" ? msg.session_name : ""
+  const text = typeof msg.text === "string" ? msg.text : ""
+  const flushId = typeof msg.flush_id === "string" ? msg.flush_id : ""
+  const itemId = typeof msg.item_id === "string" ? msg.item_id : ""
+  const pasteKey = itemId ? `qd:${itemId}:p` : ""
+  const crKey = itemId ? `qd:${itemId}:r` : ""
+  const ackFail = (reason) => {
+    if (flushId && sessionName) {
+      ctx.client.send({
+        type: "claude.tui.queue.flush.ack",
+        session_name: sessionName,
+        flush_id: flushId,
+        ok: false,
+        error: reason,
+      })
+    }
+  }
+  const ackOk = () => {
+    if (_seenFlushIds.size >= _SEEN_FLUSH_CAP) _seenFlushIds.clear()
+    _seenFlushIds.add(flushId)
+    ctx.client.send({
+      type: "claude.tui.queue.flush.ack",
+      session_name: sessionName,
+      flush_id: flushId,
+      ok: true,
+    })
+  }
+  if (!sessionName || !flushId || !text) {
+    ackFail("missing session_name / flush_id / text")
+    return
+  }
+  // 冪等化1 (message 再送)。
+  if (_seenFlushIds.has(flushId)) {
+    ackOk()
+    return
+  }
+  // 冪等化2 (cross-path): 同一 item を active 経路 (tracked PTY paste) が既に配送済みなら
+  // 再 paste せず ok を返す (browser は ack で dequeue する)。
+  if (pasteKey && seenInputIds.has(pasteKey)) {
+    ackOk()
+    return
+  }
+  // paste 前に cross-path キーを登録する。await 中に並走する active 経路の tracked paste
+  // (handleTrackedPtyData) を dedup させ二重 paste を防ぐため (登録は paste 完了より先)。
+  // 失敗時は登録解除し、再試行 (別 flush_id) で再配送できるようにする。
+  if (pasteKey) {
+    rememberInputId(pasteKey)
+    rememberInputId(crKey)
+  }
+  const result = await paste(sessionName, text, { logger: ctx.logger })
+  if (result.ok) {
+    if (_seenFlushIds.size >= _SEEN_FLUSH_CAP) _seenFlushIds.clear()
+    _seenFlushIds.add(flushId)
+  } else if (pasteKey) {
+    seenInputIds.delete(pasteKey)
+    seenInputIds.delete(crKey)
+  }
+  ctx.client.send({
+    type: "claude.tui.queue.flush.ack",
+    session_name: sessionName,
+    flush_id: flushId,
+    ok: !!result.ok,
+    error: result.ok ? undefined : result.error,
+  })
+}
+
+/**
+ * claude.tui.interrupt の処理 (確認付き中断 = Phase3)。生 ESC を tracked pty.data で best-effort
+ * 送出する旧経路 (stream 欠落で無言ドロップ + ESC 到達でも止まったか未検証) を、agent が ESC を
+ * tmux send-keys (PTY 非依存で確実配達) で送り、ペイン status で生成停止を観測してから結果を返す
+ * 方式に置き換える。
+ *
+ * 不変条件: 中断指示は、claude と同一ホストで生成状態を観測できる agent が「停止を観測 or deadline」
+ * まで責任を持つ。止まらなければ deadline 内で 1 度だけ ESC を再送する (初回が一過性 mode に吸われた
+ * 場合の保険)。Ctrl+C へはエスカレーションしない (claude を終了させる事故を避けるため。別判断)。
+ *
+ * テストから直接呼べるよう export。送出 (sendInterruptKey) / 観測 (detectSessionState) / 待機 (delay)
+ * は ctx で差し替え可能。
+ */
+export async function handleTuiInterrupt(msg, ctx) {
+  const sessionName =
+    typeof msg.session_name === "string" ? msg.session_name : ""
+  const cwd = typeof msg.cwd === "string" ? msg.cwd : ""
+  const requestId = typeof msg.request_id === "string" ? msg.request_id : ""
+  if (!sessionName || !requestId) return
+  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
+  const reply = (stopped, attempts) =>
+    ctx.client.send({
+      type: "claude.tui.interrupt.result",
+      request_id: requestId,
+      session_name: sessionName,
+      cwd: cwd || undefined,
+      stopped: !!stopped,
+      attempts,
+    })
+  const stillProcessing = async () => {
+    try {
+      const snap = await detect(sessionName, { logger: ctx.logger })
+      return (snap?.status ?? "processing") === "processing"
+    } catch {
+      // capture 失敗時は判定不能 → 保守的に「まだ生成中かも」と扱う (誤って stopped=true にしない)。
+      return true
+    }
+  }
+  // 1) ESC 送出 (= 中断)。
+  await sendEsc(sessionName, { logger: ctx.logger })
+  let attempts = 1
+  let resent = false
+  // 2) 停止 (status が processing を離れる) を deadline 内で観測。
+  for (let i = 0; i < maxPolls; i++) {
+    await delay(i === 0 ? settleMs : pollMs)
+    if (!(await stillProcessing())) {
+      reply(true, attempts)
+      return
+    }
+    // deadline 中盤でまだ processing なら 1 度だけ ESC 再送 (初回が一過性 mode に吸われた保険)。
+    // status=processing 継続が条件なので、アイドル箱への二度押し (rewind ダイアログ誤爆) は避ける。
+    if (!resent && i >= Math.floor(maxPolls / 2)) {
+      resent = true
+      await sendEsc(sessionName, { logger: ctx.logger })
+      attempts++
+    }
+  }
+  // 3) deadline 到達でまだ processing。停止確認できず (frontend は旧 tracked-ESC 経路へ degrade)。
+  reply(false, attempts)
+}
+
 async function dispatch(msg, ctx) {
   const t = msg?.type || ""
   try {
@@ -1865,6 +2016,15 @@ async function dispatch(msg, ctx) {
                 old_session_id: viewSid,
                 new_session_id: newSessionId,
               })
+              // ★回転 (= session_id 境界) で、この NAME に紐づく状態トラッカーを忘れる。生成中を表す
+              // 信号 (proc_busy=ReadinessTracker baseline/armed, status gate ラッチ, spinner freeze,
+              // capture cache) は全て NAME キーのため、忘れないと旧ターンの busy が新 session_id の
+              // 表示へ漏れ、フロントで「/clear 後に三点リーダーが固着」する。発生源 (agent) で断つのが
+              // 主、フロントの再点灯レジスタ reset が従の不変条件 (name→id 漏れの根治)。
+              if (viewName) {
+                ctx.readinessTracker?.forget(viewName)
+                invalidateSessionCache(viewName)
+              }
               logger.info(
                 { session: viewName, cwd: viewCwd, old: viewSid, new: newSessionId },
                 "tui session rotated (/clear) → notified browser",
@@ -2060,56 +2220,17 @@ async function dispatch(msg, ctx) {
         })()
         return
       }
+      case "claude.tui.interrupt": {
+        // 確認付き中断 (Phase3)。ロジックは handleTuiInterrupt に切り出し (テスト可能化)。
+        // async だが dispatch をブロックしないよう投げっぱなし (自前で result を返す)。
+        void handleTuiInterrupt(msg, ctx)
+        return
+      }
       case "claude.tui.queue.flush": {
         // 離席中(非アクティブセッション)キュー自動投入 (案A の agent 側受け口)。
-        // ブラウザの常駐ドレイン (useCockpitTuiQueueDrainAll) が、表示していないセッションの
-        // 送信待ちキュー先頭 1 件をここへ送る。tmux send-keys (PTY 非依存) で投入するので、
-        // そのセッションのターミナルを誰も開いていなくても届く。結果を flush.ack で返し、
-        // ブラウザは ack を受けてはじめて localStorage から 1 件 dequeue する (冪等)。
-        const sessionName =
-          typeof msg.session_name === "string" ? msg.session_name : ""
-        const text = typeof msg.text === "string" ? msg.text : ""
-        const flushId = typeof msg.flush_id === "string" ? msg.flush_id : ""
-        const ackFail = (reason) => {
-          if (flushId && sessionName) {
-            ctx.client.send({
-              type: "claude.tui.queue.flush.ack",
-              session_name: sessionName,
-              flush_id: flushId,
-              ok: false,
-              error: reason,
-            })
-          }
-        }
-        if (!sessionName || !flushId || !text) {
-          ackFail("missing session_name / flush_id / text")
-          return
-        }
-        // 冪等化: 同一 flush_id を既に処理済みなら再 paste せず ok を返す (ブラウザは dequeue 済み
-        // を再確認するだけ)。WS 再送・ack 取りこぼしによる二重投入を防ぐ。
-        if (_seenFlushIds.has(flushId)) {
-          ctx.client.send({
-            type: "claude.tui.queue.flush.ack",
-            session_name: sessionName,
-            flush_id: flushId,
-            ok: true,
-          })
-          return
-        }
-        ;(async () => {
-          const result = await pasteToSessionByName(sessionName, text, { logger })
-          if (result.ok) {
-            if (_seenFlushIds.size >= _SEEN_FLUSH_CAP) _seenFlushIds.clear()
-            _seenFlushIds.add(flushId)
-          }
-          ctx.client.send({
-            type: "claude.tui.queue.flush.ack",
-            session_name: sessionName,
-            flush_id: flushId,
-            ok: !!result.ok,
-            error: result.ok ? undefined : result.error,
-          })
-        })()
+        // ロジックは handleQueueFlush に切り出し (テスト可能化 + cross-path 冪等)。async だが
+        // dispatch をブロックしないよう投げっぱなしにする (自前で flush.ack を返す)。
+        void handleQueueFlush(msg, ctx)
         return
       }
       case "claude.tui.rehydratePermissions": {

package/src/tmux.mjs

@@ -964,6 +964,35 @@ export async function resumeWithMessage(name, text, opts = {}) {
   }
 }
 
+/**
+ * 対話 TUI claude へ中断 (Esc) を tmux send-keys で送る。PTY stream に依存しないため、
+ * zombie WS / stream 欠落窓でも確実に届く (生 ESC を tracked pty.data で送る旧経路は stream
+ * 欠落で無言ドロップしていた = 症状4の配達失敗半分)。copy-mode 等に入っているとキーが奪われる
+ * ので先に -X cancel で抜ける。ESC は claude TUI の生成中断キー (素の入力欄では概ね無害)。
+ * @param {string} name tmux セッション名
+ * @param {{logger?:object,tmuxBin?:string}} [opts]
+ * @returns {Promise<{ok:boolean, error?:string}>}
+ */
+export async function sendInterruptKey(name, opts = {}) {
+  if (!isSafeSessionName(name)) return { ok: false, error: "unsafe session name" }
+  const bin = tmuxBin(opts)
+  try {
+    try {
+      await execFileP(bin, ["send-keys", "-t", name, "-X", "cancel"])
+    } catch {
+      /* copy-mode でなければ "not in a mode" エラー = 無視 */
+    }
+    await execFileP(bin, ["send-keys", "-t", name, "Escape"])
+    return { ok: true }
+  } catch (err) {
+    opts.logger?.warn(
+      { session: name, err: err?.message },
+      "sendInterruptKey failed",
+    )
+    return { ok: false, error: err?.message || "send_failed" }
+  }
+}
+
 /**
  * 離席中キュー投入: 送信待ちメッセージ 1 件を、ブラウザがそのセッションのターミナルを
  * マウントしていなくても tmux send-keys で対話 claude TUI へ投入する (案A の agent 側実体)。