openAgent

1import { Billing } from "../src/billing.js"
2import { and, Database, eq, isNull } from "../src/drizzle/index.js"
3import { UserTable } from "../src/schema/user.sql.js"
4import { BillingTable, SubscriptionTable } from "../src/schema/billing.sql.js"
5import { Identifier } from "../src/identifier.js"
6import { AuthTable } from "../src/schema/auth.sql.js"
7import { BlackData } from "../src/black.js"
8 
9const plan = "200"
10const couponID = "JAIr0Pe1"
11const workspaceID = process.argv[2]
12const seats = parseInt(process.argv[3])
13 
14console.log(`Gifting ${seats} seats of Black to workspace ${workspaceID}`)
15 
16if (!workspaceID || !seats) throw new Error("Usage: bun foo.ts <workspaceID> <seats>")
17 
18// Get workspace user
19const users = await Database.use((tx) =>
20  tx
21    .select({
22      id: UserTable.id,
23      role: UserTable.role,
24      email: AuthTable.subject,
25    })
26    .from(UserTable)
27    .leftJoin(AuthTable, and(eq(AuthTable.accountID, UserTable.accountID), eq(AuthTable.provider, "email")))
28    .where(and(eq(UserTable.workspaceID, workspaceID), isNull(UserTable.timeDeleted))),
29)
30if (users.length === 0) throw new Error(`Error: No users found in workspace ${workspaceID}`)
31if (users.length !== seats)
32  throw new Error(`Error: Workspace ${workspaceID} has ${users.length} users, expected ${seats}`)
33const adminUser = users.find((user) => user.role === "admin")
34if (!adminUser) throw new Error(`Error: No admin user found in workspace ${workspaceID}`)
35if (!adminUser.email) throw new Error(`Error: Admin user ${adminUser.id} has no email`)
36 
37// Get Billing
38const billing = await Database.use((tx) =>
39  tx
40    .select({
41      customerID: BillingTable.customerID,
42      subscriptionID: BillingTable.subscriptionID,
43    })
44    .from(BillingTable)
45    .where(eq(BillingTable.workspaceID, workspaceID))
46    .then((rows) => rows[0]),
47)
48if (!billing) throw new Error(`Error: Workspace ${workspaceID} has no billing record`)
49if (billing.subscriptionID) throw new Error(`Error: Workspace ${workspaceID} already has a subscription`)
50 
51// Look up the Stripe customer by email
52const customerID =
53  billing.customerID ??
54  (await (() =>
55    Billing.stripe()
56      .customers.create({
57        email: adminUser.email,
58        metadata: {
59          workspaceID,
60        },
61      })
62      .then((customer) => customer.id))())
63console.log(`Customer ID: ${customerID}`)
64 
65const subscription = await Billing.stripe().subscriptions.create({
66  customer: customerID!,
67  items: [
68    {
69      price: BlackData.planToPriceID({ plan }),
70      discounts: [{ coupon: couponID }],
71      quantity: seats,
72    },
73  ],
74  metadata: {
75    workspaceID,
76  },
77})
78console.log(`Subscription ID: ${subscription.id}`)
79 
80await Database.transaction(async (tx) => {
81  // Set customer id, subscription id, and payment method on workspace billing
82  await tx
83    .update(BillingTable)
84    .set({
85      customerID,
86      subscriptionID: subscription.id,
87      subscription: { status: "subscribed", coupon: couponID, seats, plan },
88    })
89    .where(eq(BillingTable.workspaceID, workspaceID))
90 
91  // Create a row in subscription table
92  for (const user of users) {
93    await tx.insert(SubscriptionTable).values({
94      workspaceID,
95      id: Identifier.create("subscription"),
96      userID: user.id,
97    })
98  }
99  //
100  //  // Create a row in payments table
101  //  await tx.insert(PaymentTable).values({
102  //    workspaceID,
103  //    id: Identifier.create("payment"),
104  //    amount: centsToMicroCents(amountInCents),
105  //    customerID,
106  //    invoiceID,
107  //    paymentID,
108  //    enrichment: {
109  //      type: "subscription",
110  //      couponID,
111  //    },
112  //  })
113})
114 
115console.log(`done`)
116