@cocorograph/hub-agent 0.7.15 → 0.7.16

package/package.json

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

package/src/tmux.mjs

@@ -27,7 +27,7 @@ import {
   detectSessionState,
   getSessionCwd,
 } from "./state.mjs"
-import { getSessionUsages } from "./usage.mjs"
+import { getSessionUsages, jsonlContextForCwd } from "./usage.mjs"
 
 const execFileP = promisify(execFile)
 
@@ -634,10 +634,23 @@ export async function listSessions(opts = {}) {
         detectSessionState(s.name, opts),
         getSessionCwd(s.name, opts),
       ])
-      // statusLine 由来 context% を最優先 (USED %), なければ pane 解析の fallback
+      // context% (USED %) は以下の順で採用する:
+      //  1. jsonl per-cwd で「`/clear` 直後 (uuid 変化) かつ新 jsonl に応答未着」
+      //     と確定したら 0% を即採用 (= リセット最速反映)。statusLine cache は
+      //     前セッションの値を抱えたまま居座り、次のプロンプトまで更新されない。
+      //  2. それ以外は statusLine cache (USED %) を最優先 (claude 公式値)。
+      //  3. statusLine が無ければ jsonl per-cwd の末尾 assistant.usage 由来 %。
+      //  4. それも無ければ pane scrape の正規表現フォールバック。
       const fromStatusLine = cwd ? ctxByCwd.get(cwd)?.contextPercent : null
+      const jsonlInfo = cwd ? await jsonlContextForCwd(cwd) : null
       const context_pct =
-        typeof fromStatusLine === "number" ? fromStatusLine : state.context_pct
+        jsonlInfo?.isReset && typeof jsonlInfo.percent === "number"
+          ? jsonlInfo.percent
+          : typeof fromStatusLine === "number"
+            ? fromStatusLine
+            : typeof jsonlInfo?.percent === "number"
+              ? jsonlInfo.percent
+              : state.context_pct
       return {
         ...s,
         status: state.status,

package/src/usage.mjs

@@ -21,6 +21,8 @@ import { randomUUID } from "node:crypto"
 import os from "node:os"
 import path from "node:path"
 
+import { encodeCwdToDirName } from "./claude-history.mjs"
+
 function configPath(envKey, ...fallback) {
   return process.env[envKey] || path.join(...fallback)
 }
@@ -529,6 +531,132 @@ async function latestJsonlContext(now) {
   return result
 }
 
+/**
+ * cwd → 最後に観測した newest jsonl uuid。`/clear` 検知 (uuid 変化 → 新セッション)
+ * に使う。`jsonlContextForCwd` の呼び出しごとに更新される。
+ */
+const _jsonlSessionUuidByCwd = new Map()
+
+/**
+ * 指定 cwd の Claude セッション jsonl から context% (USED %) を算出する。
+ *
+ * 役割:
+ *  - `latestJsonlContext` が「直近全プロジェクト共通の最新 1 件」を返すのに対し、
+ *    こちらは「この cwd の最新セッション」を返す。tmux セッション 1 つに対する
+ *    ドーナツ表示の per-session ソースとして使う。
+ *  - `/clear` 直後は新規 jsonl ファイル (新 uuid) が作成される。assistant.usage が
+ *    まだ書かれていない瞬間は前回 uuid と異なれば「コンテキスト全消し」と確定でき、
+ *    statusLine cache が前セッションの値を抱えたまま 0% に落ちない不具合を回避する。
+ *
+ * 動作:
+ *  1. cwd → encoded dir → `*.jsonl` のうち最新 mtime を newest として選ぶ
+ *  2. newest jsonl の末尾 assistant.usage を `(input + cache_read + cache_creation
+ *     + output) / window_size` で % 化
+ *  3. 末尾に assistant.usage が無い (= 新セッション直後で 1 通も応答していない)
+ *     場合、前回観測 uuid と異なれば `{ percent: 0, isReset: true }` を返す。
+ *     同じ uuid のまま (= 初回ロード直後) なら `{ percent: null, isReset: false }`。
+ *
+ * 呼び出し側 (`tmux.mjs listSessionStates`) の優先順位:
+ *  - `isReset === true` のとき statusLine cache の前セッション値より優先 (= 0% を採用)
+ *  - そうでなければ statusLine > jsonl > pane scrape の順
+ *
+ * @param {string} cwd
+ * @returns {Promise<{ percent: number | null, uuid: string | null, isReset: boolean }>}
+ */
+export async function jsonlContextForCwd(cwd) {
+  const empty = { percent: null, uuid: null, isReset: false }
+  if (!cwd) return empty
+  const dirName = encodeCwdToDirName(cwd)
+  if (!dirName) return empty
+  const dir = path.join(projectsDir(), dirName)
+  const files = await fs.readdir(dir).catch(() => null)
+  if (!files) return empty
+
+  let newest = null // { fp, mtimeMs, size, uuid }
+  await Promise.all(
+    files.map(async (f) => {
+      if (!f.endsWith(".jsonl")) return
+      const fp = path.join(dir, f)
+      const st = await fs.stat(fp).catch(() => null)
+      if (!st) return
+      if (!newest || st.mtimeMs > newest.mtimeMs) {
+        newest = {
+          fp,
+          mtimeMs: st.mtimeMs,
+          size: st.size,
+          uuid: f.replace(/\.jsonl$/, ""),
+        }
+      }
+    }),
+  )
+  if (!newest) return empty
+
+  // 末尾 assistant.usage を探す。`_jsonlCtxMemo` は fp+mtime+size をキーにしているので
+  // 同 jsonl の連続呼び出しは I/O ゼロでヒットする。
+  const memo = _jsonlCtxMemo.get(newest.fp)
+  let percent = null
+  if (memo && memo.mtimeMs === newest.mtimeMs && memo.size === newest.size) {
+    percent = memo.result ? memo.result.percent : null
+  } else {
+    const windowSize = await contextWindowSize()
+    const scan = (text) => {
+      const lines = text.split("\n")
+      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
+        return Math.min(100, Math.round((tokens / windowSize) * 1000) / 10)
+      }
+      return null
+    }
+    const tail = await readTail(newest.fp)
+    if (tail != null) percent = scan(tail)
+    if (percent == null) {
+      const full = await readOrNull(newest.fp)
+      if (full != null) percent = scan(full)
+    }
+    const result = percent == null ? null : { percent, mtimeMs: newest.mtimeMs }
+    _jsonlCtxMemo.set(newest.fp, {
+      mtimeMs: newest.mtimeMs,
+      size: newest.size,
+      result,
+    })
+  }
+
+  const prevUuid = _jsonlSessionUuidByCwd.get(cwd)
+  _jsonlSessionUuidByCwd.set(cwd, newest.uuid)
+  const uuidChanged = prevUuid !== undefined && prevUuid !== newest.uuid
+
+  if (percent == null) {
+    // assistant.usage が見つからない = まだ 1 通も応答していない新セッション。
+    // 直前まで別 uuid を観測していたなら `/clear` 直後と確定でき 0% を返す。
+    if (uuidChanged) {
+      return { percent: 0, uuid: newest.uuid, isReset: true }
+    }
+    return { percent: null, uuid: newest.uuid, isReset: false }
+  }
+  return { percent, uuid: newest.uuid, isReset: uuidChanged }
+}
+
+/** テスト用: per-cwd uuid 観測ストアをクリアする。 */
+export function _resetJsonlSessionUuids() {
+  _jsonlSessionUuidByCwd.clear()
+}
+
 /**
  * 5h / 7d 集計の UsageStats を返す。statusLine が無い環境では transcript
  * から推定する。