openAgent

1import { Billing } from "../src/billing.js"
2import { and, Database, desc, eq, isNotNull, lt, sql } from "../src/drizzle/index.js"
3import { BillingTable, PaymentTable, SubscriptionTable } from "../src/schema/billing.sql.js"
4 
5const fromWrkID = process.argv[2]
6const toWrkID = process.argv[3]
7 
8if (!fromWrkID || !toWrkID) {
9  console.error("Usage: bun foo.ts <fromWrkID> <toWrkID>")
10  process.exit(1)
11}
12 
13console.log(`Transferring subscription from ${fromWrkID} to ${toWrkID}`)
14 
15// Look up the FROM workspace billing
16const fromBilling = await Database.use((tx) =>
17  tx
18    .select({
19      customerID: BillingTable.customerID,
20      subscriptionID: BillingTable.subscriptionID,
21      subscription: BillingTable.subscription,
22      paymentMethodID: BillingTable.paymentMethodID,
23      paymentMethodType: BillingTable.paymentMethodType,
24      paymentMethodLast4: BillingTable.paymentMethodLast4,
25    })
26    .from(BillingTable)
27    .where(eq(BillingTable.workspaceID, fromWrkID))
28    .then((rows) => rows[0]),
29)
30if (!fromBilling) throw new Error(`Error: FROM workspace has no billing record`)
31if (!fromBilling.customerID) throw new Error(`Error: FROM workspace has no Stripe customer ID`)
32if (!fromBilling.subscriptionID) throw new Error(`Error: FROM workspace has no subscription`)
33 
34const fromSubscription = await Database.use((tx) =>
35  tx
36    .select({ userID: SubscriptionTable.userID })
37    .from(SubscriptionTable)
38    .where(eq(SubscriptionTable.workspaceID, fromWrkID))
39    .then((rows) => rows[0]),
40)
41if (!fromSubscription) throw new Error(`Error: FROM workspace has no subscription`)
42 
43// Look up the previous customer ID in FROM workspace
44const subscriptionPayment = await Database.use((tx) =>
45  tx
46    .select({
47      customerID: PaymentTable.customerID,
48      timeCreated: PaymentTable.timeCreated,
49    })
50    .from(PaymentTable)
51    .where(and(eq(PaymentTable.workspaceID, fromWrkID), sql`JSON_EXTRACT(enrichment, '$.type') = 'subscription'`))
52    .then((rows) => {
53      if (rows.length > 1) {
54        console.error(`Error: Multiple subscription payments found for workspace ${fromWrkID}`)
55        process.exit(1)
56      }
57      return rows[0]
58    }),
59)
60const fromPrevPayment = await Database.use((tx) =>
61  tx
62    .select({ customerID: PaymentTable.customerID })
63    .from(PaymentTable)
64    .where(
65      and(
66        eq(PaymentTable.workspaceID, fromWrkID),
67        isNotNull(PaymentTable.customerID),
68        lt(PaymentTable.timeCreated, subscriptionPayment.timeCreated),
69      ),
70    )
71    .orderBy(desc(PaymentTable.timeCreated))
72    .limit(1)
73    .then((rows) => rows[0]),
74)
75if (!fromPrevPayment?.customerID) throw new Error(`Error: FROM workspace has no previous Stripe customer to revert to`)
76if (fromPrevPayment.customerID === fromBilling.customerID)
77  throw new Error(`Error: FROM workspace has the same Stripe customer ID as the current one`)
78 
79const fromPrevPaymentMethods = await Billing.stripe().customers.listPaymentMethods(fromPrevPayment.customerID, {})
80if (fromPrevPaymentMethods.data.length === 0)
81  throw new Error(`Error: FROM workspace has no previous Stripe payment methods`)
82 
83// Look up the TO workspace billing
84const toBilling = await Database.use((tx) =>
85  tx
86    .select({
87      customerID: BillingTable.customerID,
88      subscriptionID: BillingTable.subscriptionID,
89    })
90    .from(BillingTable)
91    .where(eq(BillingTable.workspaceID, toWrkID))
92    .then((rows) => rows[0]),
93)
94if (!toBilling) throw new Error(`Error: TO workspace has no billing record`)
95if (toBilling.subscriptionID) throw new Error(`Error: TO workspace already has a subscription`)
96 
97console.log(`FROM:`)
98console.log(`  Old Customer ID: ${fromBilling.customerID}`)
99console.log(`  New Customer ID: ${fromPrevPayment.customerID}`)
100console.log(`TO:`)
101console.log(`  Old Customer ID: ${toBilling.customerID}`)
102console.log(`  New Customer ID: ${fromBilling.customerID}`)
103 
104// Clear workspaceID from Stripe customer metadata
105await Billing.stripe().customers.update(fromPrevPayment.customerID, {
106  metadata: {
107    workspaceID: fromWrkID,
108  },
109})
110await Billing.stripe().customers.update(fromBilling.customerID, {
111  metadata: {
112    workspaceID: toWrkID,
113  },
114})
115 
116await Database.transaction(async (tx) => {
117  await tx
118    .update(BillingTable)
119    .set({
120      customerID: fromPrevPayment.customerID,
121      subscriptionID: null,
122      subscription: null,
123      paymentMethodID: fromPrevPaymentMethods.data[0].id,
124      paymentMethodLast4: fromPrevPaymentMethods.data[0].card?.last4 ?? null,
125      paymentMethodType: fromPrevPaymentMethods.data[0].type,
126    })
127    .where(eq(BillingTable.workspaceID, fromWrkID))
128 
129  await tx
130    .update(BillingTable)
131    .set({
132      customerID: fromBilling.customerID,
133      subscriptionID: fromBilling.subscriptionID,
134      subscription: fromBilling.subscription,
135      paymentMethodID: fromBilling.paymentMethodID,
136      paymentMethodLast4: fromBilling.paymentMethodLast4,
137      paymentMethodType: fromBilling.paymentMethodType,
138    })
139    .where(eq(BillingTable.workspaceID, toWrkID))
140 
141  await tx
142    .update(SubscriptionTable)
143    .set({
144      workspaceID: toWrkID,
145      userID: fromSubscription.userID,
146    })
147    .where(eq(SubscriptionTable.workspaceID, fromWrkID))
148 
149  await tx
150    .update(PaymentTable)
151    .set({
152      workspaceID: toWrkID,
153    })
154    .where(
155      and(
156        eq(PaymentTable.workspaceID, fromWrkID),
157        sql`JSON_EXTRACT(enrichment, '$.type') = 'subscription'`,
158        eq(PaymentTable.amount, 20000000000),
159      ),
160    )
161})
162 
163console.log(`done`)
164