@cocorograph/hub-agent 0.6.4 → 0.6.6

package/package.json

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

package/src/claude-history-watch.mjs

@@ -0,0 +1,171 @@
+/**
+ * Claude Code セッション jsonl のライブ追従 watcher (Sprint G 0.6.5)。
+ *
+ * 用途: Cockpit ChatView で「外部 (tmux の Claude 等) が同じ session の jsonl に
+ * 追記した内容」をリアルタイムに反映する。per-message query は「自分が送った
+ * ターン」しか stream しないため、外部進行を拾うには jsonl を tail する必要がある。
+ *
+ * 設計:
+ * - 対象ファイルのバイトオフセットを記録し、増分だけ読んで行単位でパース
+ * - fs.watch(file) の change イベントで増分読み取り (取りこぼし対策に軽いポーリング併用)
+ * - DISPLAY_TYPES (user/assistant/system/result) の行だけ onEvent に渡す
+ * - 各行の uuid を含めて渡す (frontend 側で重複排除に使う)
+ * - ファイル不在時は出現を待つ (ディレクトリ監視はせず、ポーリングで存在チェック)
+ */
+import { watch as fsWatch } from "node:fs"
+import { open, stat } from "node:fs/promises"
+
+const DISPLAY_TYPES = new Set(["user", "assistant", "system", "result"])
+const POLL_INTERVAL_MS = 1500
+
+/**
+ * 1 つの jsonl ファイルを tail する watcher を生成する。
+ *
+ * @param {object} args
+ * @param {string} args.filePath - 監視対象の jsonl 絶対パス
+ * @param {(event: object) => void} args.onEvent - DISPLAY_TYPES の行ごとに呼ばれる
+ * @param {boolean} [args.fromEnd=true] - 既存内容は飛ばし、監視開始後の追記のみ拾う
+ *   (起動時の履歴は history.request で別途 hydrate 済みのため、tail は新規分だけでよい)
+ * @param {import('pino').Logger} [args.logger]
+ * @returns {{ stop: () => void }}
+ */
+export function watchSessionFile({ filePath, onEvent, fromEnd = true, logger }) {
+  let offset = 0
+  let reading = false
+  let stopped = false
+  let leftover = ""
+  let fsWatcher = null
+  let pollTimer = null
+  let initialized = false
+
+  async function initOffset() {
+    try {
+      const st = await stat(filePath)
+      offset = fromEnd ? st.size : 0
+      initialized = true
+    } catch {
+      // ファイル未存在: offset=0 のまま、出現したら先頭から (fromEnd は初回出現には適用しない)
+      offset = 0
+      initialized = false
+    }
+  }
+
+  async function readIncrement() {
+    if (stopped || reading) return
+    reading = true
+    try {
+      let st
+      try {
+        st = await stat(filePath)
+      } catch {
+        return // まだ無い
+      }
+      if (!initialized) {
+        // 初回出現: fromEnd でも「出現直後の全文」は新規とみなして先頭から読む
+        // (監視開始時点で既存だったファイルは initOffset で末尾にセット済み)
+        initialized = true
+      }
+      if (st.size < offset) {
+        // truncate / rotate された → 先頭から読み直す
+        offset = 0
+        leftover = ""
+      }
+      if (st.size === offset) return
+      const fh = await open(filePath, "r")
+      try {
+        const len = st.size - offset
+        const buf = Buffer.alloc(len)
+        await fh.read(buf, 0, len, offset)
+        offset = st.size
+        const text = leftover + buf.toString("utf-8")
+        const lines = text.split("\n")
+        leftover = lines.pop() ?? "" // 最終要素は未完行として保持
+        for (const line of lines) {
+          if (!line) continue
+          let obj
+          try {
+            obj = JSON.parse(line)
+          } catch {
+            continue
+          }
+          if (!obj || !DISPLAY_TYPES.has(obj.type)) continue
+          const event = normalizeEvent(obj)
+          try {
+            onEvent(event)
+          } catch (err) {
+            logger?.warn({ err: err.message }, "watchSessionFile onEvent threw")
+          }
+        }
+      } finally {
+        await fh.close()
+      }
+    } catch (err) {
+      logger?.warn({ err: err.message, filePath }, "watchSessionFile read failed")
+    } finally {
+      reading = false
+    }
+  }
+
+  ;(async () => {
+    await initOffset()
+    if (stopped) return
+    // fs.watch (change で増分読み取り)
+    try {
+      fsWatcher = fsWatch(filePath, { persistent: false }, () => {
+        readIncrement().catch(() => {})
+      })
+      fsWatcher.on?.("error", () => {})
+    } catch {
+      // ファイル未存在等で watch 不可 → ポーリングに委ねる
+    }
+    // 取りこぼし / ファイル出現待ち用の軽いポーリング
+    pollTimer = setInterval(() => readIncrement().catch(() => {}), POLL_INTERVAL_MS)
+    pollTimer.unref?.()
+  })()
+
+  return {
+    stop() {
+      stopped = true
+      try {
+        fsWatcher?.close()
+      } catch {
+        /* ignore */
+      }
+      if (pollTimer) clearInterval(pollTimer)
+    },
+    /**
+     * offset を現在のファイル末尾に進める (= 未読分を push せず捨てる)。
+     * per-message query 実行中に書かれた行は query stream 側で既に browser に
+     * 届いているため、ターン完了時にこれを呼んで二重 push を防ぐ。
+     */
+    async skipToEnd() {
+      try {
+        const st = await stat(filePath)
+        offset = st.size
+        leftover = ""
+        initialized = true
+      } catch {
+        /* ファイル無し: 次の出現で先頭から */
+      }
+    },
+  }
+}
+
+/** jsonl の生 object を SDK message 風の表示用イベントに正規化 (history.mjs と揃える)。 */
+function normalizeEvent(obj) {
+  const event = { type: obj.type }
+  if (obj.message !== undefined) event.message = obj.message
+  if (obj.subtype !== undefined) event.subtype = obj.subtype
+  if (obj.uuid !== undefined) event.uuid = obj.uuid
+  if (obj.session_id !== undefined) event.session_id = obj.session_id
+  else if (obj.sessionId !== undefined) event.session_id = obj.sessionId
+  if (obj.model !== undefined) event.model = obj.model
+  if (obj.cwd !== undefined) event.cwd = obj.cwd
+  if (obj.tools !== undefined) event.tools = obj.tools
+  if (obj.permissionMode !== undefined) event.permissionMode = obj.permissionMode
+  if (obj.total_cost_usd !== undefined) event.total_cost_usd = obj.total_cost_usd
+  if (obj.duration_ms !== undefined) event.duration_ms = obj.duration_ms
+  if (obj.num_turns !== undefined) event.num_turns = obj.num_turns
+  if (obj.usage !== undefined) event.usage = obj.usage
+  return event
+}

