@cocorograph/hub-agent 0.5.28 → 0.5.29

package/package.json

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

package/src/main.mjs

@@ -403,6 +403,45 @@ export async function maybeSyncClaudeSettings(msg, ctx) {
   }
 }
 
+/**
+ * Plan ε: `agent.streams.sync.response` を受けて backend ↔ agent の現存
+ * stream_id を能動同期する。
+ *
+ * - local にあって backend にない → ptyBridge.detach で即時 kill (孤児)
+ * - backend にあって local にない → backend に pty.error を通知 (browser に
+ *   "pty 死んでます" を表示するため)
+ *
+ * dispatch から呼ぶ用のヘルパーだが、テストから直接叩けるよう export している。
+ */
+export function handleStreamsSyncResponse(msg, ctx) {
+  const activeIds = new Set(
+    Array.isArray(msg?.active_stream_ids) ? msg.active_stream_ids : [],
+  )
+  const localIds = new Set(ctx.ptyBridge.list())
+  for (const sid of localIds) {
+    if (!activeIds.has(sid)) {
+      ctx.logger?.warn?.(
+        { stream_id: sid, request_id: msg?.request_id },
+        "Plan ε: orphan stream detected, detaching",
+      )
+      ctx.ptyBridge.detach({ stream_id: sid })
+    }
+  }
+  for (const sid of activeIds) {
+    if (!localIds.has(sid)) {
+      ctx.logger?.warn?.(
+        { stream_id: sid, request_id: msg?.request_id },
+        "Plan ε: stream missing on agent, notifying backend (pty.error)",
+      )
+      ctx.client.send({
+        type: "pty.error",
+        stream_id: sid,
+        error: "stream_not_found_on_agent",
+      })
+    }
+  }
+}
+
 async function dispatch(msg, ctx) {
   const t = msg?.type || ""
   try {
@@ -456,6 +495,15 @@ async function dispatch(msg, ctx) {
       case "pty.detach":
         ctx.ptyBridge.detach({ stream_id: msg.stream_id })
         return
+      case "agent.streams.sync.response":
+        // Plan ε: backend ↔ agent の現存 stream_id 能動同期 (WS reconnect 直後に
+        // ws-client が agent.streams.sync.request を送って取得した結果)。
+        // ロジックは handleStreamsSyncResponse へ抽出 (テストから直接呼べる)。
+        // backend が旧 (= Plan ε 未対応) で response が一切来ないケースは何もしない。
+        // 5s タイムアウトのような明示的な timer は持たず、再接続時に再度 request
+        // が飛ぶので無害。
+        handleStreamsSyncResponse(msg, ctx)
+        return
       case "tmux.exec": {
         const args = Array.isArray(msg.args) ? msg.args : []
         const hookResult = await runHookChain(ctx.plugins, "interceptTmuxExec", {

package/src/ws-client.mjs

@@ -11,6 +11,7 @@
  *
  * 仕様書: ナレッジ/インフラ/cockpit-hub-hosted-integration-spec (id=6080)
  */
+import { randomUUID } from "node:crypto"
 import { EventEmitter } from "node:events"
 import fs from "node:fs"
 import os from "node:os"
@@ -68,22 +69,7 @@ export class WsClient extends EventEmitter {
     const ws = new WebSocket(wsUrl, { headers })
     this.ws = ws
 
-    ws.on("open", () => {
-      this.backoff = MIN_BACKOFF_MS
-      this.logger?.info("ws open")
-      // hello 前に最新の bundle version を読み直す (再接続時の鮮度確保)
-      this._refreshBundleVersion()
-      this._sendJson({
-        type: "hello",
-        agent_id: this.config.agent_id,
-        hostname: this.hostname,
-        version: this.version,
-        bundle_version: this.bundleVersion,
-      })
-      this._startHeartbeat()
-      this._startBundleWatcher()
-      this.emit("open")
-    })
+    ws.on("open", () => this._onOpen())
 
     ws.on("message", (data) => {
       let msg
@@ -111,6 +97,43 @@ export class WsClient extends EventEmitter {
     })
   }
 
+  /**
+   * WS open 時の初期送信 + 後続タスク起動。
+   *
+   * - backoff をリセット
+   * - bundle version を最新化 → hello 送信
+   * - Plan ε: agent.streams.sync.request を送信 (毎 reconnect で能動同期)
+   * - heartbeat + bundle watcher を起動
+   *
+   * test からも直接呼べるように method として切り出している (`ws.on("open")`
+   * のクロージャだと spy しづらい)。
+   */
+  _onOpen() {
+    this.backoff = MIN_BACKOFF_MS
+    this.logger?.info("ws open")
+    // hello 前に最新の bundle version を読み直す (再接続時の鮮度確保)
+    this._refreshBundleVersion()
+    this._sendJson({
+      type: "hello",
+      agent_id: this.config.agent_id,
+      hostname: this.hostname,
+      version: this.version,
+      bundle_version: this.bundleVersion,
+    })
+    // Plan ε: 毎回 reconnect 直後に backend ↔ agent の stream_id を能動同期する
+    // (orphan stream の即時 kill 用)。response は main.mjs の dispatch が拾い、
+    // ptyBridge.list() と差分を取って kill する。request_id は uuid で
+    // 重複排除。backend が古い (= Plan ε 未対応) なら "unknown_type" error が
+    // 返るだけで動作には影響しない。
+    this._sendJson({
+      type: "agent.streams.sync.request",
+      request_id: randomUUID(),
+    })
+    this._startHeartbeat()
+    this._startBundleWatcher()
+    this.emit("open")
+  }
+
   /** メッセージを送る。未接続なら no-op (logger.warn)。 */
   send(obj) {
     if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {