@cocorograph/hub-agent 0.6.12 → 0.6.13

package/package.json

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

package/src/claude-stream-bridge.mjs

@@ -30,6 +30,52 @@ import { watchSessionFile } from "./claude-history-watch.mjs"
  *  完走するまで絶対に撤去しない。明示終了 / 新規セッションでは即時撤去される。 */
 const IDLE_DETACH_TTL_MS = 7 * 24 * 60 * 60 * 1000
 
+/** 改修2: チャット常駐query化のフラグ。env HUB_AGENT_CHAT_RESIDENT="0" で無効化し
+ *  従来の「1メッセージ=1query」へ即ロールバックできる。デフォルト有効。 */
+const CHAT_RESIDENT_ENABLED = process.env.HUB_AGENT_CHAT_RESIDENT !== "0"
+
+/** 文字列を SDK streaming input の SDKUserMessage に包む。 */
+function toSDKUserMessage(text) {
+  return { type: "user", message: { role: "user", content: text } }
+}
+
+/** 常駐 query の streaming input キュー。push された SDKUserMessage を AsyncIterable として
+ *  query() に供給する。push が無ければ次を待ち (query は終了しない)、close で generator を
+ *  終わらせて query を完走させる。これにより 1 セッション = 1 query を実現し、system/init を
+ *  最初の 1 回だけにする (毎ターン init = 文脈/キャッシュ温め直しの解消)。 */
+class InputQueue {
+  constructor() {
+    this._q = []
+    this._wake = null
+    this._closed = false
+  }
+  push(item) {
+    if (this._closed) return
+    this._q.push(item)
+    this._flush()
+  }
+  close() {
+    this._closed = true
+    this._flush()
+  }
+  _flush() {
+    if (this._wake) {
+      const w = this._wake
+      this._wake = null
+      w()
+    }
+  }
+  async *[Symbol.asyncIterator]() {
+    while (true) {
+      while (this._q.length) yield this._q.shift()
+      if (this._closed) return
+      await new Promise((r) => {
+        this._wake = r
+      })
+    }
+  }
+}
+
 /** message (`{role, content}` or string) からプロンプト文字列を取り出す。 */
 function extractPromptText(message) {
   if (typeof message === "string") return message
@@ -57,6 +103,7 @@ class ClaudeStreamSession {
     maxTurns,
     maxThinkingTokens,
     resumeSessionId,
+    resident,
     sdk,
     logger,
     onEvent,
@@ -89,6 +136,20 @@ class ClaudeStreamSession {
     /** detached のまま放置されたセッションを撤去する idle タイマー。再アタッチでキャンセル。 */
     this._idleTimer = null
 
+    /** 改修2: 常駐query対象か。明示 resident 指定があれば優先 (テスト/将来の per-session 設定用)。
+     *  未指定なら resume なしの新規セッションのみ常駐 (既存 resume セッションは過去の
+     *  「Continue from where you left off」暴走を避けて従来 per-message を維持)。 */
+    this._residentEligible =
+      typeof resident === "boolean"
+        ? resident
+        : !resumeSessionId && CHAT_RESIDENT_ENABLED
+    /** 常駐query の input キュー (streaming input)。初回 input 時に生成・query 起動。 */
+    this._inputQueue = null
+    /** 起動済みの常駐 query ハンドル (interrupt() 用)。 */
+    this._residentQuery = null
+    /** 常駐 query を起動済みか (二重起動防止)。 */
+    this._residentStarted = false
+
     /** @type {Map<string, {resolve: (decision: object) => void}>} permission 応答待ち */
     this._permissionResolvers = new Map()
     /** 現在ターン実行中か (多重 query 防止) */
@@ -214,6 +275,28 @@ class ClaudeStreamSession {
    */
   async sendMessage(message) {
     if (this._closed) return
+    const prompt = extractPromptText(message)
+    if (!prompt) return
+
+    // 改修2: 常駐query対象セッションは streaming input キューへ積む (busy 中でも積み、
+    // SDK が順次処理する)。初回 input で常駐 query を 1 回だけ起動する。
+    if (this._residentEligible) {
+      if (!this._inputQueue) this._inputQueue = new InputQueue()
+      this._busy = true
+      this._inputQueue.push(toSDKUserMessage(prompt))
+      if (!this._residentStarted) {
+        this._residentStarted = true
+        this._runResidentQuery().catch((err) => {
+          this.logger?.error(
+            { stream_id: this.stream_id, err: err?.message },
+            "resident query threw",
+          )
+        })
+      }
+      return
+    }
+
+    // 従来 per-message: 1 メッセージ = 1 query (resume チェーン)。既存セッション (resume) 用。
     if (this._busy) {
       this.logger?.warn(
         { stream_id: this.stream_id },
@@ -221,8 +304,6 @@ class ClaudeStreamSession {
       )
       return
     }
-    const prompt = extractPromptText(message)
-    if (!prompt) return
 
     this._busy = true
     this._abortController = new AbortController()
@@ -320,8 +401,92 @@ class ClaudeStreamSession {
     }
   }
 
+  /** 改修2: 常駐 query を 1 回だけ起動し、streaming input キューから複数ターンを処理する。
+   *  system/init は最初の 1 回のみ (毎ターン init = 文脈/キャッシュ温め直しを解消)。各ターンは
+   *  result で _busy=false にするが query は継続し次 input を待つ。close (inputQueue.close) で
+   *  generator が終わり query が完走する。resume は使わない (新規セッションのみが対象)。
+   *  常駐では query stream が唯一のイベント源のため watcher は張らない。 */
+  async _runResidentQuery() {
+    const options = {
+      cwd: this.cwd,
+      canUseTool: (toolName, input) => this._canUseTool(toolName, input),
+      includePartialMessages: true,
+    }
+    if (this.model) options.model = this.model
+    if (this.permissionMode) options.permissionMode = this.permissionMode
+    if (this.maxTurns != null) options.maxTurns = this.maxTurns
+    if (this.maxThinkingTokens != null) options.maxThinkingTokens = this.maxThinkingTokens
+    const denyPending = (reason) => {
+      for (const [, resolver] of this._permissionResolvers) {
+        try {
+          resolver.resolve({ behavior: "deny", message: reason })
+        } catch {
+          /* ignore */
+        }
+      }
+      this._permissionResolvers.clear()
+    }
+    try {
+      const q = this.sdk.query({ prompt: this._inputQueue, options })
+      this._residentQuery = q
+      for await (const msg of q) {
+        if (
+          msg?.type === "system" &&
+          msg?.subtype === "init" &&
+          typeof msg.session_id === "string"
+        ) {
+          this.sessionId = msg.session_id
+        }
+        if (msg?.type === "result") {
+          if (typeof msg.session_id === "string") this.sessionId = msg.session_id
+          // ターン完了: 未解決 permission を閉じ、次 input 受付へ (query は継続)。
+          denyPending("turn ended")
+          this._busy = false
+        }
+        try {
+          this.onEvent?.(msg)
+        } catch (err) {
+          this.logger?.warn(
+            { err: err.message, stream_id: this.stream_id },
+            "onEvent callback threw",
+          )
+        }
+      }
+    } catch (err) {
+      if (!this._closed) {
+        try {
+          this.onError?.(err)
+        } catch {
+          /* ignore */
+        }
+      }
+    } finally {
+      this._residentQuery = null
+      this._busy = false
+      denyPending("closed")
+      // 常駐 query の終了 = セッション終了。detached 後にここへ来たら reap する。
+      if (this._reapAfterTurn && !this._closed) {
+        try {
+          this.close()
+        } finally {
+          this.onReap?.()
+        }
+      }
+    }
+  }
+
   /** 実行中ターンを中断する (次の input は引き続き受け付ける)。 */
   abortTurn() {
+    // 常駐 query は interrupt() で現ターンのみ中断する (query は継続、次 input 受付継続)。
+    if (this._residentQuery && typeof this._residentQuery.interrupt === "function") {
+      try {
+        this._residentQuery.interrupt()
+      } catch {
+        /* ignore */
+      }
+      this._busy = false
+      return
+    }
     if (this._abortController) {
       try {
         this._abortController.abort()
@@ -348,6 +513,8 @@ class ClaudeStreamSession {
   /** セッション終了。新規ターンを止め、実行中なら中断し、watcher も停止する。 */
   close() {
     this._closed = true
+    // 常駐 query は input キューを閉じて generator を終わらせ query を完走させる。
+    if (this._inputQueue) this._inputQueue.close()
     this.abortTurn()
     if (this._idleTimer) {
       clearTimeout(this._idleTimer)
@@ -415,6 +582,7 @@ export class ClaudeStreamBridge extends EventEmitter {
     maxTurns,
     maxThinkingTokens,
     resumeSessionId,
+    resident,
   }) {
     if (!stream_id) throw new TypeError("attach requires stream_id")
     if (this.sessions.has(stream_id)) {
@@ -446,6 +614,7 @@ export class ClaudeStreamBridge extends EventEmitter {
       maxThinkingTokens:
         typeof maxThinkingTokens === "number" ? maxThinkingTokens : null,
       resumeSessionId: resumeSessionId || null,
+      resident,
       sdk: this.sdk,
       logger: this.logger,
       onEvent: (event) => {