@cocorograph/hub-agent 0.6.21 → 0.6.24

package/package.json

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

package/src/chat-signals.mjs

@@ -0,0 +1,85 @@
+/**
+ * チャット(SDK stream-json)モードの per-session 状態を、Hub ホスト型 Cockpit の
+ * サイドバー信号 (session.state / session.event) に橋渡しするための in-process ストア。
+ *
+ * 背景:
+ *   Hub ホスト型 Cockpit のサイドバーは以下を「TUI 専用シグナル」から導出している:
+ *     - ステータスドット / 各行 context%: hub-agent の startStateLoop が
+ *       detectSessionState (tmux capture-pane スクレイプ) で算出して session.state を push
+ *     - 時系列ソート: bundle hook (cockpit_session_event_hook.py) が
+ *       /tmp/cockpit_session_events/<tmux名>.json を書き、hub-agent が watch して
+ *       session.event を push → frontend の eventAt sort key
+ *   チャットモードは tmux ペインに Claude TUI を描画せず、hook も TUI 経由でしか
+ *   発火しないため、これらが更新されずドット/ドーナツ/ソートが固まる。
+ *
+ * 解決:
+ *   claude-stream-bridge の SDK event (assistant / result) を main.mjs で捕捉し、
+ *   ここに cwd キーで status / context_pct / 最終アクティビティ時刻を記録する。
+ *   startStateLoop が tmux セッションを走査する際、session の cwd に一致する新鮮な
+ *   チャット信号があれば status / context_pct を上書きし、turnAt 更新時には
+ *   session-event ファイルを書いてソートを動かす。
+ *
+ * cwd をキーにする理由: チャットセッションは tmux セッション名を持たず (SDK 子
+ * プロセス)、サイドバー行 (= tmux セッション) とは cwd で対応するため。
+ */
+
+// cwd → { status, context_pct, turnAt, updatedAtMs }
+const _byCwd = new Map()
+
+// チャット信号を「生きている」と見なす最大経過時間 (ms)。これを過ぎたら tmux
+// スクレイプ結果へフォールバックさせる (チャット終了/別モード移行の検知)。
+const CHAT_SIGNAL_STALE_MS = 15 * 60 * 1000
+
+const VALID_STATUS = new Set(["processing", "waiting", "idle"])
+
+function _now() {
+  return Date.now()
+}
+
+/**
+ * チャットのアクティビティを記録する。status / context_pct は与えられた分だけ更新し、
+ * turnAt (= 最終アクティビティ時刻 / ソート用) は呼ばれるたびに前進させる。
+ *
+ * @param {string} cwd セッションの作業ディレクトリ
+ * @param {{ status?: string, contextPct?: number|null }} patch
+ */
+export function recordChatActivity(cwd, patch = {}) {
+  if (!cwd || typeof cwd !== "string") return
+  const now = _now()
+  const prev = _byCwd.get(cwd) || {
+    status: null,
+    context_pct: null,
+    turnAt: 0,
+    updatedAtMs: 0,
+  }
+  const next = { ...prev, turnAt: now, updatedAtMs: now }
+  if (typeof patch.status === "string" && VALID_STATUS.has(patch.status)) {
+    next.status = patch.status
+  }
+  if (typeof patch.contextPct === "number" && Number.isFinite(patch.contextPct)) {
+    next.context_pct = Math.max(0, Math.min(100, patch.contextPct))
+  }
+  _byCwd.set(cwd, next)
+}
+
+/**
+ * cwd に対応する生きているチャット信号を返す。stale / 不在なら null。
+ *
+ * @param {string} cwd
+ * @param {number} [now]
+ * @returns {{ status: string|null, context_pct: number|null, turnAt: number, updatedAtMs: number } | null}
+ */
+export function getChatSignal(cwd, now = _now()) {
+  if (!cwd) return null
+  const sig = _byCwd.get(cwd)
+  if (!sig) return null
+  if (now - sig.updatedAtMs > CHAT_SIGNAL_STALE_MS) return null
+  return sig
+}
+
+/** テスト用: ストアを空にする。 */
+export function _resetChatSignals() {
+  _byCwd.clear()
+}
+
+export { CHAT_SIGNAL_STALE_MS }

