@cocorograph/hub-agent 0.8.75 → 0.8.77

package/src/codex-stream-bridge.mjs

@@ -43,8 +43,9 @@
 import { EventEmitter } from "node:events"
 import { randomUUID } from "node:crypto"
 import path from "node:path"
+import { isUnderRemoval } from "./chat-removal-guard.mjs"
 
-import { recordInjectSourcePersistent } from "./inject-source-store.mjs"
+import { lookupInjectSource, recordInjectSourcePersistent } from "./inject-source-store.mjs"
 
 import { AsyncTtlCache } from "./async-ttl-cache.mjs"
 import { CodexAppServerClient } from "./codex-appserver-client.mjs"
@@ -203,6 +204,8 @@ function siblingEnvFor(cwd) {
     ...process.env,
     COCKPIT_SIBLING_TOKEN: getOrCreateSiblingToken(cwd),
     COCKPIT_SIBLING_CWD: cwd,
+    // Spawn-time display capability; rejection by an older daemon does not stop the review.
+    COCKPIT_REVIEW_EVENTS: "1",
     // ask-sibling スキル (cockpit_orch.py) が「自分は claude/codex どちらか」を
     // 環境変数から確実に判定できるようにする (ヒューリスティック不要)。
     COCKPIT_SIBLING_SELF_KIND: "codex",
@@ -798,6 +801,7 @@ export class CodexStreamBridge extends EventEmitter {
     if (this.sessions.has(stream_id)) {
       throw new Error(`stream_id already attached: ${stream_id}`)
     }
+    if (isUnderRemoval(cwd)) throw new Error(`worktree removal in progress: ${cwd}`)
     // 計測 (2026-07-31): attach 1 回 = 1 行の内訳ログ (`codex attach timing`)。
     // frontend の attach_sent→ready_received は 1 つの数字でしか見えず、その中身が
     // 「app-server の cold spawn + initialize」なのか「thread/resume の rollout 読み込み」なのか、
@@ -921,6 +925,11 @@ export class CodexStreamBridge extends EventEmitter {
       }
     }
 
+    // ⚠️ attach の入口でも同じ検査をしているが、そこから **_tryReattach の await** を挟んで
+    // ここへ来る（既存 session の thread 確定待ち）。その間に削除が始まっていることがあるので、
+    // プロセスを起こす直前にもう一度見る。ここを外すと「停止対象を列挙し終えたあとに新しい
+    // プロセスが生まれる」経路が残り、削除された cwd を指す会話ができる。
+    if (isUnderRemoval(cwd)) throw new Error(`worktree removal in progress: ${cwd}`)
     const client = this._clientFactory({
       cwd,
       codexCmd: this.codexCmd,
@@ -1206,13 +1215,13 @@ export class CodexStreamBridge extends EventEmitter {
         : { items: [], start: 0, truncated: false, total: 0 }
       // hub-agent 側の履歴整形コスト (巨大スレッドで効く)。thread_op_ms とは別勘定にする。
       timing.history_flatten_ms = lap()
-      const history = flat.items
+      const history = await this._restoreHistoryInjectSources(flat.items, cwd)
       const historyTruncated = flat.truncated
       // T06541 案1: cold attach で得た高忠実度履歴をセッションに保持する。以降ライブ item
       // 完了ごとに追記し、cwd reattach 時にこれを返す（新プロセス thread/resume を再実行しない）。
       // major#2: byte 予算判定を O(1) にするため概算 byte 合計も持つ (flatten は既に byte 上限適用済み)。
       // app-server reap 後の cold resume は rollout が正本になるため、未 rollout の空 fresh session や
-      // in-memory の sibling source タグは復元できない。空 session の滞留を避けるため fresh 例外は作らない。
+      // 未保存の sibling source タグは復元できない。空 session の滞留を避けるため fresh 例外は作らない。
       session.historyItems = history.slice()
       session.historyBytes = history.reduce((n, it) => n + estimateItemBytes(it), 0)
       session.historyTruncated = historyTruncated
@@ -1800,7 +1809,7 @@ export class CodexStreamBridge extends EventEmitter {
     })
     return {
       threadId: session.threadId,
-      history: page.items,
+      history: await this._restoreHistoryInjectSources(page.items, session.cwd),
       historyStart: page.start,
       historyTotal: page.total,
       historyTruncated: page.truncated,
@@ -1972,6 +1981,14 @@ export class CodexStreamBridge extends EventEmitter {
     return addCodexSessionTitles(sessions, sessionsDir)
   }
 
+  /** 指定 stream の live 会話を停止する。履歴ファイルは削除しない。 */
+  stopSessionForStream({ stream_id, reason }) {
+    const session = this.sessions.get(stream_id)
+    if (!session || session.dead) return 0
+    this._reapSession(session, Date.now(), reason || "explicit-stop")
+    return 1
+  }
+
   async operateSession(options) {
     const invalid = validateCodexSessionOperation(options)
     if (invalid) return { ok: false, error: invalid }
@@ -2398,6 +2415,21 @@ export class CodexStreamBridge extends EventEmitter {
     this._settleInjectDelivery(handle, { delivered: false, reason: "input_failed" })
   }
 
+  /** 履歴にも既存の cwd・本文一致ストアから注入元を戻す。元 item と既存 source は保持する。 */
+  async _restoreHistoryInjectSources(items, cwd) {
+    return Promise.all(items.map(async (item) => {
+      if (item?.type !== "userMessage" || item.source != null) return item
+      const content = item.content
+      const text = typeof content === "string"
+        ? content
+        : Array.isArray(content)
+          ? content.map((block) => typeof block?.text === "string" ? block.text : "").join("")
+          : ""
+      const source = await lookupInjectSource(cwd, text)
+      return source ? { ...item, source } : item
+    }))
+  }
+
   /** userMessage の echo (item/completed) が控えた注入本文と一致すれば、params を **クローン**して
    *  item.source を付けたものを返す (元の app-server オブジェクトは変更しない)。一致しなければ null。
    *  frontend の userMessageItemToEvent がこの item.source を読み専用カードへ分岐する。 */
@@ -3306,8 +3338,7 @@ export class CodexStreamBridge extends EventEmitter {
     // ライブ emit だけタグしていたため、cwd reattach で historyItems が返ると連携カードが通常の青い
     // 自分バブルへ戻っていた。同一プロセス内 reattach で表示が維持されるよう、以降 effectiveParams
     // (タグ済み or 素) を history と emit の両方で使う。一致しなければ null → 素の params のまま。
-    // ※ hub-agent 再起動 (cold resume) では historyItems が揮発するため青バブルへ戻る (claude の
-    //    jsonl at-rest と同じ許容範囲。source 永続マップは TTL/ID 再利用の複雑さに見合わないため不採用)。
+    // cold resume と過去履歴 page は既存の永続ストアから source を復元する。
     const eventThreadId = msg?.params?.threadId
     if (
       typeof eventThreadId === "string" &&
@@ -4136,6 +4167,22 @@ export class CodexStreamBridge extends EventEmitter {
     return out
   }
 
+  /** Display events never start a turn and only target an exact thread in this cwd. */
+  emitToStream(cwd, sessionId, event) {
+    if (!cwd || !sessionId) return { status: "not_found" }
+    const matches = [...this._liveSessions()].filter((s) => s.cwd === cwd && s.threadId === sessionId)
+    if (matches.length !== 1) return { status: matches.length ? "ambiguous" : "not_found" }
+    const session = matches[0]
+    this.emit("event", {
+      stream_id: session.stream_id,
+      producer_seq: session.seq,
+      thread_id: session.threadId,
+      cwd: session.cwd,
+      notification: { method: "cockpit/review_event", params: event },
+    })
+    return { status: "delivered", session: this._describe(session) }
+  }
+
   /**
    * 指定 cwd の「権威セッション」を返す（T06407 増分2, axis2 の単一正本）。同一 cwd に複数
    * ライブ codex セッションがあるとき、**最終活性が最新のもの**（＝いまユーザーが使っている
@@ -4698,7 +4745,12 @@ export class CodexStreamBridge extends EventEmitter {
       producer_seq: session.seq,
       thread_id: session.threadId,
       code: 0,
-      reason: reason === "session-deleted" ? "session-deleted" : "idle-reaped",
+      // ⚠️ ここは「明示的に止めた理由」の許可リスト。載っていない理由はすべて "idle-reaped"
+      // へ丸められ、利用者には「放置されて片付いた」と通知される。明示停止を足したら必ず
+      // ここへも足すこと（足し忘れても停止自体は動くので、通知だけが静かに嘘になる）。
+      reason: ["session-deleted", "chat-stop", "chat-stop-all", "worktree-remove"].includes(reason)
+        ? reason
+        : "idle-reaped",
     })
   }
 

package/src/dispatch-scheduler.mjs

@@ -83,8 +83,14 @@ function globalMutationMode(type) {
   return null
 }
 
-function attachLaneKey(mode, msg) {
-  const cwd = typeof msg?.cwd === "string" ? msg.cwd.trim() : ""
+// ⚠️ cwd を省略した attach は、ハンドラ側で config.claude_cwd に補われ、さらに claude では
+// bridge が HOME へ補完して「その cwd の会話」になる (main.mjs の claude.attach / codex.attach
+// と claude-stream-bridge の attach)。レーン分類がその補完を知らないと、同じ会話への attach が
+// stream レーンへ、個別停止 (cwd を持つ) が cwd レーンへ散り、直列化されないまま同時に走る。
+// scheduler へ設定も HOME も読ませず、補完後の既定 cwd を呼び出し側から mode 別に渡してもらう
+// (補完の規則は CLI 種別で違うため、mode を知らない scheduler 側では決められない)。
+function attachLaneKey(mode, msg, defaultCwd = "") {
+  const cwd = (typeof msg?.cwd === "string" ? msg.cwd.trim() : "") || defaultCwd
   if (cwd) return `${mode}:cwd:${cwd}`
   const resumeId =
     mode === "claude"
@@ -100,14 +106,18 @@ function attachLaneKey(mode, msg) {
  *   run: (msg: object) => Promise<void>|void,
  *   logger?: {debug?: Function, warn?: Function, error?: Function},
  *   shouldBypass?: (type: string|undefined) => boolean,
+ *   defaultCwd?: (mode: "claude"|"codex") => string,
  * }} options
  */
 export class DispatchScheduler {
-  constructor({ run, logger, shouldBypass }) {
+  constructor({ run, logger, shouldBypass, defaultCwd }) {
     if (typeof run !== "function") throw new TypeError("DispatchScheduler run is required")
     this.run = run
     this.logger = logger
     this.shouldBypass = typeof shouldBypass === "function" ? shouldBypass : () => false
+    // cwd 省略時にハンドラ／bridge が使う既定 cwd を CLI 種別ごとに返す関数。展開済みの値を
+    // 返すこと (scheduler は `~` 展開も HOME 参照もしない)。
+    this.defaultCwd = typeof defaultCwd === "function" ? defaultCwd : () => ""
     this.generation = 0
     this.sequence = 0
     this.laneTails = new Map()
@@ -154,6 +164,21 @@ export class DispatchScheduler {
     }
   }
 
+  /**
+   * cwd 省略時にハンドラ／bridge が使う既定 cwd を、CLI 種別ごとに引く。
+   * ⚠️ レーン分類は全メッセージが通る経路なので、設定の読み取りが失敗しても dispatch を
+   * 止めない。取れなければ省略時の従来分類 (stream / control) へ戻す。
+   * @param {"claude"|"codex"} mode
+   */
+  _resolveDefaultCwd(mode) {
+    try {
+      const cwd = this.defaultCwd(mode)
+      return typeof cwd === "string" ? cwd.trim() : ""
+    } catch {
+      return ""
+    }
+  }
+
   _laneFor(msg) {
     const type = msg?.type
     const mutationMode = globalMutationMode(type)
@@ -163,20 +188,35 @@ export class DispatchScheduler {
     // frontend の 10 秒打ち切りで捨てられていたため分離する (2026-09-04 本番計測)。
     if (type === "usage.request") return "usage"
 
+    if (type === "chat.session.stop") {
+      if (msg.cli_kind === "claude") {
+        const cwd = typeof msg.cwd === "string" ? msg.cwd.trim() : ""
+        return cwd ? `claude:cwd:${cwd}` : "control"
+      }
+      if (msg.cli_kind === "codex" && msg.stream_id) {
+        return this.streamLaneKeys.get("codex:" + msg.stream_id) || `codex:stream:${msg.stream_id}`
+      }
+      return "control"
+    }
+    // chat.stop_all は全レーンに跨るため control に置く。その時点で live な会話だけを
+    // 止める契約であり、接続処理中のまだ live でない会話は対象外。sessions.request も control。
     const mode = chatMode(type)
     if (!mode) return "control"
 
     const streamId = typeof msg?.stream_id === "string" ? msg.stream_id : ""
     const streamKey = streamId ? `${mode}:${streamId}` : ""
+    const fallbackCwd = this._resolveDefaultCwd(mode)
     if (type === `${mode}.attach`) {
-      const laneKey = attachLaneKey(mode, msg)
+      const laneKey = attachLaneKey(mode, msg, fallbackCwd)
       this._rememberStreamLane(streamId, laneKey)
       return laneKey
     }
     if (streamKey && this.streamLaneKeys.has(streamKey)) {
       return this.streamLaneKeys.get(streamKey)
     }
-    const cwd = typeof msg?.cwd === "string" ? msg.cwd.trim() : ""
+    // attach と同じ補完を使う。片方だけ既定 cwd を見ると、同じ会話宛の upload / history 等が
+    // attach とは別レーンへ散り、attach で直したはずの取り違えが型ごと再発する。
+    const cwd = (typeof msg?.cwd === "string" ? msg.cwd.trim() : "") || fallbackCwd
     if (cwd) return `${mode}:cwd:${cwd}`
     if (streamId) return `${mode}:stream:${streamId}`
     // codex.profile.list 等の stream 非依存 read は main 側の read-only bypass