@cocorograph/hub-agent 0.6.20 → 0.6.22

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cocorograph/hub-agent",
-  "version": "0.6.20",
+  "version": "0.6.22",
   "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/claude-stream-bridge.mjs

@@ -902,7 +902,13 @@ export class ClaudeStreamBridge extends EventEmitter {
       logger: this.logger,
       onEvent: (event) => {
         // stream_id は再アタッチで変わるため session.stream_id (最新) を使う。
-        this.emit("event", { stream_id: session.stream_id, session_id: session.sessionId, event })
+        // cwd は per-session の context% ドーナツ書き出し (usage.mjs) で使う。
+        this.emit("event", {
+          stream_id: session.stream_id,
+          session_id: session.sessionId,
+          cwd: session.cwd,
+          event,
+        })
         // session_id 確定後は索引に登録 (冪等)。再接続時の再アタッチに使う。
         if (session.sessionId) this._liveBySession.set(session.sessionId, session)
       },

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,6 +38,7 @@ import {
   removeWorktree as removeWorktreeDir,
 } from "./tmux.mjs"
 import { getSessionUsages, getUsage, recordChatRateLimit } from "./usage.mjs"
+import { getChatSignal, recordChatActivity } from "./chat-signals.mjs"
 
 const logger = pino({ name: "hub-agent" })
 
@@ -178,7 +179,7 @@ export async function startDaemon({ version, ptyModule, claudeSdk } = {}) {
   // そのまま browser に転送する。SDK 未インストール時は claudeBridge=null で全 attach が
   // claude.error を返す経路に分岐 (dispatch 側で判定)。
   if (claudeBridge) {
-    claudeBridge.on("event", ({ stream_id, session_id, event }) => {
+    claudeBridge.on("event", ({ stream_id, session_id, cwd, event }) => {
       // SDK の rate_limit_event を捕捉して usage の 5h/7d 実値に反映する
       // (チャットモードでは statusLine が更新されないため、これが live 値の唯一の源)。
       if (event?.type === "rate_limit_event" && event.rate_limit_info) {
@@ -188,6 +189,26 @@ export async function startDaemon({ version, ptyModule, claudeSdk } = {}) {
           /* ignore */
         }
       }
+      // Hub ホスト型 Cockpit 用: SDK event から per-session の状態をチャット信号として
+      // 記録する。startStateLoop が cwd 一致で session.state (ステータスドット/各行
+      // ドーナツ) と session.event (ソート) に橋渡しする。チャットは tmux ペインに
+      // TUI を出さず、capture-pane スクレイプも bundle hook も発火しないための補完。
+      //   - assistant (生成中) → processing + usage から context%
+      //   - result   (ターン完了/入力待ち) → waiting
+      if (event?.type === "assistant") {
+        const pct = event.message?.usage ? contextPctFromUsage(event.message.usage) : null
+        try {
+          recordChatActivity(cwd, { status: "processing", contextPct: pct })
+        } catch {
+          /* ignore */
+        }
+      } else if (event?.type === "result") {
+        try {
+          recordChatActivity(cwd, { status: "waiting" })
+        } catch {
+          /* ignore */
+        }
+      }
       client.send({ type: "claude.event", stream_id, session_id, event })
     })
     claudeBridge.on("permission", ({ stream_id, session_id, request_id, tool_name, input }) => {
@@ -298,6 +319,48 @@ export async function startDaemon({ version, ptyModule, claudeSdk } = {}) {
 const SESSION_EVENTS_DIR =
   process.env.COCKPIT_SESSION_EVENTS_DIR || "/tmp/cockpit_session_events"
 
+// context 窓サイズ (トークン)。usage.mjs と同じ既定。1M ベータ等は env で上書き。
+const CONTEXT_WINDOW_TOKENS = Number(process.env.HUB_CONTEXT_WINDOW) || 200000
+
+/**
+ * SDK assistant メッセージの usage から context 使用率 (%) を概算する。
+ * context = input + cache_read + cache_creation + output(直近応答)。
+ * プロンプトキャッシュ有効時は cache_* に大半が乗るため 4 種を合算する。
+ */
+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 / CONTEXT_WINDOW_TOKENS) * 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 レスポンスに含めて配信する用)。
@@ -414,6 +477,7 @@ 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 () => {
@@ -421,21 +485,39 @@ function startStateLoop({ client, plugins, logger, intervalMs }) {
     try {
       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

@@ -124,9 +124,72 @@ function utilizationToPercent(u) {
   return u <= 1 ? Math.round(u * 1000) / 10 : Math.round(u * 10) / 10
 }
 
+// persist の in-flight promise。テストやシャットダウンで書き込み完了を待てるよう保持。
+let _persistInFlight = Promise.resolve()
+
+/**
+ * 捕捉済みの chatRateLimits を statusLine cache (`~/.hub/usage/latest.json`) に
+ * マージ書き込みする。
+ *
+ * webapp 側 (`cockpit/webapp/lib/usage.ts`) のフッターは hub-agent の getUsage を
+ * 呼ばず、このファイルを直接読む (readOfficial)。そのファイルは TUI の statusLine
+ * でしか更新されないため、チャットモードでは 5h/7d が固まる。ここで rate_limit_event
+ * の実値をファイルへ書き戻すことで、webapp のフッターもチャット中にライブ更新される。
+ *
+ * 既存フィールド (context_window 等、TUI が書いた値) は保持し、rate_limits のみ
+ * 更新する。書き込みは tmp → rename の atomic write。全エラーは握りつぶす。
+ */
+async function persistChatRateLimitsToCache() {
+  const p = statuslineCache()
+  let base = {}
+  const existing = await readOrNull(p)
+  if (existing) {
+    try {
+      const parsed = JSON.parse(existing)
+      if (parsed && typeof parsed === "object") base = parsed
+    } catch {
+      /* 壊れたファイルは作り直す */
+    }
+  }
+  const rl =
+    base.rate_limits && typeof base.rate_limits === "object" ? { ...base.rate_limits } : {}
+  if (chatRateLimits.five_hour) {
+    rl.five_hour = {
+      used_percentage: chatRateLimits.five_hour.percent,
+      resets_at: chatRateLimits.five_hour.resetAtMs ?? rl.five_hour?.resets_at ?? null,
+    }
+  }
+  if (chatRateLimits.seven_day) {
+    rl.seven_day = {
+      used_percentage: chatRateLimits.seven_day.percent,
+      resets_at: chatRateLimits.seven_day.resetAtMs ?? rl.seven_day?.resets_at ?? null,
+    }
+  }
+  base.rate_limits = rl
+  try {
+    await fs.mkdir(path.dirname(p), { recursive: true })
+  } catch {
+    /* ignore */
+  }
+  const tmp = `${p}.tmp.${process.pid}`
+  try {
+    await fs.writeFile(tmp, JSON.stringify(base))
+    await fs.rename(tmp, p)
+  } catch {
+    try {
+      await fs.unlink(tmp)
+    } catch {
+      /* ignore */
+    }
+  }
+}
+
 /**
  * SDK の SDKRateLimitInfo を取り込む (main.mjs の rate_limit_event ハンドラから呼ぶ)。
  * rateLimitType: 'five_hour' / 'seven_day*' を 5h / 7d スロットに振り分ける。
+ *
+ * 取り込んだ値はプロセス内 (hub-agent の getUsage 用) と statusLine cache ファイル
+ * (webapp フッター用) の両方に反映する。ファイル書き込みは fire-and-forget。
  */
 export function recordChatRateLimit(info) {
   if (!info || typeof info !== "object") return
@@ -141,6 +204,17 @@ export function recordChatRateLimit(info) {
   }
   chatRateLimits[slot] = { percent, resetAtMs }
   chatRateLimits.updatedAtMs = Date.now()
+  // webapp フッター (ファイルベース readOfficial) 用に latest.json へ書き戻す。
+  // fire-and-forget だが in-flight promise は保持 (flush 可能にする)。
+  _persistInFlight = persistChatRateLimitsToCache()
+}
+
+/**
+ * 直近の statusLine cache 書き込みの完了を待つ。テストや graceful shutdown 用。
+ * @returns {Promise<void>}
+ */
+export function whenChatRateLimitsPersisted() {
+  return _persistInFlight
 }
 
 async function readOfficial(now) {