openAgent

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