@cocorograph/hub-agent 0.7.93 → 0.7.95

package/package.json

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

package/src/codex-stream-bridge.mjs

@@ -333,6 +333,7 @@ export class CodexStreamBridge extends EventEmitter {
     }
     /** @type {Map<string, object>} stream_id -> session record */
     this.sessions = new Map()
+    this._accountRateLimitsByProfile = new Map()
   }
 
   /**
@@ -787,6 +788,27 @@ export class CodexStreamBridge extends EventEmitter {
     }
   }
 
+  /**
+   * アカウント単位の利用制限を取得する。thread を起動しないため、全セッション停止中でも
+   * Cockpit フッターを最新化できる。呼び出し側は認証済みプロファイルの場合だけ呼ぶ。
+   */
+  async readAccountRateLimits({ cwd, profileId = "default" } = {}) {
+    const client = this._clientFactory({ cwd: cwd || process.cwd(), codexCmd: this.codexCmd, logger: this.logger })
+    client.start()
+    try {
+      await client.request("initialize", { clientInfo: CLIENT_INFO })
+      const result = await client.request("account/rateLimits/read", {})
+      this._accountRateLimitsByProfile.set(profileId, result)
+      return result
+    } catch (err) {
+      const cached = this._accountRateLimitsByProfile.get(profileId)
+      if (cached) return cached
+      throw err
+    } finally {
+      client.stop()
+    }
+  }
+
   _wireClient(session) {
     const { client } = session
     client.on("notification", (msg) => this._onNotification(session, msg))

package/src/main.mjs

@@ -118,6 +118,7 @@ import { SiblingInjectBridge } from "./sibling-inject.mjs"
 import {
   boundSessionId,
   codexContextPercentFromTokenUsage,
+  codexUsageFromAccountRateLimits,
   codexLiveContextForProducer,
   codexSessionUsages,
   codexTurnActiveForThread,
@@ -130,6 +131,8 @@ import {
   jsonlContextForCwd,
   recordChatRateLimit,
   recordChatUsageSnapshot,
+  readAccountUsageSnapshot,
+  recordAccountUsageSnapshot,
   recordCodexLiveTokenUsage,
   turnActiveForCwd,
 } from "./usage.mjs"
@@ -3076,6 +3079,50 @@ export function handleClaudeChatInput(msg, ctx) {
     })
 }
 
