@cocorograph/hub-agent 0.8.2 → 0.8.3

package/package.json

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

package/src/usage.mjs

@@ -20,6 +20,7 @@ import { promises as fs } from "node:fs"
 import { randomUUID } from "node:crypto"
 import os from "node:os"
 import path from "node:path"
+import { StringDecoder } from "node:string_decoder"
 
 import { encodeCwdToDirName } from "./claude-history.mjs"
 
@@ -147,41 +148,6 @@ async function readTail(fp, maxBytes = 64 * 1024) {
   }
 }
 
-/**
- * ファイル先頭だけを読む (head)。readTail の対称。rollout の session_meta ヘッダー行
- * (先頭行) + 先頭近傍の最初の user_message (preview) を取り出すのに使う。maxBytes まで
- * 読み、末尾の不完全な行を捨てる (最後の改行以降を落とす)。size <= maxBytes なら全体。失敗時 null。
- */
-async function readHead(fp, maxBytes = 32 * 1024) {
-  let fh
-  try {
-    fh = await fs.open(fp, "r")
-    const st = await fh.stat()
-    const size = st.size
-    if (size === 0) return ""
-    const len = size > maxBytes ? maxBytes : size
-    const buf = Buffer.allocUnsafe(len)
-    // ⚠️ bytesRead で切る (allocUnsafe は未初期化。並行 truncate 等で len 未満しか読めなかった
-    // とき、残りのゴミ領域を文字列化しない。cross-review minor)。
-    const { bytesRead } = await fh.read(buf, 0, len, 0)
-    let text = buf.toString("utf-8", 0, bytesRead)
-    // EOF まで読み切っていない (maxBytes で頭だけ / 途中 truncate) 場合、末尾の不完全な行を捨てる。
-    if (bytesRead < size) {
-      const nl = text.lastIndexOf("\n")
-      text = nl >= 0 ? text.slice(0, nl) : text
-    }
-    return text
-  } catch {
-    return null
-  } finally {
-    try {
-      await fh?.close()
-    } catch {
-      /* ignore */
-    }
-  }
-}
-
 /** P5(perf): latestJsonlContext の結果を {mtimeMs,size} キーでメモ化し再パースを避ける。 */
 const _jsonlCtxMemo = new Map() // fp → { mtimeMs, size, result }
 
@@ -1513,21 +1479,37 @@ function threadIdFromRolloutFilename(fp) {
   return m ? m[1] : ""
 }
 
-/** rollout の先頭 (head) から一覧用メタ (thread_id / cwd / preview) を取り出す。
- *  header (session_meta) は先頭行、preview は先頭近傍の最初の user_message。null=読めない。 */
-async function parseCodexRolloutHead(fp) {
-  const text = await readHead(fp, 32 * 1024)
-  if (text == null) return null
+const CODEX_ROLLOUT_HEAD_CHUNK_BYTES = 32 * 1024
+const CODEX_ROLLOUT_PREVIEW_MAX_BYTES = 1024 * 1024
+
+/**
+ * rollout の先頭から一覧用メタ (thread_id / cwd / preview) を取り出す。
+ *
+ * 現行 Codex は最初の user_message より前に system / developer / AGENTS コンテキストを
+ * response_item として書くため、user_message が 200KB 以降に現れることがある。固定 32KB の
+ * head 読みでは preview が常に空になったため、32KB 単位で最大 1MB まで段階的に読み進める。
+ * ファイル全体は読まず、cwd が対象外と分かった時点、または preview が見つかった時点で止める。
+ *
+ * @param {string} fp
+ * @param {string} expectedCwd cwd-scoped 一覧の対象。別 cwd は session_meta 読み取り後に早期終了。
+ * @returns {Promise<{thread_id:string,cwd:string,preview:string}|null>}
+ */
+async function parseCodexRolloutHead(fp, expectedCwd = "") {
   let thread_id = threadIdFromRolloutFilename(fp)
   let cwd = ""
   let preview = ""
-  for (const line of text.split("\n")) {
-    if (!line) continue
+  let fh
+  const decoder = new StringDecoder("utf8")
+  let pending = ""
+  let offset = 0
+
+  const parseLine = (line) => {
+    if (!line) return
     let o
     try {
       o = JSON.parse(line)
     } catch {
-      continue
+      return
     }
     if (o.type === "session_meta" && o.payload && typeof o.payload === "object") {
       if (typeof o.payload.cwd === "string") cwd = o.payload.cwd
@@ -1543,8 +1525,58 @@ async function parseCodexRolloutHead(fp) {
     ) {
       preview = o.payload.message.slice(0, 80)
     }
-    if (cwd && preview) break // 両方揃ったら以降は読まない (先頭近傍で足りる)。
   }
+
+  try {
+    fh = await fs.open(fp, "r")
+    while (offset < CODEX_ROLLOUT_PREVIEW_MAX_BYTES) {
+      const readSize = Math.min(
+        CODEX_ROLLOUT_HEAD_CHUNK_BYTES,
+        CODEX_ROLLOUT_PREVIEW_MAX_BYTES - offset,
+      )
+      const buf = Buffer.allocUnsafe(readSize)
+      const { bytesRead } = await fh.read(buf, 0, readSize, offset)
+      if (bytesRead === 0) {
+        pending += decoder.end()
+        if (pending) parseLine(pending)
+        break
+      }
+      offset += bytesRead
+      pending += decoder.write(buf.subarray(0, bytesRead))
+
+      let newline
+      while ((newline = pending.indexOf("\n")) >= 0) {
+        const line = pending.slice(0, newline)
+        pending = pending.slice(newline + 1)
+        parseLine(line)
+        // session_meta は先頭行にある。別 workspace の候補に対して巨大な初期コンテキストを
+        // 読み進めないことで、maxScan が大きい場合も一覧走査コストを抑える。
+        if (cwd && expectedCwd && cwd !== expectedCwd) {
+          return thread_id ? { thread_id, cwd, preview } : null
+        }
+        if (cwd && preview) {
+          return thread_id ? { thread_id, cwd, preview } : null
+        }
+      }
+
+      // 通常ファイルの EOF。decoder.end() で分割された UTF-8 の末尾を確定し、改行なしの
+      // 最終行も一度だけ解析する。
+      if (bytesRead < readSize) {
+        pending += decoder.end()
+        if (pending) parseLine(pending)
+        break
+      }
+    }
+  } catch {
+    return null
+  } finally {
+    try {
+      await fh?.close()
+    } catch {
+      /* ignore */
+    }
+  }
+
   if (!thread_id) return null
   return { thread_id, cwd, preview }
 }
@@ -1610,7 +1642,7 @@ export async function listCodexThreadsFromDisk({
     if (limit && out.length >= limit) break
     if (opened >= maxScan) break
     opened++
-    const meta = await parseCodexRolloutHead(fp)
+    const meta = await parseCodexRolloutHead(fp, cwd)
     if (!meta) continue
     // header の cwd が一致するものだけ (cwd 未読/不一致は除外 = 無関係スレッドの混入防止)。
     if (meta.cwd !== cwd) continue