fix(console): recover canceled Go access
This commit is contained in:
parent
19231fce4b
commit
585df6864d
4 changed files with 247 additions and 21 deletions
74
packages/console/app/script/recover-lite-cancellations.ts
Normal file
74
packages/console/app/script/recover-lite-cancellations.ts
Normal file
|
|
@ -0,0 +1,74 @@
|
|||
import type { Stripe } from "stripe"
|
||||
import { Billing } from "@opencode-ai/console-core/billing.js"
|
||||
import { BlackData } from "@opencode-ai/console-core/black.js"
|
||||
import { Database, and, inArray, isNotNull } from "@opencode-ai/console-core/drizzle/index.js"
|
||||
import { LiteData } from "@opencode-ai/console-core/lite.js"
|
||||
import { BillingTable } from "@opencode-ai/console-core/schema/billing.sql.js"
|
||||
import { subscriptionKind } from "../src/routes/stripe/lite-cancellation.js"
|
||||
|
||||
const after = option("after")
|
||||
const before = option("before")
|
||||
if (!after || !before) throw new Error("Usage: recover-lite-cancellations.ts --after=<ISO> --before=<ISO> [--apply --expect=<count>]")
|
||||
|
||||
const start = new Date(after)
|
||||
const end = new Date(before)
|
||||
if (!Number.isFinite(start.getTime()) || !Number.isFinite(end.getTime())) throw new Error("Invalid recovery window")
|
||||
if (start >= end) throw new Error("Recovery window must end after it starts")
|
||||
if (end.getTime() - start.getTime() > 24 * 60 * 60 * 1000) throw new Error("Recovery window cannot exceed 24 hours")
|
||||
|
||||
const subscriptionIDs: string[] = []
|
||||
for await (const event of Billing.stripe().events.list({
|
||||
type: "customer.subscription.deleted",
|
||||
created: {
|
||||
gte: Math.floor(start.getTime() / 1000),
|
||||
lt: Math.floor(end.getTime() / 1000),
|
||||
},
|
||||
limit: 100,
|
||||
})) {
|
||||
const subscription = event.data.object as Stripe.Subscription
|
||||
if (
|
||||
subscriptionKind(subscription, {
|
||||
liteProductID: LiteData.productID(),
|
||||
blackProductID: BlackData.productID(),
|
||||
}) !== "lite"
|
||||
)
|
||||
continue
|
||||
subscriptionIDs.push(subscription.id)
|
||||
}
|
||||
|
||||
const stale = subscriptionIDs.length
|
||||
? await Database.use((tx) =>
|
||||
tx
|
||||
.select({ subscriptionID: BillingTable.liteSubscriptionID })
|
||||
.from(BillingTable)
|
||||
.where(and(isNotNull(BillingTable.liteSubscriptionID), inArray(BillingTable.liteSubscriptionID, subscriptionIDs)))
|
||||
.then((rows) => rows.map((row) => row.subscriptionID).filter((value): value is string => !!value)),
|
||||
)
|
||||
: []
|
||||
|
||||
const apply = process.argv.includes("--apply")
|
||||
const expected = option("expect")
|
||||
if (!apply) {
|
||||
console.log(JSON.stringify({ after: start.toISOString(), before: end.toISOString(), deletedGoSubscriptions: subscriptionIDs.length, staleEntitlements: stale.length, applied: false }))
|
||||
process.exit(0)
|
||||
}
|
||||
if (!expected || Number(expected) !== stale.length) {
|
||||
throw new Error(`Expected stale entitlement count must equal ${stale.length}`)
|
||||
}
|
||||
|
||||
let cursor = 0
|
||||
await Promise.all(
|
||||
Array.from({ length: Math.min(4, stale.length) }, async () => {
|
||||
while (true) {
|
||||
const index = cursor++
|
||||
if (index >= stale.length) return
|
||||
await Billing.unsubscribeLite({ subscriptionID: stale[index] })
|
||||
}
|
||||
}),
|
||||
)
|
||||
|
||||
console.log(JSON.stringify({ after: start.toISOString(), before: end.toISOString(), deletedGoSubscriptions: subscriptionIDs.length, staleEntitlements: stale.length, applied: true }))
|
||||
|
||||
function option(name: string) {
|
||||
return process.argv.find((value) => value.startsWith(`--${name}=`))?.slice(name.length + 3)
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue