@cocorograph/hub-agent 0.6.15 → 0.6.17

package/package.json

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

package/src/claude-history-watch.mjs

@@ -15,7 +15,8 @@
 import { watch as fsWatch } from "node:fs"
 import { open, stat } from "node:fs/promises"
 
-const DISPLAY_TYPES = new Set(["user", "assistant", "system", "result"])
+// 履歴 hydrate と live watch で jsonl→SDK message 整形を共通化 (フィールド取りこぼし防止)。
+import { DISPLAY_TYPES, normalizeHistoryEvent } from "./claude-history.mjs"
 const POLL_INTERVAL_MS = 1500
 
 /**
@@ -89,7 +90,7 @@ export function watchSessionFile({ filePath, onEvent, fromEnd = true, logger })
             continue
           }
           if (!obj || !DISPLAY_TYPES.has(obj.type)) continue
-          const event = normalizeEvent(obj)
+          const event = normalizeHistoryEvent(obj)
           try {
             onEvent(event)
           } catch (err) {
@@ -151,21 +152,3 @@ export function watchSessionFile({ filePath, onEvent, fromEnd = true, logger })
   }
 }
 
-/** jsonl の生 object を SDK message 風の表示用イベントに正規化 (history.mjs と揃える)。 */
-function normalizeEvent(obj) {
-  const event = { type: obj.type }
-  if (obj.message !== undefined) event.message = obj.message
-  if (obj.subtype !== undefined) event.subtype = obj.subtype
-  if (obj.uuid !== undefined) event.uuid = obj.uuid
-  if (obj.session_id !== undefined) event.session_id = obj.session_id
-  else if (obj.sessionId !== undefined) event.session_id = obj.sessionId
-  if (obj.model !== undefined) event.model = obj.model
-  if (obj.cwd !== undefined) event.cwd = obj.cwd
-  if (obj.tools !== undefined) event.tools = obj.tools
-  if (obj.permissionMode !== undefined) event.permissionMode = obj.permissionMode
-  if (obj.total_cost_usd !== undefined) event.total_cost_usd = obj.total_cost_usd
-  if (obj.duration_ms !== undefined) event.duration_ms = obj.duration_ms
-  if (obj.num_turns !== undefined) event.num_turns = obj.num_turns
-  if (obj.usage !== undefined) event.usage = obj.usage
-  return event
-}

package/src/claude-history.mjs

@@ -20,7 +20,36 @@ import path from "node:path"
 export const MAX_HISTORY_LINES = 500
 
 /** UI 表示対象の SDK message type (それ以外は jsonl 内部メタなので除外)。 */
-const DISPLAY_TYPES = new Set(["user", "assistant", "system", "result"])
+export const DISPLAY_TYPES = new Set(["user", "assistant", "system", "result"])
+
+/**
+ * jsonl の生 object を、Browser が SDK message として受け取れる表示用 shape に正規化する。
+ * 余分な内部メタは落とし、表示・集計に必要なフィールドだけを superset で拾う。
+ *
+ * ⚠️ 履歴 hydrate (readSessionHistory) と live watch (claude-history-watch) の両方が
+ * これを使う。フィールドを増やすときは必ずここ 1 箇所に足すこと (過去、2 箇所に分散して
+ * いて uuid が片方で取りこぼされるドリフトが発生したため共通化した。2026-05-29)。
+ *
+ * @param {Record<string, unknown>} obj jsonl 1 行をパースした生 object
+ * @returns {Record<string, unknown>} SDK message 風の表示用イベント
+ */
+export function normalizeHistoryEvent(obj) {
+  const event = { type: obj.type }
+  if (obj.message !== undefined) event.message = obj.message
+  if (obj.subtype !== undefined) event.subtype = obj.subtype
+  if (obj.uuid !== undefined) event.uuid = obj.uuid
+  if (obj.session_id !== undefined) event.session_id = obj.session_id
+  else if (obj.sessionId !== undefined) event.session_id = obj.sessionId
+  if (obj.model !== undefined) event.model = obj.model
+  if (obj.cwd !== undefined) event.cwd = obj.cwd
+  if (obj.tools !== undefined) event.tools = obj.tools
+  if (obj.permissionMode !== undefined) event.permissionMode = obj.permissionMode
+  if (obj.total_cost_usd !== undefined) event.total_cost_usd = obj.total_cost_usd
+  if (obj.duration_ms !== undefined) event.duration_ms = obj.duration_ms
+  if (obj.num_turns !== undefined) event.num_turns = obj.num_turns
+  if (obj.usage !== undefined) event.usage = obj.usage
+  return event
+}
 
 /**
  * cwd 文字列を Claude Code の project dir 名に変換する。
@@ -82,21 +111,8 @@ export async function readSessionHistory(filePath, { maxLines = MAX_HISTORY_LINE
     }
     if (!obj || typeof obj !== "object") continue
     if (!DISPLAY_TYPES.has(obj.type)) continue
-    // SDK message と同じ shape にする (余分な meta は落とす)
-    const event = { type: obj.type }
-    if (obj.message !== undefined) event.message = obj.message
-    if (obj.subtype !== undefined) event.subtype = obj.subtype
-    if (obj.session_id !== undefined) event.session_id = obj.session_id
-    else if (obj.sessionId !== undefined) event.session_id = obj.sessionId
-    if (obj.model !== undefined) event.model = obj.model
-    if (obj.cwd !== undefined) event.cwd = obj.cwd
-    if (obj.tools !== undefined) event.tools = obj.tools
-    if (obj.permissionMode !== undefined) event.permissionMode = obj.permissionMode
-    if (obj.total_cost_usd !== undefined) event.total_cost_usd = obj.total_cost_usd
-    if (obj.duration_ms !== undefined) event.duration_ms = obj.duration_ms
-    if (obj.num_turns !== undefined) event.num_turns = obj.num_turns
-    if (obj.usage !== undefined) event.usage = obj.usage
-    events.push(event)
+    // SDK message と同じ shape にする (余分な meta は落とす)。整形は共通関数に集約。
+    events.push(normalizeHistoryEvent(obj))
   }
   return { events, total_lines, truncated }
 }

package/src/claude-stream-bridge.mjs

@@ -34,9 +34,29 @@ const IDLE_DETACH_TTL_MS = 7 * 24 * 60 * 60 * 1000
  *  従来の「1メッセージ=1query」へ即ロールバックできる。デフォルト有効。 */
 const CHAT_RESIDENT_ENABLED = process.env.HUB_AGENT_CHAT_RESIDENT !== "0"
 
