@cocorograph/hub-agent 0.6.16 → 0.6.18

package/package.json

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

package/src/claude-stream-bridge.mjs

@@ -47,6 +47,21 @@ const CHAT_RESIDENT_ENABLED = process.env.HUB_AGENT_CHAT_RESIDENT !== "0"
 const CHAT_RESIDENT_RESUME_ENABLED =
   process.env.HUB_AGENT_CHAT_RESIDENT_RESUME !== "0"
 
+/** 多端末共有セッション (2026-05-30)。同じ Claude セッション (session_id) を複数端末で
+ *  同時に開いて、どの端末でもライブ表示・入力・許可応答できるようにする。env
+ *  HUB_AGENT_CHAT_SHARED="1" で有効化 (デフォルト無効 = 従来の「最後の端末だけが live」)。
+ *
+ *  有効時の差分:
+ *   - reattach で旧 stream_id を sessions Map から消さず、1 セッションに複数の端末
+ *     stream_id を購読登録する (this.streamIds)。これにより旧端末からの input /
+ *     permission.reply / interrupt も sessions.get(stream_id) で解決できる。
+ *   - detach は購読端末の参照カウントを下げ、最後の 1 台が外れて初めて idle softDetach
+ *     する (1 台閉じただけで走行中セッションを撤去しない)。
+ *  配信のファンアウト自体は Hub backend が session_id 単位の group broadcast で行うため、
+ *  agent 側はイベントに session_id を載せるだけでよい (既に emit 済み)。無効時は従来の
+ *  単一 stream_id 挙動と完全に一致する (this.streamIds は未使用のまま)。 */
+const CHAT_SHARED_ENABLED = process.env.HUB_AGENT_CHAT_SHARED === "1"
+
 /** 文字列を SDK streaming input の SDKUserMessage に包む。
  *  SDKUserMessage は parent_tool_use_id: string|null が必須フィールド。現行 SDK は入力側で
  *  寛容なので省略しても動くが、将来の型厳格化に備えて明示する。トップレベルのユーザー入力
@@ -133,6 +148,10 @@ class ClaudeStreamSession {
     onReap,
   }) {
     this.stream_id = stream_id
+    /** 多端末共有: このセッションを購読中の全端末 stream_id。CHAT_SHARED_ENABLED 時のみ
+     *  参照され、reattach で増え detach で減る。最後の 1 台が外れると idle softDetach に入る。
+     *  this.stream_id は「最も新しく attach した端末」= legacy emit / primary 用に残す。 */
+    this.streamIds = new Set([stream_id])
     this.cwd = cwd
     this.model = model || null
     this.permissionMode = permissionMode || null
