@cocorograph/hub-agent 0.6.37 → 0.6.39

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cocorograph/hub-agent",
-  "version": "0.6.37",
+  "version": "0.6.39",
   "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),

package/src/ws-client.mjs

@@ -23,6 +23,12 @@ const MIN_BACKOFF_MS = 1_000
 const MAX_BACKOFF_MS = 30_000
 const BUNDLE_WATCH_DEBOUNCE_MS = 500
 
+// 接続が安定したとみなすまでの猶予。open 直後に即 backoff をリセットすると、
+// open→即 close のフラッピング時に毎回 backoff が最小 (1s) へ戻り、1〜2 秒間隔の
+// 再接続ストロボに陥る (CF/backend 側の瞬断時に hub-agent↔backend WS で観測)。
+// この時間だけ接続を維持できて初めて backoff をリセットする。
+const STABLE_CONNECTION_MS = 10_000
+
 // CF / origin が 5xx を返した直後の再接続は短い backoff だと 5xx キャッシュに
 // 当たって連発失敗するため、最低 5s 待つ。30s リトライまで段階的に伸びる。
 const MIN_BACKOFF_AFTER_5XX_MS = 5_000
@@ -76,6 +82,9 @@ export class WsClient extends EventEmitter {
     this.ptyOutboundBuffer = []
     // 直前の close 原因が CF/origin の 5xx か。次回 backoff の下限を引き上げる。
     this.lastCloseWas5xx = false
+    // 接続が STABLE_CONNECTION_MS 維持できたら backoff/5xx フラグをリセットする
+    // タイマー。close (_clearStableReset) でキャンセルされる。
+    this.stableResetTimer = null
   }
 
   /** WSS 接続を開始する。`stop()` まで自動で reconnect 続行。 */
@@ -104,6 +113,7 @@ export class WsClient extends EventEmitter {
     })
 
     ws.on("close", (code, reason) => {
+      this._clearStableReset()
       this._stopHeartbeat()
       this._stopBundleWatcher()
       this.logger?.info({ code, reason: reason?.toString() }, "ws close")
@@ -136,8 +146,11 @@ export class WsClient extends EventEmitter {
    * のクロージャだと spy しづらい)。
    */
   _onOpen() {
-    this.backoff = MIN_BACKOFF_MS
-    this.lastCloseWas5xx = false
+    // backoff の即時リセットはしない。接続が STABLE_CONNECTION_MS 維持できてから
+    // _armStableReset() でリセットする (open→即 close フラッピング時の再接続
+    // ストロボ対策)。すぐ切れた接続では backoff がリセットされず指数バックオフが
+    // 効き続けるため、1〜2 秒間隔の再接続ループに陥らない。
+    this._armStableReset()
     this.logger?.info("ws open")
     // 切断中に積んだ pty.data を flush。hello より先に送ると backend で stream
     // 未登録のため unknown_stream error になる可能性があるが、hello の応答待ち
@@ -232,6 +245,7 @@ export class WsClient extends EventEmitter {
   /** Reconnect を止めて切断する。 */
   stop() {
     this.stopped = true
+    this._clearStableReset()
     this._stopHeartbeat()
     this._stopBundleWatcher()
     if (this.reconnectTimer) {
@@ -428,4 +442,29 @@ export class WsClient extends EventEmitter {
     }, delay)
     this.reconnectTimer.unref?.()
   }
+
+  /**
+   * 接続が STABLE_CONNECTION_MS 維持できたら backoff と 5xx フラグをリセットする
+   * タイマーを張る (open 直後に呼ぶ)。close で _clearStableReset によりキャンセル
+   * されるため、安定しなかった (= すぐ切れた) 接続では backoff がリセットされず
+   * 指数バックオフが効き続け、再接続ストロボを防ぐ。
+   */
+  _armStableReset() {
+    this._clearStableReset()
+    this.stableResetTimer = setTimeout(() => {
+      this.stableResetTimer = null
+      this.backoff = MIN_BACKOFF_MS
+      this.lastCloseWas5xx = false
+      this.logger?.debug("ws connection stable, backoff reset")
+    }, STABLE_CONNECTION_MS)
+    this.stableResetTimer.unref?.()
+  }
+
+  /** 安定リセットタイマーをキャンセルする (close / stop 時)。 */
+  _clearStableReset() {
+    if (this.stableResetTimer) {
+      clearTimeout(this.stableResetTimer)
+      this.stableResetTimer = null
+    }
+  }
 }