-/** 文字列を SDK streaming input の SDKUserMessage に包む。 */
+/** 改修4 (2026-05-29): resume セッションも常駐query化する (init を毎ターン送らず、
+ *  1 セッション=1 query で system/init を初回 1 回のみにする。VS Code 拡張 / Claude Web
+ *  と同じ挙動)。query 起動時に options.resume で過去文脈を引き継ぐ。
+ *  env HUB_AGENT_CHAT_RESIDENT_RESUME="0" で従来の per-message (毎ターン init) に
+ *  個別ロールバック可能 (新規セッションの常駐化は CHAT_RESIDENT_ENABLED 側で独立制御)。
+ *  デフォルト有効。
+ *  注意: 過去に「resume + 常駐」で『Continue from where you left off』暴走が疑われたが、
+ *  原因は continue オプション (直近会話の自動継続) の誤用と Hub 不調の重なりと推定。
+ *  本実装では continue は使わず resume (明示 session 指定) のみを用い、初回 input は
+ *  ユーザーの実メッセージのみとすることで自動継続の暴走を回避する。 */
+const CHAT_RESIDENT_RESUME_ENABLED =
+  process.env.HUB_AGENT_CHAT_RESIDENT_RESUME !== "0"
+
+/** 文字列を SDK streaming input の SDKUserMessage に包む。
+ *  SDKUserMessage は parent_tool_use_id: string|null が必須フィールド。現行 SDK は入力側で
+ *  寛容なので省略しても動くが、将来の型厳格化に備えて明示する。トップレベルのユーザー入力
+ *  なので親 tool_use は無く null。 */
 function toSDKUserMessage(text) {
-  return { type: "user", message: { role: "user", content: text } }
+  return {
+    type: "user",
+    message: { role: "user", content: text },
+    parent_tool_use_id: null,
+  }
 }
 
 /** 常駐 query の streaming input キュー。push された SDKUserMessage を AsyncIterable として
@@ -136,13 +156,17 @@ class ClaudeStreamSession {
     /** detached のまま放置されたセッションを撤去する idle タイマー。再アタッチでキャンセル。 */
     this._idleTimer = null
 
-    /** 改修2: 常駐query対象か。明示 resident 指定があれば優先 (テスト/将来の per-session 設定用)。
-     *  未指定なら resume なしの新規セッションのみ常駐 (既存 resume セッションは過去の
-     *  「Continue from where you left off」暴走を避けて従来 per-message を維持)。 */
+    /** 改修2+4: 常駐query対象か。明示 resident 指定があれば優先 (テスト/将来の per-session 設定用)。
+     *  未指定の場合:
+     *   - 新規セッション (resume なし): CHAT_RESIDENT_ENABLED で常駐 (改修2)。
+     *   - 既存セッション (resume あり): CHAT_RESIDENT_ENABLED かつ CHAT_RESIDENT_RESUME_ENABLED
+     *     なら常駐 (改修4)。query 起動時に options.resume で文脈を引き継ぎ init は初回 1 回のみ。
+     *  どちらも無効なら従来 per-message (毎ターン init) にフォールバックする。 */
     this._residentEligible =
       typeof resident === "boolean"
         ? resident
