@cocorograph/hub-agent 0.7.3 → 0.7.5

package/package.json

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

package/src/main.mjs

@@ -68,11 +68,13 @@ import {
   listSessions as listTmuxSessions,
   listWorktreeNameHistory,
   listWorktreeStubs,
+  isPaneRunningClaude,
   rebindClaudeSession,
   shouldSkipRebindRespawn,
   recoverTuiInput,
   removeWorktree as removeWorktreeDir,
   resumeWithMessage,
+  pasteToSessionByName,
   setSessionMouse,
   setTmuxGlobalEnv,
   setTuiEffort,
@@ -1432,6 +1434,13 @@ export function handleUntrackedPtyData(msg, ctx) {
   }
 }
 
+/**
+ * 離席中キュー投入 (claude.tui.queue.flush) の冪等化用に、処理済み flush_id を記憶する。
+ * WS 再送で同じ flush が二重に届いても二重 paste しないため。上限を超えたら全消去 (best-effort)。
+ */
+const _seenFlushIds = new Set()
+const _SEEN_FLUSH_CAP = 2000
+
 async function dispatch(msg, ctx) {
   const t = msg?.type || ""
   try {
@@ -1973,6 +1982,58 @@ async function dispatch(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,
+          })
+        })()
+        return
+      }
       case "claude.tui.rehydratePermissions": {
         // セッション切替でビューが再マウントすると、その時点の承認/質問カードは React
         // state ごと消える (agent は claude.permission.request を 1 回しか push しない)。
@@ -2080,13 +2141,23 @@ async function dispatch(msg, ctx) {
           //    resume は targetId をキー、fresh は session_id 不在のため bind の request_id を
           //    キーにする (同一マウントの再送は同 request_id で skip、新規要求は再マウントで
           //    新 request_id になり 1 回だけ起動する)。
+          // 実プロセス由来の「claude 実行中か」。respawn-pane -k による生成中 claude の誤 kill
+          // (症状D 再発 D2: armed 140s cap / capture gap で generating=false に倒れ、--resume bind
+          // でも実行中 claude を kill していた)を、capture/armed に依存しない pane_current_command で防ぐ。
+          const paneRunningClaude = await isPaneRunningClaude(sessionName, { logger })
           let rebind = { ok: false, skipped: false }
           const bindKey = fresh ? `fresh:${request_id || ""}` : targetId
           if (bindKey && sessionName && (fresh || targetId)) {
             if (ctx.tuiReboundSessions.get(sessionName) === bindKey) {
               rebind = { ok: true, skipped: true }
             } else if (
-              shouldSkipRebindRespawn({ generating, fresh, targetId, newestId })
+              shouldSkipRebindRespawn({
+                generating,
+                fresh,
+                targetId,
+                newestId,
+                paneRunningClaude,
+              })
             ) {
               // 生成中 claude を再接続/remount の bind で kill しない (謎停止対策)。respawn を抑止して
               // 既存の生成中 claude を温存する。bindKey は記録して以降の remount を冪等化する

package/src/tmux.mjs

@@ -796,6 +796,17 @@ export function isSafeSessionId(sessionId) {
   return typeof sessionId === "string" && /^[A-Za-z0-9_-]{8,128}$/.test(sessionId)
 }
 
+/** tmux セッション名として安全な形だけ許可する。`-t <name>` の引数に流すため、先頭ハイフン
+ *  (フラグ誤認) や予期せぬ文字を弾く。agent が作るセッション名は英数・`_`・`-`・`.` のみ。 */
+export function isSafeSessionName(name) {
+  return typeof name === "string" && /^[A-Za-z0-9_][A-Za-z0-9_.-]{0,127}$/.test(name)
+}
+
+/** bracketed paste マーカーと貼り付け確定の待ち時間 (pasteToSessionByName 用)。 */
+const PASTE_START = "\x1b[200~"
+const PASTE_END = "\x1b[201~"
+const PASTE_CONFIRM_DELAY_MS = 120
+
 /** agent 設定 (model / permissionMode) を claude のフラグ文字列に変換する ("" or "--model X ...")。 */
 export function composeClaudeFlags(opts = {}) {
   const model = (opts.model || "").trim()
@@ -953,6 +964,55 @@ export async function resumeWithMessage(name, text, opts = {}) {
   }
 }
 
+/**
+ * 離席中キュー投入: 送信待ちメッセージ 1 件を、ブラウザがそのセッションのターミナルを
+ * マウントしていなくても tmux send-keys で対話 claude TUI へ投入する (案A の agent 側実体)。
+ *
+ * resumeWithMessage は改行を空白へ畳む単一行用 (誤確定防止) でチャット本文の複数行を壊すため、
+ * こちらは **bracketed paste** (ESC[200~ … ESC[201~) で本文を「貼り付け」として送り、改行を
+ * 保ったまま入力欄へ入れてから、確定の Enter (CR) を別途送る。これは frontend の通常送信
+ * (`\x1b[200~${text}\x1b[201~` → CR) と同じ手順を agent 側で再現したもの。PTY 非依存
+ * (tmux send-keys) なので、当該セッションのターミナルを誰も開いていなくても届く。
+ *
+ * @param {string} name tmux セッション名
+ * @param {string} text 投入する本文 (複数行可)
+ * @param {{logger?:object,tmuxBin?:string}} [opts]
+ * @returns {Promise<{ok:boolean, error?:string}>}
+ */
+export async function pasteToSessionByName(name, text, opts = {}) {
+  const bin = tmuxBin(opts)
+  const body = String(text ?? "")
+  if (!body) return { ok: false, error: "empty paste text" }
+  if (!isSafeSessionName(name)) return { ok: false, error: "unsafe session name" }
+  try {
+    // copy-mode 等に入っているとキーが奪われるので先に抜ける (cyclePermission と同じ防御)。
+    try {
+      await execFileP(bin, ["send-keys", "-t", name, "-X", "cancel"])
+    } catch {
+      /* copy-mode でなければ "not in a mode" エラー = 無視 */
+    }
+    // bracketed paste の開始/終了マーカーと本文を literal (-l) で送る。改行を含む本文も
+    // 「貼り付け」として扱われ、Claude TUI は改行で確定しない (CR を別送するまで未確定)。
+    await execFileP(bin, ["send-keys", "-t", name, "-l", PASTE_START])
+    await execFileP(bin, ["send-keys", "-t", name, "-l", body])
+    await execFileP(bin, ["send-keys", "-t", name, "-l", PASTE_END])
+    // 貼り付けが入力欄へ反映される猶予を置いてから確定 (CR を本文より先着させない)。
+    await _delay(PASTE_CONFIRM_DELAY_MS)
+    await execFileP(bin, ["send-keys", "-t", name, "Enter"])
+    opts.logger?.info(
+      { session: name, len: body.length },
+      "tui queue flush: pasted queued message (headless)",
+    )
+    return { ok: true }
+  } catch (err) {
+    opts.logger?.warn(
+      { session: name, err: err?.message },
+      "pasteToSessionByName failed",
+    )
+    return { ok: false, error: err?.message || String(err) }
+  }
+}
+
 /**
  * 対話 claude TUI に shift+tab (BACKTAB) を 1 回送って権限モードを循環させる。
  *
@@ -1237,8 +1297,49 @@ export async function recoverTuiInput(name, expectText, opts = {}) {
  * @param {{generating?: boolean, fresh?: boolean, targetId?: string|null, newestId?: string|null}} a
  * @returns {boolean}
  */
-export function shouldSkipRebindRespawn({ generating, fresh, targetId, newestId } = {}) {
+/**
+ * ペインの前景プロセスが claude を実行中か（= respawn-pane -k で kill すると in-flight 応答を失う）
+ * を pane_current_command から判定する。capture-pane スクレイプや armed(140s cap) の脆い推論に
+ * 依存しない実プロセス由来の判定で、hub-agent 再起動・armed cap・capture gap を跨いで有効。
+ * シェル（fish/bash/zsh/sh/dash/tmux/login）なら false。判定不能時は安全側 false（= respawn 許可）。
+ */
+export async function isPaneRunningClaude(name, opts = {}) {
+  if (!name) return false
+  try {
+    const { stdout } = await execFileP(tmuxBin(opts), [
+      "display-message",
+      "-p",
+      "-t",
+      `${name}:`,
+      "-F",
+      "#{pane_current_command}",
+    ])
+    const cmd = (stdout || "").trim().toLowerCase()
+    if (!cmd) return false
+    const SHELLS = new Set(["fish", "bash", "zsh", "sh", "dash", "tmux", "login"])
+    // TUI チャットのペインで走るのは claude（claude / claude.exe / node 等）かシェルのいずれか。
+    // シェルでない前景プロセス = claude 実行中とみなす（命名差 claude.exe/node に強い）。
+    return !SHELLS.has(cmd)
+  } catch {
+    return false
+  }
+}
+
+export function shouldSkipRebindRespawn({
+  generating,
+  fresh,
+  targetId,
+  newestId,
+  paneRunningClaude,
+} = {}) {
   if (fresh) return false
+  // 実行中 claude を kill しない最優先ガード（症状D / 再発 D2 の根治）: ペインが現に claude を
+  // 実行中で、最新セッション（= 今動いている会話）への再 bind（remount/reconnect）なら、respawn
+  // (respawn-pane -k) は破壊的なだけで不要。generating の脆い検出（isArmed 140s cap / capture gap）
+  // で false に倒れても、実プロセス由来の paneRunningClaude が守る。targetId 未指定（= newest 解決前）
+  // も「最新への再 bind」とみなす。明示的な別会話への切替（targetId !== newestId）と +新規（fresh）は
+  // 従来どおり respawn する。
+  if (paneRunningClaude && (!targetId || targetId === newestId)) return true
   if (!generating) return false
   return !!targetId && targetId === newestId
 }