36 lines
1023 B
TypeScript
36 lines
1023 B
TypeScript
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}`)
|