@@ -268,14 +287,92 @@ 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
+    // 多端末共有: この端末を購読集合に追加 (無効時は未使用)。旧端末の stream_id は
+    // bridge.attach 側で sessions Map に残されるため、ここでは増やすだけでよい。
+    this.streamIds.add(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 に
@@ -704,6 +801,23 @@ export class ClaudeStreamBridge extends EventEmitter {
     this._liveBySession = new Map()
   }
 
+  /** 多端末共有: あるセッションに紐づく全端末 stream_id を sessions Map から外し、
+   *  session_id 索引も掃除する。撤去 (reap / idle / shutdown) で 1 端末分だけ消すと
+   *  他端末キーが Map に残って孤児化するため、まとめて消す。 */
+  _dropSessionMappings(session) {
+    // session.streamIds に載っている端末キーをすべて外す。
+    for (const sid of Array.from(session.streamIds || [])) {
+      if (this.sessions.get(sid) === session) this.sessions.delete(sid)
+    }
+    // 念のため primary stream_id も (streamIds に無い経路があった場合の保険)。
+    if (this.sessions.get(session.stream_id) === session) {
+      this.sessions.delete(session.stream_id)
+    }
+    if (session.sessionId && this._liveBySession.get(session.sessionId) === session) {
+      this._liveBySession.delete(session.sessionId)
+    }
+  }
+
   /**
    * 新しい Claude セッションを開始する。
    *
@@ -739,11 +853,31 @@ export class ClaudeStreamBridge extends EventEmitter {
     if (resumeSessionId) {
       const live = this._liveBySession.get(resumeSessionId)
       if (live && !live._closed) {
-        this.sessions.delete(live.stream_id)
-        live.reattach(stream_id)
+        // 多端末共有 (CHAT_SHARED_ENABLED): 旧 stream_id を Map から消さない。同一セッションを
+        // 複数端末が購読する状態を作り、どの端末の input/permission/interrupt も解決できるように
+        // する。無効時は従来どおり旧 stream_id を撤去し「最後の 1 端末だけ live」に戻す。
+        if (!CHAT_SHARED_ENABLED) {
+          this.sessions.delete(live.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 }
@@ -768,24 +902,27 @@ export class ClaudeStreamBridge extends EventEmitter {
         if (session.sessionId) this._liveBySession.set(session.sessionId, session)
       },
       onPermission: ({ tool_name, input, request_id }) => {
+        // 多端末共有: session_id を載せて backend が session group へ broadcast できるように
+        // する (全端末に許可ダイアログを出し、先着 1 件の reply を採用する)。
         this.emit("permission", {
           stream_id: session.stream_id,
+          session_id: session.sessionId,
           request_id,
           tool_name,
           input,
         })
       },
       onError: (err) => {
-        this.emit("error", { stream_id: session.stream_id, error: err?.message || String(err) })
+        this.emit("error", {
+          stream_id: session.stream_id,
+          session_id: session.sessionId,
+          error: err?.message || String(err),
+        })
       },
       onReap: () => {
         // ターン完走後の遅延クローズ。Map / 索引から撤去し exit を emit する。
-        if (this.sessions.get(session.stream_id) === session) {
-          this.sessions.delete(session.stream_id)
-        }
-        if (session.sessionId && this._liveBySession.get(session.sessionId) === session) {
-          this._liveBySession.delete(session.sessionId)
-        }
+        // 多端末共有では複数端末キーが Map に載っているのでまとめて外す。
+        this._dropSessionMappings(session)
         this.emit("exit", {
           stream_id: session.stream_id,
           code: 0,
@@ -851,14 +988,25 @@ export class ClaudeStreamBridge extends EventEmitter {
   detach({ stream_id }) {
     const s = this.sessions.get(stream_id)
     if (!s) return false
+    // 多端末共有: この端末分の購読だけ外す。まだ他端末が見ているならセッションは生かし、
+    // softDetach (idle 撤去タイマー) には入らない。最後の 1 台が外れたときだけ従来の
+    // idle softDetach に進む。無効時は streamIds が常に 1 要素なので即 softDetach に落ちる。
+    if (CHAT_SHARED_ENABLED) {
+      s.streamIds.delete(stream_id)
+      if (this.sessions.get(stream_id) === s) this.sessions.delete(stream_id)
+      if (s.streamIds.size > 0) {
+        // primary を生存端末へ寄せ替え (legacy emit / softDetach 撤去ログ用)。
+        if (s.stream_id === stream_id) {
+          s.stream_id = Array.from(s.streamIds)[s.streamIds.size - 1]
+        }
+        return true
+      }
+    }
     // 常駐化: browser 切断ではセッションを止めない (reap しない)。走行中ターンは完走し、
     // idle のまま IDLE_DETACH_TTL_MS 経過して初めて撤去する。TTL 内に再接続 (再アタッチ)
     // されればキャンセルされる。明示的に止めたい場合は interrupt()/明示終了を使う。
     s.softDetach(IDLE_DETACH_TTL_MS, () => {
-      if (this.sessions.get(s.stream_id) === s) this.sessions.delete(s.stream_id)
-      if (s.sessionId && this._liveBySession.get(s.sessionId) === s) {
-        this._liveBySession.delete(s.sessionId)
-      }
+      this._dropSessionMappings(s)
       s.close()
       this.emit("exit", {
         stream_id: s.stream_id,

package/src/main.mjs

@@ -190,10 +190,13 @@ export async function startDaemon({ version, ptyModule, claudeSdk } = {}) {
       }
       client.send({ type: "claude.event", stream_id, session_id, event })
     })
-    claudeBridge.on("permission", ({ stream_id, request_id, tool_name, input }) => {
+    claudeBridge.on("permission", ({ stream_id, session_id, request_id, tool_name, input }) => {
+      // 多端末共有: session_id を載せて backend が session group へ broadcast できるように
+      // する (全端末に許可ダイアログ → 先着 1 件採用)。
       client.send({
         type: "claude.permission.request",
         stream_id,
+        session_id,
         request_id,
         tool_name,
         input,
@@ -202,8 +205,8 @@ export async function startDaemon({ version, ptyModule, claudeSdk } = {}) {
     claudeBridge.on("exit", ({ stream_id, code, reason, session_id }) => {
       client.send({ type: "claude.exit", stream_id, code, reason, session_id })
     })
-    claudeBridge.on("error", ({ stream_id, error }) => {
-      client.send({ type: "claude.error", stream_id, error })
+    claudeBridge.on("error", ({ stream_id, session_id, error }) => {
+      client.send({ type: "claude.error", stream_id, session_id, error })
     })
   }