fix(console): recover canceled Go access

This commit is contained in:
Jack 2026-08-01 06:39:12 +00:00
commit 585df6864d
4 changed files with 247 additions and 21 deletions

View 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)
}