package/src/main.mjs

@@ -11,7 +11,7 @@
  * 仕様書: ナレッジ/インフラ/cockpit-hub-hosted-integration-spec (id=6080)
  */
 import { readFileSync, watch as fsWatch } from "node:fs"
-import { mkdir, readFile, readdir } from "node:fs/promises"
+import { mkdir, readFile, readdir, rename, unlink, writeFile } from "node:fs/promises"
 import os from "node:os"
 import path from "node:path"
 
@@ -38,12 +38,12 @@ import {
   removeWorktree as removeWorktreeDir,
 } from "./tmux.mjs"
 import {
+  contextWindowSize,
   getSessionUsages,
   getUsage,
   recordChatRateLimit,
-  recordChatSessionContext,
-  recordChatSessionStatus,
 } from "./usage.mjs"
+import { getChatSignal, recordChatActivity } from "./chat-signals.mjs"
 
 const logger = pino({ name: "hub-agent" })
 
@@ -194,30 +194,22 @@ export async function startDaemon({ version, ptyModule, claudeSdk } = {}) {
           /* ignore */
         }
       }
-      // assistant メッセージの usage から session 別 context% を算出してファイルへ
-      // 書き出す (サイドバー各行のドーナツ用。これも statusLine がチャットでは更新
-      // されないための補完)。あわせてターン状態を更新する:
-      //   - assistant (生成中) → processing
+      // Hub ホスト型 Cockpit 用: SDK event から per-session の状態をチャット信号として
+      // 記録する。startStateLoop が cwd 一致で session.state (ステータスドット/各行
+      // ドーナツ) と session.event (ソート) に橋渡しする。チャットは tmux ペインに
+      // TUI を出さず、capture-pane スクレイプも bundle hook も発火しないための補完。
+      //   - assistant (生成中) → processing + usage から context%
       //   - result   (ターン完了/入力待ち) → waiting
