tensorlake 0.5.4 → 0.5.6

package/dist/sandbox-image.js

@@ -3593,15 +3593,53 @@ var init_sandbox = __esm({
       }
       /** Connect to a sandbox VNC session for programmatic desktop control. */
       async connectDesktop(options) {
+        const port = options?.port ?? 5901;
+        const connectTimeout = options?.connectTimeout ?? 10;
+        const startMs = Date.now();
+        const deadlineMs = startMs + connectTimeout * 1e3;
+        await this.waitForPortReady(port, deadlineMs);
+        const remainingSecs = Math.max(0.1, (deadlineMs - Date.now()) / 1e3);
         return Desktop.connect({
           baseUrl: this.baseUrl,
           wsHeaders: this.wsHeaders,
-          port: options?.port,
+          port,
           password: options?.password,
           shared: options?.shared,
-          connectTimeout: options?.connectTimeout
+          connectTimeout: remainingSecs
         });
       }
+      /**
+       * Poll the in-sandbox daemon until `127.0.0.1:port` accepts a TCP connection.
+       * Uses `bash`'s `/dev/tcp` builtin via `processes/run` — no extra deps in
+       * the sandbox image. `bash` is present on every image we ship.
+       */
+      async waitForPortReady(port, deadlineMs) {
+        const probeIntervalMs = 250;
+        const probeProcessTimeoutSecs = 2;
+        let lastError;
+        while (Date.now() < deadlineMs) {
+          try {
+            const result = await this.run("/bin/bash", {
+              args: ["-c", `exec 3<>/dev/tcp/127.0.0.1/${port}`],
+              timeout: probeProcessTimeoutSecs
+            });
+            if (result.exitCode === 0) {
+              return;
+            }
+          } catch (error) {
+            lastError = error;
+          }
+          const remainingMs = deadlineMs - Date.now();
+          if (remainingMs <= 0) break;
+          await new Promise(
+            (resolve) => setTimeout(resolve, Math.min(probeIntervalMs, remainingMs))
+          );
+        }
+        const detail = lastError instanceof Error ? `: ${lastError.message}` : "";
+        throw new SandboxError(
+          `port ${port} did not become reachable inside sandbox within the connect timeout${detail}`
+        );
+      }
       ptyWsUrl(sessionId, token) {
         let wsBase;
         if (this.baseUrl.startsWith("https://")) {