feat: optional backfill of related entities

supabase · Oct 30, 2023 · 9f12577 · 9f12577
1 parent abfa564
commit 9f12577
Show file tree

Hide file tree

Showing 11 changed files with 161 additions and 90 deletions.
diff --git a/src/lib/charges.ts b/src/lib/charges.ts
@@ -9,11 +9,16 @@ import { stripe } from '../utils/StripeClientManager'
 
 const config = getConfig()
 
-export const upsertCharges = async (charges: Stripe.Charge[]): Promise<Stripe.Charge[]> => {
-  await Promise.all([
-    backfillCustomers(getUniqueIds(charges, 'customer')),
-    backfillInvoices(getUniqueIds(charges, 'invoice')),
-  ])
+export const upsertCharges = async (
+  charges: Stripe.Charge[],
+  backfillRelatedEntities: boolean = true
+): Promise<Stripe.Charge[]> => {
+  if (backfillRelatedEntities) {
+    await Promise.all([
+      backfillCustomers(getUniqueIds(charges, 'customer')),
+      backfillInvoices(getUniqueIds(charges, 'invoice')),
+    ])
+  }
 
   return upsertMany(charges, () =>
     constructUpsertSql(config.SCHEMA || 'stripe', 'charges', chargeSchema)

diff --git a/src/lib/disputes.ts b/src/lib/disputes.ts
@@ -7,8 +7,13 @@ import { getUniqueIds, upsertMany } from './database_utils'
 
 const config = getConfig()
 
-export const upsertDisputes = async (disputes: Stripe.Dispute[]): Promise<Stripe.Dispute[]> => {
-  await backfillCharges(getUniqueIds(disputes, 'charge'))
+export const upsertDisputes = async (
+  disputes: Stripe.Dispute[],
+  backfillRelatedEntities: boolean = true
+): Promise<Stripe.Dispute[]> => {
+  if (backfillRelatedEntities) {
+    await backfillCharges(getUniqueIds(disputes, 'charge'))
+  }
 
   return upsertMany(disputes, () =>
     constructUpsertSql(config.SCHEMA || 'stripe', 'disputes', disputeSchema)

diff --git a/src/lib/invoices.ts b/src/lib/invoices.ts
@@ -10,11 +10,16 @@ import { stripe } from '../utils/StripeClientManager'
 
 const config = getConfig()
 
-export const upsertInvoices = async (invoices: Invoice.Invoice[]): Promise<Invoice.Invoice[]> => {
-  await Promise.all([
-    backfillCustomers(getUniqueIds(invoices, 'customer')),
-    backfillSubscriptions(getUniqueIds(invoices, 'subscription')),
-  ])
+export const upsertInvoices = async (
+  invoices: Invoice.Invoice[],
+  backfillRelatedEntities: boolean = true
+): Promise<Invoice.Invoice[]> => {
+  if (backfillRelatedEntities) {
+    await Promise.all([
+      backfillCustomers(getUniqueIds(invoices, 'customer')),
+      backfillSubscriptions(getUniqueIds(invoices, 'subscription')),
+    ])
+  }
 
   return upsertMany(invoices, () =>
     constructUpsertSql(config.SCHEMA || 'stripe', 'invoices', invoiceSchema)

diff --git a/src/lib/payment_intents.ts b/src/lib/payment_intents.ts
@@ -9,12 +9,15 @@ import { paymentIntentSchema } from '../schemas/payment_intent'
 const config = getConfig()
 
 export const upsertPaymentIntents = async (
-  paymentIntents: Stripe.PaymentIntent[]
+  paymentIntents: Stripe.PaymentIntent[],
+  backfillRelatedEntities: boolean = true
 ): Promise<Stripe.PaymentIntent[]> => {
-  await Promise.all([
-    backfillCustomers(getUniqueIds(paymentIntents, 'customer')),
-    backfillInvoices(getUniqueIds(paymentIntents, 'invoice')),
-  ])
+  if (backfillRelatedEntities) {
+    await Promise.all([
+      backfillCustomers(getUniqueIds(paymentIntents, 'customer')),
+      backfillInvoices(getUniqueIds(paymentIntents, 'invoice')),
+    ])
+  }
 
   return upsertMany(paymentIntents, () =>
     constructUpsertSql(config.SCHEMA || 'stripe', 'payment_intents', paymentIntentSchema)

diff --git a/src/lib/payment_methods.ts b/src/lib/payment_methods.ts
@@ -8,9 +8,12 @@ import { getUniqueIds, upsertMany } from './database_utils'
 const config = getConfig()
 
 export const upsertPaymentMethods = async (
-  paymentMethods: Stripe.PaymentMethod[]
+  paymentMethods: Stripe.PaymentMethod[],
+  backfillRelatedEntities: boolean = true
 ): Promise<Stripe.PaymentMethod[]> => {
-  await backfillCustomers(getUniqueIds(paymentMethods, 'customer'))
+  if (backfillRelatedEntities) {
+    await backfillCustomers(getUniqueIds(paymentMethods, 'customer'))
+  }
 
   return upsertMany(paymentMethods, () =>
     constructUpsertSql(config.SCHEMA || 'stripe', 'payment_methods', paymentMethodsSchema)

diff --git a/src/lib/plans.ts b/src/lib/plans.ts
@@ -9,8 +9,13 @@ import { planSchema } from '../schemas/plan'
 
 const config = getConfig()
 
-export const upsertPlans = async (plans: Stripe.Plan[]): Promise<Stripe.Plan[]> => {
-  await backfillProducts(getUniqueIds(plans, 'product'))
+export const upsertPlans = async (
+  plans: Stripe.Plan[],
+  backfillRelatedEntities: boolean = true
+): Promise<Stripe.Plan[]> => {
+  if (backfillRelatedEntities) {
+    await backfillProducts(getUniqueIds(plans, 'product'))
+  }
 
   return upsertMany(plans, () => constructUpsertSql(config.SCHEMA || 'stripe', 'plans', planSchema))
 }

diff --git a/src/lib/prices.ts b/src/lib/prices.ts
@@ -9,8 +9,13 @@ import { getUniqueIds, upsertMany } from './database_utils'
 
 const config = getConfig()
 
-export const upsertPrices = async (prices: Price.Price[]): Promise<Price.Price[]> => {
-  await backfillProducts(getUniqueIds(prices, 'product'))
+export const upsertPrices = async (
+  prices: Price.Price[],
+  backfillRelatedEntities: boolean = true
+): Promise<Price.Price[]> => {
+  if (backfillRelatedEntities) {
+    await backfillProducts(getUniqueIds(prices, 'product'))
+  }
 
   return upsertMany(prices, () =>
     constructUpsertSql(config.SCHEMA || 'stripe', 'prices', priceSchema)

diff --git a/src/lib/setup_intents.ts b/src/lib/setup_intents.ts
@@ -8,9 +8,12 @@ import { getUniqueIds, upsertMany } from './database_utils'
 const config = getConfig()
 
 export const upsertSetupIntents = async (
-  setupIntents: Stripe.SetupIntent[]
+  setupIntents: Stripe.SetupIntent[],
+  backfillRelatedEntities: boolean = true
 ): Promise<Stripe.SetupIntent[]> => {
-  await backfillCustomers(getUniqueIds(setupIntents, 'customer'))
+  if (backfillRelatedEntities) {
+    await backfillCustomers(getUniqueIds(setupIntents, 'customer'))
+  }
 
   return upsertMany(setupIntents, () =>
     constructUpsertSql(config.SCHEMA || 'stripe', 'setup_intents', setupIntentsSchema)

diff --git a/src/lib/subscription_schedules.ts b/src/lib/subscription_schedules.ts
@@ -10,11 +10,14 @@ import { subscriptionScheduleSchema } from '../schemas/subscription_schedules'
 const config = getConfig()
 
 export const upsertSubscriptionSchedules = async (
-  subscriptionSchedules: Subscription.SubscriptionSchedule[]
+  subscriptionSchedules: Subscription.SubscriptionSchedule[],
+  backfillRelatedEntities: boolean = true
 ): Promise<Subscription.SubscriptionSchedule[]> => {
-  const customerIds = getUniqueIds(subscriptionSchedules, 'customer')
+  if (backfillRelatedEntities) {
+    const customerIds = getUniqueIds(subscriptionSchedules, 'customer')
 
-  await backfillCustomers(customerIds)
+    await backfillCustomers(customerIds)
+  }
 
   // Run it
   const rows = await upsertMany(subscriptionSchedules, () =>

diff --git a/src/lib/subscriptions.ts b/src/lib/subscriptions.ts
@@ -11,11 +11,14 @@ import Stripe from 'stripe'
 const config = getConfig()
 
 export const upsertSubscriptions = async (
-  subscriptions: Subscription.Subscription[]
+  subscriptions: Subscription.Subscription[],
+  backfillRelatedEntities: boolean = true
 ): Promise<Subscription.Subscription[]> => {
-  const customerIds = getUniqueIds(subscriptions, 'customer')
+  if (backfillRelatedEntities) {
+    const customerIds = getUniqueIds(subscriptions, 'customer')
 
-  await backfillCustomers(customerIds)
+    await backfillCustomers(customerIds)
+  }
 
   // Run it
   const rows = await upsertMany(subscriptions, () =>