@cocorograph/hub-agent 0.6.5 → 0.6.7

package/package.json

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

package/src/claude-stream-bridge.mjs

@@ -55,6 +55,7 @@ class ClaudeStreamSession {
     onPermission,
     onExit,
     onError,
+    onReap,
   }) {
     this.stream_id = stream_id
     this.cwd = cwd
@@ -66,6 +67,8 @@ class ClaudeStreamSession {
     this.onPermission = onPermission
     this.onExit = onExit
     this.onError = onError
+    /** ターン完走後に遅延クローズする際、manager にセッション撤去を依頼するコールバック */
+    this.onReap = onReap
 
     /** 次の query() で resume に使う session_id。各ターンの system/init で更新。 */
     this.sessionId = resumeSessionId || null
@@ -76,6 +79,10 @@ class ClaudeStreamSession {
     this._busy = false
     /** detach 済みフラグ (新規ターン受付停止) */
     this._closed = false
+    /** 実行中ターンの完走を待ってからクローズする予約フラグ (graceful detach 用)。
+     *  端末スリープ / ネット断で browser が切れても、明示的な中断 (interrupt) が
+     *  無い限りターンを落とさず最後まで走らせるために使う。 */
+    this._reapAfterTurn = false
     /** 現在ターンの AbortController (interrupt 用) */
     this._abortController = null
 
@@ -243,6 +250,19 @@ class ClaudeStreamSession {
       if (aborted) {
         this.logger?.info({ stream_id: this.stream_id }, "claude turn aborted")
       }
+      // graceful detach: browser が切れている間にターンが完走したら、ここで遅延
+      // クローズする。manager 側で sessions Map から撤去 + exit を emit する。
+      if (this._reapAfterTurn && !this._closed) {
+        this.logger?.info(
+          { stream_id: this.stream_id },
+          "claude turn finished after detach, reaping session",
+        )
+        try {
+          this.close()
+        } finally {
+          this.onReap?.()
+        }
+      }
     }
   }
 
@@ -257,6 +277,20 @@ class ClaudeStreamSession {
     }
   }
 
+  /**
+   * graceful detach: 実行中ターンがあれば中断せず完走を待つ (完走後に finally で
+   * 自動クローズ + onReap)。アイドルなら即クローズする。
+   * @returns {boolean} 即時クローズしたら true / 完走待ちにしたら false
+   */
+  gracefulClose() {
+    if (this._busy) {
+      this._reapAfterTurn = true
+      return false
+    }
+    this.close()
+    return true
+  }
+
   /** セッション終了。新規ターンを止め、実行中なら中断し、watcher も停止する。 */
   close() {
     this._closed = true
@@ -332,6 +366,18 @@ export class ClaudeStreamBridge extends EventEmitter {
       onError: (err) => {
         this.emit("error", { stream_id, error: err?.message || String(err) })
       },
+      onReap: () => {
+        // ターン完走後の遅延クローズ。Map から撤去し exit を emit する。
+        if (this.sessions.get(stream_id) === session) {
+          this.sessions.delete(stream_id)
+        }
+        this.emit("exit", {
+          stream_id,
+          code: 0,
+          reason: "detached-after-turn",
+          session_id: session.sessionId,
+        })
+      },
     })
     this.sessions.set(stream_id, session)
     this.logger?.info(
@@ -381,20 +427,31 @@ export class ClaudeStreamBridge extends EventEmitter {
     return true
   }
 
-  /** セッション停止。Map から即時削除し、実行中ターンを中断する。 */
+  /**
+   * セッション停止 (graceful)。実行中ターンは中断せず完走させ、完走後に
+   * onReap で Map から撤去する。アイドルなら即時撤去。
+   * 端末スリープ / ネット断による browser 切断でもターンを落とさないための入口。
+   * 明示的に止めたい場合は interrupt() を使う。
+   */
   detach({ stream_id }) {
     const s = this.sessions.get(stream_id)
     if (!s) return false
-    s.close()
-    this.sessions.delete(stream_id)
-    this.emit("exit", { stream_id, code: 0, reason: "detached", session_id: s.sessionId })
+    const reaped = s.gracefulClose()
+    if (reaped) {
+      this.sessions.delete(stream_id)
+      this.emit("exit", { stream_id, code: 0, reason: "detached", session_id: s.sessionId })
+    }
     return true
   }
 
-  /** 全セッションを停止 (agent shutdown 用)。 */
+  /** 全セッションを強制停止 (agent shutdown 用。実行中ターンも中断する)。 */
   shutdown() {
     for (const stream_id of Array.from(this.sessions.keys())) {
-      this.detach({ stream_id })
+      const s = this.sessions.get(stream_id)
+      if (!s) continue
+      s.close()
+      this.sessions.delete(stream_id)
+      this.emit("exit", { stream_id, code: 0, reason: "shutdown", session_id: s.sessionId })
     }
   }
 

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,104 @@ async function readEstimate(now) {
   }
 }
 
+/** statusLine cache ファイルの mtime (ms)。存在しなければ 0。 */
+async function statuslineCacheMtime() {
+  try {
+    return (await fs.stat(statuslineCache())).mtimeMs
+  } catch {
+    return null
+  }
+}
+
+/**
+ * 直近にアクティブな 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 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 / CONTEXT_WINDOW) * 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
 }
 
 /**