@qazuor/qzpay-drizzle 1.2.0 → 1.4.0

package/dist/index.cjs

@@ -1262,6 +1262,7 @@ var billingPromoCodes = pgCore.pgTable(
     maxUses: pgCore.integer("max_uses"),
     usedCount: pgCore.integer("used_count").default(0),
     maxPerCustomer: pgCore.integer("max_per_customer").default(1),
+    maxUsesPerUser: pgCore.integer("max_uses_per_user").notNull().default(1),
     validPlans: pgCore.text("valid_plans").array(),
     newCustomersOnly: pgCore.boolean("new_customers_only").default(false),
     existingCustomersOnly: pgCore.boolean("existing_customers_only").default(false),
@@ -1291,7 +1292,8 @@ var billingPromoCodeUsage = pgCore.pgTable(
   },
   (table) => ({
     codeIdx: pgCore.index("idx_promo_usage_code").on(table.promoCodeId),
-    customerIdx: pgCore.index("idx_promo_usage_customer").on(table.customerId)
+    customerIdx: pgCore.index("idx_promo_usage_customer").on(table.customerId),
+    customerPromoUnique: pgCore.uniqueIndex("promo_code_usage_customer_promo_unique").on(table.customerId, table.promoCodeId)
   })
 );
 var billingPromoCodeInsertSchema = drizzleZod.createInsertSchema(billingPromoCodes);
@@ -4384,6 +4386,28 @@ var QZPayPromoCodesRepository = class {
     }).where(drizzleOrm.eq(billingPromoCodes.id, id)).returning();
     return firstOrThrow(result, "PromoCode", id);
   }
+  /**
+   * Atomically increment the usage count if (and only if) the promo
+   * code has not reached its `maxUses` limit yet.
+   *
+   * Implemented as a single conditional `UPDATE ... WHERE used_count <
+   * max_uses RETURNING *` statement so it is race-safe under
+   * concurrency. Returns `null` when the increment would exceed the
+   * limit; callers should treat that as "redemption limit reached".
+   *
+   * Promo codes with `maxUses = null` (no limit) always succeed.
+   */
+  async atomicIncrementUsage(id) {
+    const result = await this.db.update(billingPromoCodes).set({
+      usedCount: drizzleOrm.sql`COALESCE(${billingPromoCodes.usedCount}, 0) + 1`
+    }).where(
+      drizzleOrm.and(
+        drizzleOrm.eq(billingPromoCodes.id, id),
+        drizzleOrm.or(drizzleOrm.isNull(billingPromoCodes.maxUses), drizzleOrm.sql`COALESCE(${billingPromoCodes.usedCount}, 0) < ${billingPromoCodes.maxUses}`)
+      )
+    ).returning();
+    return firstOrNull(result);
+  }
   /**
    * Deactivate promo code
    */
@@ -5949,6 +5973,10 @@ var QZPayDrizzleStorageAdapter = class {
       async incrementRedemptions(id) {
         await repo.incrementUsage(id);
       },
+      async atomicIncrementRedemptions(id) {
+        const result = await repo.atomicIncrementUsage(id);
+        return result ? mapDrizzlePromoCodeToCore(result) : null;
+      },
       async list(options) {
         const limit = options?.limit ?? 20;
         const offset = options?.offset ?? 0;