-      // webapp のステータスドットは tmux ペインのスクレイプで判定するが、チャットは
-      // ペインに TUI を出さないため、この chat_status を優先させて補完する。
-      const sid = session_id || event?.session_id
       if (event?.type === "assistant") {
-        if (event.message?.usage) {
-          try {
-            recordChatSessionContext({ sessionId: sid, cwd, message: event.message })
-          } catch {
-            /* ignore */
-          }
-        }
+        const pct = event.message?.usage ? contextPctFromUsage(event.message.usage) : null
         try {
-          recordChatSessionStatus({ sessionId: sid, cwd, status: "processing" })
+          recordChatActivity(cwd, { status: "processing", contextPct: pct })
         } catch {
           /* ignore */
         }
       } else if (event?.type === "result") {
         try {
-          recordChatSessionStatus({ sessionId: sid, cwd, status: "waiting" })
+          recordChatActivity(cwd, { status: "waiting" })
         } catch {
           /* ignore */
         }
@@ -332,6 +324,71 @@ export async function startDaemon({ version, ptyModule, claudeSdk } = {}) {
 const SESSION_EVENTS_DIR =
   process.env.COCKPIT_SESSION_EVENTS_DIR || "/tmp/cockpit_session_events"
 
+// context 窓サイズ (トークン) の既定。env HUB_CONTEXT_WINDOW があれば最優先 (手動上書き)。
+// 無ければ起動後に statusLine cache の context_window_size (Opus 4.8 の 1M 等) を
+// refreshContextWindow() で解決して resolvedContextWindow を上書きする。
+const CONTEXT_WINDOW_FALLBACK = Number(process.env.HUB_CONTEXT_WINDOW) || 200000
+
+// 実際に context% 計算で使う窓サイズ。startStateLoop の tick で定期リフレッシュされる。
+// 200k 固定だと 1M ベータのセッションで分母が 5 倍小さく、ドーナツが常に振り切れる
+// (273k トークン → 137% を 100% にクランプ表示) 不具合があったため、実窓サイズを使う。
+let resolvedContextWindow = CONTEXT_WINDOW_FALLBACK
+
+/**
+ * 実コンテキスト窓サイズ (1M ベータ等) を statusLine cache から解決して
+ * resolvedContextWindow を更新する。env HUB_CONTEXT_WINDOW 指定時は手動上書きを
+ * 尊重して何もしない。ファイル読み失敗時は前回値を保持する。
+ */
+export async function refreshContextWindow() {
+  if (process.env.HUB_CONTEXT_WINDOW) return // 手動上書きが最優先
+  try {
+    const size = await contextWindowSize()
+    if (typeof size === "number" && size > 0) resolvedContextWindow = size
+  } catch {
+    /* 前回値を保持 */
+  }
+}
+
+/**
+ * SDK assistant メッセージの usage から context 使用率 (%) を概算する。
+ * context = input + cache_read + cache_creation + output(直近応答)。
+ * プロンプトキャッシュ有効時は cache_* に大半が乗るため 4 種を合算する。
+ * 分母は resolvedContextWindow (実窓サイズ。1M ベータ等を自動反映)。
+ */
+export function contextPctFromUsage(u) {
+  if (!u) return null
+  const tokens =
+    (u.input_tokens || 0) +
+    (u.cache_read_input_tokens || 0) +
+    (u.cache_creation_input_tokens || 0) +
+    (u.output_tokens || 0)
+  if (tokens <= 0) return null
+  return Math.min(100, Math.round((tokens / resolvedContextWindow) * 1000) / 10)
+}
+
+/**
+ * sort 用の session-event ファイルをチャットのターン境界で書き出す
+ * (`/tmp/cockpit_session_events/<tmux名>.json`)。bundle hook が TUI で書くものと
+ * 同形式 (`{ event, at }`)。既存の watcher がこれを拾って session.event を push し、
+ * frontend の eventAt sort key を更新する。atomic write、全エラー握りつぶし。
+ */
+async function writeSessionEventFile(sessionName, event, at) {
+  if (!sessionName || /[/\\]/.test(sessionName)) return
+  const fp = path.join(SESSION_EVENTS_DIR, `${sessionName}.json`)
+  const tmp = `${fp}.tmp.${process.pid}`
+  try {
+    await mkdir(SESSION_EVENTS_DIR, { recursive: true })
+    await writeFile(tmp, JSON.stringify({ event, at }))
+    await rename(tmp, fp)
+  } catch {
+    try {
+      await unlink(tmp)
+    } catch {
+      /* ignore */
+    }
+  }
+}
+
 /**
  * `/tmp/cockpit_session_events/` 全 .json を読んで session_name → {event, at}
  * の Map を返す (Phase 4: tmux.list_sessions レスポンスに含めて配信する用)。
@@ -448,28 +505,50 @@ async function startSessionEventWatcher({ client, logger }) {
  */
 function startStateLoop({ client, plugins, logger, intervalMs }) {
   const lastByName = new Map() // session_name → {status, context_pct}
+  const lastTurnAtByName = new Map() // session_name → 最後に event ファイル化した turnAt
   let stopped = false
 
   const tick = async () => {
     if (stopped) return
     try {
+      // 実コンテキスト窓サイズ (1M ベータ等) を反映。contextPctFromUsage の分母を
+      // 5s ごとに最新化し、ドーナツが 200k 固定で振り切れるのを防ぐ。
+      await refreshContextWindow()
       const states = await listSessionStates({ plugins, logger })
       for (const s of states) {
+        let status = s.status
+        let contextPct = s.context_pct
+        // チャットモード補完: tmux ペインのスクレイプは TUI 前提で効かないため、
+        // session の cwd に一致する新鮮なチャット信号があれば status/context% を
+        // 上書きする (Hub ホスト型 Cockpit のステータスドット/各行ドーナツ用)。
+        const chat = s.cwd ? getChatSignal(s.cwd) : null
+        if (chat) {
+          if (chat.status) status = chat.status
+          if (typeof chat.context_pct === "number") contextPct = chat.context_pct
+          // チャットのターン境界 (turnAt 前進) を sort 用 session-event に橋渡し。
+          // tmux ペインが動かず bundle hook が発火しないため、ここで代替発火する。
+          const prevTurnAt = lastTurnAtByName.get(s.session_name) || 0
+          if (chat.turnAt > prevTurnAt) {
+            lastTurnAtByName.set(s.session_name, chat.turnAt)
+            const ev = chat.status === "processing" ? "prompt_submit" : "stop"
+            writeSessionEventFile(s.session_name, ev, chat.turnAt).catch(() => {})
+          }
+        }
         const prev = lastByName.get(s.session_name)
         if (
           !prev ||
-          prev.status !== s.status ||
-          prev.context_pct !== s.context_pct
+          prev.status !== status ||
+          prev.context_pct !== contextPct
         ) {
           lastByName.set(s.session_name, {
-            status: s.status,
-            context_pct: s.context_pct,
+            status,
+            context_pct: contextPct,
           })
           client.send({
             type: "session.state",
             session_name: s.session_name,
-            status: s.status,
-            context_pct: s.context_pct,
+            status,
+            context_pct: contextPct,
           })
         }
       }

package/src/state.mjs

@@ -83,6 +83,28 @@ export async function capturePane(sessionName, opts = {}) {
   }
 }
 
+/**
+ * tmux session の active pane の current_path を取得する (chat 信号との cwd 照合用)。
+ * 取れなければ null。
+ */
+export async function getSessionCwd(sessionName, opts = {}) {
+  const tmuxBin = opts.tmuxBin || "tmux"
+  try {
+    const { stdout } = await execFileP(tmuxBin, [
+      "display-message",
+      "-p",
+      "-t",
+      `${sessionName}:`,
+      "-F",
+      "#{pane_current_path}",
+    ])
+    const s = stdout.trim()
+    return s || null
+  } catch {
+    return null
+  }
+}
+
 /**
  * 1 session の現在状態を取得する。plugin hook で上書き可。
  *
@@ -111,14 +133,17 @@ export async function detectSessionState(sessionName, opts = {}) {
   return { status: defaultStatus, context_pct: defaultContextPct }
 }
 
-/** 全 session の現在状態を取得する。 */
+/** 全 session の現在状態を取得する。cwd も含める (chat 信号照合用)。 */
 export async function listSessionStates(opts = {}) {
   const names = await listSessionNames(opts)
   return Promise.all(
-    names.map(async (name) => ({
-      session_name: name,
-      ...(await detectSessionState(name, opts)),
-    })),
+    names.map(async (name) => {
+      const [state, cwd] = await Promise.all([
+        detectSessionState(name, opts),
+        getSessionCwd(name, opts),
+      ])
+      return { session_name: name, cwd, ...state }
+    }),
   )
 }
 

package/src/usage.mjs

@@ -217,130 +217,6 @@ export function whenChatRateLimitsPersisted() {
   return _persistInFlight
 }
 
-// session 別ファイル書き込みの in-flight promise (テスト/shutdown 用)。
-// context% (assistant) と chat_status (result) の書き込みは read-modify-write の
-// ため、この chain に直列化して lost-update / 競合を防ぐ。
-let _sessionPersistInFlight = Promise.resolve()
-
-/** session_id を安全なファイル名としてバリデーション (path traversal 防止)。 */
-function isSafeSessionId(sid) {
-  return typeof sid === "string" && /^[A-Za-z0-9_-]{8,64}$/.test(sid)
-}
-
-const VALID_CHAT_STATUS = new Set(["processing", "waiting", "idle"])
-
-/**
- * session 別ファイル (`~/.hub/usage/sessions/<sid>.json`) へ patch を
- * マージ書き込みする (read-modify-write, atomic rename)。
- *
- * webapp サイドバーは getSessionUsages (context ドーナツ) と listSessions
- * (ステータスドット) がこのディレクトリを読むが、ファイルは TUI statusLine
- * でしか更新されないため、チャットモードでは両方が固まる。チャット中は SDK の
- * usage / busy 状態からここへ書き出すことで両方ライブ更新される。
- *
- * context_window と chat_status は別タイミング (assistant / result) で書かれる
- * ため、既存値を破壊しないよう deep merge する。
- */
-async function mergeSessionFile(sessionId, cwd, patch) {
-  if (!isSafeSessionId(sessionId) || !cwd) return
-  const dir = statuslineSessionsDir()
-  const fp = path.join(dir, `${sessionId}.json`)
-  let base = {}
-  const existing = await readOrNull(fp)
-  if (existing) {
-    try {
-      const p = JSON.parse(existing)
-      if (p && typeof p === "object") base = p
-    } catch {
-      /* 壊れたファイルは作り直す */
-    }
-  }
-  const next = {
-    ...base,
-    ...patch,
-    session_id: sessionId,
-    workspace: { ...(base.workspace || {}), project_dir: cwd },
-    _source: "chat",
-  }
-  // context_window はネストオブジェクトなので明示マージ。
-  if (patch.context_window) {
-    next.context_window = { ...(base.context_window || {}), ...patch.context_window }
-  }
-  try {
-    await fs.mkdir(dir, { recursive: true })
-  } catch {
-    /* ignore */
-  }
-  const tmp = `${fp}.tmp.${process.pid}`
-  try {
-    await fs.writeFile(tmp, JSON.stringify(next))
-    await fs.rename(tmp, fp)
-  } catch {
-    try {
-      await fs.unlink(tmp)
-    } catch {
-      /* ignore */
-    }
-  }
-}
-
-/** session 別ファイル書き込みを直列 chain に積む。 */
-function enqueueSessionWrite(fn) {
-  _sessionPersistInFlight = _sessionPersistInFlight.catch(() => {}).then(fn)
-  return _sessionPersistInFlight
-}
-
-/**
- * チャット assistant メッセージを取り込んで session 別 context% を更新する
- * (main.mjs の event ハンドラから assistant メッセージで呼ぶ)。fire-and-forget。
- *
- * context トークン = input + cache_read + cache_creation + output(直近応答)。
- * プロンプトキャッシュ有効時は cache_* に大半が乗るため 4 種を必ず合算する。
- *
- * @param {{ sessionId?: string, cwd?: string, message?: { usage?: object } }} arg
- */
-export function recordChatSessionContext(arg) {
-  if (!arg || typeof arg !== "object") return
-  const { sessionId, cwd, message } = arg
-  const u = message?.usage
-  if (!u) return
-  const tokens =
-    (u.input_tokens || 0) +
-    (u.cache_read_input_tokens || 0) +
-    (u.cache_creation_input_tokens || 0) +
-    (u.output_tokens || 0)
-  if (tokens <= 0) return
-  enqueueSessionWrite(async () => {
-    const windowSize = await contextWindowSize()
-    const percent = Math.min(100, Math.round((tokens / windowSize) * 1000) / 10)
-    await mergeSessionFile(sessionId, cwd, { context_window: { used_percentage: percent } })
-  })
-}
-
-/**
- * チャットのターン状態 (processing/waiting/idle) を session 別ファイルへ書き出す。
- * main.mjs の event ハンドラから、assistant→processing / result→waiting で呼ぶ。
- * webapp listSessions が tmux スクレイプの代わりにこれを優先する (チャットモードは
- * ペインに TUI を出さずスクレイプが効かないため)。fire-and-forget。
- *
- * @param {{ sessionId?: string, cwd?: string, status?: string }} arg
- */
-export function recordChatSessionStatus(arg) {
-  if (!arg || typeof arg !== "object") return
-  const { sessionId, cwd, status } = arg
-  if (!VALID_CHAT_STATUS.has(status)) return
-  enqueueSessionWrite(() => mergeSessionFile(sessionId, cwd, { chat_status: status }))
-}
-
-/**
- * 直近の session 別ファイル書き込み (context%/status 両方) の完了を待つ。
- * テスト/shutdown 用。
- * @returns {Promise<void>}
- */
-export function whenChatSessionContextPersisted() {
-  return _sessionPersistInFlight
-}
-
 async function readOfficial(now) {
   const text = await readOrNull(statuslineCache())
   if (!text) return null
@@ -484,7 +360,7 @@ async function statuslineCacheMtime() {
  * あればそれを使う (1M ベータ等を自動反映)。無ければ HUB_CONTEXT_WINDOW / 200000。
  * ※ cache が stale でも窓サイズ自体は不変なので利用できる。
  */
-async function contextWindowSize() {
+export async function contextWindowSize() {
   const text = await readOrNull(statuslineCache())
   if (text) {
     try {