@cocorograph/hub-agent 0.8.117 → 0.8.119

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cocorograph/hub-agent",
-  "version": "0.8.117",
+  "version": "0.8.119",
   "description": "Hub Hosted Cockpit のローカル常駐 agent。Hub と outbound WSS で接続し、ローカルの Claude / Codex チャットと tmux セッションを中継する。",
   "type": "module",
   "license": "UNLICENSED",

package/src/claude-history.mjs

@@ -781,7 +781,7 @@ async function attachBackgroundSubagentHistory(filePath, { result, requests }, t
  *
  * 末尾 maxLines 行に絞ってから JSON.parse + DISPLAY_TYPES でフィルタ。
  * 大きいファイルでも先頭から全行 parse する必要はないが、line buffer を最小化
- * するため簡易実装で全文 readFile → split → 末尾 slice する (典型 1MB 未満想定)。
+ * するため前方チャンク走査で末尾窓だけ保持する。
  *
  * @param {string} filePath
  * @param {{maxLines?: number, logger?: import('pino').Logger}} [opts]
@@ -807,9 +807,78 @@ export async function readSessionHistory(filePath, { maxLines = MAX_HISTORY_LINE
   }
 }
 
+// 前方に一度だけ走査し、末尾窓と全体の tool_use 索引を同時に作る。
+// byteLimit は呼び出し側が stat したサイズ。走査中の追記を追い続けないよう、そこで止める
+// (キャッシュも同じ stat の mtime/size で登録するので、読んだ範囲と照合キーが一致する)。
+async function scanParentHistory(filePath, maxLines, byteLimit) {
+  const chunkSize = 1024 * 1024
+  // slice(-maxLines) の従来の挙動を 0・負数・小数等でも維持する。
+  const windowSize = Math.trunc(Number(maxLines))
+  const capacity = windowSize > 0 ? windowSize : Infinity
+  const lines = []
+  const toolNames = new Map()
+  let total_lines = 0
+  let next = 0
+  let fragments = []
+  let fragmentBytes = 0
+  const acceptLine = (bytes) => {
+    if (bytes.length === 0) return
+    const line = bytes.toString("utf-8")
+    total_lines++
+    if (lines.length < capacity) lines.push(line)
+    else {
+      lines[next] = line
+      next = (next + 1) % capacity
+    }
+    if (MASK_ENABLED && line.includes('"tool_use"')) {
+      try {
+        collectToolUseNames(JSON.parse(line), toolNames)
+      } catch {
+        /* ignore */
+      }
+    }
+  }
+  const handle = await open(filePath, "r")
+  try {
+    let remaining = byteLimit
+    while (remaining > 0) {
+      // 未完行の断片が参照するので、次の read でバッファを上書きしない。
+      const want = Math.min(chunkSize, remaining)
+      const buffer = Buffer.allocUnsafe(want)
+      const { bytesRead } = await handle.read(buffer, 0, want, null)
+      if (bytesRead === 0) break // 並行 truncate
+      remaining -= bytesRead
+      const chunk = buffer.subarray(0, bytesRead)
+      let start = 0
+      let end
+      while ((end = chunk.indexOf(10, start)) !== -1) {
+        const part = chunk.subarray(start, end)
+        if (fragments.length) {
+          fragments.push(part)
+          acceptLine(Buffer.concat(fragments, fragmentBytes + part.length))
+          fragments = []
+          fragmentBytes = 0
+        } else acceptLine(part)
+        start = end + 1
+      }
+      if (start < bytesRead) {
+        const part = chunk.subarray(start)
+        fragments.push(part)
+        fragmentBytes += part.length
+      }
+    }
+    if (fragments.length) acceptLine(Buffer.concat(fragments, fragmentBytes))
+  } finally {
+    await handle.close()
+  }
+  const ordered = next ? lines.slice(next).concat(lines.slice(0, next)) : lines
+  const truncated = total_lines > maxLines
+  const slice = truncated ? ordered.slice(-maxLines) : ordered
+  return { slice, total_lines, truncated, toolNames }
+}
+
 async function readParentSessionHistory(filePath, { maxLines, logger, timing, attach }) {
-  // stat は readFile よりずっと軽い (メタデータのみ)。mtimeMs+size が前回と同じなら
-  // ファイル内容も同じとみなし、以降の readFile + 全行 parse をまるごとスキップする。
+  // mtimeMs+size が前回と同じなら、親ファイルの走査と parse をスキップする。
   let st
   try {
     st = await stat(filePath)
@@ -826,10 +895,12 @@ async function readParentSessionHistory(filePath, { maxLines, logger, timing, at
     return attach(cached)
   }
 
+  // ⚠️ read_ms は走査とマスク用 tool_use 索引の JSON.parse を含む (2026-10-27 のチャンク
+  // 走査化以降)。それ以前の read_ms / parse_ms とは内訳が違うので、前後比較は parent_ms で行う。
   const readStarted = performance.now()
-  let text
+  let scanned
   try {
-    text = await readFile(filePath, "utf-8")
+    scanned = await scanParentHistory(filePath, maxLines, st.size)
   } catch (err) {
     if (err.code === "ENOENT") {
       return { events: [], total_lines: 0, truncated: false }
@@ -838,25 +909,7 @@ async function readParentSessionHistory(filePath, { maxLines, logger, timing, at
     throw err
   }
   const parseStarted = performance.now()
-  // 末尾 newline は捨てる
-  const lines = text.split("\n").filter((l) => l.length > 0)
-  const total_lines = lines.length
-  const truncated = total_lines > maxLines
-  const slice = truncated ? lines.slice(-maxLines) : lines
-
-  // マスク (2026-06-23): tool_use 行が表示スライス外へ truncate されても tool_result の
-  // スキル判定が効くよう、全行を事前走査して tool_use_id→name を集めておく。
-  const toolNames = new Map()
-  if (MASK_ENABLED) {
-    for (const line of lines) {
-      if (!line.includes('"tool_use"')) continue
-      try {
-        collectToolUseNames(JSON.parse(line), toolNames)
-      } catch {
-        /* ignore */
-      }
-    }
-  }
+  const { slice, total_lines, truncated, toolNames } = scanned
 
   const events = []
   const requests = new Map()

package/src/grok-stream-bridge.mjs

@@ -479,6 +479,10 @@ export class GrokStreamBridge extends EventEmitter {
     session.cancelledTurn = session.turnSeq
     this._clearTurnPermissions(session, session.turnSeq, true)
     session.client.notify('session/cancel', { sessionId: session.sessionId })
+    if (!session.promptInFlight && !session.dispatchingInput) {
+      this.emit('grok_turn_complete', { stream_id: session.stream_id, session_id: session.sessionId,
+        stopReason: 'cancelled', lastSeq: session.updateSeq })
+    }
   }
 
   permissionReply({ stream_id, request_id, result }) {

package/src/main.mjs

@@ -2804,7 +2804,7 @@ export async function handleClaudeHistoryRequest(msg, ctx, {
         // ログ失敗でも応答を送る。
       }
     }
-    sendOk = ctx.client.send({
+    sendOk = sendHistoryFrame(ctx, {
       type: "claude.history.response",
       stream_id,
       session_id,
@@ -2822,7 +2822,7 @@ export async function handleClaudeHistoryRequest(msg, ctx, {
       total_lines: result.total_lines,
       truncated: result.truncated,
       error: result.error,
-    })
+    }, timing)
   } finally {
     logHistoryDelivery(ctx, "request", msg.stream_id, timing, payload, sendOk)
   }
@@ -2857,6 +2857,11 @@ function logHistoryDelivery(ctx, via, stream_id, timing, payload, sendOk) {
       cache_hit: timing.cache_hit, parent_ms: timing.parent_ms,
       subagent_ms: timing.subagent_ms, subagent_files: timing.subagent_files,
       subagent_events: timing.subagent_events, tag_ms: timing.tag_ms,
+      // 送信側の内訳 (2026-10-27): 送る前に既に詰まっていた量・frame の大きさ・文字列化時間。
+      // ws_buffered (送信直後) との差で、この 1 通が送信待ちに積んだ量も読める。
+      ws_buffered_before: timing.ws_buffered_before ?? null,
+      payload_bytes: timing.payload_bytes ?? null,
+      stringify_ms: timing.stringify_ms ?? null,
       events: payload?.events?.length || 0, total_lines: payload?.total_lines || 0,
       truncated: payload?.truncated === true, send_ok: Boolean(sendOk),
       ws_buffered: ctx.client.bufferedAmount ?? null,
@@ -2868,16 +2873,35 @@ function logHistoryDelivery(ctx, via, stream_id, timing, payload, sendOk) {
   }
 }
 
+function sendHistoryFrame(ctx, frame, timing) {
+  // 計測値の取得失敗で配送を止めない (cross-review major)。取れなかった値は 0 ではなく null。
+  try {
+    timing.ws_buffered_before = ctx.client.bufferedAmount ?? null
+  } catch {
+    timing.ws_buffered_before = null
+  }
+  const stats = {}
+  const ok = ctx.client.send(frame, stats)
+  if (typeof stats.bytes === "number") timing.payload_bytes = stats.bytes
+  if (typeof stats.stringify_ms === "number") timing.stringify_ms = stats.stringify_ms
+  return ok
+}
+
 export function sendClaudeAttachHistory(ctx, stream_id, payload, timing) {
   if (!payload) return
   let sendOk = false
   try {
-    sendOk = ctx.client.send({ ...payload, stream_id })
+    sendOk = sendHistoryFrame(ctx, { ...payload, stream_id }, timing)
   } finally {
     logHistoryDelivery(ctx, "push", stream_id, timing, payload, sendOk)
   }
 }
 
+// attach 時の先回り push の窓 (元行数)。⚠️ 縮めるときはブラウザ側を先に直す: 追加取得は
+// 「今の先頭イベント」を境界にするため、窓に境界候補 (照合キーを持つイベント) が無いと
+// 追加取得が止まる (2026-10-27 cross-review。100 行化は Hub 側の対応後に行う)。
+export const ATTACH_HISTORY_PUSH_LINES = 500
+
 /**
  * claude.attach の ready 直後にpushする履歴を組み立てる。
  * Browserからの claude.history.request 往復を省くため、resume時だけ先回りして読む。
@@ -2922,7 +2946,7 @@ export async function loadClaudeAttachHistory({
     const result = await timeHistoryPhase(timing, "fetch_ms", () => fetchHistoryFn({
       cwd,
       session_id: sessionId,
-      maxLines: 500,
+      maxLines: ATTACH_HISTORY_PUSH_LINES,
       projectsRoot,
       logger,
       timing,
@@ -3794,10 +3818,11 @@ export async function dispatch(msg, ctx) {
           runInLane: scheduler && ((lane, fn) => scheduler.runInLane(lane, fn)),
           claudeBridge: ctx.claudeBridge,
           codexBridge: ctx.codexBridge,
+          grokBridge: ctx.grokBridge,
           claudeAttachOptions: () => resolveClaudeAttachOptions(ctx.config),
           available: (kind) => kind === "codex"
             ? codexAvailableForDispatch(ctx)
-            : ctx.availableClis?.includes("claude") === true,
+            : ctx.availableClis?.includes(kind) === true,
           logger: ctx.logger,
         })
         return
@@ -4990,6 +5015,7 @@ export async function dispatch(msg, ctx) {
               noteRestoreStop(ctx.sessionRestoreState, "claude", cwd)
               result.stopped = await ctx.claudeBridge.stopSessionsFor({ cwd, reason: "chat-stop" })
             } else if (msg.cli_kind === "grok") {
+              noteRestoreStop(ctx.sessionRestoreState, "grok", cwd)
               result.stopped = await ctx.grokBridge?.stopSessionsForCwd({ cwd, reason: "chat-stop" }) ?? 0
             } else {
               noteRestoreStop(ctx.sessionRestoreState, "codex", cwd)
@@ -5010,7 +5036,11 @@ export async function dispatch(msg, ctx) {
             }
           } else if (msg.cli_kind === "grok") {
             if (typeof msg.stream_id !== "string" || !msg.stream_id) result.error = "stream_id required"
-            else result.stopped = await ctx.grokBridge?.stopSessionForStream({ stream_id: msg.stream_id, reason: "chat-stop" }) ?? 0
+            else {
+              const streamCwd = ctx.grokBridge?.sessions?.get?.(msg.stream_id)?.cwd
+              if (streamCwd) noteRestoreStop(ctx.sessionRestoreState, "grok", streamCwd)
+              result.stopped = await ctx.grokBridge?.stopSessionForStream({ stream_id: msg.stream_id, reason: "chat-stop" }) ?? 0
+            }
           } else if (msg.cli_kind === "codex") {
             if (typeof msg.stream_id !== "string" || !msg.stream_id) result.error = "stream_id required"
             else {

package/src/session-restore.mjs

@@ -31,7 +31,7 @@ export function noteRestoreStop(state, cliKind, cwd) {
  * 受付状態と Hub の「接続ごとに 1 回」の送信がかみ合わず、復元が途中で止まる (2 巡目 blocker)。
  */
 export async function restoreSessions(message, {
-  state, claudeBridge, codexBridge, available, logger,
+  state, claudeBridge, codexBridge, grokBridge, available, logger,
   runInLane = (_lane, fn) => fn(), statCwd = stat,
   claudeAttachOptions = () => ({}),
 }) {
@@ -40,7 +40,7 @@ export async function restoreSessions(message, {
     return
   }
   state.accepted = true
-  const bridges = { claude: claudeBridge, codex: codexBridge }
+  const bridges = { claude: claudeBridge, codex: codexBridge, grok: grokBridge }
   const logSkip = (row, reason) => logger?.info?.({
     cli_kind: row?.cli_kind, cwd: row?.cwd, session_id: row?.session_id, reason,
   }, "agent.sessions.restore skipped")
@@ -48,7 +48,7 @@ export async function restoreSessions(message, {
     logSkip(null, "sessions must be an array")
     return
   }
-  const snapshot = () => describeLiveSessionsUnion({ claudeBridge, codexBridge }).map((s) => ({
+  const snapshot = () => describeLiveSessionsUnion({ claudeBridge, codexBridge, grokBridge }).map((s) => ({
     cli_kind: s.kind, cwd: s.cwd, session_id: s.session_id || s.thread_id,
   }))
   const same = (a, b) => a.cli_kind === b.cli_kind && a.cwd === b.cwd && a.session_id === b.session_id
@@ -73,7 +73,7 @@ export async function restoreSessions(message, {
       const live = [...current, ...selected.filter((r) => !current.some((s) => same(s, r)))]
       if (live.length >= WARM_TOTAL_MAX) break
       try {
-        if (!row || !["claude", "codex"].includes(row.cli_kind)) throw new Error("unknown CLI")
+        if (!row || !["claude", "codex", "grok"].includes(row.cli_kind)) throw new Error("unknown CLI")
         if (!bridges[row.cli_kind] || !available(row.cli_kind)) throw new Error("CLI unavailable")
         if (typeof row.session_id !== "string" || !row.session_id.trim()) throw new Error("empty session_id")
         if (typeof row.cwd !== "string" || !path.isAbsolute(row.cwd)) throw new Error("cwd must be absolute")
@@ -115,9 +115,11 @@ export async function restoreSessions(message, {
             // ここで欠けると復元した会話だけ権限モードが既定に落ちる (0.8.91 の実測)。
             ...(row.cli_kind === "claude"
               ? { ...claudeAttachOptions(), resumeSessionId: row.session_id }
-              // Codex は承認ポリシー / サンドボックスを thread 開始時にしか受け取らない。Hub が
-              // Cockpit の既定を行に載せる (空なら載せない = config.toml 既定。通常 attach と同じ)。
-              : {
+              : row.cli_kind === "grok"
+                ? { resumeSessionId: row.session_id }
+                // Codex は承認ポリシー / サンドボックスを thread 開始時にしか受け取らない。Hub が
+                // Cockpit の既定を行に載せる (空なら載せない = config.toml 既定。通常 attach と同じ)。
+                : {
                   resumeThreadId: row.session_id,
                   approvalPolicy: nonEmpty(row.approval_policy),
                   sandbox: nonEmpty(row.sandbox),

package/src/ws-client.mjs

@@ -333,7 +333,13 @@ export class WsClient extends EventEmitter {
    * heartbeat / hello / agent.streams.sync.* など「古くなると意味が無い」制御系は
    * buffer しない (warn のみ)。ack / event 系は 1 件ずつ意味を持つのでバッファに保持する。
    */
-  send(obj) {
+  /**
+   * @param {object} obj
+   * @param {{bytes?: number, stringify_ms?: number}} [stats] 渡されたときだけ、実際に送れた
+   *   frame の UTF-8 バイト数と JSON 文字列化時間を書き込む (履歴配信の計測用。通常の
+   *   frame で毎回 byteLength を数えないよう、呼び出し側が明示したときに限る)。
+   */
+  send(obj, stats) {
     const type = obj?.type
     const wsOpen = this.ws?.readyState === WebSocket.OPEN
     // sync barrier 中に新しく発生した frame も同じ buffer へ積み、切断中 frame より
@@ -365,7 +371,7 @@ export class WsClient extends EventEmitter {
         return false
       }
     }
-    return this._sendJson(obj)
+    return this._sendJson(obj, stats)
   }
 
   /** 制御/イベント系 (ack / jsonl.event 等) を outbound buffer に積む。リング (drop oldest)。 */
@@ -548,9 +554,17 @@ export class WsClient extends EventEmitter {
     return `${scheme}//${host}/ws/cockpit/agent/`
   }
 
-  _sendJson(obj) {
+  _sendJson(obj, stats) {
     try {
-      this.ws.send(JSON.stringify(obj))
+      const started = stats ? performance.now() : 0
+      const text = JSON.stringify(obj)
+      const stringifyMs = stats ? performance.now() - started : 0
+      this.ws.send(text)
+      // 送れた frame だけを記録する (送信例外時は stats を書かない)。
+      if (stats) {
+        stats.stringify_ms = stringifyMs
+        stats.bytes = Buffer.byteLength(text)
+      }
       return true
     } catch (err) {
       this.logger?.warn({ err: err.message }, "ws send failed")