@cocorograph/hub-agent 0.6.5 → 0.6.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cocorograph/hub-agent",
-  "version": "0.6.5",
+  "version": "0.6.6",
   "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 })
     }
   }