freeze-workspace.ts 1.1 KB

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