@mastra/libsql 1.23.4-alpha.0 → 1.24.0-alpha.1

package/dist/index.js

@@ -2953,7 +2953,9 @@ function rowToTask(row) {
 		createdAt: new Date(String(row.createdAt)),
 		startedAt: row.startedAt ? new Date(String(row.startedAt)) : void 0,
 		suspendedAt: row.suspendedAt ? new Date(String(row.suspendedAt)) : void 0,
-		completedAt: row.completedAt ? new Date(String(row.completedAt)) : void 0
+		completedAt: row.completedAt ? new Date(String(row.completedAt)) : void 0,
+		ownerId: row.ownerId != null ? String(row.ownerId) : void 0,
+		leaseExpiresAt: row.leaseExpiresAt ? new Date(String(row.leaseExpiresAt)) : void 0
 	};
 }
 var BackgroundTasksLibSQL = class BackgroundTasksLibSQL extends BackgroundTasksStorage {
@@ -2987,7 +2989,12 @@ var BackgroundTasksLibSQL = class BackgroundTasksLibSQL extends BackgroundTasksS
 		await this.#db.alterTable({
 			tableName: TABLE_BACKGROUND_TASKS,
 			schema: TABLE_SCHEMAS[TABLE_BACKGROUND_TASKS],
-			ifNotExists: ["suspend_payload", "suspendedAt"]
+			ifNotExists: [
+				"suspend_payload",
+				"suspendedAt",
+				"ownerId",
+				"leaseExpiresAt"
+			]
 		});
 	}
 	async dangerouslyClearAll() {
@@ -3030,7 +3037,9 @@ var BackgroundTasksLibSQL = class BackgroundTasksLibSQL extends BackgroundTasksS
 				createdAt: task.createdAt.toISOString(),
 				startedAt: task.startedAt?.toISOString() ?? null,
 				suspendedAt: task.suspendedAt?.toISOString() ?? null,
-				completedAt: task.completedAt?.toISOString() ?? null
+				completedAt: task.completedAt?.toISOString() ?? null,
+				ownerId: task.ownerId ?? null,
+				leaseExpiresAt: task.leaseExpiresAt?.toISOString() ?? null
 			}
 		});
 	}
@@ -3069,6 +3078,14 @@ var BackgroundTasksLibSQL = class BackgroundTasksLibSQL extends BackgroundTasksS
 			setClauses.push("completedAt = ?");
 			params.push(update.completedAt?.toISOString() ?? null);
 		}
+		if ("ownerId" in update) {
+			setClauses.push("ownerId = ?");
+			params.push(update.ownerId ?? null);
+		}
+		if ("leaseExpiresAt" in update) {
+			setClauses.push("leaseExpiresAt = ?");
+			params.push(update.leaseExpiresAt?.toISOString() ?? null);
+		}
 		if (setClauses.length === 0) return false;
 		params.push(taskId);
 		let where = "id = ?";
@@ -3076,6 +3093,16 @@ var BackgroundTasksLibSQL = class BackgroundTasksLibSQL extends BackgroundTasksS
 			where += " AND status = ?";
 			params.push(options.expectedStatus);
 		}
+		if (options?.expectedOwnerId !== void 0) if (options.expectedOwnerId === null) where += " AND ownerId IS NULL";
+		else {
+			where += " AND ownerId = ?";
+			params.push(options.expectedOwnerId);
+		}
+		if (options?.expectedLeaseExpiresAt !== void 0) if (options.expectedLeaseExpiresAt === null) where += " AND leaseExpiresAt IS NULL";
+		else {
+			where += " AND leaseExpiresAt = ?";
+			params.push(options.expectedLeaseExpiresAt.toISOString());
+		}
 		return (await this.#client.execute({
 			sql: `UPDATE ${TABLE_BACKGROUND_TASKS} SET ${setClauses.join(", ")} WHERE ${where}`,
 			args: params