@cocorograph/hub-agent 0.8.97 → 0.8.99

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cocorograph/hub-agent",
-  "version": "0.8.97",
+  "version": "0.8.99",
   "description": "Hub Hosted Cockpit のローカル常駐 agent。Hub と outbound WSS で接続し、ローカルの Claude / Codex チャットと tmux セッションを中継する。",
   "type": "module",
   "license": "UNLICENSED",
@@ -18,7 +18,7 @@
   },
   "scripts": {
     "start": "node bin/hub-agent.mjs start",
-    "test": "node --test --test-concurrency=1 test/*.test.mjs",
+    "test": "node --import ./test/setup-env.mjs --test --test-concurrency=1 test/*.test.mjs",
     "check:publish-on-main": "node scripts/check-publish-on-main.mjs",
     "prepublishOnly": "node scripts/check-publish-on-main.mjs && npm test"
   },

package/src/main.mjs

@@ -50,6 +50,7 @@ import { WsClient } from "./ws-client.mjs"
 import { DispatchScheduler } from "./dispatch-scheduler.mjs"
 import { noteRestoreStop, restoreSessions, sendBridgeMessage } from "./session-restore.mjs"
 import { AsyncTtlCache } from "./async-ttl-cache.mjs"
+import { makePersistentSeenInputIdRegistry } from "./seen-input-id-store.mjs"
 import { ClaudeStreamBridge } from "./claude-stream-bridge.mjs"
 import { installClaudeDisplayMask } from "./claude-display-mask.mjs"
 import { CodexMcpControl } from "./codex-mcp-control.mjs"
