@blade-hq/agent-client 1.2.1-beta.13 → 1.2.1-beta.15

package/dist/index.js

@@ -3166,7 +3166,7 @@ var AgentSession = class _AgentSession {
         countFiles: async () => readonlyError(),
         downloadFile: async () => readonlyError(),
         attachAppCli: async () => readonlyError(),
-        onDispose: () => void 0
+        onDispose: () => true
       },
       "connected"
     );
@@ -3424,15 +3424,18 @@ var AgentSession = class _AgentSession {
   downloadFile(filePath, downloadName) {
     return this.runtime.downloadFile(this.sessionId, filePath, downloadName);
   }
-  /** 释放订阅与监听器。之后该实例不再接收任何事件。 */
+  /**
+   * 归还一份持有。同一个会话被多处 connect 时按引用计数释放：只有最后
+   * 一份归还才真正断开订阅与监听器，之前的归还不影响仍在使用的一方。
+   */
   dispose() {
     if (this.disposed) return;
+    if (!this.runtime.onDispose(this)) return;
     this.disposed = true;
     this.cancelPatchFlush();
     this.listeners.clear();
     this.emitter.clear();
     this.commands.clear();
-    this.runtime.onDispose(this);
   }
   get isDisposed() {
     return this.disposed;
@@ -3882,6 +3885,8 @@ var SessionHub = class {
   client;
   sessions = /* @__PURE__ */ new Map();
   bootstrapPending = /* @__PURE__ */ new Map();
+  /** 每个会话当前有多少方持有（connect 一次 +1，dispose 一次 -1）。 */
+  holdCounts = /* @__PURE__ */ new Map();
   connection = "disconnected";
   connectionListeners = /* @__PURE__ */ new Set();
   bound = false;
@@ -3900,22 +3905,30 @@ var SessionHub = class {
       this.connectionListeners.delete(listener);
     };
   }
-  /** 取得（或复用）一个会话的 AgentSession，并完成历史加载 + 房间订阅。 */
+  /**
+   * 取得（或复用）一个会话的 AgentSession，并完成历史加载 + 房间订阅。
+   *
+   * 每次调用都算一份持有，调用方用完必须 `dispose()` 归还。复用同一实例时
+   * 只有最后一份归还才退订房间——否则 StrictMode 双跑、路由抖动这类"连两次
+   * 只释放一次"的场景会把仍在用的会话退订掉，之后所有 turn/chat 事件静默丢失。
+   */
   async connect(sessionId) {
     const existing = this.sessions.get(sessionId);
     if (existing && !existing.isDisposed) {
+      this.hold(sessionId);
       return this.bootstrapPending.get(sessionId) ?? existing;
     }
     this.ensureBound();
     const session = new AgentSession(sessionId, this.runtimeFor(), this.connection);
     this.sessions.set(sessionId, session);
+    this.hold(sessionId);
     this.ensureConnected();
     const bootstrap = session._bootstrap().then(() => session);
     this.bootstrapPending.set(sessionId, bootstrap);
     try {
       return await bootstrap;
     } catch (error) {
-      session.dispose();
+      this.forceRelease(session);
       throw error;
     } finally {
       if (this.bootstrapPending.get(sessionId) === bootstrap) {
@@ -3946,7 +3959,7 @@ var SessionHub = class {
   disposeAll() {
     this.bootstrapPending.clear();
     for (const session of [...this.sessions.values()]) {
-      session.dispose();
+      this.forceRelease(session);
     }
   }
   /**
@@ -3983,15 +3996,36 @@ var SessionHub = class {
       countFiles: (sessionId, dirPath) => this.client.sessions.countFiles(sessionId, dirPath),
       downloadFile: (sessionId, filePath, downloadName) => this.client.sessions.downloadFile(sessionId, filePath, downloadName),
       attachAppCli: (sessionId, definition) => this.client.sessions.attachAppCli(sessionId, definition),
-      onDispose: (session) => {
-        if (this.sessions.get(session.sessionId) === session) {
-          this.sessions.delete(session.sessionId);
-        }
-        ;
-        this.client.socket().emit("session:unsubscribe", { session_id: session.sessionId });
-      }
+      onDispose: (session) => this.release(session)
     };
   }
+  /** 记一份持有。 */
+  hold(sessionId) {
+    this.holdCounts.set(sessionId, (this.holdCounts.get(sessionId) ?? 0) + 1);
+  }
+  /**
+   * 归还一份持有；还剩别的持有者时返回 false，让 AgentSession 保持可用。
+   * 归零才退订房间并从路由表移除。
+   */
+  release(session) {
+    const remaining = (this.holdCounts.get(session.sessionId) ?? 0) - 1;
+    if (remaining > 0) {
+      this.holdCounts.set(session.sessionId, remaining);
+      return false;
+    }
+    this.holdCounts.delete(session.sessionId);
+    if (this.sessions.get(session.sessionId) === session) {
+      this.sessions.delete(session.sessionId);
+    }
+    ;
+    this.client.socket().emit("session:unsubscribe", { session_id: session.sessionId });
+    return true;
+  }
+  /** 不管还有几方持有都立即释放（启动失败、切换身份）。 */
+  forceRelease(session) {
+    this.holdCounts.delete(session.sessionId);
+    session.dispose();
+  }
   ensureConnected() {
     const socket = this.client.socket();
     if (socket.connected || socket.active) return;
@@ -4179,7 +4213,7 @@ function resolveAuthToken(options) {
 
 // src/version.ts
 var SDK_NAME = "agent-client";
-var SDK_VERSION = true ? "1.2.1-beta.13" : "1.1.1";
+var SDK_VERSION = true ? "1.2.1-beta.15" : "1.1.1";
 
 // src/socket.ts
 function withSdkIdentity(auth) {