@cocorograph/hub-agent 0.6.6 → 0.6.8

package/package.json

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

package/src/usage.mjs

@@ -55,6 +55,12 @@ function statuslineSessionsDir() {
 
 const SESSION_STALE_MS = 15 * 60 * 1000
 
+// コンテキスト窓サイズ (トークン)。Opus/Sonnet 4.x は 200k が標準。
+// 1M ベータ等を使う場合は HUB_CONTEXT_WINDOW で上書き。
+const CONTEXT_WINDOW = Number(process.env.HUB_CONTEXT_WINDOW) || 200000
+// チャットモード(SDK)の context% 推定で参照する jsonl の鮮度窓 (直近 N 分)。
+const CONTEXT_JSONL_RECENT_MS = 30 * 60 * 1000
+
 const PLAN_LIMITS = {
   pro: { msg5h: 45, msg7d: 315 },
   max_5x: { msg5h: 280, msg7d: 1960 },
@@ -230,15 +236,125 @@ async function readEstimate(now) {
   }
 }
 
+/** statusLine cache ファイルの mtime (ms)。存在しなければ 0。 */
+async function statuslineCacheMtime() {
+  try {
+    return (await fs.stat(statuslineCache())).mtimeMs
+  } catch {
+    return null
+  }
+}
+
+/**
+ * context 窓サイズ (トークン) を解決する。statusLine cache に context_window_size が
+ * あればそれを使う (1M ベータ等を自動反映)。無ければ HUB_CONTEXT_WINDOW / 200000。
+ * ※ cache が stale でも窓サイズ自体は不変なので利用できる。
+ */
+async function contextWindowSize() {
+  const text = await readOrNull(statuslineCache())
+  if (text) {
+    try {
+      const j = JSON.parse(text)
+      const cw = j.context_window ?? j.contextWindow
+      const size = cw?.context_window_size ?? cw?.contextWindowSize
+      if (typeof size === "number" && size > 0) return size
+    } catch {
+      /* ignore */
+    }
+  }
+  return CONTEXT_WINDOW
+}
+
+/**
+ * 直近にアクティブな session jsonl の最後の assistant.usage から context% を
+ * 推定する。チャットモード(SDK headless)は statusLine を書き出さないため、
+ * ドーナツ(usage.context)が固まるのを防ぐフォールバック。
+ *
+ * context トークン = input + cache_read + cache_creation (+ 直近応答の output)。
+ * プロンプトキャッシュ有効時は cache_creation / cache_read に大半が乗るため、
+ * input_tokens だけだと大幅に過小評価になる点に注意 (3 種を必ず合算する)。
+ *
+ * @returns {Promise<{ percent: number, mtimeMs: number } | null>}
+ */
+async function latestJsonlContext(now) {
+  const projects = await fs.readdir(projectsDir()).catch(() => null)
+  if (!projects) return null
+  const recent = now - CONTEXT_JSONL_RECENT_MS
+  let best = null // { mtimeMs, fp }
+  await Promise.all(
+    projects.map(async (p) => {
+      const dir = path.join(projectsDir(), p)
+      const files = await fs.readdir(dir).catch(() => [])
+      for (const f of files) {
+        if (!f.endsWith(".jsonl")) continue
+        const fp = path.join(dir, f)
+        try {
+          const st = await fs.stat(fp)
+          if (st.mtimeMs < recent) continue
+          if (!best || st.mtimeMs > best.mtimeMs) best = { mtimeMs: st.mtimeMs, fp }
+        } catch {
+          /* ignore */
+        }
+      }
+    }),
+  )
+  if (!best) return null
+  const windowSize = await contextWindowSize()
+  const text = await readOrNull(best.fp)
+  if (!text) return null
+  const lines = text.split("\n")
+  // 末尾から最初に見つかった assistant.usage を採用 (= 現在の文脈サイズ)
+  for (let i = lines.length - 1; i >= 0; i--) {
+    const line = lines[i]
+    if (!line || !line.includes('"usage"')) continue
+    let d
+    try {
+      d = JSON.parse(line)
+    } catch {
+      continue
+    }
+    if (d.type !== "assistant") continue
+    const u = d.message?.usage
+    if (!u) continue
+    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) continue
+    const percent = Math.min(100, Math.round((tokens / windowSize) * 1000) / 10)
+    return { percent, mtimeMs: best.mtimeMs }
+  }
+  return null
+}
+
 /**
  * 5h / 7d 集計の UsageStats を返す。statusLine が無い環境では transcript
  * から推定する。
+ *
+ * context%: statusLine 値が最新 jsonl より新しければ official を優先。
+ * 古い / 無い (= チャットモードで statusLine 未更新) 場合は jsonl 推定で上書き。
  */
 export async function getUsage() {
   const now = Date.now()
   const official = await readOfficial(now)
-  if (official) return official
-  return readEstimate(now)
+  const result = official || (await readEstimate(now))
+
+  const est = await latestJsonlContext(now)
+  if (est) {
+    const cacheMtime = await statuslineCacheMtime()
+    // statusLine が jsonl と同程度以上に新しければ official.context が信頼できる
+    // (ターミナルモード)。jsonl の方が新しい (チャットモード) なら推定で上書き。
+    const officialFresh =
+      official &&
+      typeof official.context === "number" &&
+      cacheMtime !== null &&
+      cacheMtime >= est.mtimeMs - 5000
+    if (!officialFresh) {
+      result.context = est.percent
+    }
+  }
+  return result
 }
 
 /**