summaryrefslogtreecommitdiffhomepage
path: root/packages/console/core/script/credit-workspace.ts
blob: ad2404f799abc544ff70300aa5c90b207f0a3cba (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import { Billing } from "../src/billing.js"
import { Database, eq } from "../src/drizzle/index.js"
import { WorkspaceTable } from "../src/schema/workspace.sql.js"

// get input from command line
const workspaceID = process.argv[2]
const dollarAmount = process.argv[3]

if (!workspaceID || !dollarAmount) {
  console.error("Usage: bun credit-workspace.ts <workspaceID> <dollarAmount>")
  process.exit(1)
}

// check workspace exists
const workspace = await Database.use((tx) =>
  tx
    .select()
    .from(WorkspaceTable)
    .where(eq(WorkspaceTable.id, workspaceID))
    .then((rows) => rows[0]),
)
if (!workspace) {
  console.error("Error: Workspace not found")
  process.exit(1)
}

const amountInDollars = parseFloat(dollarAmount)
if (isNaN(amountInDollars) || amountInDollars <= 0) {
  console.error("Error: dollarAmount must be a positive number")
  process.exit(1)
}

await Billing.grantCredit(workspaceID, amountInDollars)

console.log(`Added payment of $${amountInDollars.toFixed(2)} to workspace ${workspaceID}`)