openAgent

1import { Billing } from "../src/billing.js"
2import { Database, eq } from "../src/drizzle/index.js"
3import { BillingTable } from "../src/schema/billing.sql.js"
4import { WorkspaceTable } from "../src/schema/workspace.sql.js"
5import { microCentsToCents } from "../src/util/price.js"
6 
7// get input from command line
8const workspaceID = process.argv[2]
9 
10if (!workspaceID) {
11  console.error("Usage: bun freeze-workspace.ts <workspaceID>")
12  process.exit(1)
13}
14 
15// check workspace exists
16const workspace = await Database.use((tx) =>
17  tx
18    .select()
19    .from(WorkspaceTable)
20    .where(eq(WorkspaceTable.id, workspaceID))
21    .then((rows) => rows[0]),
22)
23if (!workspace) {
24  console.error("Error: Workspace not found")
25  process.exit(1)
26}
27 
28const billing = await Database.use((tx) =>
29  tx
30    .select()
31    .from(BillingTable)
32    .where(eq(BillingTable.workspaceID, workspaceID))
33    .then((rows) => rows[0]),
34)
35 
36const amountInDollars = microCentsToCents(billing.balance) / 100
37await Billing.grantCredit(workspaceID, 0 - amountInDollars)
38 
39console.log(`Removed payment of $${amountInDollars.toFixed(2)} from workspace ${workspaceID}`)
40