@qazuor/qzpay-drizzle 1.1.0 → 1.2.0

package/dist/index.cjs

@@ -473,18 +473,7 @@ function mapCoreGrantEntitlementToDrizzle(input, livemode) {
     grantedAt: /* @__PURE__ */ new Date(),
     expiresAt: input.expiresAt ?? null,
     source: input.source ?? "manual",
-    sourceId: null,
-    livemode
-  };
-}
-function mapCoreGrantEntitlementWithSourceToDrizzle(customerId, entitlementKey, source, sourceId, expiresAt, livemode) {
-  return {
-    customerId,
-    entitlementKey,
-    grantedAt: /* @__PURE__ */ new Date(),
-    expiresAt,
-    source,
-    sourceId,
+    sourceId: input.sourceId ?? null,
     livemode
   };
 }
@@ -630,19 +619,7 @@ function mapCoreSetLimitToDrizzle(input, livemode) {
     currentValue: 0,
     resetAt: input.resetAt ?? null,
     source: input.source ?? "manual",
-    sourceId: null,
-    livemode
-  };
-}
-function mapCoreSetLimitWithSourceToDrizzle(customerId, limitKey, maxValue, source, sourceId, resetAt, livemode) {
-  return {
-    customerId,
-    limitKey,
-    maxValue,
-    currentValue: 0,
-    resetAt,
-    source,
-    sourceId,
+    sourceId: input.sourceId ?? null,
     livemode
   };
 }
@@ -1134,6 +1111,15 @@ function mapCoreSubscriptionUpdateToDrizzle(input) {
   if (input.cancelAt !== void 0) {
     update.cancelAt = input.cancelAt;
   }
+  if (input.currentPeriodStart !== void 0) {
+    update.currentPeriodStart = input.currentPeriodStart;
+  }
+  if (input.currentPeriodEnd !== void 0) {
+    update.currentPeriodEnd = input.currentPeriodEnd;
+  }
+  if (input.trialEnd !== void 0) {
+    update.trialEnd = input.trialEnd;
+  }
   return update;
 }
 
@@ -1490,7 +1476,8 @@ var billingCustomerEntitlements = pgCore.pgTable(
     customerIdIdx: pgCore.index("idx_customer_entitlements_customer_id").on(table.customerId),
     entitlementKeyIdx: pgCore.index("idx_customer_entitlements_key").on(table.entitlementKey),
     customerKeyIdx: pgCore.index("idx_customer_entitlements_customer_key").on(table.customerId, table.entitlementKey),
-    expiresAtIdx: pgCore.index("idx_customer_entitlements_expires_at").on(table.expiresAt)
+    expiresAtIdx: pgCore.index("idx_customer_entitlements_expires_at").on(table.expiresAt),
+    sourceIdx: pgCore.index("idx_customer_entitlements_source").on(table.source, table.sourceId)
   })
 );
 var billingCustomerEntitlementInsertSchema = drizzleZod.createInsertSchema(billingCustomerEntitlements);
@@ -1626,7 +1613,8 @@ var billingCustomerLimits = pgCore.pgTable(
     customerIdIdx: pgCore.index("idx_customer_limits_customer_id").on(table.customerId),
     limitKeyIdx: pgCore.index("idx_customer_limits_key").on(table.limitKey),
     customerKeyIdx: pgCore.index("idx_customer_limits_customer_key").on(table.customerId, table.limitKey),
-    resetAtIdx: pgCore.index("idx_customer_limits_reset_at").on(table.resetAt)
+    resetAtIdx: pgCore.index("idx_customer_limits_reset_at").on(table.resetAt),
+    sourceIdx: pgCore.index("idx_customer_limits_source").on(table.source, table.sourceId)
   })
 );
 var billingCustomerLimitInsertSchema = drizzleZod.createInsertSchema(billingCustomerLimits);
@@ -3501,10 +3489,7 @@ var QZPayLimitsRepository = class {
    * Delete a customer limit
    */
   async delete(customerId, limitKey) {
-    const result = await this.db.delete(billingCustomerLimits).where(drizzleOrm.and(drizzleOrm.eq(billingCustomerLimits.customerId, customerId), drizzleOrm.eq(billingCustomerLimits.limitKey, limitKey))).returning();
-    if (result.length === 0) {
-      throw new Error(`Customer limit not found for customer ${customerId} and key ${limitKey}`);
-    }
+    await this.db.delete(billingCustomerLimits).where(drizzleOrm.and(drizzleOrm.eq(billingCustomerLimits.customerId, customerId), drizzleOrm.eq(billingCustomerLimits.limitKey, limitKey)));
   }
   /**
    * Delete all limits from a source
@@ -6045,6 +6030,9 @@ var QZPayDrizzleStorageAdapter = class {
       async revoke(customerId, entitlementKey) {
         await repo.revoke(customerId, entitlementKey);
       },
+      async revokeBySource(source, sourceId) {
+        return repo.revokeBySource(source, sourceId);
+      },
       async findByCustomerId(customerId) {
         const result = await repo.findByCustomerId(customerId);
         return result.map(mapDrizzleCustomerEntitlementToCore);
@@ -6078,6 +6066,12 @@ var QZPayDrizzleStorageAdapter = class {
         const result = await repo.set(drizzleInput);
         return mapDrizzleCustomerLimitToCore(result);
       },
+      async delete(customerId, limitKey) {
+        await repo.delete(customerId, limitKey);
+      },
+      async deleteBySource(source, sourceId) {
+        return repo.deleteBySource(source, sourceId);
+      },
       async increment(input) {
         const result = await repo.increment(input.customerId, input.limitKey, input.incrementBy);
         return mapDrizzleCustomerLimitToCore(result);
@@ -6656,7 +6650,6 @@ exports.mapCoreCustomerCreateToDrizzle = mapCoreCustomerCreateToDrizzle;
 exports.mapCoreCustomerUpdateToDrizzle = mapCoreCustomerUpdateToDrizzle;
 exports.mapCoreEntitlementToDrizzle = mapCoreEntitlementToDrizzle;
 exports.mapCoreGrantEntitlementToDrizzle = mapCoreGrantEntitlementToDrizzle;
-exports.mapCoreGrantEntitlementWithSourceToDrizzle = mapCoreGrantEntitlementWithSourceToDrizzle;
 exports.mapCoreInvoiceCreateToDrizzle = mapCoreInvoiceCreateToDrizzle;
 exports.mapCoreInvoiceLineCreateToDrizzle = mapCoreInvoiceLineCreateToDrizzle;
 exports.mapCoreInvoiceUpdateToDrizzle = mapCoreInvoiceUpdateToDrizzle;
@@ -6672,7 +6665,6 @@ exports.mapCorePriceUpdateToDrizzle = mapCorePriceUpdateToDrizzle;
 exports.mapCorePromoCodeCreateToDrizzle = mapCorePromoCodeCreateToDrizzle;
 exports.mapCorePromoCodeUpdateToDrizzle = mapCorePromoCodeUpdateToDrizzle;
 exports.mapCoreSetLimitToDrizzle = mapCoreSetLimitToDrizzle;
-exports.mapCoreSetLimitWithSourceToDrizzle = mapCoreSetLimitWithSourceToDrizzle;
 exports.mapCoreSubscriptionAddonCreateToDrizzle = mapCoreSubscriptionAddonCreateToDrizzle;
 exports.mapCoreSubscriptionAddonUpdateToDrizzle = mapCoreSubscriptionAddonUpdateToDrizzle;
 exports.mapCoreSubscriptionCreateToDrizzle = mapCoreSubscriptionCreateToDrizzle;