testdriverai 7.9.0-test.39 → 7.9.0-test.40

package/agent/lib/sandbox.js

@@ -12,6 +12,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
       this._ably = null;
       this._sessionChannel = null;
       this._channelName = null;
+      this._membersChannelName = null;
       this.ps = {};
       this._execBuffers = {}; // accumulate streamed exec.output chunks per requestId
       this.heartbeat = null;
@@ -53,7 +54,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
       return this._publishCount;
     }
 
-    async _initAbly(ablyToken, channelName) {
+    async _initAbly(ablyToken, channelName, membersChannelName) {
       if (this._ably) {
         try {
           this._ably.close();
@@ -62,6 +63,19 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
         }
       }
       this._channelName = channelName;
+
+      // Derive the members channel from the session channel if not supplied.
+      // Format: testdriver:{env}:{teamId}:{sandboxId} → testdriver:{env}:{teamId}:members
+      if (!membersChannelName) {
+        const parts = channelName.split(":");
+        if (parts.length >= 3) {
+          membersChannelName = parts.slice(0, 3).join(":") + ":members";
+        } else {
+          logger.warn("[ably] Channel name format unexpected (" + channelName + "), cannot derive members channel");
+        }
+      }
+      this._membersChannelName = membersChannelName;
+
       var self = this;
 
       this._ably = new Ably.Realtime({
@@ -105,7 +119,24 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
 
       logger.debug(`[realtime] Channel initialized: ${channelName}`);
 
-      // Enter presence on the session channel so the API can count connected SDK clients
+      // Enter presence on the team members channel so the API can count
+      // connected SDK clients with a single direct lookup per team.
+      if (membersChannelName) {
+        try {
+          var membersChannel = this._ably.channels.get(membersChannelName);
+          await membersChannel.presence.enter({
+            sandboxId: this._sandboxId,
+            connectedAt: Date.now(),
+          });
+          logger.debug(`[realtime] Entered presence on members channel (sandbox=${this._sandboxId})`);
+        } catch (e) {
+          // Non-fatal — presence is used for concurrency counting, not critical path
+          logger.warn("Failed to enter presence on members channel: " + (e.message || e));
+        }
+      }
+
+      // Enter presence on the session channel so the API's session monitor can
+      // detect SDK connect/disconnect events for this sandbox.
       try {
         await this._sessionChannel.presence.enter({
           sandboxId: this._sandboxId,
@@ -113,7 +144,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
         });
         logger.debug(`[realtime] Entered presence on session channel (sandbox=${this._sandboxId})`);
       } catch (e) {
-        // Non-fatal — presence is used for concurrency counting, not critical path
+        // Non-fatal — presence is used for disconnect detection, not critical path
         logger.warn("Failed to enter presence on session channel: " + (e.message || e));
       }
 
@@ -522,7 +553,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
       this._teamId = reply.teamId;
 
       if (reply.ably && reply.ably.token) {
-        await this._initAbly(reply.ably.token, reply.ably.channel);
+        await this._initAbly(reply.ably.token, reply.ably.channel, reply.ably.membersChannel);
         this.instanceSocketConnected = true;
 
         // Tell the runner to enable debug log forwarding if debug mode is on
@@ -648,7 +679,7 @@ const createSandbox = function (emitter, analytics, sessionInstance) {
           if (reply.status === 'pending' && reply.ably && reply.ably.token) {
             this._sandboxId = reply.sandboxId;
             this._teamId = reply.teamId;
-            await this._initAbly(reply.ably.token, reply.ably.channel);
+            await this._initAbly(reply.ably.token, reply.ably.channel, reply.ably.membersChannel);
             this.instanceSocketConnected = true;
           }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "testdriverai",
-  "version": "7.9.0-test.39",
+  "version": "7.9.0-test.40",
   "description": "Next generation autonomous AI agent for end-to-end testing of web & desktop",
   "main": "sdk.js",
   "types": "sdk.d.ts",