@cocorograph/hub-agent 0.8.97 → 0.8.98

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cocorograph/hub-agent",
-  "version": "0.8.97",
+  "version": "0.8.98",
   "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)
@@ -2388,9 +2394,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 +2471,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)")
@@ -2491,10 +2522,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({

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
+    },
+  }
+}