@cocorograph/hub-agent 0.6.37 → 0.6.38

package/package.json

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

package/src/claude-stream-bridge.mjs

@@ -998,6 +998,17 @@ class ClaudeStreamSession {
     }
   }
 
+  /** このセッションの常駐 query 内で MCP サーバーを再接続する (動作中チャットへ波及)。
+   *  常駐 query が無い (per-message セッション) 場合は false。per-message は毎ターン
+   *  新プロセスで MCP を張り直すため、次メッセージで自動的に再接続される。
+   *  @returns {Promise<boolean>} 再接続を発行できたら true */
+  async reconnectMcp(serverName) {
+    const q = this._residentQuery
+    if (!q || typeof q.reconnectMcpServer !== "function") return false
+    await q.reconnectMcpServer(serverName)
+    return true
+  }
+
   /**
    * graceful detach: 実行中ターンがあれば中断せず完走を待つ (完走後に finally で
    * 自動クローズ + onReap)。アイドルなら即クローズする。
@@ -1360,7 +1371,7 @@ export class ClaudeStreamBridge extends EventEmitter {
     return await q.mcpServerStatus()
   }
 
-  /** 指定 MCP サーバーを再接続し、最新状態一覧を返す。 */
+  /** 指定 MCP サーバーを control 専用 query 内で再接続し、最新状態一覧を返す。 */
   async reconnectMcp(serverName) {
     const q = await this._ensureMcpControlQuery()
     this._armMcpControlIdle()
@@ -1368,6 +1379,32 @@ export class ClaudeStreamBridge extends EventEmitter {
     return await q.mcpServerStatus()
   }
 
+  /** 指定 MCP サーバーを「動作中の全 resident セッション」+ control query で再接続する。
+   *  MCP 接続はプロセス単位のため、control query だけ張り直しても動作中チャットには
+   *  波及しない。そこで全 resident セッションの常駐 query にも reconnect を発行し、
+   *  会話中の MCP 接続を実際に張り直す。reconnect は冪等なので多重発行は無害。
+   *  @returns {Promise<{servers: object[], targeted: number}>}
+   *    servers: control query の最新状態 (popover 表示用) / targeted: 波及したセッション数 */
+  async reconnectMcpAllSessions(serverName) {
+    const seen = new Set()
+    let targeted = 0
+    for (const session of this.sessions.values()) {
+      if (seen.has(session)) continue
+      seen.add(session)
+      try {
+        if (await session.reconnectMcp(serverName)) targeted += 1
+      } catch (err) {
+        this.logger?.warn(
+          { err: err?.message, server: serverName },
+          "session mcp reconnect failed",
+        )
+      }
+    }
+    // control query も張り直して popover 表示用の最新状態を返す。
+    const servers = await this.reconnectMcp(serverName)
+    return { servers, targeted }
+  }
+
   /** 指定 MCP サーバーの OAuth 認証を開始する。認可 URL を含む応答を返す。 */
   async mcpAuthenticate(serverName, redirectUri) {
     const q = await this._ensureMcpControlQuery()

package/src/main.mjs

@@ -990,11 +990,14 @@ async function dispatch(msg, ctx) {
             case "claude.mcp.status":
               result = { servers: await ctx.claudeBridge.mcpStatus() }
               break
-            case "claude.mcp.reconnect":
-              result = {
-                servers: await ctx.claudeBridge.reconnectMcp(msg.server_name),
-              }
+            case "claude.mcp.reconnect": {
+              // 動作中の全 resident セッション + control query を再接続 (会話中の MCP に波及)。
+              const r = await ctx.claudeBridge.reconnectMcpAllSessions(
+                msg.server_name,
+              )
+              result = { servers: r.servers, targeted: r.targeted }
               break
+            }
             case "claude.mcp.authenticate":
               result = {
                 auth: await ctx.claudeBridge.mcpAuthenticate(
@@ -1003,14 +1006,20 @@ async function dispatch(msg, ctx) {
                 ),
               }
               break
-            case "claude.mcp.callback":
-              result = {
-                auth: await ctx.claudeBridge.mcpSubmitOAuthCallback(
-                  msg.server_name,
-                  msg.callback_url,
-                ),
-              }
+            case "claude.mcp.callback": {
+              const auth = await ctx.claudeBridge.mcpSubmitOAuthCallback(
+                msg.server_name,
+                msg.callback_url,
+              )
+              // 再認証で得たトークンはディスクに永続するが、動作中セッションは別プロセス
+              // のため自動では拾わない。全 resident セッション + control query を再接続し、
+              // 会話中もすぐ新トークンで使えるようにする。
+              const r = await ctx.claudeBridge.reconnectMcpAllSessions(
+                msg.server_name,
+              )
+              result = { auth, servers: r.servers, targeted: r.targeted }
               break
+            }
             case "claude.mcp.clearauth":
               result = {
                 cleared: await ctx.claudeBridge.mcpClearAuth(msg.server_name),