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

package/dist/index.js

@@ -423,9 +423,14 @@ function handleLogicalOperator(key, value, parentPath) {
 	}
 	const values = [];
 	const joinOperator = key === "$or" || key === "$nor" ? "OR" : "AND";
-	const joined = (Array.isArray(value) ? value.map((f) => {
-		return (!!f ? Object.entries(f) : []).map(([k, v]) => buildCondition$1(k, v, key));
-	}) : [buildCondition$1(key, value, parentPath)]).flat().map((c) => {
+	const joined = (Array.isArray(value) ? value.flatMap((f) => {
+		const branch = (!!f ? Object.entries(f) : []).map(([k, v]) => buildCondition$1(k, v, key));
+		if (branch.length <= 1) return branch;
+		return [{
+			sql: `(${branch.map((c) => c.sql).join(" AND ")})`,
+			values: branch.flatMap((c) => c.values)
+		}];
+	}) : [buildCondition$1(key, value, parentPath)]).map((c) => {
 		values.push(...c.values);
 		return c.sql;
 	}).join(` ${joinOperator} `);
@@ -2953,7 +2958,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 +2994,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 +3042,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 +3083,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 +3098,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