@cocorograph/hub-agent 0.5.6 → 0.5.8

package/package.json

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

package/src/main.mjs

@@ -10,6 +10,7 @@
  *
  * 仕様書: ナレッジ/インフラ/cockpit-hub-hosted-integration-spec (id=6080)
  */
+import { readFileSync } from "node:fs"
 import { readFile } from "node:fs/promises"
 import os from "node:os"
 import path from "node:path"
@@ -53,6 +54,20 @@ async function readBundleVersion() {
   }
 }
 
+/**
+ * WsClient の runtime refresh / fs.watch コールバックから同期で呼ばれる版。
+ * 失敗時は null を返す (silent fail)。
+ */
+function readBundleVersionSync() {
+  try {
+    const text = readFileSync(BUNDLE_MANIFEST_PATH, "utf-8")
+    const data = JSON.parse(text)
+    return typeof data?.version === "string" ? data.version : null
+  } catch {
+    return null
+  }
+}
+
 export async function startDaemon({ version, ptyModule } = {}) {
   const config = await readConfig()
   if (!config) {
@@ -74,7 +89,13 @@ export async function startDaemon({ version, ptyModule } = {}) {
     logger.info({ bundleVersion }, "hub bundle version detected")
   }
 
-  const client = new WsClient(config, { logger, version, bundleVersion })
+  const client = new WsClient(config, {
+    logger,
+    version,
+    bundleVersion,
+    bundleVersionProvider: readBundleVersionSync,
+    bundleManifestPath: BUNDLE_MANIFEST_PATH,
+  })
 
   // EventEmitter の 'error' は listener が無いと process が落ちる。
   // ws-client は close 側で reconnect を予約しているので、ここでは log だけ。
@@ -298,6 +319,7 @@ async function dispatch(msg, ctx) {
         try {
           await createTmuxSession(name, cwd, {
             claudeCmd: typeof msg.claude_cmd === "string" ? msg.claude_cmd : undefined,
+            logger: ctx.logger,
           })
           ctx.client.send({
             type: "tmux.create_session.result",
@@ -342,6 +364,7 @@ async function dispatch(msg, ctx) {
           // worktree dir で tmux session を作成 (claude_cmd は createSession の default)
           await createTmuxSession(wtName, wtPath, {
             claudeCmd: typeof msg.claude_cmd === "string" ? msg.claude_cmd : undefined,
+            logger: ctx.logger,
           })
           ctx.client.send({
             type: "worktree.create.result",

package/src/tmux.mjs

@@ -33,6 +33,20 @@ function sanitizeTmuxName(s) {
   return s.replace(/[.:]/g, "-")
 }
 
+/**
+ * 先頭の `~` / `~/` を `os.homedir()` に展開する。
+ *
+ * tmux は `-c` に渡された cwd をシェル展開しないため、フロントから
+ * 受け取った `~/hub/projects/...` を絶対パスに変換する必要がある。
+ * 既に絶対パス / 相対パスなら noop。
+ */
+export function expandTilde(p) {
+  if (typeof p !== "string" || p.length === 0) return p
+  if (p === "~") return os.homedir()
+  if (p.startsWith("~/")) return path.join(os.homedir(), p.slice(2))
+  return p
+}
+
 /**
  * git branch 名から worktree dir 名 (= tmux session 名) を導出する。
  * 例: "feature/multi-area" -> "feature-multi-area"
@@ -70,7 +84,10 @@ async function branchExists(repoDir, branch) {
  * 移植元: D00000_cockpit/webapp/lib/worktrees.ts (createWorktree)
  */
 export async function createWorktreeDir(parentDir, branch) {
-  const repoDir = path.resolve(HUB_PROJECTS_BASE, parentDir)
+  // parentDir に `~/...` 形式が来ても受け付けられるよう先に展開する。
+  // 展開後が絶対パスなら path.resolve は HUB_PROJECTS_BASE を無視して
+  // そのまま使う (Node の path.resolve 仕様)。
+  const repoDir = path.resolve(HUB_PROJECTS_BASE, expandTilde(parentDir))
   if (repoDir !== HUB_PROJECTS_BASE && !repoDir.startsWith(HUB_PROJECTS_BASE + path.sep)) {
     throw new Error("dir outside projects base")
   }
@@ -302,6 +319,20 @@ export async function killManySessions(names, opts = {}) {
  * - opts.claudeCmd で send-keys 内容を上書き可 (空文字なら claude 自動起動しない)
  */
 export async function createSession(name, cwd, opts = {}) {
+  // tmux は `-c` をシェル展開しないため、ここで `~` を絶対パスに展開する。
+  // 展開後の cwd は同名引数で以降の検証 / tmux に渡す。
+  const resolvedCwd = expandTilde(cwd)
+  // cwd 存在チェック (旧 Cockpit 単体実装と同等の防御)
+  // 無いまま tmux に渡すと pane の current_path が HOME など別ディレクトリに
+  // fallback して、SessionStart hook が project を解決できなくなる。
+  try {
+    const st = await fs.stat(resolvedCwd)
+    if (!st.isDirectory()) throw new Error("cwd not a directory")
+  } catch (err) {
+    if (err?.code === "ENOENT") throw new Error(`cwd not found: ${resolvedCwd}`)
+    if (err?.message === "cwd not a directory") throw err
+    throw new Error(`cwd stat failed: ${err?.message || String(err)}`)
+  }
   // 既存チェック
   try {
     await execFileP(tmuxBin(opts), ["has-session", "-t", name])
@@ -311,7 +342,30 @@ export async function createSession(name, cwd, opts = {}) {
     if (msg === "duplicate session") throw err
     // has-session が非 0 = セッション無し
   }
-  await execFileP(tmuxBin(opts), ["new-session", "-d", "-s", name, "-c", cwd])
+  await execFileP(tmuxBin(opts), ["new-session", "-d", "-s", name, "-c", resolvedCwd])
+  // 起動直後 pane の current_path を確認し、想定 cwd と異なれば WARN ログを残す
+  // (Phase 3: 任意検証。tmux 側のクオート / 権限問題で fallback した場合を検知する)
+  if (opts.logger) {
+    try {
+      const { stdout } = await execFileP(tmuxBin(opts), [
+        "display-message",
+        "-p",
+        "-t",
+        `${name}:`,
+        "-F",
+        "#{pane_current_path}",
+      ])
+      const actual = (stdout || "").trim()
+      if (actual && actual !== resolvedCwd) {
+        opts.logger.warn(
+          { session: name, requested: resolvedCwd, actual },
+          "tmux pane cwd differs from requested cwd",
+        )
+      }
+    } catch {
+      // display-message 失敗は致命的でないので飲み込む
+    }
+  }
   const claudeCmd = opts.claudeCmd ?? DEFAULT_CLAUDE_CMD
   if (claudeCmd) {
     await execFileP(tmuxBin(opts), ["send-keys", "-t", name, claudeCmd, "Enter"])

package/src/ws-client.mjs

@@ -5,10 +5,14 @@
  * - 起動時に `hello`、30s おきに `heartbeat` を送信
  * - 切断時は exponential backoff (1s, 2s, 4s, ..., max 30s) で再接続
  * - サーバから受け取った JSON は `onMessage` callback に渡す
+ * - `bundleVersionProvider` + `bundleManifestPath` を渡すと、manifest.json の
+ *   変更を fs.watch で検知して即時 heartbeat を送信し Cockpit UI に最新版を
+ *   リアルタイム反映する (heartbeat 30s 待ちを回避)
  *
  * 仕様書: ナレッジ/インフラ/cockpit-hub-hosted-integration-spec (id=6080)
  */
 import { EventEmitter } from "node:events"
+import fs from "node:fs"
 import os from "node:os"
 
 import WebSocket from "ws"
@@ -16,11 +20,19 @@ import WebSocket from "ws"
 const HEARTBEAT_INTERVAL_MS = 30_000
 const MIN_BACKOFF_MS = 1_000
 const MAX_BACKOFF_MS = 30_000
+const BUNDLE_WATCH_DEBOUNCE_MS = 500
 
 export class WsClient extends EventEmitter {
   /**
    * @param {{ hub_url: string, agent_id: string, agent_token: string }} config
-   * @param {{ logger?: import('pino').Logger, version?: string, bundleVersion?: string|null, hostname?: string }} opts
+   * @param {{
+   *   logger?: import('pino').Logger,
+   *   version?: string,
+   *   bundleVersion?: string|null,
+   *   bundleVersionProvider?: () => (string|null),
+   *   bundleManifestPath?: string|null,
+   *   hostname?: string,
+   * }} opts
    */
   constructor(config, opts = {}) {
     super()
@@ -28,6 +40,13 @@ export class WsClient extends EventEmitter {
     this.logger = opts.logger
     this.version = opts.version || "0.1.0"
     this.bundleVersion = opts.bundleVersion || null
+    this.bundleVersionProvider =
+      typeof opts.bundleVersionProvider === "function"
+        ? opts.bundleVersionProvider
+        : null
+    this.bundleManifestPath = opts.bundleManifestPath || null
+    this.bundleWatcher = null
+    this.bundleWatchDebounceTimer = null
     this.hostname = opts.hostname || os.hostname()
     this.ws = null
     this.heartbeatTimer = null
@@ -52,6 +71,8 @@ export class WsClient extends EventEmitter {
     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,
@@ -60,6 +81,7 @@ export class WsClient extends EventEmitter {
         bundle_version: this.bundleVersion,
       })
       this._startHeartbeat()
+      this._startBundleWatcher()
       this.emit("open")
     })
 
@@ -76,6 +98,7 @@ export class WsClient extends EventEmitter {
 
     ws.on("close", (code, reason) => {
       this._stopHeartbeat()
+      this._stopBundleWatcher()
       this.logger?.info({ code, reason: reason?.toString() }, "ws close")
       this.emit("close", { code, reason })
       if (!this.stopped) this._scheduleReconnect()
@@ -101,6 +124,7 @@ export class WsClient extends EventEmitter {
   stop() {
     this.stopped = true
     this._stopHeartbeat()
+    this._stopBundleWatcher()
     if (this.reconnectTimer) {
       clearTimeout(this.reconnectTimer)
       this.reconnectTimer = null
@@ -138,6 +162,9 @@ export class WsClient extends EventEmitter {
   _startHeartbeat() {
     this._stopHeartbeat()
     this.heartbeatTimer = setInterval(() => {
+      // heartbeat 都度 provider 経由で bundle version を最新化する。
+      // fs.watch を取り逃した変更 (atomic rename / 別マウント越し等) への保険。
+      this._refreshBundleVersion()
       // 送信失敗 (= ws not OPEN) を検知したら即時 close + reconnect 発火。
       // setInterval を放置すると死んだコネクションに heartbeat を投げ続けて
       // 30s 間気付かないので、close を強制トリガーする。
@@ -162,6 +189,99 @@ export class WsClient extends EventEmitter {
     }
   }
 
+  /**
+   * bundleVersionProvider を呼んで `this.bundleVersion` を最新化する。
+   * 値が変わったら true を返す。provider が未設定 / 失敗時は false。
+   */
+  _refreshBundleVersion() {
+    if (!this.bundleVersionProvider) return false
+    try {
+      const next = this.bundleVersionProvider()
+      const normalized = typeof next === "string" && next.length > 0 ? next : null
+      if (normalized === this.bundleVersion) return false
+      this.logger?.info(
+        { from: this.bundleVersion, to: normalized },
+        "bundle version refreshed",
+      )
+      this.bundleVersion = normalized
+      return true
+    } catch (err) {
+      this.logger?.warn({ err: err.message }, "bundle version refresh failed")
+      return false
+    }
+  }
+
+  /**
+   * `bundleManifestPath` を fs.watch で監視し、変更時に即時 heartbeat を送る。
+   * Cockpit UI への反映ラグを 30s heartbeat 周期から ~15s (UI poll) に短縮する。
+   * fs.watch の起動に失敗しても heartbeat 都度 refresh が保険として動くので no-op。
+   */
+  _startBundleWatcher() {
+    if (!this.bundleManifestPath || this.bundleWatcher) return
+    try {
+      this.bundleWatcher = fs.watch(
+        this.bundleManifestPath,
+        { persistent: false },
+        () => this._onBundleManifestChange(),
+      )
+      this.bundleWatcher.on("error", (err) => {
+        this.logger?.warn(
+          { err: err.message, path: this.bundleManifestPath },
+          "bundle manifest watcher errored",
+        )
+      })
+      this.logger?.debug(
+        { path: this.bundleManifestPath },
+        "bundle manifest watcher started",
+      )
+    } catch (err) {
+      this.logger?.warn(
+        { err: err.message, path: this.bundleManifestPath },
+        "bundle manifest watcher failed to start",
+      )
+    }
+  }
+
+  _stopBundleWatcher() {
+    if (this.bundleWatchDebounceTimer) {
+      clearTimeout(this.bundleWatchDebounceTimer)
+      this.bundleWatchDebounceTimer = null
+    }
+    if (this.bundleWatcher) {
+      try {
+        this.bundleWatcher.close()
+      } catch {
+        /* ignore */
+      }
+      this.bundleWatcher = null
+    }
+  }
+
+  /**
+   * fs.watch のコールバック。同一書き込みで多重 fire することがある (macOS) ため
+   * デバウンスしてから refresh + 即時 heartbeat する。
+   */
+  _onBundleManifestChange() {
+    if (this.bundleWatchDebounceTimer) {
+      clearTimeout(this.bundleWatchDebounceTimer)
+    }
+    this.bundleWatchDebounceTimer = setTimeout(() => {
+      this.bundleWatchDebounceTimer = null
+      const changed = this._refreshBundleVersion()
+      if (!changed) return
+      const ok = this._sendJson({
+        type: "heartbeat",
+        uptime_sec: Math.floor((Date.now() - this.startedAt) / 1000),
+        bundle_version: this.bundleVersion,
+      })
+      if (!ok) {
+        this.logger?.warn("bundle-change heartbeat send failed, forcing reconnect")
+        this._forceReconnect()
+      }
+    }, BUNDLE_WATCH_DEBOUNCE_MS)
+    this.bundleWatchDebounceTimer.unref?.()
+  }
+
   _forceReconnect() {
     // 現在の ws を強制 close。close ハンドラが _scheduleReconnect を呼ぶので
     // それ以上の処理は不要。stopped 中は no-op。