@aexhq/sdk 0.59.0 → 0.59.2

package/dist/session.js

@@ -9,6 +9,7 @@ export class Sessions {
     #webSocketFactory;
     #customerEnvironments = new Map();
     #customerEnvironmentInstances = new Map();
+    #customerEnvironmentReadinessWaiters = new Map();
     #closed = false;
     constructor(transport, webSocketFactory) {
         this.#transport = transport;
@@ -237,16 +238,42 @@ export class Sessions {
                     }
                 }
             });
-            // The request may stop waiting, but the process-scoped runner remains reconnectable for
-            // later sessions. Its grant/reconnect lifetime must never inherit one create signal.
-            await waitWithSignal(starting, signal);
+            await this.#waitForCustomerEnvironmentReadiness(clientId, starting, signal);
             return;
         }
-        const hand = await waitWithSignal(existing, signal);
+        const hand = await this.#waitForCustomerEnvironmentReadiness(clientId, existing, signal);
         if (this.#closed)
             throw new SessionError("Aex client is closed");
         await waitWithSignal(hand.register(registrations), signal);
     }
+    async #waitForCustomerEnvironmentReadiness(clientId, starting, signal) {
+        this.#customerEnvironmentReadinessWaiters.set(clientId, (this.#customerEnvironmentReadinessWaiters.get(clientId) ?? 0) + 1);
+        let settled = false;
+        void starting.then(() => { settled = true; }, () => { settled = true; });
+        try {
+            return await waitWithSignal(starting, signal);
+        }
+        catch (error) {
+            if (signal?.aborted && !settled
+                && this.#customerEnvironmentReadinessWaiters.get(clientId) === 1
+                && this.#customerEnvironments.get(clientId) === starting) {
+                const partial = this.#customerEnvironmentInstances.get(clientId);
+                partial?.close();
+                this.#customerEnvironments.delete(clientId);
+                if (this.#customerEnvironmentInstances.get(clientId) === partial) {
+                    this.#customerEnvironmentInstances.delete(clientId);
+                }
+            }
+            throw error;
+        }
+        finally {
+            const waiters = this.#customerEnvironmentReadinessWaiters.get(clientId) ?? 1;
+            if (waiters <= 1)
+                this.#customerEnvironmentReadinessWaiters.delete(clientId);
+            else
+                this.#customerEnvironmentReadinessWaiters.set(clientId, waiters - 1);
+        }
+    }
 }
 function isApplicationEnvironment(value) {
     const config = value.config;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aexhq/sdk",
-  "version": "0.59.0",
+  "version": "0.59.2",
   "description": "TypeScript SDK for Aex sessions",
   "license": "Apache-2.0",
   "homepage": "https://aex.dev",
@@ -34,7 +34,7 @@
     "test": "tsc -p tsconfig.json && node --test"
   },
   "dependencies": {
-    "@aexhq/brain": "0.3.1",
+    "@aexhq/brain": "0.3.3",
     "@aexhq/env-app": "0.2.0"
   },
   "peerDependencies": {