-        : !resumeSessionId && CHAT_RESIDENT_ENABLED
+        : CHAT_RESIDENT_ENABLED &&
+          (!resumeSessionId || CHAT_RESIDENT_RESUME_ENABLED)
     /** 常駐query の input キュー (streaming input)。初回 input 時に生成・query 起動。 */
     this._inputQueue = null
     /** 起動済みの常駐 query ハンドル (interrupt() 用)。 */
@@ -244,14 +268,89 @@ class ClaudeStreamSession {
   /** 再アタッチ: 走行中(または生存中)セッションに新しい stream_id を紐付け直し、
    *  idle 撤去タイマーを止める。以降のターンイベントはこの stream_id 経由で新しい
    *  browser 接続へライブに流れる (= 通常の生成中表示と同じ)。再アタッチ前の確定分は
-   *  browser 側の jsonl hydrate (history.request) で復元するため、ここでは replay しない。 */
-  reattach(stream_id) {
+   *  browser 側の jsonl hydrate (history.request) で復元するため、ここでは replay しない。
+   *
+   *  改修5 (2026-05-29): モデル/権限/拡張思考のターン切替。再アタッチ時に opts で
+   *  新しい値が渡され、現在値と異なれば applyRuntimeOptions で反映する。これにより
+   *  入力欄下バッジの変更が「同一セッション(常駐 query)を維持したまま次ターンから」
+   *  効くようになる。従来は reattach が stream_id だけ差し替えていたため、起動済み
+   *  query の model/permission/thinking は初期値のまま変わらなかった (バッジが効かない
+   *  不具合の原因)。 */
+  reattach(stream_id, opts = undefined) {
     this.stream_id = stream_id
     this._detached = false
     if (this._idleTimer) {
       clearTimeout(this._idleTimer)
       this._idleTimer = null
     }
+    if (opts) this.applyRuntimeOptions(opts)
+  }
+
+  /** 改修5: モデル/権限/拡張思考をランタイムに切り替える。
+   *
+   *  - 保持フィールド (this.model/permissionMode/maxThinkingTokens) を更新する。これは
+   *    常駐 query が異常終了して resume 再起動する際 (_runResidentQuery) に最新値で
+   *    再 spawn させるため、および per-message セッションが次ターンの options に反映する
+   *    ため。
+   *  - 起動済みの常駐 query があれば SDK の制御メソッド (setModel / setPermissionMode /
+   *    setMaxThinkingTokens) を呼び、プロセス再起動なしで次ターンから即反映する
+   *    (公式 streaming input mode のランタイム制御。stdin に control_request を流す)。
+   *
+   *  値が undefined のキーは「変更なし」として無視する (バッジ未送出時に既存値を消さない)。
+   *  model に空文字/null が来たら setModel(undefined) でデフォルトへ戻す。
+   *  maxThinkingTokens に 0/null が来たら setMaxThinkingTokens(null) でオフにする。 */
+  applyRuntimeOptions({ model, permissionMode, maxThinkingTokens } = {}) {
+    const q = this._residentQuery
+    // モデル
+    if (model !== undefined) {
+      const next = model || null
+      if (next !== this.model) {
+        this.model = next
+        if (q && typeof q.setModel === "function") {
+          q.setModel(next || undefined).catch((err) =>
+            this.logger?.warn(
+              { err: err?.message, stream_id: this.stream_id },
+              "setModel failed",
+            ),
+          )
+        }
+      }
+    }
+    // 権限モード
+    if (permissionMode !== undefined) {
+      const next = permissionMode || null
+      if (next !== this.permissionMode) {
+        this.permissionMode = next
+        // setPermissionMode は有効な mode を要求する。null/空 (=未指定へ戻す) は
+        // SDK 側に「解除」API が無いため、保持値の更新のみ (次回 spawn で既定に従う)。
+        if (next && q && typeof q.setPermissionMode === "function") {
+          q.setPermissionMode(next).catch((err) =>
+            this.logger?.warn(
+              { err: err?.message, stream_id: this.stream_id },
+              "setPermissionMode failed",
+            ),
+          )
+        }
+      }
+    }
+    // 拡張思考予算
+    if (maxThinkingTokens !== undefined) {
+      const next =
+        typeof maxThinkingTokens === "number" && maxThinkingTokens > 0
+          ? maxThinkingTokens
+          : null
+      if (next !== this.maxThinkingTokens) {
+        this.maxThinkingTokens = next
+        if (q && typeof q.setMaxThinkingTokens === "function") {
+          q.setMaxThinkingTokens(next).catch((err) =>
+            this.logger?.warn(
+              { err: err?.message, stream_id: this.stream_id },
+              "setMaxThinkingTokens failed",
+            ),
+          )
+        }
+      }
+    }
   }
 
   /** soft detach: browser 切断時にターンを中断せずセッションを生かしたまま detached に
@@ -286,20 +385,30 @@ class ClaudeStreamSession {
     const prompt = extractPromptText(message)
     if (!prompt) return
 
-    // 改修2: 常駐query対象セッションは streaming input キューへ積む (busy 中でも積み、
-    // SDK が順次処理する)。初回 input で常駐 query を 1 回だけ起動する。
+    // 改修2+4: 常駐query対象セッション。
     if (this._residentEligible) {
       if (!this._inputQueue) this._inputQueue = new InputQueue()
+      // 改修4 (A): ターンのシリアライズ。実行中ターン (busy) は InputQueue へ即 push せず
+      // pending へ退避し、現ターンの result 後に 1 件ずつ drain する。前ターン未完了のまま
+      // 次を push すると SDK streaming-input で割り込み扱いになり得るため (公式 interrupt 警告)。
+      // pending は queue_state で UI (送信待ちチップ) に出す (per-message と同じ体験)。
+      if (this._busy) {
+        this._pendingMessages.push(prompt)
+        this.logger?.info(
+          { stream_id: this.stream_id, queued: this._pendingMessages.length },
+          "resident busy, message queued",
+        )
+        this._emitQueueState()
+        return
+      }
       this._busy = true
       this._inputQueue.push(toSDKUserMessage(prompt))
-      if (!this._residentStarted) {
+      // 改修4 (A): 死亡ガード。常駐 query が未起動 or (エラー等で) 終了済み (_residentQuery=null)
+      // なら (再)起動する。_runResidentQuery は起動時 options.resume=this.sessionId で文脈を
+      // 復元するため、途中死からの復活でも過去コンテキストは失われない。
+      if (!this._residentQuery) {
         this._residentStarted = true
-        this._runResidentQuery().catch((err) => {
-          this.logger?.error(
-            { stream_id: this.stream_id, err: err?.message },
-            "resident query threw",
-          )
-        })
+        this._startResidentQuery()
       }
       return
     }
@@ -341,7 +450,13 @@ class ClaudeStreamSession {
     if (this.sessionId) options.resume = this.sessionId
 
     try {
-      const generator = this.sdk.query({ prompt, options })
+      // 改修4 (A): canUseTool は streaming input (AsyncIterable prompt) でのみ確実に機能する。
+      // per-message でも単発 InputQueue (1 件 push → 即 close) を prompt に渡し、1 query=1 ターンの
+      // resume チェーン挙動は維持しつつ、文字列 prompt + canUseTool の SDK 制約リスクを回避する。
+      const inputQueue = new InputQueue()
+      inputQueue.push(toSDKUserMessage(prompt))
+      inputQueue.close()
+      const generator = this.sdk.query({ prompt: inputQueue, options })
       for await (const msg of generator) {
         if (
           msg?.type === "system" &&
@@ -460,11 +575,40 @@ class ClaudeStreamSession {
     }
   }
 
-  /** 改修2: 常駐 query を 1 回だけ起動し、streaming input キューから複数ターンを処理する。
+  /** 改修4 (A): 常駐 query を resume 付きで (再)起動する。_runResidentQuery 起動時に
+   *  options.resume=this.sessionId で過去文脈を復元するため、途中死からの復活でも安全。 */
+  _startResidentQuery() {
+    this._runResidentQuery().catch((err) => {
+      this.logger?.error(
+        { stream_id: this.stream_id, err: err?.message },
+        "resident query threw",
+      )
+    })
+  }
+
+  /** 改修4 (A): ターン完了時に pending の先頭 1 件を InputQueue へ流す (ターンのシリアライズ)。
+   *  queue_state を更新して送信待ちチップを drain させる (frontend がバブル昇格する)。 */
+  _drainResidentPending() {
+    if (this._closed) return
+    if (this._pendingMessages.length === 0) {
+      this._emitQueueState()
+      return
+    }
+    const next = this._pendingMessages.shift()
+    this._busy = true
+    this._inputQueue.push(toSDKUserMessage(next))
+    this._emitQueueState()
+  }
+
+  /** 改修2+4: 常駐 query を 1 回だけ起動し、streaming input キューから複数ターンを処理する。
    *  system/init は最初の 1 回のみ (毎ターン init = 文脈/キャッシュ温め直しを解消)。各ターンは
    *  result で _busy=false にするが query は継続し次 input を待つ。close (inputQueue.close) で
-   *  generator が終わり query が完走する。resume は使わない (新規セッションのみが対象)。
-   *  常駐では query stream が唯一のイベント源のため watcher は張らない。 */
+   *  generator が終わり query が完走する。
+   *  改修4: resume セッションも対象。query 起動時 options.resume に過去 session_id を渡して
+   *  文脈を引き継ぐ (init は初回 1 回のみ)。continue は使わない (自動継続の暴走回避)。初回 input は
+   *  sendMessage が積むユーザーの実メッセージなので「Continue from where you left off」は発生しない。
+   *  常駐では query stream が唯一のイベント源のため watcher は張らない (過去確定分は browser が
+   *  history.request で hydrate 済み。resume で query stream が過去行を replay しない前提)。 */
   async _runResidentQuery() {
     const options = {
       cwd: this.cwd,
@@ -475,6 +619,9 @@ class ClaudeStreamSession {
     if (this.permissionMode) options.permissionMode = this.permissionMode
     if (this.maxTurns != null) options.maxTurns = this.maxTurns
     if (this.maxThinkingTokens != null) options.maxThinkingTokens = this.maxThinkingTokens
+    // 改修4: 起動時に sessionId (= resumeSessionId) があれば resume チェーンで文脈を引き継ぐ。
+    // query 起動時点の値のみ有効 (起動後に確定/変化する session_id は同一 query 内で継続される)。
+    if (this.sessionId) options.resume = this.sessionId
     const denyPending = (reason) => {
       for (const [, resolver] of this._permissionResolvers) {
         try {
@@ -501,6 +648,8 @@ class ClaudeStreamSession {
           // ターン完了: 未解決 permission を閉じ、次 input 受付へ (query は継続)。
           denyPending("turn ended")
           this._busy = false
+          // 改修4 (A): シリアライズした pending があれば次の 1 件を InputQueue へ流す。
+          this._drainResidentPending()
         }
         try {
           this.onEvent?.(msg)
@@ -523,13 +672,22 @@ class ClaudeStreamSession {
       this._residentQuery = null
       this._busy = false
       denyPending("closed")
-      // 常駐 query の終了 = セッション終了。detached 後にここへ来たら reap する。
+      // 常駐 query の終了。detached 後にここへ来たら reap する。
       if (this._reapAfterTurn && !this._closed) {
         try {
           this.close()
         } finally {
           this.onReap?.()
         }
+      } else if (!this._closed && this._pendingMessages.length > 0) {
+        // 改修4 (A): 異常終了 (close 以外) で pending が残っていれば resume 付きで再起動し
+        // 取りこぼしを防ぐ。_runResidentQuery が options.resume=this.sessionId で文脈を復元する。
+        const next = this._pendingMessages.shift()
+        this._busy = true
+        this._residentStarted = true
+        this._inputQueue.push(toSDKUserMessage(next))
+        this._emitQueueState()
+        this._startResidentQuery()
       }
     }
   }
@@ -657,10 +815,25 @@ export class ClaudeStreamBridge extends EventEmitter {
       const live = this._liveBySession.get(resumeSessionId)
       if (live && !live._closed) {
         this.sessions.delete(live.stream_id)
-        live.reattach(stream_id)
+        // 改修5: 再アタッチ時に model/permission/maxThinkingTokens を引き継ぎ反映する。
+        // browser はバッジ変更時に新しい値を載せた claude.attach を同一 resume で送る
+        // ため、ここで適用すると常駐 query を維持したまま次ターンから切り替わる。
+        live.reattach(stream_id, {
+          model,
+          permissionMode,
+          maxThinkingTokens:
+            typeof maxThinkingTokens === "number" ? maxThinkingTokens : null,
+        })
         this.sessions.set(stream_id, live)
         this.logger?.info(
-          { stream_id, resume: resumeSessionId, busy: live._busy },
+          {
+            stream_id,
+            resume: resumeSessionId,
+            busy: live._busy,
+            model: live.model,
+            permissionMode: live.permissionMode,
+            maxThinkingTokens: live.maxThinkingTokens,
+          },
           "claude stream reattached to live session",
         )
         return { stream_id, resuming: true, reattached: true }