@mastra/cloudflare-d1 1.3.2 → 1.4.0-alpha.0

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-cloudflare-d1
 description: Documentation for @mastra/cloudflare-d1. Use when working with @mastra/cloudflare-d1 APIs, configuration, or implementation.
 metadata:
   package: "@mastra/cloudflare-d1"
-  version: "1.3.2"
+  version: "1.4.0-alpha.0"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.3.2",
+  "version": "1.4.0-alpha.0",
   "package": "@mastra/cloudflare-d1",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -690,7 +690,9 @@ function rowToTask(row) {
 		createdAt: new Date(row.createdAt),
 		startedAt: row.startedAt ? new Date(row.startedAt) : void 0,
 		suspendedAt: row.suspendedAt ? new Date(row.suspendedAt) : void 0,
-		completedAt: row.completedAt ? new Date(row.completedAt) : void 0
+		completedAt: row.completedAt ? new Date(row.completedAt) : void 0,
+		ownerId: row.ownerId || void 0,
+		leaseExpiresAt: row.leaseExpiresAt ? new Date(row.leaseExpiresAt) : void 0
 	};
 }
 var BackgroundTasksStorageD1 = class extends _mastra_core_storage.BackgroundTasksStorage {
@@ -708,7 +710,12 @@ var BackgroundTasksStorageD1 = class extends _mastra_core_storage.BackgroundTask
 		await this.#db.alterTable({
 			tableName: _mastra_core_storage.TABLE_BACKGROUND_TASKS,
 			schema: _mastra_core_storage.TABLE_SCHEMAS[_mastra_core_storage.TABLE_BACKGROUND_TASKS],
-			ifNotExists: ["suspend_payload", "suspendedAt"]
+			ifNotExists: [
+				"suspend_payload",
+				"suspendedAt",
+				"ownerId",
+				"leaseExpiresAt"
+			]
 		});
 	}
 	async dangerouslyClearAll() {
@@ -735,7 +742,9 @@ var BackgroundTasksStorageD1 = class extends _mastra_core_storage.BackgroundTask
 			"createdAt",
 			"startedAt",
 			"suspendedAt",
-			"completedAt"
+			"completedAt",
+			"ownerId",
+			"leaseExpiresAt"
 		], [
 			task.id,
 			task.toolCallId,
@@ -755,7 +764,9 @@ var BackgroundTasksStorageD1 = class extends _mastra_core_storage.BackgroundTask
 			task.createdAt.toISOString(),
 			task.startedAt?.toISOString() ?? null,
 			task.suspendedAt?.toISOString() ?? null,
-			task.completedAt?.toISOString() ?? null
+			task.completedAt?.toISOString() ?? null,
+			task.ownerId ?? null,
+			task.leaseExpiresAt?.toISOString() ?? null
 		]).build();
 		await this.#db.executeQuery({
 			sql,
@@ -797,13 +808,34 @@ var BackgroundTasksStorageD1 = class extends _mastra_core_storage.BackgroundTask
 			sets.push("completedAt = ?");
 			params.push(update.completedAt?.toISOString() ?? null);
 		}
+		if ("ownerId" in update) {
+			sets.push("ownerId = ?");
+			params.push(update.ownerId ?? null);
+		}
+		if ("leaseExpiresAt" in update) {
+			sets.push("leaseExpiresAt = ?");
+			params.push(update.leaseExpiresAt?.toISOString() ?? null);
+		}
 		if (sets.length === 0) return false;
 		params.push(taskId);
-		const statusPredicate = options?.expectedStatus ? " AND status = ?" : "";
-		if (options?.expectedStatus) params.push(options.expectedStatus);
+		let predicates = "";
+		if (options?.expectedStatus) {
+			predicates += " AND status = ?";
+			params.push(options.expectedStatus);
+		}
+		if (options?.expectedOwnerId !== void 0) if (options.expectedOwnerId === null) predicates += " AND ownerId IS NULL";
+		else {
+			predicates += " AND ownerId = ?";
+			params.push(options.expectedOwnerId);
+		}
+		if (options?.expectedLeaseExpiresAt !== void 0) if (options.expectedLeaseExpiresAt === null) predicates += " AND leaseExpiresAt IS NULL";
+		else {
+			predicates += " AND leaseExpiresAt = ?";
+			params.push(options.expectedLeaseExpiresAt.toISOString());
+		}
 		const fullTableName = this.#db.getTableName(_mastra_core_storage.TABLE_BACKGROUND_TASKS);
 		const result = await this.#db.executeQuery({
-			sql: `UPDATE ${fullTableName} SET ${sets.join(", ")} WHERE id = ?${statusPredicate} RETURNING id`,
+			sql: `UPDATE ${fullTableName} SET ${sets.join(", ")} WHERE id = ?${predicates} RETURNING id`,
 			params
 		});
 		return Array.isArray(result) && result.length > 0;