@mastra/daytona 0.2.0 → 0.2.1-alpha.0

package/dist/index.js

@@ -1,4 +1,4 @@
-import { Daytona, DaytonaNotFoundError, DaytonaError, SandboxState } from '@daytonaio/sdk';
+import { Daytona, DaytonaNotFoundError, SandboxState } from '@daytonaio/sdk';
 import { SandboxProcessManager, MastraSandbox, SandboxNotReadyError, ProcessHandle } from '@mastra/core/workspace';
 import { createHash } from 'crypto';
 
@@ -478,6 +478,7 @@ var DaytonaSandbox = class _DaytonaSandbox extends MastraSandbox {
   autoDeleteInterval;
   volumeConfigs;
   sandboxName;
+  _daytonaSandboxId;
   sandboxUser;
   sandboxPublic;
   networkBlockAll;
@@ -562,6 +563,7 @@ var DaytonaSandbox = class _DaytonaSandbox extends MastraSandbox {
     const existing = await this.findExistingSandbox();
     if (existing) {
       this._sandbox = existing;
+      this._daytonaSandboxId = existing.id;
       this._createdAt = existing.createdAt ? new Date(existing.createdAt) : /* @__PURE__ */ new Date();
       this.logger.debug(`${LOG_PREFIX} Reconnected to existing sandbox ${existing.id} for: ${this.id}`);
       const expectedPaths = Array.from(this.mounts.entries.keys());
@@ -597,6 +599,7 @@ var DaytonaSandbox = class _DaytonaSandbox extends MastraSandbox {
       resources: this.resources
     }) : compact({ ...baseParams, snapshot: this.snapshotId });
     this._sandbox = await this._daytona.create(createParams);
+    this._daytonaSandboxId = this._sandbox.id;
     this.logger.debug(`${LOG_PREFIX} Created sandbox ${this._sandbox.id} for logical ID: ${this.id}`);
     this._createdAt = /* @__PURE__ */ new Date();
     await this.detectWorkingDir();
@@ -631,15 +634,17 @@ var DaytonaSandbox = class _DaytonaSandbox extends MastraSandbox {
       } catch {
       }
     } else if (!this._sandbox && this._daytona) {
-      try {
-        const orphan = await this._daytona.findOne({ labels: { "mastra-sandbox-id": this.id } });
-        if (orphan) {
+      const lookupKey = this._daytonaSandboxId ?? this.sandboxName;
+      if (lookupKey) {
+        try {
+          const orphan = await this._daytona.get(lookupKey);
           await this._daytona.delete(orphan);
+        } catch {
         }
-      } catch {
       }
     }
     this._sandbox = null;
+    this._daytonaSandboxId = void 0;
     this._daytona = null;
     this.mounts?.clear();
   }
@@ -1017,16 +1022,7 @@ var DaytonaSandbox = class _DaytonaSandbox extends MastraSandbox {
   // Internal Helpers
   // ---------------------------------------------------------------------------
   /**
-   * Try to find and reconnect to an existing Daytona sandbox.
-   *
-   * Uses two strategies:
-   * 1. `get()` by sandbox name — calls the getSandbox API directly, which
-   *    returns sandboxes in ANY state (including stopped). This is the
-   *    primary path and fixes reconnection after stop/start cycles.
-   * 2. `findOne()` by label — falls back to label-based search. Note: the
-   *    SDK's list() API only returns started sandboxes by default (no states
-   *    param in v0.143.0), so this only finds running sandboxes.
-   *
+   * Try to find and reconnect to an existing Daytona sandbox by ID.
    * Returns the sandbox if found and usable, or null if a fresh one should
    * be created.
    */
@@ -1050,31 +1046,19 @@ var DaytonaSandbox = class _DaytonaSandbox extends MastraSandbox {
       SandboxState.ERROR,
       SandboxState.BUILD_FAILED
     ];
-    let sandbox = null;
-    if (this.sandboxName) {
-      try {
-        sandbox = await this._daytona.get(this.sandboxName);
-      } catch (error) {
-        if (error instanceof DaytonaNotFoundError) ; else if (error instanceof DaytonaError && (error.statusCode === 401 || error.statusCode === 403)) {
-          throw error;
-        } else {
-          this.logger.debug(`${LOG_PREFIX} Transient error looking up sandbox by name: ${error}`);
-        }
-      }
+    const lookupKey = this._daytonaSandboxId ?? this.sandboxName;
+    if (!lookupKey) {
+      return null;
     }
-    if (!sandbox) {
-      try {
-        sandbox = await this._daytona.findOne({ labels: { "mastra-sandbox-id": this.id } });
-      } catch (error) {
-        if (error instanceof DaytonaNotFoundError || error instanceof Error && error.message.includes("No sandbox found")) {
-          return null;
-        }
-        if (error instanceof DaytonaError && (error.statusCode === 401 || error.statusCode === 403)) {
-          throw error;
-        }
-        this.logger.debug(`${LOG_PREFIX} Transient error looking up sandbox by label: ${error}`);
+    let sandbox;
+    try {
+      sandbox = await this._daytona.get(lookupKey);
+    } catch (error) {
+      if (error instanceof DaytonaNotFoundError) {
+        this._daytonaSandboxId = void 0;
         return null;
       }
+      throw error;
     }
     const state = sandbox.state;
     if (state && DEAD_STATES.includes(state)) {