@mastra/platform-workspace 1.2.1 → 1.3.0-alpha.1

package/dist/index.js

@@ -1,6 +1,6 @@
 import { Buffer } from "buffer";
 import nodePath from "path";
-import { FileExistsError, FileNotFoundError, MastraFilesystem, MastraSandbox, ProcessHandle, SandboxNotReadyError, SandboxProcessManager, WorkspaceReadOnlyError } from "@mastra/core/workspace";
+import { FileExistsError, FileNotFoundError, MastraFilesystem, MastraSandbox, ProcessHandle, SandboxNotReadyError, SandboxProcessManager, UnsupportedStdinCloseError, WorkspaceReadOnlyError } from "@mastra/core/workspace";
 //#region src/client.ts
 const DEFAULT_PROXY_URL = "https://workspaces.mastra.ai";
 /**
@@ -800,6 +800,9 @@ var PlatformProcessHandle = class extends ProcessHandle {
 	async sendStdin() {
 		throw new Error("Platform sandbox command execution does not support stdin");
 	}
+	async closeStdin() {
+		throw new UnsupportedStdinCloseError("Platform sandbox command execution does not support closing stdin");
+	}
 };
 var PlatformProcessManager = class extends SandboxProcessManager {
 	spawnCounter = 0;
@@ -911,6 +914,12 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 	* failed/timed out, but now coalesced via `_leaseInFlight`).
 	*/
 	_transportReadyPromise = null;
+	/**
+	* The sidecar address of the most recent `start()`, kept so a timed-out
+	* probe can be restarted by a later exec ({@link _awaitTransportReady})
+	* instead of pinning the sandbox to the lease path for its lifetime.
+	*/
+	_probeTarget = null;
 	constructor(options = {}) {
 		super({
 			...options,
@@ -1063,14 +1072,24 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 	* HTTP round-trip.
 	*
 	* `null`/absent `instanceUrl` (proxy discovery failed, or an older proxy
-	* that predates the field) leaves the registry untouched — executes fall
-	* through to the lease path with no branch here.
+	* that predates the field) evicts any stale registry address and leaves
+	* executes on the lease path.
 	*/
 	_populateAddressFromResponse(json) {
 		if (!this._addressRegistry) return;
-		if (!json.instanceUrl) return;
+		if (!json.instanceUrl) {
+			this._probeGeneration++;
+			this._probeTarget = null;
+			this._transportReadyPromise = null;
+			this._addressRegistry.delete(json.id);
+			return;
+		}
 		this._addressRegistry.delete(json.id);
 		const generation = ++this._probeGeneration;
+		this._probeTarget = {
+			sandboxId: json.id,
+			instanceUrl: json.instanceUrl
+		};
 		this._transportReadyPromise = this._probeSidecarThenRegister(json.id, json.instanceUrl, generation);
 	}
 	/**
@@ -1080,8 +1099,9 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 	* fall back to the lease path until the probe succeeds.
 	*
 	* If the sidecar never comes up within {@link SIDECAR_PROBE_TIMEOUT_MS},
-	* the registry stays unpopulated and all execs go via lease for this
-	* sandbox's lifetime (or until a future `start()` re-runs the probe).
+	* the registry stays unpopulated, `_transportReadyPromise` is cleared,
+	* and a later {@link _awaitTransportReady} call restarts the probe
+	* instead of pinning this sandbox to the lease path.
 	*
 	* @param generation - The probe generation captured at call time. If this
 	*   no longer matches `_probeGeneration` when the probe succeeds, the probe
@@ -1115,6 +1135,7 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 			} catch {}
 			await new Promise((r) => setTimeout(r, SIDECAR_PROBE_INTERVAL_MS));
 		}
+		if (generation === this._probeGeneration && this._transportReadyPromise) this._transportReadyPromise = null;
 		this.logger.warn("platform-workspace probe timed out", {
 			sandboxId,
 			sessionId: this._client.sessionId,
@@ -1135,7 +1156,12 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 	*/
 	async _awaitTransportReady() {
 		if (this._sandboxId && this._addressRegistry?.get(this._sandboxId)) return;
-		if (!this._transportReadyPromise) return;
+		if (!this._transportReadyPromise) {
+			const target = this._probeTarget;
+			if (!target || this._sandboxId !== target.sandboxId) return;
+			const generation = ++this._probeGeneration;
+			this._transportReadyPromise = this._probeSidecarThenRegister(target.sandboxId, target.instanceUrl, generation);
+		}
 		await Promise.race([this._transportReadyPromise, new Promise((r) => setTimeout(r, TRANSPORT_READY_WAIT_MS))]);
 	}
 	/**
@@ -1202,6 +1228,8 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 		if (!this._sandboxId) return;
 		const destroyedSandboxId = this._sandboxId;
 		this._probeGeneration++;
+		this._probeTarget = null;
+		this._transportReadyPromise = null;
 		await this._client.request(`/sandbox/${encodeURIComponent(destroyedSandboxId)}`, { method: "DELETE" });
 		this._sandboxId = void 0;
 		this._createdAt = null;
@@ -1212,6 +1240,8 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 	async snapshot() {
 		await this.captureCheckpoint();
 	}
+	/** Snapshots persist real checkpoints that can seed future sandboxes. */
+	supportsCheckpoints = true;
 	/**
 	* Capture the sandbox's checkpoint on demand, outside any refresh timer the
 	* workspace-proxy owns internally.
@@ -1331,6 +1361,9 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 	* the cached `'running'` state (see `MastraSandbox._start`).
 	*/
 	_clearDestroyedState(destroyedSandboxId) {
+		this._probeGeneration++;
+		this._probeTarget = null;
+		this._transportReadyPromise = null;
 		this._sandboxId = void 0;
 		this._createdAt = null;
 		this._lease = null;