openAgent

1import { Database, eq, and, sql, inArray, isNull } from "../src/drizzle/index.js"
2import { BillingTable, BlackPlans } from "../src/schema/billing.sql.js"
3import { UserTable } from "../src/schema/user.sql.js"
4import { AuthTable } from "../src/schema/auth.sql.js"
5 
6const plan = process.argv[2] as (typeof BlackPlans)[number]
7if (!BlackPlans.includes(plan)) {
8  console.error("Usage: bun foo.ts <count>")
9  process.exit(1)
10}
11 
12const workspaces = await Database.use((tx) =>
13  tx
14    .select({ workspaceID: BillingTable.workspaceID })
15    .from(BillingTable)
16    .where(and(eq(BillingTable.subscriptionPlan, plan), isNull(BillingTable.timeSubscriptionSelected)))
17    .orderBy(sql`RAND()`)
18    .limit(100),
19)
20 
21console.log(`Found ${workspaces.length} workspaces on Black ${plan} waitlist`)
22 
23console.log("== Workspace IDs ==")
24const ids = workspaces.map((w) => w.workspaceID)
25for (const id of ids) {
26  console.log(id)
27}
28 
29console.log("\n== User Emails ==")
30const emails = await Database.use((tx) =>
31  tx
32    .select({ email: AuthTable.subject })
33    .from(UserTable)
34    .innerJoin(AuthTable, and(eq(UserTable.accountID, AuthTable.accountID), eq(AuthTable.provider, "email")))
35    .where(inArray(UserTable.workspaceID, ids)),
36)
37 
38const unique = new Set(emails.map((row) => row.email))
39for (const email of unique) {
40  console.log(email)
41}
42