@cocorograph/hub-agent 0.7.14 → 0.7.16

package/package.json

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

package/src/main.mjs

@@ -1672,9 +1672,13 @@ export async function handleTuiInterrupt(msg, ctx) {
   const sendEsc = ctx.sendInterruptKey || sendInterruptKey
   const detect = ctx.detectSessionState || detectSessionState
   const delay = ctx.delay || ((ms) => new Promise((r) => setTimeout(r, ms)))
-  const settleMs = ctx.interruptSettleMs ?? 350
-  const pollMs = ctx.interruptPollMs ?? 350
-  const maxPolls = ctx.interruptMaxPolls ?? 7 // ~2.5s deadline
+  // 症状⑤: 初回チェックまでの settle と poll 間隔を 350→150ms に詰め、maxPolls を 7→16 に
+  // 増やす。総 deadline は ~2.4s で従来(~2.5s)とほぼ同じだが、実際に停止した瞬間の観測が
+  // 約2倍速くなり「停止押下→画面が止まる」体感ラグを縮める。capture-pane スクレイプ回数は
+  // 増えるが 150ms 間隔なら許容範囲。Ctrl+C 非エスカレーション方針は不変。
+  const settleMs = ctx.interruptSettleMs ?? 150
+  const pollMs = ctx.interruptPollMs ?? 150
+  const maxPolls = ctx.interruptMaxPolls ?? 16 // ~2.4s deadline (150 + 150*15)
   const reply = (stopped, attempts) =>
     ctx.client.send({
       type: "claude.tui.interrupt.result",

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
  * から推定する。

package/src/ws-client.mjs

@@ -3,7 +3,7 @@
  *
  * - 接続: `Authorization: Bearer <agent_id>:<agent_token>`
  * - 起動時に `hello`、30s おきに `heartbeat` を送信
- * - 切断時は exponential backoff (1s, 2s, 4s, ..., max 30s) で再接続
+ * - 切断時は exponential backoff (1s, 2s, 4s, 8s max) で再接続
  * - サーバから受け取った JSON は `onMessage` callback に渡す
  * - `bundleVersionProvider` + `bundleManifestPath` を渡すと、manifest.json の
  *   変更を fs.watch で検知して即時 heartbeat を送信し Cockpit UI に最新版を
@@ -20,7 +20,10 @@ import WebSocket from "ws"
 
 const HEARTBEAT_INTERVAL_MS = 30_000
 const MIN_BACKOFF_MS = 1_000
-const MAX_BACKOFF_MS = 30_000
+// 症状①(接続復帰の遅さ)対策: 旧 30s。本番ログで再接続 p50=1.9s に対し、フラッピング時に
+// backoff が指数増加して復帰レイテンシのテール(p90=9s, 最大35s)を作っていた。実測の切断は
+// ほぼ瞬断(CF/NW)で 8s も待てば回復に十分。jitter(±20%)で thundering herd は引き続き回避。
+const MAX_BACKOFF_MS = 8_000
 const BUNDLE_WATCH_DEBOUNCE_MS = 500
 
 // zombie WS (half-open TCP) 検知のしきい値。backend は heartbeat (30s) のたびに
@@ -196,15 +199,18 @@ export class WsClient extends EventEmitter {
    * のクロージャだと spy しづらい)。
    */
   _onOpen() {
-    // backoff の即時リセットはしない。接続が STABLE_CONNECTION_MS 維持できてから
-    // _armStableReset() でリセットする (open→即 close フラッピング時の再接続
-    // ストロボ対策)。すぐ切れた接続では backoff がリセットされず指数バックオフが
-    // 効き続けるため、1〜2 秒間隔の再接続ループに陥らない。
-    this._armStableReset()
     // open 自体が生存の証拠。直前接続の古い lastRecvAt で開幕直後に zombie 誤検知
-    // しないようリセットする。
+    // しないようリセットする。⚠️ _armStableReset() の armedAt 捕捉より「前」に設定すること。
+    // ack ベース判定 (lastRecvAt > armedAt) を初期状態で偽にするため (順序を逆にすると
+    // armedAt < lastRecvAt となり「inbound 受信済み」と誤判定し、half-open でも backoff を
+    // リセットしてしまう)。
     this.lastRecvAt = Date.now()
     this.lastTickAt = Date.now()
+    // backoff の即時リセットはしない。接続が STABLE_CONNECTION_MS 維持 + inbound 受信できてから
+    // _armStableReset() でリセットする (open→即 close フラッピング時の再接続ストロボ対策)。
+    // すぐ切れた / half-open の接続では backoff がリセットされず指数バックオフが効き続けるため、
+    // 1〜2 秒間隔の再接続ループに陥らない。
+    this._armStableReset()
     this.logger?.info("ws open")
     // 切断中に積んだ pty.data を flush。hello より先に送ると backend で stream
     // 未登録のため unknown_stream error になる可能性があるが、hello の応答待ち
@@ -583,11 +589,27 @@ export class WsClient extends EventEmitter {
    */
   _armStableReset() {
     this._clearStableReset()
+    // 症状①: backoff リセットを「ack ベース」にする。armedAt(≈open 時刻) を捕捉し、
+    // STABLE_CONNECTION_MS 後に「open 以降に inbound を受信したか(lastRecvAt > armedAt)」で
+    // 判定する。受信あり = 双方向に健全とみなし backoff/5xx をリセット。受信ゼロ =
+    // half-open(send は kernel buffer に書けて成功扱いだが TCP は死)とみなし、75s の zombie
+    // 検知を待たずに即 forceReconnect する(再接続後の half-open を 10s で回収し復帰を早める)。
+    // _onOpen は lastRecvAt=now を armedAt より前に設定するため、初期状態では
+    // lastRecvAt <= armedAt となり「未受信」と正しく判定される。
+    const armedAt = Date.now()
     this.stableResetTimer = setTimeout(() => {
       this.stableResetTimer = null
-      this.backoff = MIN_BACKOFF_MS
-      this.lastCloseWas5xx = false
-      this.logger?.debug("ws connection stable, backoff reset")
+      if (this.lastRecvAt > armedAt) {
+        this.backoff = MIN_BACKOFF_MS
+        this.lastCloseWas5xx = false
+        this.logger?.debug("ws connection stable (inbound received), backoff reset")
+      } else {
+        this.logger?.warn(
+          { sinceArmedMs: Date.now() - armedAt },
+          "ws stable window elapsed with no inbound (half-open suspected), forcing reconnect",
+        )
+        this._forceReconnect()
+      }
     }, STABLE_CONNECTION_MS)
     this.stableResetTimer.unref?.()
   }