@better-auth/drizzle-adapter 1.6.10 → 1.6.11

package/dist/index.mjs

@@ -40,7 +40,7 @@ function insensitiveNe(column, value) {
 //#region src/drizzle-adapter.ts
 const drizzleAdapter = (db, config) => {
 	let lazyOptions = null;
-	const createCustomAdapter = (db) => ({ getFieldName, getDefaultFieldName, options }) => {
+	const createCustomAdapter = (db, inTransaction = false) => ({ getFieldName, getDefaultFieldName, options }) => {
 		function getSchema(model) {
 			const schema = config.schema || db._.fullSchema;
 			if (!schema) throw new BetterAuthError("Drizzle adapter failed to initialize. Schema not found. Please provide a schema object in the adapter options object.");
@@ -419,6 +419,29 @@ const drizzleAdapter = (db, config) => {
 				});
 				return count;
 			},
+			async consumeOne({ model, where }) {
+				const schemaModel = getSchema(model);
+				const clause = convertWhereClause(where, model);
+				const idField = getFieldName({
+					model,
+					field: "id"
+				});
+				const idColumn = schemaModel[idField];
+				if (config.provider === "mysql") {
+					const claimFromTransaction = async (tx) => {
+						const target = (await tx.select().from(schemaModel).where(...clause).for("update").limit(1))[0];
+						if (!target) return null;
+						const targetId = target[idField] ?? target.id;
+						if (targetId === void 0 || targetId === null || !idColumn) return null;
+						const delRes = await tx.delete(schemaModel).where(eq(idColumn, targetId)).execute();
+						return ((delRes && (delRes.rowsAffected ?? delRes.affectedRows ?? delRes.changes)) ?? 0) > 0 ? target : null;
+					};
+					return inTransaction ? claimFromTransaction(db) : db.transaction(claimFromTransaction);
+				}
+				if (!idColumn) return null;
+				const targetIds = db.select({ id: idColumn }).from(schemaModel).where(...clause).limit(1);
+				return (await db.delete(schemaModel).where(inArray(idColumn, targetIds)).returning())[0] ?? null;
+			},
 			options: config
 		};
 	};
@@ -441,8 +464,11 @@ const drizzleAdapter = (db, config) => {
 			},
 			transaction: config.transaction ?? false ? (cb) => db.transaction((tx) => {
 				return cb(createAdapterFactory({
-					config: adapterOptions.config,
-					adapter: createCustomAdapter(tx)
+					config: {
+						...adapterOptions.config,
+						transaction: false
+					},
+					adapter: createCustomAdapter(tx, true)
 				})(lazyOptions));
 			}) : false
 		},

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@better-auth/drizzle-adapter",
-  "version": "1.6.10",
+  "version": "1.6.11",
   "description": "Drizzle adapter for Better Auth",
   "type": "module",
   "license": "MIT",
@@ -37,7 +37,7 @@
   "peerDependencies": {
     "@better-auth/utils": "0.4.0",
     "drizzle-orm": "^0.45.2",
-    "@better-auth/core": "^1.6.10"
+    "@better-auth/core": "^1.6.11"
   },
   "peerDependenciesMeta": {
     "drizzle-orm": {
@@ -49,7 +49,7 @@
     "drizzle-orm": "^0.45.2",
     "tsdown": "0.21.1",
     "typescript": "^5.9.3",
-    "@better-auth/core": "1.6.10"
+    "@better-auth/core": "1.6.11"
   },
   "scripts": {
     "build": "tsdown",