+/**
+ * claude.attach の ready 直後にpushする履歴を組み立てる。
+ * Browserからの claude.history.request 往復を省くため、resume時だけ先回りして読む。
+ * 失敗しても空のhistory.responseへ縮退し、attach自体を失敗させない。
+ */
+export async function loadClaudeAttachHistory({
+  cwd,
+  sessionId,
+  logger,
+  getProjectsRootFn = getActiveProjectsRoot,
+  fetchHistoryFn = fetchSessionHistory,
+}) {
+  if (!cwd || !sessionId) return null
+  try {
+    const projectsRoot = await getProjectsRootFn()
+    const result = await fetchHistoryFn({
+      cwd,
+      session_id: sessionId,
+      maxLines: 500,
+      projectsRoot,
+      logger,
+    })
+    return {
+      type: "claude.history.response",
+      session_id: sessionId,
+      cwd,
+      events: result.events || [],
+      total_lines: result.total_lines || 0,
+      truncated: result.truncated === true,
+      error: result.error,
+    }
+  } catch (err) {
+    return {
+      type: "claude.history.response",
+      session_id: sessionId,
+      cwd,
+      events: [],
+      total_lines: 0,
+      truncated: false,
+      error: err?.message || String(err),
+    }
+  }
+}
+
 /**
  * SDK チャット配達保証 (S2, 2026-07-05): `codex.input` を処理し、browser が付与した
  * `input_id` があれば `codex.input.ack` / `codex.input.nack` を返す (claude.input と
@@ -3494,6 +3541,16 @@ async function dispatch(msg, ctx) {
           // model / permissionMode は agent.json の値も使えるよう fallback。
           const cwdArg = msg.cwd || ctx.config?.claude_cwd || undefined
           const resumeSessionId = msg.resume_session_id || null
+          // attachは同期処理なので、履歴I/Oを先に開始してreadyまでの待ちを重ねる。
+          // 新Browserはhistory_pushedを見て別requestを省略し、旧Browserは従来どおり
+          // requestしてもhydrateのuuid dedupで安全に収束する。
+          const historyPromise = resumeSessionId
+            ? loadClaudeAttachHistory({
+                cwd: cwdArg || process.env.HOME || process.cwd(),
+                sessionId: resumeSessionId,
+                logger: ctx.logger,
+              })
+            : Promise.resolve(null)
           const info = ctx.claudeBridge.attach({
             stream_id,
             cwd: cwdArg,
@@ -3517,12 +3574,25 @@ async function dispatch(msg, ctx) {
             type: "claude.ready",
             stream_id,
             resuming: info.resuming,
+            history_pushed: resumeSessionId !== null,
             // id 空間統一 (2026-07-08): この agent は browser の input_id をキュー id として
             // 採用し、楽観送信待ちアイテムの確認前キャンセルに対応することを通知する capability。
             // 旧 agent にはこのフラグが無く、browser は楽観アイテムの×を出さない (ゴースト
             // 実行防止のフォールバック)。
             supports_queue_input_id: true,
           })
+          // dispatch全体はWSメッセージ順を守る直列キュー上にある。ここで履歴I/Oをawaitすると
+          // 同一接続のpermission/input等まで停止するため、readyの同期送信後は非同期でpushする。
+          void historyPromise
+            .then((pushedHistory) => {
+              if (pushedHistory) ctx.client.send({ ...pushedHistory, stream_id })
+            })
+            .catch((err) => {
+              ctx.logger?.warn?.(
+                { err: err?.message || String(err), stream_id },
+                "failed to push Claude attach history",
+              )
+            })
           // S1 (2026-07): セッション状態の identity 正本化。TUI の claude.tui.bind
           // (main.mjs 内 clearChatSignal 呼び出し箇所参照) と対称に、SDK チャットの
           // attach 時点でも前セッションのチャット信号を引き継がない。
@@ -5565,6 +5635,10 @@ async function dispatch(msg, ctx) {
       }
       case "usage.request": {
         try {
+          const activeClaudeProfile = await getActiveProfile()
+          const claudeHasCredentials = isKeychainPlatform()
+            ? await profileHasKeychainCredentials(activeClaudeProfile.configDir)
+            : await hasFileCredentials(activeClaudeProfile.configDir)
           // 5h/7d の本命ソース: SDK control request (アカウント全体、session 非依存)。
           // statusLine が rate_limits を出さなくなった環境 (2026-07 時点の Claude CLI) でも
           // ここでライブ更新できる。frontend の usage.request ポーリング (既定 60s) 頻度に
@@ -5574,15 +5648,20 @@ async function dispatch(msg, ctx) {
           if (ctx.claudeBridge) {
             try {
               const acct = await ctx.claudeBridge.getAccountUsage()
-              if (acct) recordChatUsageSnapshot(acct)
+              if (acct) recordChatUsageSnapshot(acct, Date.now(), activeClaudeProfile.id)
             } catch (err) {
               ctx.logger?.debug?.({ err: err?.message }, "account usage snapshot skipped")
             }
           }
-          const [usage, sessions] = await Promise.all([
-            getUsage(),
+          let [usage, sessions] = await Promise.all([
+            getUsage({ profileId: activeClaudeProfile.id }),
             getSessionUsages(),
           ])
+          // 未ログインで残ったローカル transcript/cache をアカウント利用量として見せない。
+          if (!claudeHasCredentials) {
+            usage = null
+            sessions = []
+          }
 
           // Codex 版 (Stream B3, 2026-07-03): rollout jsonl の token_count イベントから
           // 5h/7d + context% を取る。⚠️ Codex の 5h/7d は ChatGPT/OpenAI サブスクリプション
@@ -5591,7 +5670,9 @@ async function dispatch(msg, ctx) {
           // として返す (frontend 側でどちらのアカウントの残量か混同しないため)。
           let codexUsage = null
           let codexSessions = []
-          if (ctx.codexBridge && ctx.codexBridge.sessions.size > 0) {
+          const activeCodexProfile = await codexProfiles.getActiveProfile()
+          const codexHasCredentials = await codexProfiles.hasCodexCredentials(activeCodexProfile.configDir)
+          if (ctx.codexBridge && codexHasCredentials) {
             try {
               // T06407 増分2 (axis2): per-cwd の context% は **cwd ごとに権威 producer 1 件**へ
               // 集約する。旧実装は全 live session を codexSessions へ流していたため、同一 cwd に
@@ -5641,10 +5722,24 @@ async function dispatch(msg, ctx) {
                   codexUsage = { ...codexUsage, context: null }
                 }
               }
+              // ライブ session の有無にかかわらず、起動直後の account RPC を優先する。
+              // rollout は context の補助だけにし、5h/7d はプロファイル固有の最新実値にする。
+              const accountUsage = codexUsageFromAccountRateLimits(
+                await ctx.codexBridge.readAccountRateLimits({ profileId: activeCodexProfile.id }),
+              )
+              if (accountUsage) {
+                codexUsage = { ...accountUsage, context: codexUsage?.context ?? null }
+                recordAccountUsageSnapshot("codex", activeCodexProfile.id, codexUsage)
+              }
             } catch (err) {
               ctx.logger?.debug?.({ err: err?.message }, "codex usage snapshot skipped")
             }
           }
+          // Agent 再起動後や一時的な account RPC 失敗時も、同一 profile の最終取得値だけを
+          // 表示する。未ログインならここへ到達せず、表示も完全に隠す。
+          if (codexHasCredentials && !codexUsage) {
+            codexUsage = await readAccountUsageSnapshot("codex", activeCodexProfile.id)
+          }
 
           ctx.client.send({
             type: "usage.response",

package/src/usage.mjs

@@ -292,12 +292,26 @@ export function whenChatRateLimitsPersisted() {
 // (旧 chatRateLimits の設計的弱点)。専用ファイルなら誰にも上書きされない。
 // ---------------------------------------------------------------------------
 
-function chatUsageCachePath() {
-  return configPath("HUB_CHAT_USAGE_CACHE", os.homedir(), ".hub", "usage", "chat-usage.json")
+function profileCacheKey(profileId = "default") {
+  // Profile id は UI から作られるが、キャッシュパスに使う前に必ず無害な文字列へ
+  // 正規化する。プロフィール間の値が混ざるより、読めない cache を無視する方が安全。
+  const value = String(profileId || "default")
+  return value.replace(/[^a-zA-Z0-9._-]/g, "_").slice(0, 120) || "default"
+}
+
+function chatUsageCachePath(profileId = "default") {
+  const configured = process.env.HUB_CHAT_USAGE_CACHE
+  if (configured) {
+    // 既存の環境変数は default profile の互換パスとして維持する。追加 profile は
+    // 同じディレクトリ内に別ファイルで置き、最後に利用した別 profile を読まない。
+    if (profileId === "default") return configured
+    return path.join(path.dirname(configured), `chat-usage-${profileCacheKey(profileId)}.json`)
+  }
+  return path.join(os.homedir(), ".hub", "usage", "chat-usage", `${profileCacheKey(profileId)}.json`)
 }
 
 /** @type {{five_hour:{percent:number,resetAtMs:number|null}|null, seven_day:{percent:number,resetAtMs:number|null}|null, plan:string|null, updatedAtMs:number}|null} */