package/src/claude-stream-bridge.mjs

@@ -21,6 +21,9 @@
 import { EventEmitter } from "node:events"
 import { randomUUID } from "node:crypto"
 
+import { jsonlPath } from "./claude-history.mjs"
+import { watchSessionFile } from "./claude-history-watch.mjs"
+
 /** message (`{role, content}` or string) からプロンプト文字列を取り出す。 */
 function extractPromptText(message) {
   if (typeof message === "string") return message
@@ -52,6 +55,7 @@ class ClaudeStreamSession {
     onPermission,
     onExit,
     onError,
+    onReap,
   }) {
     this.stream_id = stream_id
     this.cwd = cwd
@@ -63,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
@@ -73,8 +79,54 @@ class ClaudeStreamSession {
     this._busy = false
     /** detach 済みフラグ (新規ターン受付停止) */
     this._closed = false
+    /** 実行中ターンの完走を待ってからクローズする予約フラグ (graceful detach 用)。
+     *  端末スリープ / ネット断で browser が切れても、明示的な中断 (interrupt) が
+     *  無い限りターンを落とさず最後まで走らせるために使う。 */
+    this._reapAfterTurn = false
     /** 現在ターンの AbortController (interrupt 用) */
     this._abortController = null
+
+    /** jsonl ライブ追従 watcher (0.6.5)。外部 (tmux 等) の追記を拾う。 */
+    this._watcher = null
+    /** watcher が監視中の session_id (変化時に張り替え) */
+    this._watchedSessionId = null
+    // resume セッションがあれば即 watch 開始 (閲覧中の外部進行をライブ反映)
+    if (this.sessionId) this._ensureWatch()
+  }
+
+  /** 現在の sessionId の jsonl を watch する (既に同じものを watch 中なら何もしない)。 */
+  _ensureWatch() {
+    if (this._closed || !this.sessionId || !this.cwd) return
+    if (this._watchedSessionId === this.sessionId && this._watcher) return
+    // 旧 watcher を畳む
+    if (this._watcher) {
+      try {
+        this._watcher.stop()
+      } catch {
+        /* ignore */
+      }
+      this._watcher = null
+    }
+    const filePath = jsonlPath({ cwd: this.cwd, session_id: this.sessionId })
+    this._watchedSessionId = this.sessionId
+    this._watcher = watchSessionFile({
+      filePath,
+      fromEnd: true, // 監視開始時点の既存内容は history hydrate 済み。新規追記のみ拾う
+      logger: this.logger,
+      onEvent: (event) => {
+        // 自分の query 実行中 (busy) は query stream が同じ内容を流すため push しない
+        // (二重表示防止)。ターン完了時に skipToEnd で読み飛ばす。
+        if (this._busy) return
+        try {
+          this.onEvent?.(event)
+        } catch (err) {
+          this.logger?.warn(
+            { err: err.message, stream_id: this.stream_id },
+            "watch onEvent threw",
+          )
+        }
+      },
+    })
   }
 
   /** SDK options.canUseTool: tool 起動を許可するかを browser に問い合わせる。 */
@@ -145,6 +197,8 @@ class ClaudeStreamSession {
           typeof msg.session_id === "string"
         ) {
           this.sessionId = msg.session_id
+          // session_id が確定/変化したら watch をその jsonl に張り替える
+          this._ensureWatch()
         }
         // result イベントでも session_id が来ることがある (念のため拾う)
         if (msg?.type === "result" && typeof msg.session_id === "string") {
@@ -170,7 +224,6 @@ class ClaudeStreamSession {
         }
       }
     } finally {
-      this._busy = false
       this._abortController = null
       // 未解決 permission は閉じる
       for (const [, resolver] of this._permissionResolvers) {
@@ -181,9 +234,35 @@ class ClaudeStreamSession {
         }
       }
       this._permissionResolvers.clear()
+      // このターンで jsonl に書かれた行は query stream で既に push 済みなので、
+      // watcher の offset を末尾に飛ばして二重 push を防ぐ。busy=true のまま
+      // skipToEnd を待ち、完了後に busy=false にすることで、その間に watcher poll が
+      // 走ってもターン行を push しない (二重表示防止)。
+      this._ensureWatch()
+      if (this._watcher?.skipToEnd) {
+        try {
+          await this._watcher.skipToEnd()
+        } catch {
+          /* ignore */
+        }
+      }
+      this._busy = false
       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?.()
+        }
+      }
     }
   }
 
@@ -198,10 +277,32 @@ 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
     this.abortTurn()
+    if (this._watcher) {
+      try {
+        this._watcher.stop()
+      } catch {
+        /* ignore */
+      }
+      this._watcher = null
+    }
     for (const [, resolver] of this._permissionResolvers) {
       try {
         resolver.resolve({ behavior: "deny", message: "closed" })
@@ -265,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(
@@ -314,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 })
     }
   }