@cocorograph/hub-agent 0.6.13 → 0.6.15

package/package.json

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

package/src/claude-stream-bridge.mjs

@@ -150,6 +150,13 @@ class ClaudeStreamSession {
     /** 常駐 query を起動済みか (二重起動防止)。 */
     this._residentStarted = false
 
+    /** 改修3: per-message セッションで busy 中に届いた送信を退避する pending キュー。
+     *  常駐 query 化 (改修2) とは別レイヤー。resume チェーンは維持したまま、現ターン
+     *  完了時 (finally) に先頭から drain して次ターンを自動発火する。 */
+    this._pendingMessages = []
+    /** 改修3: 直近 browser へ通知した pending 件数 (変化時のみ queue_state を emit する)。 */
+    this._lastEmittedQueueCount = 0
+
     /** @type {Map<string, {resolve: (decision: object) => void}>} permission 応答待ち */
     this._permissionResolvers = new Map()
     /** 現在ターン実行中か (多重 query 防止) */
@@ -271,7 +278,8 @@ class ClaudeStreamSession {
 
   /**
    * ユーザーメッセージ 1 件を 1 query() として実行する。
-   * 既存ターン実行中は無視 (UI 側で送信を抑止する想定だが念のため)。
+   * 既存ターン実行中 (busy) は破棄せず pending キューへ退避し、現ターン完了時に drain する
+   * (改修3)。常駐 query 対象 (新規セッション) は InputQueue へ積む (改修2)。
    */
   async sendMessage(message) {
     if (this._closed) return
@@ -297,14 +305,23 @@ class ClaudeStreamSession {
     }
 
     // 従来 per-message: 1 メッセージ = 1 query (resume チェーン)。既存セッション (resume) 用。
+    // 改修3: busy 中の送信は破棄せず pending キューへ退避し、現ターン完了時に drain する
+    // (ターミナル流の「積む→待機→順次実行」)。常駐 query 化はしないので暴走リスクは増えない。
     if (this._busy) {
-      this.logger?.warn(
-        { stream_id: this.stream_id },
-        "claude session busy, message ignored",
+      this._pendingMessages.push(prompt)
+      this.logger?.info(
+        { stream_id: this.stream_id, queued: this._pendingMessages.length },
+        "claude busy, message queued",
       )
+      this._emitQueueState()
       return
     }
+    return this._runPerMessage(prompt)
+  }
 
+  /** per-message 1 ターンを実行する (resume チェーン)。busy 中に届いた送信は sendMessage
+   *  が pending キューへ退避し、本メソッドの finally で drain する。 */
+  async _runPerMessage(prompt) {
     this._busy = true
     this._abortController = new AbortController()
     let aborted = false
@@ -397,10 +414,52 @@ class ClaudeStreamSession {
         } finally {
           this.onReap?.()
         }
+      } else {
+        // 改修3: busy 中に積まれた pending を順次発火 (per-message resume チェーン維持)。
+        // 停止 (abort) 後も温存したキューはそのまま流す。
+        this._drainPending()
       }
     }
   }
 
+  /** 改修3: pending キューの先頭を取り出して次の per-message ターンを発火する。
+   *  各ターンの finally から呼ばれ、キューが空になるまで resume チェーンが続く。 */
+  _drainPending() {
+    if (this._closed || this._busy) return
+    if (this._pendingMessages.length === 0) {
+      this._emitQueueState()
+      return
+    }
+    const next = this._pendingMessages.shift()
+    this._emitQueueState()
+    this._runPerMessage(next).catch((err) => {
+      this.logger?.error(
+        { stream_id: this.stream_id, err: err?.message },
+        "drain runPerMessage threw",
+      )
+    })
+  }
+
+  /** 改修3: pending キューの現状を browser へ通知する (送信待ちチップ表示用)。
+   *  onEvent 経由で claude.event(event.type="queue_state") として届く。 */
+  _emitQueueState() {
+    const count = this._pendingMessages.length
+    // 件数が変わらないなら通知しない (空→空の冗長 emit を抑止)。
+    if (count === this._lastEmittedQueueCount) return
+    this._lastEmittedQueueCount = count
+    try {
+      // messages は全文を載せる。frontend は実行開始 (drain) 時にこれを user バブルへ
+      // 昇格させるため、ここで切り詰めると本文が欠ける。チップの省略表示は CSS 側で行う。
+      this.onEvent?.({
+        type: "queue_state",
+        pending: count,
+        messages: [...this._pendingMessages],
+      })
+    } catch {
+      /* ignore */
+    }
+  }
+
   /** 改修2: 常駐 query を 1 回だけ起動し、streaming input キューから複数ターンを処理する。
    *  system/init は最初の 1 回のみ (毎ターン init = 文脈/キャッシュ温め直しを解消)。各ターンは
    *  result で _busy=false にするが query は継続し次 input を待つ。close (inputQueue.close) で
@@ -513,6 +572,8 @@ class ClaudeStreamSession {
   /** セッション終了。新規ターンを止め、実行中なら中断し、watcher も停止する。 */
   close() {
     this._closed = true
+    // 改修3: 未処理の pending を破棄 (セッション終了後に drain しないため)。
+    this._pendingMessages = []
     // 常駐 query は input キューを閉じて generator を終わらせ query を完走させる。
     if (this._inputQueue) this._inputQueue.close()
     this.abortTurn()