-let chatUsageSnapshot = null
+let chatUsageSnapshots = new Map()
 let _chatUsagePersistInFlight = Promise.resolve()
 
 /** SDKControlGetUsageResponse.rate_limits.{five_hour,seven_day} の 1 窓を正規化する。 */
@@ -322,8 +336,9 @@ function normalizeRateLimitWindow(w) {
  *
  * @param {{rate_limits_available?: boolean, rate_limits?: {five_hour?: object|null, seven_day?: object|null}|null, subscription_type?: string|null}|null} usageResponse
  * @param {number} [now]
+ * @param {string} [profileId]
  */
-export function recordChatUsageSnapshot(usageResponse, now = Date.now()) {
+export function recordChatUsageSnapshot(usageResponse, now = Date.now(), profileId = "default") {
   if (!usageResponse || typeof usageResponse !== "object") return
   if (usageResponse.rate_limits_available === false) return
   const rl = usageResponse.rate_limits
@@ -331,24 +346,25 @@ export function recordChatUsageSnapshot(usageResponse, now = Date.now()) {
   const five = normalizeRateLimitWindow(rl.five_hour)
   const seven = normalizeRateLimitWindow(rl.seven_day)
   if (!five && !seven) return
-  chatUsageSnapshot = {
+  const previous = chatUsageSnapshots.get(profileId)
+  chatUsageSnapshots.set(profileId, {
     five_hour: five,
     seven_day: seven,
-    plan: usageResponse.subscription_type || chatUsageSnapshot?.plan || null,
+    plan: usageResponse.subscription_type || previous?.plan || null,
     updatedAtMs: now,
-  }
+  })
   // P5(bug) と同じ直列化パターン (recordChatRateLimit 参照): 並行 persist の tmp 破壊/
   // 中間状態読みを防ぐため、常に前回の persist 完了を待ってから次を実行する。
   _chatUsagePersistInFlight = _chatUsagePersistInFlight.then(
-    () => persistChatUsageSnapshot(),
-    () => persistChatUsageSnapshot(),
+    () => persistChatUsageSnapshot(profileId),
+    () => persistChatUsageSnapshot(profileId),
   )
 }
 
-async function persistChatUsageSnapshot() {
-  const snapshot = chatUsageSnapshot
+async function persistChatUsageSnapshot(profileId) {
+  const snapshot = chatUsageSnapshots.get(profileId)
   if (!snapshot) return
-  const p = chatUsageCachePath()
+  const p = chatUsageCachePath(profileId)
   try {
     await fs.mkdir(path.dirname(p), { recursive: true })
   } catch {
@@ -373,13 +389,17 @@ export function whenChatUsageSnapshotPersisted() {
 }
 
 /** プロセス内メモリを優先し、無ければディスク永続化 (hub-agent 再起動直後等) を読む。 */
-async function readChatUsageSnapshot() {
-  if (chatUsageSnapshot) return chatUsageSnapshot
-  const text = await readOrNull(chatUsageCachePath())
+async function readChatUsageSnapshot(profileId = "default") {
+  const inMemory = chatUsageSnapshots.get(profileId)
+  if (inMemory) return inMemory
+  const text = await readOrNull(chatUsageCachePath(profileId))
   if (!text) return null
   try {
     const j = JSON.parse(text)
-    if (j && typeof j === "object" && typeof j.updatedAtMs === "number") return j
+    if (j && typeof j === "object" && typeof j.updatedAtMs === "number") {
+      chatUsageSnapshots.set(profileId, j)
+      return j
+    }
   } catch {
     /* ignore */
   }
@@ -388,7 +408,7 @@ async function readChatUsageSnapshot() {
 
 /** テスト用: chatUsageSnapshot をクリアする。 */
 export function _resetChatUsageSnapshot() {
-  chatUsageSnapshot = null
+  chatUsageSnapshots = new Map()
 }
 
 async function readOfficial(now) {
@@ -1129,7 +1149,7 @@ export function _resetTurnActiveMemo() {
  * context%: statusLine 値が最新 jsonl より新しければ official を優先。
  * 古い / 無い (= チャットモードで statusLine 未更新) 場合は jsonl 推定で上書き。
  */
-export async function getUsage() {
+export async function getUsage({ profileId = "default" } = {}) {
   const now = Date.now()
   const official = await readOfficial(now)
   const result = official || (await readEstimate(now))
@@ -1184,28 +1204,30 @@ export async function getUsage() {
   // chatRateLimits (rate_limit_event) より本命の情報源 (utilization 欠落問題が無い、
   // 2026-07 時点で実測確認済み)。statusLine / rate_limit_event のどちらよりも新しければ
   // 上書きする。5h/7d が実際に埋まったときだけ source を official に昇格させる。
-  const chatSnap = await readChatUsageSnapshot()
+  const chatSnap = await readChatUsageSnapshot(profileId)
   if (chatSnap && chatSnap.updatedAtMs > 0) {
-    const baseline = Math.max(cacheMtime ?? 0, chatRateLimits.updatedAtMs || 0)
-    // `>=` (上と同じ理由): ファイル mtime とプロセス内タイムスタンプがタイになり得る。
-    // chatSnap は最も直接的な情報源 (SDK control request 実値) なのでタイ時は優先する。
-    if (chatSnap.updatedAtMs >= baseline) {
-      if (chatSnap.five_hour) {
-        result.last5h = {
-          ...result.last5h,
-          percent: chatSnap.five_hour.percent,
-          resetAtMs: chatSnap.five_hour.resetAtMs ?? result.last5h?.resetAtMs ?? null,
-        }
+    // statusLine/latest.json と rate_limit_event は profile を持たない旧経路。別 profile
+    // の更新時刻がより新しくても、profile 専用 SDK snapshot を負かしてはならない。
+    if (chatSnap.five_hour) {
+      result.last5h = {
+        ...result.last5h,
+        percent: chatSnap.five_hour.percent,
+        resetAtMs: chatSnap.five_hour.resetAtMs ?? result.last5h?.resetAtMs ?? null,
       }
-      if (chatSnap.seven_day) {
-        result.last7d = {
-          ...result.last7d,
-          percent: chatSnap.seven_day.percent,
-          resetAtMs: chatSnap.seven_day.resetAtMs ?? result.last7d?.resetAtMs ?? null,
-        }
+    }
+    if (chatSnap.seven_day) {
+      result.last7d = {
+        ...result.last7d,
+        percent: chatSnap.seven_day.percent,
+        resetAtMs: chatSnap.seven_day.resetAtMs ?? result.last7d?.resetAtMs ?? null,
       }
-      if (chatSnap.plan) result.plan = chatSnap.plan
-      if (chatSnap.five_hour || chatSnap.seven_day) result.source = "official"
+    }
+    if (chatSnap.plan) result.plan = chatSnap.plan
+    if (chatSnap.five_hour || chatSnap.seven_day) {
+      result.source = "official"
+      // スナップショットを再表示しただけでは「今取得した」ことにしない。フッターの
+      // 最終更新時刻は実アカウント値を得た時刻を示す。
+      result.computedAt = chatSnap.updatedAtMs
     }
   }
   return result
@@ -1594,6 +1616,99 @@ export function classifyCodexRateLimitWindows(snapshot) {
   return result
 }
 
+/** App Server のアカウント利用制限レスポンスを Cockpit wire 形式へ正規化する。 */
+export function codexUsageFromAccountRateLimits(response) {
+  const rateLimits = response?.rateLimits
+  if (!rateLimits || typeof rateLimits !== "object") return null
+  const { last5h, last7d } = classifyCodexRateLimitWindows(rateLimits)
+  return {
+    plan: rateLimits.planType || "codex",
+    computedAt: Date.now(),
+    source: "official",
+    context: null,
+    last5h,
+    last7d,
+  }
+}
+
+// Codex app-server の account/rateLimits/read はセッション不要だが、一時的な CLI 障害や
+// hub-agent 再起動直後には読めないことがある。最後に確認できた値を provider + profile
+// ごとに保存し、別アカウントの残量を表示しないようにする。
+function accountUsageCachePath(provider, profileId = "default") {
+  const root = process.env.HUB_ACCOUNT_USAGE_CACHE_DIR || path.join(os.homedir(), ".hub", "usage", "accounts")
+  return path.join(root, profileCacheKey(provider), `${profileCacheKey(profileId)}.json`)
+}
+
+const accountUsageSnapshots = new Map()
+let _accountUsagePersistInFlight = Promise.resolve()
+
+function accountUsageSnapshotKey(provider, profileId) {
+  return `${profileCacheKey(provider)}:${profileCacheKey(profileId)}`
+}
+
+function isUsageSnapshot(value) {
+  return (
+    value &&
+    typeof value === "object" &&
+    typeof value.computedAt === "number" &&
+    typeof value.source === "string" &&
+    value.last5h &&
+    value.last7d
+  )
+}
+
+/** 最後に取得できたアカウント利用量を profile 単位で永続化する。 */
+export function recordAccountUsageSnapshot(provider, profileId, usage) {
+  if (!isUsageSnapshot(usage)) return
+  const key = accountUsageSnapshotKey(provider, profileId)
+  accountUsageSnapshots.set(key, usage)
+  _accountUsagePersistInFlight = _accountUsagePersistInFlight.then(
+    () => persistAccountUsageSnapshot(provider, profileId, usage),
+    () => persistAccountUsageSnapshot(provider, profileId, usage),
+  )
+}
+
+async function persistAccountUsageSnapshot(provider, profileId, usage) {
+  const target = accountUsageCachePath(provider, profileId)
+  try {
+    await fs.mkdir(path.dirname(target), { recursive: true })
+    const tmp = `${target}.tmp.${randomUUID()}`
+    try {
+      await fs.writeFile(tmp, JSON.stringify(usage))
+      await fs.rename(tmp, target)
+    } catch {
+      await fs.unlink(tmp).catch(() => {})
+    }
+  } catch {
+    // 表示用キャッシュの書込み失敗は利用セッションを妨げない。
+  }
+}
+
+/** メモリ優先、hub-agent 再起動後は provider/profile 専用のディスク値を返す。 */
+export async function readAccountUsageSnapshot(provider, profileId = "default") {
+  const key = accountUsageSnapshotKey(provider, profileId)
+  const inMemory = accountUsageSnapshots.get(key)
+  if (inMemory) return inMemory
+  const text = await readOrNull(accountUsageCachePath(provider, profileId))
+  if (!text) return null
+  try {
+    const parsed = JSON.parse(text)
+    if (!isUsageSnapshot(parsed)) return null
+    accountUsageSnapshots.set(key, parsed)
+    return parsed
+  } catch {
+    return null
+  }
+}
+
+export function whenAccountUsageSnapshotPersisted() {
+  return _accountUsagePersistInFlight
+}
+
+export function _resetAccountUsageSnapshots() {
+  accountUsageSnapshots.clear()
+}
+
 // ---------------------------------------------------------------------------
 // Codex context%（ドーナツ）の live source（T06407 増分2, 症状⑥ ドーナツ前後動の根治）
 //