wip: zen black

packages/console/core/script/black-select-workspaces.ts

@@ -0,0 +1,41 @@
+import { Database, eq, and, sql, inArray, isNull, count } from "../src/drizzle/index.js"
+import { BillingTable, SubscriptionPlan } from "../src/schema/billing.sql.js"
+import { UserTable } from "../src/schema/user.sql.js"
+import { AuthTable } from "../src/schema/auth.sql.js"
+
+const plan = process.argv[2] as typeof SubscriptionPlan[number]
+if (!SubscriptionPlan.includes(plan)) {
+  console.error("Usage: bun foo.ts <count>")
+  process.exit(1)
+}
+
+const workspaces = await Database.use((tx) =>
+  tx
+    .select({ workspaceID: BillingTable.workspaceID })
+    .from(BillingTable)
+    .where(and(eq(BillingTable.subscriptionPlan, plan), isNull(BillingTable.timeSubscriptionSelected)))
+    .orderBy(sql`RAND()`)
+    .limit(100),
+)
+
+console.log(`Found ${workspaces.length} workspaces on Black ${plan} waitlist`)
+
+console.log("== Workspace IDs ==")
+const ids = workspaces.map((w) => w.workspaceID)
+for (const id of ids) {
+  console.log(id)
+}
+
+console.log("\n== User Emails ==")
+const emails = await Database.use((tx) =>
+  tx
+    .select({ email: AuthTable.subject })
+    .from(UserTable)
+    .innerJoin(AuthTable, and(eq(UserTable.accountID, AuthTable.accountID), eq(AuthTable.provider, "email")))
+    .where(inArray(UserTable.workspaceID, ids)),
+)
+
+const unique = new Set(emails.map((row) => row.email))
+for (const email of unique) {
+  console.log(email)
+}