@@ -1179,6 +1180,11 @@ export async function startDaemon({ version, claudeSdk } = {}) {
     restartCoordinator?.stop?.()
     claudeBridge?.shutdown?.()
     codexBridge?.shutdown?.()
+    // 受理済み input_id の保存を待ってから exit する（再起動後の再送で二重実行しないため）。
+    const persistence = await waitForClaudeInputPersistence()
+    if (persistence === "timeout") {
+      logger.warn({ signal }, "claude input_id の保存待ちが上限に達したため終了する")
+    }
     client.stop()
     // 0.6.2 fix: watchdog を解放して event loop を抜けられるようにする
     clearInterval(keepaliveTimer)
@@ -1405,7 +1411,7 @@ export function updateSawUserTurn(name, event) {
  * @param {number} at epoch ms
  * @returns {{ sortAffecting: boolean, sortAffectingAt: number|undefined, stale: boolean }}
  */
-export function ingestSessionEventForSort(name, event, at, { fresh = false } = {}) {
+export function ingestSessionEventForSort(name, event, at, { fresh = false, sortable = true } = {}) {
   const prevAt = lastEventAtByName.get(name) ?? 0
   // stale (この session で処理済みの at より古い) read は状態を汚さない。arm や status は
   // 呼び出し側で従来通り処理してよいが、sort へは一切効かせない。
@@ -1431,7 +1437,9 @@ export function ingestSessionEventForSort(name, event, at, { fresh = false } = {
   }
   if (typeof at === "number") lastEventAtByName.set(name, Math.max(prevAt, at))
   updateSawUserTurn(name, event)
-  const sortAffecting = computeSortAffecting(event, sawUserTurnByName.get(name) === true)
+  // `sortable=false`: 並び順の供給源ではない経路 (bundle hook のイベントファイル)。状態・readiness には
+  // 使うが sort 時刻は動かさない (publishChatPromptSubmit / publishHookSessionEvent 参照)。
+  const sortAffecting = sortable && computeSortAffecting(event, sawUserTurnByName.get(name) === true)
   // sort_affecting=true で前進したときだけ sort 時刻を更新し、sidecar に永続化する (major#1)。
   // event ファイルは揮発 (最新イベントで上書き) のため、再起動後に「最後の実ターン時刻」を
   // 復元できるよう別ファイルに残す。冪等な再取込 (at 同値) では書かない。
@@ -1451,6 +1459,26 @@ export function getSortAffectingAt(name) {
   return lastSortAffectingAtByName.get(name)
 }
 
+/**
+ * `chat.sessions` の各 live に、その行の「最後にユーザーが送信した時刻」(`sort_affecting_at`) を載せる。
+ *
+ * サイドバーの並び順の正本は送信時刻だけ (2026-07-29 決定)。frontend は `session.event` の push で
+ * これを受け取るが、push はページを開いている間しか届かないため、再読み込み後に復元する手段が
+ * 一覧応答しかない。値は sidecar で永続化済みのもの (seedSortAtFromSidecars) をそのまま返す。
+ * 送信履歴の無い行はキーごと付けない (0 を入れると「送信済み」と区別できない)。
+ */
+export function withSortAffectingAt(snapshot, getSortAt = getSortAffectingAt) {
+  const live = Array.isArray(snapshot?.live) ? snapshot.live : []
+  return {
+    ...snapshot,
+    live: live.map((item) => {
+      const name = typeof item?.session_name === "string" ? item.session_name : ""
+      const at = name ? getSortAt(name) : undefined
+      return typeof at === "number" && at > 0 ? { ...item, sort_affecting_at: at } : item
+    }),
+  }
+}
+
 // sort 時刻 sidecar の接尾辞。`.json` にしない (readAllSessionEvents / pushFile が
 // `.endsWith(".json")` で event ファイルと誤認しないように)。
 const SORT_AT_SUFFIX = ".sortat"
@@ -1633,6 +1661,24 @@ async function readAllSessionEvents() {
   return out
 }
 
+/**
+ * bundle hook が書いたイベントファイル 1 件 (`{ event, at }`) を取り込んで push する。
+ *
+ * ⚠️ hook のイベントは並び順に使わない (sortable: false)。並び順の正本は「ユーザーが送信した
+ * 時刻」で、その供給源は agent の入力受付 (publishChatPromptSubmit) だけにする。hook の
+ * prompt_submit は (1) 実行を始めた時刻なので、キュー待ちの送信では受付より後ろへずれて
+ * 送信順を逆転させ、(2) 行名を cwd の basename から作るので、Cockpit 外で同じフォルダの
+ * claude (headless `claude -p` 等) が動いただけでも行を浮かせる。状態・readiness には従来どおり使う。
+ */
+export async function publishHookSessionEvent(ctx, sessionName, data) {
+  if (!data || typeof data.event !== "string") return undefined
+  const at = typeof data.at === "number" ? data.at : Date.now()
+  // チャットの行は state loop が返答付きの stop を送る。ここでも送ると Hub が完了通知を
+  // 2 通 (定型文だけ + 返答付き) 出すため、送信だけ省く (readiness は取り込む)。
+  const send = !(data.event === "stop" && chatStopNotifyingSessionNames.has(sessionName))
+  return publishSessionEvent(ctx, sessionName, data.event, at, { send, sortable: false })
+}
+
 /**
  * `/tmp/cockpit_session_events/<tmux_session>.json` を fs.watch して
  * UserPromptSubmit / Stop の event を `session.event` 型で Hub に push する。
@@ -1662,13 +1708,7 @@ async function startSessionEventWatcher({ client, logger, readinessTracker }) {
     const full = path.join(SESSION_EVENTS_DIR, filename)
     try {
       const text = await readFile(full, "utf-8")
-      const data = JSON.parse(text)
-      if (!data || typeof data.event !== "string") return
-      const at = typeof data.at === "number" ? data.at : Date.now()
-      // チャットの行は state loop が返答付きの stop を送る。ここでも送ると Hub が完了通知を
-      // 2 通 (定型文だけ + 返答付き) 出すため、送信だけ省く (並び順・readiness は取り込む)。
-      const send = !(data.event === "stop" && chatStopNotifyingSessionNames.has(sessionName))
-      await publishSessionEvent({ client, readinessTracker, logger }, sessionName, data.event, at, { send })
+      await publishHookSessionEvent({ client, readinessTracker, logger }, sessionName, JSON.parse(text))
     } catch {
       // 一時欠如 / parse 失敗等は無視 (次の変更で読み直す)
     }
@@ -1716,20 +1756,39 @@ async function startSessionEventWatcher({ client, logger, readinessTracker }) {
  */
 export const chatStopNotifyingSessionNames = new Set()
 
+/**
+ * チャット入力の受付を並び順へ反映する (`prompt_submit`, 受付時刻, fresh)。
+ *
+ * 並び順の正本は「ユーザーが送信した時刻」だけ (2026-07-29 決定)。**これが唯一の供給源**で、
+ * bundle hook のイベントファイルは並び順に使わない (publishHookSessionEvent の sortable: false)。
+ * この直接 publish は tmux 撤去 (189c555) で、tmux 行を走査する旧経路ごと消えていた。行名は
+ * cwd から直接解決する (chatRowSessionName が行名の正本)。publish の失敗で入力の配送を止めない。
+ */
+export function publishChatPromptSubmit(ctx, cwd, cliKind, at) {
+  const name = chatRowSessionName(cwd, cliKind)
+  if (!name) return
+  const publish = ctx._publishSessionEvent ?? publishSessionEvent
+  Promise.resolve()
+    .then(() => publish({ client: ctx.client, logger: ctx.logger }, name, "prompt_submit", at, { fresh: true }))
+    .catch((err) => {
+      ctx.logger?.warn?.({ err: err?.message, cwd, cli_kind: cliKind, session_name: name }, "chat prompt_submit publish failed")
+    })
+}
+
 /** session-event をプロセス内で取り込み、並び順を更新して WS push する。 */
 export async function publishSessionEvent(
   { client, readinessTracker, logger },
   sessionName,
   event,
   at,
-  { fresh = false, arm = !fresh, text = "", send = true } = {},
+  { fresh = false, arm = !fresh, text = "", send = true, sortable = true } = {},
 ) {
   // readiness (proc 内省の ready 判定) を動かすのはファイル監視経路だけ。直接 publish (fresh) は
   // 「人間が送った」事実を並び順へ流すだけで、ターンの開始を意味しない (bridge が拒否すれば
   // ターンは始まらないし、走査が逆順に完了した後着の古い方は既に終わったターン)。ここで arm すると
   // 次の stop が来ない限り busy が残る。ターン境界の arm は従来どおり state loop / bundle hook が
   // 書く event ファイル経由で行う。
-  const { sortAffecting } = ingestSessionEventForSort(sessionName, event, at, { fresh })
+  const { sortAffecting } = ingestSessionEventForSort(sessionName, event, at, { fresh, sortable })
   if (arm) {
     if (event === "prompt_submit") {
       readinessTracker?.arm(sessionName, { newTurn: true })
@@ -2388,9 +2447,34 @@ function makeSeenInputIdRegistry(max = SEEN_INPUT_IDS_MAX) {
     },
   }
 }
-const seenClaudeChatInputIds = makeSeenInputIdRegistry()
+let seenClaudeChatInputIds = makePersistentSeenInputIdRegistry(SEEN_INPUT_IDS_MAX)
+// テストでは処理完了後に呼び、プロセス再起動時のメモリ破棄を模擬する。
+export function resetSeenClaudeChatInputIdsForTest() {
+  seenClaudeChatInputIds = makePersistentSeenInputIdRegistry(SEEN_INPUT_IDS_MAX)
+}
 const seenCodexChatInputIds = makeSeenInputIdRegistry()
 
+/**
+ * 通常終了（SIGTERM・安全な再起動）の直前に、処理中の claude.input と受理済み ID の保存を待つ。
+ * 保存前に exit すると、再起動後の同じ input_id の再送が Claude へ再投入されるため。
+ * ディスクや bridge が止まっていても終了を無期限に延ばさないよう timeoutMs で打ち切る
+ * （打ち切り・強制終了では保証しない）。reject しない。
+ */
+export async function waitForClaudeInputPersistence(timeoutMs = 2_000) {
+  let timer
+  const timeout = new Promise((resolve) => {
+    timer = setTimeout(() => resolve("timeout"), timeoutMs)
+  })
+  const settled = Promise.allSettled([...claudeInflightByInputId.values()])
+    .then(() => seenClaudeChatInputIds.flush())
+    .then(() => "flushed")
+  try {
+    return await Promise.race([settled, timeout])
+  } finally {
+    clearTimeout(timer)
+  }
+}
+
 // round4 major#1 (T06542): codex.input の **in-flight** dedup 用レジストリ (input_id -> Promise)。
 // `seen` は settle 後にしか登録されないため、frontend の ack watchdog (5s) が bridge の barrier/RPC
 // (turn/start→turn/started は数秒〜) より短く、同一 input_id が settle 前に再送されると bridge.input を
@@ -2440,7 +2524,7 @@ export function handleClaudeChatInput(msg, ctx) {
     }
     return undefined
   }
-  if (input_id && seenClaudeChatInputIds.has(input_id)) {
+  if (input_id && seenClaudeChatInputIds.has(input_id, ctx.logger) && !claudeInflightByInputId.has(input_id)) {
     // 再送 (ack 喪失想定)。同じターンを 2 回実行しないよう sendMessage を
     // 呼び直さず ack のみ返す。
     ctx.logger?.info?.({ stream_id, input_id }, "claude.input dedup (ack only)")
@@ -2479,6 +2563,7 @@ export function handleClaudeChatInput(msg, ctx) {
   const cwdSnapshot = ctx.claudeBridge.getCwd?.(stream_id) ?? null
   if (cwdSnapshot) {
     recordPromptAccepted(cwdSnapshot, "claude", acceptedAt, ctx.claudeBridge.sessions?.get(stream_id), ctx.claudeBridge)
+    publishChatPromptSubmit(ctx, cwdSnapshot, "claude", acceptedAt)
   }
   const p = ctx.claudeBridge
     .input({
@@ -2491,10 +2576,10 @@ export function handleClaudeChatInput(msg, ctx) {
       // 確認前キャンセルを可能にする)。
       input_id,
     })
-    .then((result) => {
+    .then(async (result) => {
       if (!input_id) return
       if (result?.ok) {
-        seenClaudeChatInputIds.remember(input_id)
+        await seenClaudeChatInputIds.remember(input_id, ctx.logger)
         ctx.client.send({ type: "claude.input.ack", stream_id, input_id })
       } else {
         ctx.client.send({
@@ -2694,6 +2779,7 @@ export function handleCodexChatInput(msg, ctx) {
   const cwdSnapshot = ctx.codexBridge.getCwd?.(stream_id) ?? null
   if (cwdSnapshot) {
     recordPromptAccepted(cwdSnapshot, "codex", acceptedAt, ctx.codexBridge.sessions?.get(stream_id), ctx.codexBridge)
+    publishChatPromptSubmit(ctx, cwdSnapshot, "codex", acceptedAt)
   }
   const p = ctx.codexBridge
     .input({
@@ -4552,7 +4638,7 @@ export async function dispatch(msg, ctx) {
             bridges: { claudeBridge: ctx.claudeBridge, codexBridge: ctx.codexBridge },
             logger: ctx.logger,
           })
-          ctx.client.send({ type: "chat.sessions", request_id: msg.request_id, ...refreshChatSessionsSignals(snapshot, ctx.logger) })
+          ctx.client.send({ type: "chat.sessions", request_id: msg.request_id, ...withSortAffectingAt(refreshChatSessionsSignals(snapshot, ctx.logger)) })
         } catch (err) {
           ctx.client.send({
             type: "chat.sessions", request_id: msg.request_id, targets: [], live: [],

package/src/seen-input-id-store.mjs

@@ -0,0 +1,66 @@
+import { readFileSync } from "node:fs"
+import { mkdir, writeFile, rename, unlink } from "node:fs/promises"
+import { randomUUID } from "node:crypto"
+import path from "node:path"
+import { paths } from "./config.mjs"
+
+/** Claude の受理済み ID。初回だけ同期読み込みし、受付の同期的な順序を保つ。 */
+export function makePersistentSeenInputIdRegistry(max = 1_000) {
+  const seen = new Set()
+  let storePath
+  let writeChain = Promise.resolve()
+
+  function trim() {
+    while (seen.size > max) seen.delete(seen.values().next().value)
+  }
+
+  function ensureLoaded(logger) {
+    if (storePath !== undefined) return
+    storePath = process.env.HUB_CLAUDE_SEEN_INPUT_IDS_PATH ||
+      path.join(paths.configDir, "claude-seen-input-ids.json")
+    try {
+      const ids = JSON.parse(readFileSync(storePath, "utf8"))
+      if (!Array.isArray(ids)) throw new Error("expected an input_id array")
+      if (ids.some((id) => typeof id !== "string")) {
+        logger?.warn?.("claude input_id store contains invalid entries")
+      }
+      for (const id of ids) if (typeof id === "string") seen.add(id)
+      trim()
+    } catch (err) {
+      if (err.code !== "ENOENT") {
+        logger?.warn?.({ err: err.message }, "claude input_id store load failed")
+      }
+    }
+  }
+
+  return {
+    has(id, logger) {
+      ensureLoaded(logger)
+      return seen.has(id)
+    },
+    remember(id, logger) {
+      ensureLoaded(logger)
+      seen.add(id)
+      trim()
+      const snapshot = JSON.stringify([...seen])
+      // rename まで直列化し、古い snapshot が新しい記録を消さないようにする。
+      writeChain = writeChain.then(async () => {
+        const tmp = `${storePath}.${process.pid}.${randomUUID()}.tmp`
+        try {
+          await mkdir(path.dirname(storePath), { recursive: true, mode: 0o700 })
+          await writeFile(tmp, snapshot, { mode: 0o600, flag: "wx" })
+          await rename(tmp, storePath)
+        } catch (err) {
+          logger?.warn?.({ err: err.message }, "claude input_id store write failed")
+        } finally {
+          await unlink(tmp).catch(() => {})
+        }
+      })
+      return writeChain
+    },
+    /** 予約済みの保存がすべて終わるまで待つ（失敗は内部で warn 済みなので reject しない）。 */
+    flush() {
+      return writeChain
+    },
+  }
+}