@cocorograph/hub-agent 0.7.94 → 0.7.95

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cocorograph/hub-agent",
-  "version": "0.7.94",
+  "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"
@@ -5632,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) 頻度に
@@ -5641,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 サブスクリプション
@@ -5658,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 に
@@ -5708,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, 症状⑥ ドーナツ前後動の根治）
 //