@cocorograph/hub-agent 0.5.12 → 0.5.14

package/bin/hub-agent.mjs

@@ -34,7 +34,11 @@ import {
   uninstallPlugin,
 } from "../src/plugin-install.mjs"
 import { loadPlugins } from "../src/plugin-loader.mjs"
-import { installService, uninstallService } from "../src/service-install.mjs"
+import {
+  installService,
+  restartService,
+  uninstallService,
+} from "../src/service-install.mjs"
 
 const here = path.dirname(fileURLToPath(import.meta.url))
 const pkg = JSON.parse(readFileSync(path.join(here, "..", "package.json"), "utf-8"))
@@ -164,6 +168,24 @@ program
     }
   })
 
+program
+  .command("restart")
+  .description("install-service 済みの daemon を再起動 (macOS=launchctl kickstart / Linux=systemctl --user restart)")
+  .action(async () => {
+    try {
+      const r = await restartService()
+      console.log(`restarted: ${r.platform} ${r.path}`)
+      if (r.platform === "darwin") {
+        console.log(`label: ${r.label}`)
+      } else {
+        console.log(`unit: ${r.unit}`)
+      }
+    } catch (err) {
+      console.error(`restart failed: ${err.message}`)
+      process.exit(1)
+    }
+  })
+
 program
   .command("uninstall-service")
   .description("OS サービス登録を解除")

package/package.json

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

package/scripts/install.sh

@@ -30,14 +30,54 @@ present() { command -v "$1" >/dev/null 2>&1; }
 
 ensure_brew() {
   if [[ "$(uname)" != "Darwin" ]]; then return 0; fi
-  if present brew; then color_ok "brew already installed"; return 0; fi
-  color_step "Homebrew をインストール"
-  /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
-  # Apple Silicon の brew はデフォルト PATH に入らないので追加
-  if [[ -d /opt/homebrew/bin ]]; then
-    export PATH="/opt/homebrew/bin:$PATH"
+  if present brew; then
+    color_ok "brew already installed"
+  else
+    color_step "Homebrew をインストール"
+    /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"
+    # Apple Silicon の brew はデフォルト PATH に入らないので追加
+    if [[ -d /opt/homebrew/bin ]]; then
+      export PATH="/opt/homebrew/bin:$PATH"
+    fi
+    present brew || { color_err "brew install failed"; exit 1; }
   fi
-  present brew || { color_err "brew install failed"; exit 1; }
+  # Apple Silicon Mac は `/opt/homebrew/bin` がデフォルト PATH に入らないので、
+  # `~/.zprofile` / `~/.bash_profile` に `brew shellenv` の eval を **永続化** する。
+  # これを入れないと install.sh 終了後の新規ターミナルから `npm` / `node` /
+  # `hub-agent` が `command not found` になる (葛西氏 Mac で 0.5.11 リリース直後に踏んだ)。
+  persist_brew_shellenv
+}
+
+# Apple Silicon Mac で brew のパスを zsh / bash 永続化する。既に追記済みなら no-op。
+# `eval "$(/opt/homebrew/bin/brew shellenv)"` を書くことで、Homebrew が用意する
+# HOMEBREW_PREFIX / PATH / MANPATH / INFOPATH すべてを新規シェルで自動セットする。
+persist_brew_shellenv() {
+  local brew_bin=""
+  if [[ -x /opt/homebrew/bin/brew ]]; then
+    brew_bin="/opt/homebrew/bin/brew"
+  elif [[ -x /usr/local/bin/brew ]]; then
+    # Intel Mac は `/usr/local/bin` がデフォルト PATH に入るので追記不要
+    return 0
+  else
+    return 0
+  fi
+  local snippet="eval \"\$(${brew_bin} shellenv)\""
+  local marker="# >>> hub-agent: brew shellenv (Apple Silicon PATH) >>>"
+  local end_marker="# <<< hub-agent: brew shellenv <<<"
+  local target
+  for target in "$HOME/.zprofile" "$HOME/.bash_profile"; do
+    if [[ -f "$target" ]] && grep -Fq "$snippet" "$target"; then
+      color_ok "brew shellenv は既に $target にあります"
+      continue
+    fi
+    color_step "$target に brew shellenv を追記"
+    {
+      printf '\n%s\n' "$marker"
+      printf '%s\n' "$snippet"
+      printf '%s\n' "$end_marker"
+    } >> "$target"
+    color_ok "$target に追記しました (新規シェルから有効)"
+  done
 }
 
 ensure_pkg() {

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 {

package/src/service-install.mjs

@@ -105,6 +105,47 @@ export async function installService({ bin } = {}) {
   throw new Error(`unsupported platform: ${process.platform}`)
 }
 
+/**
+ * install-service 済みの hub-agent を再起動する。
+ *
+ * - macOS: `launchctl kickstart -k gui/<uid>/<label>` で SIGTERM + 再起動
+ * - Linux: `systemctl --user restart hub-agent.service`
+ *
+ * service が install されていない場合は分かりやすい error を投げる。
+ * (cockpit web UI から `hub-agent restart` 案内が出ても、enroll 直後の
+ *  ユーザーが叩くケースを想定)
+ */
+export async function restartService() {
+  if (process.platform === "darwin") {
+    const dest = macPlistPath()
+    try {
+      await fs.access(dest)
+    } catch {
+      throw new Error(
+        `service not installed (${dest} not found). run \`hub-agent install-service\` first`,
+      )
+    }
+    const uid = ensureUnixUid()
+    run("launchctl", ["kickstart", "-k", `gui/${uid}/${PLIST_LABEL}`])
+    return { platform: "darwin", path: dest, label: PLIST_LABEL }
+  }
+
+  if (process.platform === "linux") {
+    const dest = linuxUnitPath()
+    try {
+      await fs.access(dest)
+    } catch {
+      throw new Error(
+        `service not installed (${dest} not found). run \`hub-agent install-service\` first`,
+      )
+    }
+    run("systemctl", ["--user", "restart", SYSTEMD_UNIT_NAME])
+    return { platform: "linux", path: dest, unit: SYSTEMD_UNIT_NAME }
+  }
+
+  throw new Error(`unsupported platform: ${process.platform}`)
+}
+
 export async function uninstallService() {
   if (process.platform === "darwin") {
     const uid = ensureUnixUid()