@cocorograph/hub-agent 0.5.13 → 0.5.14

package/package.json

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

package/src/pty-bridge.mjs

@@ -20,6 +20,14 @@ import { runHookChain } from "./plugin-loader.mjs"
 const DEFAULT_COLS = 120
 const DEFAULT_ROWS = 32
 
+// pty.onData の chunk を coalesce する待ち時間 (ms)。
+// 0 だと chunk 1 つにつき 1 WebSocket メッセージ + 1 JSON.stringify になり、
+// claude TUI の連続 redraw で秒間 100+ メッセージ → Hub → channels_redis pub/sub
+// → ブラウザ JSON.parse が詰まって瞬断・固まりの原因になる。12ms バッファに
+// 溜めて 80Hz 程度に丸めるとメッセージ数が 1/5〜1/10 に減り、画面は十分滑らか
+// に見える (1 フレーム 16.7ms より短いので体感レイテンシは増えない)。
+const DEFAULT_COALESCE_MS = 12
+
 function resolveBin(name) {
   try {
     return execFileSync("/usr/bin/which", [name], { encoding: "utf-8" }).trim()
@@ -43,8 +51,9 @@ export class PtyBridge extends EventEmitter {
    * @param {(args: {sessionName: string, cols: number, rows: number, env: object}) => {command: string, args: string[], env?: object}} [opts.defaultSpawnCommand]
    *   - plugin が null を返したときに使うデフォルト spawn 仕様。省略時は
    *     `/bin/sh -c "exec tmux attach -t <sessionName>"`
+   * @param {number} [opts.coalesceMs] - pty 出力を coalesce する間隔 (ms)。0 で無効。
    */
-  constructor({ ptyModule, logger, plugins = [], defaultSpawnCommand } = {}) {
+  constructor({ ptyModule, logger, plugins = [], defaultSpawnCommand, coalesceMs } = {}) {
     super()
     if (!ptyModule || typeof ptyModule.spawn !== "function") {
       throw new TypeError("PtyBridge requires { ptyModule: { spawn } }")
@@ -59,8 +68,30 @@ export class PtyBridge extends EventEmitter {
         args: ["-c", `exec ${tmuxBin()} attach -t ${sessionName}`],
         env: process.env,
       }))
+    this.coalesceMs = coalesceMs ?? DEFAULT_COALESCE_MS
     /** @type {Map<string, import('node-pty').IPty>} */
     this.streams = new Map()
+    /** @type {Map<string, {buf: string, timer: ReturnType<typeof setTimeout>|null}>} */
+    this.coalesceState = new Map()
+  }
+
+  /**
+   * stream の coalesce buffer を即時 flush して emit する。
+   * detach / exit / shutdown のタイミングで残バッファの取りこぼしを防ぐ。
+   * @param {string} stream_id
+   */
+  _flushCoalesce(stream_id) {
+    const state = this.coalesceState.get(stream_id)
+    if (!state) return
+    if (state.timer) {
+      clearTimeout(state.timer)
+      state.timer = null
+    }
+    if (state.buf) {
+      const out = state.buf
+      state.buf = ""
+      this.emit("output", { stream_id, data: out })
+    }
   }
 
   /**
@@ -108,10 +139,30 @@ export class PtyBridge extends EventEmitter {
       "pty attached",
     )
 
-    pty.onData((data) => {
-      this.emit("output", { stream_id, data })
-    })
+    if (this.coalesceMs > 0) {
+      const state = { buf: "", timer: null }
+      this.coalesceState.set(stream_id, state)
+      pty.onData((data) => {
+        state.buf += data
+        if (!state.timer) {
+          state.timer = setTimeout(() => {
+            state.timer = null
+            const out = state.buf
+            if (!out) return
+            state.buf = ""
+            this.emit("output", { stream_id, data: out })
+          }, this.coalesceMs)
+        }
+      })
+    } else {
+      pty.onData((data) => {
+        this.emit("output", { stream_id, data })
+      })
+    }
     pty.onExit(({ exitCode }) => {
+      // 残バッファを取りこぼさないよう flush してから exit を emit する。
+      this._flushCoalesce(stream_id)
+      this.coalesceState.delete(stream_id)
       this.streams.delete(stream_id)
       this.emit("exit", { stream_id, code: exitCode })
     })
@@ -154,6 +205,9 @@ export class PtyBridge extends EventEmitter {
   detach({ stream_id }) {
     const pty = this.streams.get(stream_id)
     if (!pty) return false
+    // kill 前に残バッファを emit しておく (onExit にも flush はあるが、
+    // detach は browser 側都合なので最新を確実に届けたい)。
+    this._flushCoalesce(stream_id)
     try {
       pty.kill()
     } catch {