@cocorograph/hub-agent 0.5.16 → 0.5.18

package/package.json

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

package/src/main.mjs

@@ -17,7 +17,7 @@ import path from "node:path"
 
 import pino from "pino"
 
-import { readConfig } from "./config.mjs"
+import { readConfig, writeConfig } from "./config.mjs"
 import { loadPlugins, runHookBroadcast, runHookChain } from "./plugin-loader.mjs"
 import { WsClient } from "./ws-client.mjs"
 import { PtyBridge } from "./pty-bridge.mjs"
@@ -169,6 +169,40 @@ export async function startDaemon({ version, ptyModule } = {}) {
 const SESSION_EVENTS_DIR =
   process.env.COCKPIT_SESSION_EVENTS_DIR || "/tmp/cockpit_session_events"
 
+/**
+ * `/tmp/cockpit_session_events/` 全 .json を読んで session_name → {event, at}
+ * の Map を返す (Phase 4: tmux.list_sessions レスポンスに含めて配信する用)。
+ *
+ * fs.watch 由来の `session.event` push はファイル変更時のみのため、frontend が
+ * 新規マウント (ハードリロード等) すると過去 event を取得できず CockpitStatusDot
+ * が全グレー表示になる問題があった。これを list 同梱で復元する。
+ */
+async function readAllSessionEvents() {
+  const out = new Map()
+  let files
+  try {
+    files = await readdir(SESSION_EVENTS_DIR)
+  } catch {
+    return out
+  }
+  for (const f of files) {
+    if (!f.endsWith(".json")) continue
+    const sessionName = f.slice(0, -5)
+    try {
+      const text = await readFile(path.join(SESSION_EVENTS_DIR, f), "utf-8")
+      const data = JSON.parse(text)
+      if (!data || typeof data.event !== "string") continue
+      out.set(sessionName, {
+        event: data.event,
+        at: typeof data.at === "number" ? data.at : Date.now(),
+      })
+    } catch {
+      // 一時欠如 / parse 失敗は無視
+    }
+  }
+  return out
+}
+
 /**
  * `/tmp/cockpit_session_events/<tmux_session>.json` を fs.watch して
  * UserPromptSubmit / Stop の event を `session.event` 型で Hub に push する。
@@ -302,15 +336,56 @@ function startStateLoop({ client, plugins, logger, intervalMs }) {
  * `pty.*` は PtyBridge にそのまま流す。`tmux.exec` は Sprint E で tmux 抽象に
  * 渡す予定なので、現状は plugin hook chain だけ走らせて log に残す。
  */
+/**
+ * Hub から `ready` / `ack` で配信された Claude CLI 設定を agent.json と
+ * in-memory config に反映する (Phase 3 live-sync)。
+ *
+ * 旧設計 (Phase 2) では enrollment 時のみ取得していたため、cockpit 管理画面で
+ * 設定を変更しても既存 agent には再 enroll しない限り反映されなかった。Phase 3 で
+ * 全 WS 接続/heartbeat 毎に最新値が同期されるようにした。
+ *
+ * 値が変化していない場合は writeConfig を呼ばない (書込負荷軽減)。
+ * 失敗しても agent は止めず log だけ出して継続。
+ */
+export async function maybeSyncClaudeSettings(msg, ctx) {
+  if (typeof msg?.claude_model !== "string" && typeof msg?.claude_permission_mode !== "string") {
+    return
+  }
+  const nextModel = typeof msg.claude_model === "string" ? msg.claude_model : ctx.config.claude_model || ""
+  const nextMode = typeof msg.claude_permission_mode === "string" ? msg.claude_permission_mode : ctx.config.claude_permission_mode || ""
+  const prevModel = ctx.config.claude_model || ""
+  const prevMode = ctx.config.claude_permission_mode || ""
+  if (nextModel === prevModel && nextMode === prevMode) return
+  ctx.config.claude_model = nextModel
+  ctx.config.claude_permission_mode = nextMode
+  try {
+    await writeConfig({
+      agent_id: ctx.config.agent_id,
+      agent_token: ctx.config.agent_token,
+      hub_url: ctx.config.hub_url,
+      claude_model: nextModel,
+      claude_permission_mode: nextMode,
+    })
+    ctx.logger.info(
+      { claude_model: nextModel, claude_permission_mode: nextMode },
+      "claude settings synced from hub",
+    )
+  } catch (err) {
+    ctx.logger.warn({ err: err.message }, "failed to persist claude settings to agent.json")
+  }
+}
+
 async function dispatch(msg, ctx) {
   const t = msg?.type || ""
   try {
     switch (t) {
       case "ready":
         ctx.logger.info({ msg }, "hub ready")
+        await maybeSyncClaudeSettings(msg, ctx)
         return
       case "ack":
         ctx.logger.debug({ echo: msg.echo }, "hub ack")
+        await maybeSyncClaudeSettings(msg, ctx)
         return
       case "error":
         ctx.logger.warn({ msg }, "hub error")
@@ -396,10 +471,19 @@ async function dispatch(msg, ctx) {
       case "tmux.list_sessions": {
         try {
           const sessions = await listTmuxSessions({ plugins: ctx.plugins, logger: ctx.logger })
+          // 各 session の最新 hook event を /tmp/cockpit_session_events/ から読んで
+          // レスポンスに含める (Phase 4: ハードリロード時の lastEvent 復元用)。
+          // fs.watch 由来の session.event push は揮発性のため、frontend が新規
+          // マウントすると過去 event を取れず全グレー表示になる事象を解消する。
+          const lastEventByName = await readAllSessionEvents()
+          const enriched = sessions.map((s) => ({
+            ...s,
+            last_event: lastEventByName.get(s.name) || null,
+          }))
           ctx.client.send({
             type: "tmux.sessions",
             request_id: msg.request_id,
-            sessions,
+            sessions: enriched,
           })
         } catch (err) {
           ctx.client.send({