mirror of
https://github.com/yu-i-i/overleaf-cep.git
synced 2026-06-04 14:49:01 +02:00
[web] add script for updating prices for existing Stripe subscriptions (#30216)
GitOrigin-RevId: 8f20a6ba9c7a737bb304d8698c4fbcb06b6ddc8c
This commit is contained in:
@@ -0,0 +1,746 @@
|
||||
#!/usr/bin/env node
|
||||
|
||||
/**
|
||||
* This script changes prices for existing Stripe subscriptions.
|
||||
* It schedules changes to apply at the next renewal.
|
||||
*
|
||||
* Usage:
|
||||
* node scripts/stripe/change_existing_subscription_prices.mjs [OPTS] [INPUT-FILE]
|
||||
*
|
||||
* Options:
|
||||
* --region REGION Either 'uk' or 'us' (required)
|
||||
* --timeframe TIMEFRAME Either 'renewal' or 'now' (default: renewal)
|
||||
* --output PATH Output file path (default: /tmp/change_prices_output_<timestamp>.csv)
|
||||
* Use '-' to write to stdout
|
||||
* --commit Apply changes (without this flag, runs in dry-run mode)
|
||||
* --throttle DURATION Minimum time (in ms) between subscriptions processed (default: 100)
|
||||
* --force Overwrite any existing pending changes
|
||||
* --help Show a help message
|
||||
*
|
||||
* CSV Input Format:
|
||||
* The CSV must have the following columns:
|
||||
* - subscription_id: Stripe subscription id
|
||||
* - current_lookup_key: Current price lookup key
|
||||
* - new_lookup_key: New price lookup key
|
||||
* - current_add_on_lookup_key: Current price lookup key for add-on (optional)
|
||||
* - new_add_on_lookup_key: New price lookup key for add-on (optional)
|
||||
*
|
||||
* Output:
|
||||
* Writes a CSV with columns:
|
||||
* - subscription_id: The subscription id processed
|
||||
* - status: Result status (changed, validated, not-found, inactive, mismatch, pending-change, or error)
|
||||
* - note: Additional information about the status (includes dry run notice when not using --commit)
|
||||
*
|
||||
* Running on a Pod:
|
||||
* This script may run for multiple days. When running using `rake run:longpod[ENV,web]`,
|
||||
* use one of these strategies to preserve output:
|
||||
*
|
||||
* 1. Tail the output file from another session (the filename is logged when the script starts):
|
||||
* kubectl exec -it <pod-name> -- tail -f /tmp/change_prices_output_<timestamp>.csv > local_backup.csv
|
||||
*
|
||||
* 2. Periodically copy the output file to your laptop:
|
||||
* kubectl cp <pod-name>:/tmp/change_prices_output_<timestamp>.csv ./backup.csv
|
||||
*
|
||||
* 3. Write to stdout and capture locally:
|
||||
* kubectl exec -it <pod-name> -- node scripts/stripe/change_existing_subscription_prices.mjs \
|
||||
* --timeframe renewal --commit --output - input.csv > output.csv
|
||||
*
|
||||
* For monitoring handoffs, have the next person start tailing (or copying periodically)
|
||||
* before the current monitor disconnects to ensure no records are lost.
|
||||
*/
|
||||
|
||||
import fs from 'node:fs'
|
||||
import path from 'node:path'
|
||||
import { setTimeout } from 'node:timers/promises'
|
||||
import * as csv from 'csv'
|
||||
import minimist from 'minimist'
|
||||
import AnalyticsManager from '../../app/src/Features/Analytics/AnalyticsManager.mjs'
|
||||
import { z } from '../../app/src/infrastructure/Validation.mjs'
|
||||
import { scriptRunner } from '../lib/ScriptRunner.mjs'
|
||||
import { getRegionClient } from '../../modules/subscriptions/app/src/StripeClient.mjs'
|
||||
import {
|
||||
ReportError,
|
||||
getProductIdFromPrice,
|
||||
getProductIdFromItem,
|
||||
getPriceIdFromItem,
|
||||
} from './helpers.mjs'
|
||||
|
||||
/**
|
||||
* @import { CSVSubscriptionChange, Timeframe, StripeClient } from './helpers.mjs'
|
||||
* @import Stripe from 'stripe'
|
||||
* @import { ReadStream } from 'node:fs'
|
||||
* @import { Parser } from 'csv-parse'
|
||||
* @import { Stringifier } from 'csv-stringify'
|
||||
*/
|
||||
|
||||
// 100 ms corresponds to 10 requests per second (cautious rate within Stripe's 100 req/s limit)
|
||||
const DEFAULT_THROTTLE = 100
|
||||
|
||||
/**
|
||||
* cache for price objects to avoid redundant Stripe API calls
|
||||
* @type {Map<string, Stripe.Price>}
|
||||
*/
|
||||
const priceCache = new Map()
|
||||
|
||||
/**
|
||||
* Print usage information to stderr
|
||||
*/
|
||||
function usage() {
|
||||
console.error(`Usage: node scripts/stripe/change_existing_subscription_prices.mjs [OPTS] [INPUT-FILE]
|
||||
|
||||
Options:
|
||||
--region REGION Either 'uk' or 'us' (required)
|
||||
--timeframe TIMEFRAME Either 'renewal' or 'now' (default: renewal)
|
||||
--output PATH Output file path (default: /tmp/change_prices_output_<timestamp>.csv)
|
||||
Use '-' to write to stdout
|
||||
--commit Apply changes (without this, runs in dry-run mode)
|
||||
--throttle DURATION Minimum time between requests in ms (default: ${DEFAULT_THROTTLE})
|
||||
--force Overwrite any existing pending changes
|
||||
--help Show this help message
|
||||
|
||||
See the source file header for detailed documentation on CSV format and pod usage.
|
||||
`)
|
||||
}
|
||||
|
||||
/**
|
||||
* Main script entry point
|
||||
* @param {function(string): Promise<void>} trackProgress - Function to log progress messages
|
||||
*/
|
||||
async function main(trackProgress) {
|
||||
const opts = parseArgs()
|
||||
const timestamp = new Date().toISOString().replace(/[:.]/g, '-')
|
||||
const outputFile = opts.output ?? `/tmp/change_prices_output_${timestamp}.csv`
|
||||
|
||||
const stripeClient = getRegionClient(opts.region)
|
||||
|
||||
await trackProgress('Starting price change script for Stripe')
|
||||
await trackProgress(`Region: ${opts.region}`)
|
||||
await trackProgress(
|
||||
`Timeframe: ${opts.timeframe === 'now' ? 'now (immediate)' : 'renewal (at next cycle)'}`
|
||||
)
|
||||
await trackProgress(`Run mode: ${opts.commit ? 'COMMIT' : 'DRY RUN'}`)
|
||||
await trackProgress(`Throttle: ${opts.throttle}ms between requests`)
|
||||
await trackProgress(`Force mode: ${opts.force ? 'enabled' : 'disabled'}`)
|
||||
|
||||
const inputStream = opts.inputFile
|
||||
? fs.createReadStream(opts.inputFile)
|
||||
: process.stdin
|
||||
const csvReader = getCsvReader(inputStream)
|
||||
const csvWriter = getCsvWriter(outputFile)
|
||||
|
||||
await trackProgress(`Output: ${outputFile === '-' ? 'stdout' : outputFile}`)
|
||||
|
||||
let processedCount = 0
|
||||
let successCount = 0
|
||||
let errorCount = 0
|
||||
|
||||
let lastLoopTimestamp = 0
|
||||
for await (const change of csvReader) {
|
||||
const timeSinceLastLoop = Date.now() - lastLoopTimestamp
|
||||
if (timeSinceLastLoop < opts.throttle) {
|
||||
await setTimeout(opts.throttle - timeSinceLastLoop)
|
||||
}
|
||||
lastLoopTimestamp = Date.now()
|
||||
|
||||
processedCount++
|
||||
|
||||
try {
|
||||
const subscription = await processChange(
|
||||
change,
|
||||
stripeClient,
|
||||
opts.commit,
|
||||
opts.force,
|
||||
opts.timeframe
|
||||
)
|
||||
|
||||
if (opts.commit && subscription) {
|
||||
try {
|
||||
const userId = subscription.customer.metadata?.userId
|
||||
await AnalyticsManager.recordEventForUser(
|
||||
userId,
|
||||
'script_price_change',
|
||||
{
|
||||
subscriptionId: change.subscription_id,
|
||||
}
|
||||
)
|
||||
} catch (err) {
|
||||
await trackProgress(
|
||||
`Warning: failed to record analytics event after successful price change for ${change.subscription_id}: ${err.message}`
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
csvWriter.write({
|
||||
subscription_id: change.subscription_id,
|
||||
status: opts.commit ? 'changed' : 'validated',
|
||||
note: opts.commit ? undefined : 'dry run - no changes applied',
|
||||
})
|
||||
successCount++
|
||||
|
||||
if (processedCount % 10 === 0) {
|
||||
await trackProgress(
|
||||
`Processed ${processedCount} subscriptions (${successCount} ${opts.commit ? 'changed' : 'validated'}, ${errorCount} errors)`
|
||||
)
|
||||
}
|
||||
} catch (err) {
|
||||
errorCount++
|
||||
if (err instanceof ReportError) {
|
||||
csvWriter.write({
|
||||
subscription_id: change.subscription_id,
|
||||
status: err.status,
|
||||
note: err.message,
|
||||
})
|
||||
} else {
|
||||
csvWriter.write({
|
||||
subscription_id: change.subscription_id,
|
||||
status: 'error',
|
||||
note: err.message,
|
||||
})
|
||||
await trackProgress(
|
||||
`Error processing ${change.subscription_id}: ${err.message}`
|
||||
)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
await trackProgress('\n✨ FINAL SUMMARY ✨')
|
||||
await trackProgress(`📊 Total processed: ${processedCount}`)
|
||||
if (opts.commit) {
|
||||
await trackProgress(`✅ Successfully changed: ${successCount}`)
|
||||
} else {
|
||||
await trackProgress(`✅ Successfully validated: ${successCount}`)
|
||||
await trackProgress('ℹ️ DRY RUN: No changes were applied to Stripe')
|
||||
}
|
||||
await trackProgress(`❌ Errors: ${errorCount}`)
|
||||
await trackProgress('🎉 Script completed!')
|
||||
|
||||
csvWriter.end()
|
||||
}
|
||||
|
||||
/**
|
||||
* Get a CSV parser configured for subscription change input
|
||||
* @param {ReadStream | NodeJS.ReadableStream} inputStream - The input stream to parse
|
||||
* @returns {Parser} The configured CSV parser
|
||||
*/
|
||||
function getCsvReader(inputStream) {
|
||||
const parser = csv.parse({
|
||||
columns: true,
|
||||
})
|
||||
inputStream.pipe(parser)
|
||||
return parser
|
||||
}
|
||||
|
||||
/**
|
||||
* Get a CSV stringifier configured for output
|
||||
* @param {string} outputFile - The output file path to write to, or '-' for stdout
|
||||
* @returns {Stringifier} The configured CSV stringifier
|
||||
*/
|
||||
function getCsvWriter(outputFile) {
|
||||
let outputStream
|
||||
if (outputFile === '-') {
|
||||
outputStream = process.stdout
|
||||
} else {
|
||||
fs.mkdirSync(path.dirname(outputFile), { recursive: true })
|
||||
outputStream = fs.createWriteStream(outputFile)
|
||||
}
|
||||
const writer = csv.stringify({
|
||||
columns: ['subscription_id', 'status', 'note'],
|
||||
header: true,
|
||||
})
|
||||
writer.on('error', err => {
|
||||
console.error(err)
|
||||
process.exit(1)
|
||||
})
|
||||
writer.pipe(outputStream)
|
||||
return writer
|
||||
}
|
||||
|
||||
/**
|
||||
* Process a single subscription change
|
||||
* @param {CSVSubscriptionChange} change - The subscription change to process
|
||||
* @param {StripeClient} stripeClient - The Stripe client for the region
|
||||
* @param {boolean} commit - Whether to commit changes or run in dry-run mode
|
||||
* @param {boolean} force - Whether to overwrite existing pending changes
|
||||
* @param {Timeframe} timeframe - When to apply the change
|
||||
* @returns {Promise<Stripe.Subscription | undefined>} The subscription if commit mode, undefined otherwise
|
||||
*/
|
||||
async function processChange(change, stripeClient, commit, force, timeframe) {
|
||||
const subscription = await fetchSubscription(
|
||||
change.subscription_id,
|
||||
stripeClient
|
||||
)
|
||||
|
||||
const nextPrices = await fetchPrices(
|
||||
[change.new_lookup_key, change.new_add_on_lookup_key].filter(Boolean),
|
||||
stripeClient
|
||||
)
|
||||
|
||||
validateChange(change, subscription, nextPrices, force)
|
||||
|
||||
if (!commit) {
|
||||
// dry run mode - validation passed, no changes applied
|
||||
return
|
||||
}
|
||||
|
||||
let updatedSubscription
|
||||
if (timeframe === 'now') {
|
||||
updatedSubscription = await updateSubscriptionImmediately(
|
||||
subscription,
|
||||
stripeClient,
|
||||
nextPrices
|
||||
)
|
||||
} else {
|
||||
updatedSubscription = await createSubscriptionSchedule(
|
||||
subscription,
|
||||
stripeClient,
|
||||
nextPrices
|
||||
)
|
||||
}
|
||||
|
||||
return updatedSubscription
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch a subscription from Stripe
|
||||
* @param {string} subscriptionId - The Stripe subscription id
|
||||
* @param {StripeClient} stripeClient - The Stripe client for the region
|
||||
* @returns {Promise<Stripe.Subscription>} The subscription with expanded items and schedule
|
||||
* @throws {ReportError} If subscription is not found
|
||||
*/
|
||||
async function fetchSubscription(subscriptionId, stripeClient) {
|
||||
try {
|
||||
const subscription = await stripeClient.stripe.subscriptions.retrieve(
|
||||
subscriptionId,
|
||||
{
|
||||
expand: ['schedule', 'discounts'],
|
||||
}
|
||||
)
|
||||
return subscription
|
||||
} catch (err) {
|
||||
if (err.type === 'StripeInvalidRequestError' && err.statusCode === 404) {
|
||||
throw new ReportError('not-found', 'subscription not found')
|
||||
}
|
||||
throw err
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch price entities from Stripe with caching
|
||||
* @param {string[]} lookupKeys
|
||||
* @param {StripeClient} stripeClient - The Stripe client for the region
|
||||
* @returns {Promise<Stripe.Price[]>} the fetched prices
|
||||
* @throws {ReportError} If price is not found
|
||||
*/
|
||||
async function fetchPrices(lookupKeys, stripeClient) {
|
||||
if (lookupKeys.length === 0) {
|
||||
throw new ReportError('not-found', 'no lookup keys provided')
|
||||
}
|
||||
|
||||
const cachedPrices = []
|
||||
const keysToFetch = []
|
||||
|
||||
for (const key of lookupKeys) {
|
||||
if (priceCache.has(key)) {
|
||||
cachedPrices.push(priceCache.get(key))
|
||||
} else {
|
||||
keysToFetch.push(key)
|
||||
}
|
||||
}
|
||||
|
||||
if (keysToFetch.length === 0) {
|
||||
return cachedPrices
|
||||
}
|
||||
|
||||
try {
|
||||
const results = await stripeClient.stripe.prices.list({
|
||||
lookup_keys: keysToFetch,
|
||||
limit: keysToFetch.length,
|
||||
})
|
||||
|
||||
if (results.data.length === 0) {
|
||||
throw new ReportError(
|
||||
'not-found',
|
||||
`${keysToFetch.join(', ')} prices not found in Stripe`
|
||||
)
|
||||
}
|
||||
|
||||
for (const price of results.data) {
|
||||
priceCache.set(price.lookup_key, price)
|
||||
}
|
||||
|
||||
return [...cachedPrices, ...results.data]
|
||||
} catch (err) {
|
||||
if (err.type === 'StripeInvalidRequestError' && err.statusCode === 404) {
|
||||
throw new ReportError(
|
||||
'not-found',
|
||||
`${keysToFetch.join(', ')} prices not found in Stripe`
|
||||
)
|
||||
}
|
||||
throw err
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Helper function to validate that the proposed price looks correct
|
||||
* @param {Stripe.Price} currentPrice - The current price
|
||||
* @param {Stripe.Price[]} nextPrices - Available next prices
|
||||
* @throws {ReportError} If validation fails
|
||||
*/
|
||||
function validateProposedPrice(currentPrice, nextPrices) {
|
||||
const currentProductId = getProductIdFromPrice(currentPrice)
|
||||
|
||||
if (!currentProductId) {
|
||||
throw new ReportError('mismatch', 'current price has no associated product')
|
||||
}
|
||||
|
||||
// Find matching new price by product ID
|
||||
const matchingNewPrice = nextPrices.find(p => {
|
||||
const nextProductId = getProductIdFromPrice(p)
|
||||
return nextProductId === currentProductId
|
||||
})
|
||||
|
||||
// Verify new price with matching product exists
|
||||
if (!matchingNewPrice) {
|
||||
throw new ReportError(
|
||||
'mismatch',
|
||||
`no new prices found belonging to the same product: current ${currentProductId}`
|
||||
)
|
||||
}
|
||||
|
||||
// Verify currency matches
|
||||
if (currentPrice.currency !== matchingNewPrice.currency) {
|
||||
throw new ReportError(
|
||||
'mismatch',
|
||||
`currency mismatch found: current ${currentPrice.currency}, new ${matchingNewPrice.currency}`
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Validate that the subscription matches the expected state
|
||||
* @param {CSVSubscriptionChange} change - The subscription change to validate
|
||||
* @param {Stripe.Subscription} subscription - The Stripe subscription
|
||||
* @param {Stripe.Price[]} nextPrices - The next prices for the subscription
|
||||
* @param {boolean} force - Whether to ignore existing pending changes
|
||||
* @throws {ReportError} If validation fails
|
||||
*/
|
||||
function validateChange(change, subscription, nextPrices, force) {
|
||||
// Check subscription is updatable
|
||||
const inactiveStatuses = [
|
||||
'incomplete',
|
||||
'incomplete_expired',
|
||||
'canceled',
|
||||
'trialing',
|
||||
]
|
||||
if (inactiveStatuses.includes(subscription.status)) {
|
||||
throw new ReportError(
|
||||
'inactive',
|
||||
`subscription status: ${subscription.status}`
|
||||
)
|
||||
}
|
||||
|
||||
// Skip subscriptions that are scheduled to be canceled
|
||||
if (subscription.cancel_at_period_end) {
|
||||
throw new ReportError(
|
||||
'inactive',
|
||||
'subscription is scheduled to be canceled at period end'
|
||||
)
|
||||
}
|
||||
|
||||
// Skip subscriptions that already have a schedule (unless force mode)
|
||||
if (subscription.schedule && !force) {
|
||||
if (typeof subscription.schedule === 'string') {
|
||||
throw new ReportError(
|
||||
'pending-change',
|
||||
`subscription has a schedule (${subscription.schedule}) - re-run with expanded schedule data`
|
||||
)
|
||||
}
|
||||
if (subscription.schedule.status !== 'released') {
|
||||
throw new ReportError(
|
||||
'pending-change',
|
||||
'subscription already has an active schedule'
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
// Verify all requested next prices were found
|
||||
const requestedLookupKeys = [
|
||||
change.new_lookup_key,
|
||||
change.new_add_on_lookup_key,
|
||||
].filter(Boolean)
|
||||
|
||||
for (const requestedKey of requestedLookupKeys) {
|
||||
if (!nextPrices.find(p => p.lookup_key === requestedKey)) {
|
||||
throw new ReportError(
|
||||
'not-found',
|
||||
`requested price with lookup key ${requestedKey} not found in Stripe`
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
// Verify current lookup keys exist in subscription
|
||||
const currentLookupKey = change.current_lookup_key
|
||||
const currentItem = subscription.items.data.find(
|
||||
item => item.price.lookup_key === currentLookupKey
|
||||
)
|
||||
if (!currentItem) {
|
||||
throw new ReportError(
|
||||
'mismatch',
|
||||
`current_lookup_key ${currentLookupKey} not found in subscription items`
|
||||
)
|
||||
}
|
||||
|
||||
if (change.current_add_on_lookup_key) {
|
||||
const currentAddOnItem = subscription.items.data.find(
|
||||
item => item.price.lookup_key === change.current_add_on_lookup_key
|
||||
)
|
||||
if (!currentAddOnItem) {
|
||||
throw new ReportError(
|
||||
'mismatch',
|
||||
`current_add_on_lookup_key ${change.current_add_on_lookup_key} not found in subscription items`
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
// Verify the proposed new prices match the current prices' products
|
||||
// For plan price
|
||||
const currentPrice = currentItem.price
|
||||
validateProposedPrice(currentPrice, nextPrices)
|
||||
|
||||
// For add-on price (if present)
|
||||
if (change.current_add_on_lookup_key) {
|
||||
const currentAddOnItem = subscription.items.data.find(
|
||||
item => item.price.lookup_key === change.current_add_on_lookup_key
|
||||
)
|
||||
const currentAddOnPrice = currentAddOnItem.price
|
||||
validateProposedPrice(currentAddOnPrice, nextPrices)
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Helper function to find a matching new price for a subscription item
|
||||
* @param {Stripe.SubscriptionItem} item - The subscription item
|
||||
* @param {Stripe.Price[]} nextPrices - Available next prices
|
||||
* @returns {string} The matching price ID or the current price ID if no match
|
||||
*/
|
||||
function findMatchingPriceId(item, nextPrices) {
|
||||
const itemProductId = getProductIdFromItem(item)
|
||||
|
||||
if (!itemProductId) {
|
||||
// No product ID available, keep current price
|
||||
return item.price.id
|
||||
}
|
||||
|
||||
// Find matching new price by product ID
|
||||
const matchingPrice = nextPrices.find(price => {
|
||||
const priceProductId = getProductIdFromPrice(price)
|
||||
return priceProductId === itemProductId
|
||||
})
|
||||
|
||||
return matchingPrice?.id || item.price.id
|
||||
}
|
||||
|
||||
/**
|
||||
* Update subscription prices immediately without proration
|
||||
* @param {Stripe.Subscription} subscription - The Stripe subscription
|
||||
* @param {StripeClient} stripeClient - The Stripe client for the region
|
||||
* @param {Stripe.Price[]} nextPrices - The next prices for the subscription
|
||||
* @returns {Promise<Stripe.Subscription>} The updated subscription
|
||||
* @throws {Error} If unable to build lookup key for plan
|
||||
* @throws {ReportError} If new price doesn't match expected price
|
||||
*/
|
||||
async function updateSubscriptionImmediately(
|
||||
subscription,
|
||||
stripeClient,
|
||||
nextPrices
|
||||
) {
|
||||
// Release any existing schedule. This only executes when --force is enabled,
|
||||
// as validation would have rejected subscriptions with active schedules otherwise.
|
||||
if (
|
||||
subscription.schedule &&
|
||||
typeof subscription.schedule !== 'string' &&
|
||||
subscription.schedule.status !== 'released'
|
||||
) {
|
||||
await stripeClient.stripe.subscriptionSchedules.release(
|
||||
subscription.schedule.id
|
||||
)
|
||||
}
|
||||
|
||||
// NOTE: The `id` field is required for all items when using this endpoint,
|
||||
// otherwise Stripe will append the subscription with the new item instead of
|
||||
// replacing the current item.
|
||||
const subscriptionItems = subscription.items.data.map(item => ({
|
||||
id: item.id,
|
||||
price: findMatchingPriceId(item, nextPrices),
|
||||
quantity: item.quantity || 1,
|
||||
}))
|
||||
|
||||
const updatedSubscription = await stripeClient.stripe.subscriptions.update(
|
||||
subscription.id,
|
||||
{
|
||||
items: subscriptionItems,
|
||||
proration_behavior: 'none',
|
||||
expand: ['customer'],
|
||||
}
|
||||
)
|
||||
|
||||
return updatedSubscription
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a subscription schedule to change prices at renewal
|
||||
* @param {Stripe.Subscription} subscription - The Stripe subscription
|
||||
* @param {StripeClient} stripeClient - The Stripe client for the region
|
||||
* @param {Stripe.Price[]} nextPrices - The next prices for the subscription
|
||||
* @returns {Promise<Stripe.Subscription>} The updated subscription
|
||||
* @throws {Error} If unable to build lookup key for plan
|
||||
* @throws {ReportError} If new price doesn't match expected price
|
||||
*/
|
||||
async function createSubscriptionSchedule(
|
||||
subscription,
|
||||
stripeClient,
|
||||
nextPrices
|
||||
) {
|
||||
// Release any existing schedule. This only executes when --force is enabled,
|
||||
// as validation would have rejected subscriptions with active schedules otherwise.
|
||||
if (
|
||||
subscription.schedule &&
|
||||
typeof subscription.schedule !== 'string' &&
|
||||
subscription.schedule.status !== 'released'
|
||||
) {
|
||||
await stripeClient.stripe.subscriptionSchedules.release(
|
||||
subscription.schedule.id
|
||||
)
|
||||
}
|
||||
|
||||
// NOTE: The `id` field cannot be used with this endpoint
|
||||
const nextPhaseItems = subscription.items.data.map(item => ({
|
||||
price: findMatchingPriceId(item, nextPrices),
|
||||
quantity: item.quantity || 1,
|
||||
}))
|
||||
|
||||
// Create a subscription schedule
|
||||
const schedule = await stripeClient.stripe.subscriptionSchedules.create({
|
||||
from_subscription: subscription.id,
|
||||
expand: ['subscription', 'subscription.customer'],
|
||||
})
|
||||
|
||||
const currentPhase = schedule.phases[0]
|
||||
|
||||
// Update the schedule to include the new phase starting at the end of the current billing period
|
||||
// If the update fails, release the schedule to clean up the state
|
||||
try {
|
||||
const currentPhaseConfig = {
|
||||
start_date: currentPhase.start_date,
|
||||
end_date: currentPhase.end_date,
|
||||
items: currentPhase.items.map(item => ({
|
||||
price: getPriceIdFromItem(item),
|
||||
quantity: item.quantity,
|
||||
})),
|
||||
}
|
||||
|
||||
const nextPhaseConfig = {
|
||||
start_date: currentPhase.end_date,
|
||||
items: nextPhaseItems,
|
||||
}
|
||||
|
||||
// Stripe doesn't copy discount settings from subscription to schedule
|
||||
// so we need to manually preserve them
|
||||
if (subscription.discounts) {
|
||||
const discounts = subscription.discounts
|
||||
.map(discount => {
|
||||
if (discount.promotion_code) {
|
||||
return {
|
||||
promotion_code: discount.promotion_code,
|
||||
}
|
||||
} else if (discount.coupon) {
|
||||
return {
|
||||
coupon:
|
||||
typeof discount.coupon === 'string'
|
||||
? discount.coupon
|
||||
: discount.coupon.id,
|
||||
}
|
||||
}
|
||||
return {}
|
||||
})
|
||||
.filter(d => d.coupon || d.promotion_code)
|
||||
|
||||
currentPhaseConfig.discounts = discounts
|
||||
nextPhaseConfig.discounts = discounts
|
||||
}
|
||||
|
||||
await stripeClient.stripe.subscriptionSchedules.update(schedule.id, {
|
||||
phases: [currentPhaseConfig, nextPhaseConfig],
|
||||
end_behavior: 'release',
|
||||
})
|
||||
} catch (err) {
|
||||
// If the update fails, release the schedule to prevent it from remaining in an invalid state
|
||||
try {
|
||||
await stripeClient.stripe.subscriptionSchedules.release(schedule.id)
|
||||
} catch (releaseErr) {
|
||||
// Do nothing, and throw the original error
|
||||
}
|
||||
throw err
|
||||
}
|
||||
|
||||
return schedule.subscription
|
||||
}
|
||||
|
||||
const paramsSchema = z.object({
|
||||
region: z.enum(['uk', 'us']),
|
||||
timeframe: z.enum(['renewal', 'now']).default('renewal'),
|
||||
output: z.string().optional(),
|
||||
commit: z.boolean().default(false),
|
||||
force: z.boolean().default(false),
|
||||
throttle: z
|
||||
.string()
|
||||
.optional()
|
||||
.transform(val => (val ? parseInt(val, 10) : DEFAULT_THROTTLE)),
|
||||
_: z.array(z.string()).max(1),
|
||||
help: z.boolean().optional(),
|
||||
})
|
||||
|
||||
/**
|
||||
* Parse command line arguments
|
||||
* @returns {{inputFile: string | undefined, output: string | undefined, force: boolean, commit: boolean, timeframe: 'renewal' | 'now', throttle: number, region: 'uk' | 'us'}} Parsed options
|
||||
*/
|
||||
function parseArgs() {
|
||||
const argv = minimist(process.argv.slice(2), {
|
||||
string: ['throttle', 'timeframe', 'output', 'region'],
|
||||
boolean: ['help', 'force', 'commit'],
|
||||
})
|
||||
|
||||
if (argv.help) {
|
||||
usage()
|
||||
process.exit(0)
|
||||
}
|
||||
|
||||
const parseResult = paramsSchema.safeParse(argv)
|
||||
|
||||
if (!parseResult.success) {
|
||||
console.error(`Invalid parameters: ${parseResult.error.message}`)
|
||||
usage()
|
||||
process.exit(1)
|
||||
}
|
||||
|
||||
const { region, timeframe, output, commit, force, throttle, _ } =
|
||||
parseResult.data
|
||||
|
||||
return {
|
||||
inputFile: _[0],
|
||||
output,
|
||||
force,
|
||||
commit,
|
||||
timeframe,
|
||||
throttle,
|
||||
region,
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
await scriptRunner(main)
|
||||
process.exit(0)
|
||||
} catch (error) {
|
||||
console.error(error)
|
||||
process.exit(1)
|
||||
}
|
||||
@@ -0,0 +1,80 @@
|
||||
/* eslint-disable @overleaf/require-script-runner */
|
||||
// This file contains helper functions used by other scripts.
|
||||
// The scripts that import these helpers should use Script Runner.
|
||||
|
||||
/**
|
||||
* @import Stripe from 'stripe'
|
||||
* @import { getRegionClient } from '../../modules/subscriptions/app/src/StripeClient.mjs'
|
||||
*/
|
||||
|
||||
/**
|
||||
* @export
|
||||
* @typedef {Object} CSVSubscriptionChange
|
||||
* @property {string} subscription_id
|
||||
* @property {string} current_lookup_key
|
||||
* @property {string} new_lookup_key
|
||||
* @property {string} current_add_on_lookup_key
|
||||
* @property {string} new_add_on_lookup_key
|
||||
*/
|
||||
|
||||
/**
|
||||
* @export
|
||||
* @typedef {'renewal' | 'now'} Timeframe
|
||||
*/
|
||||
|
||||
/**
|
||||
* @export
|
||||
* @typedef {ReturnType<typeof getRegionClient>} StripeClient
|
||||
*/
|
||||
|
||||
/**
|
||||
* Custom error class for reportable errors that should be written to CSV output
|
||||
*/
|
||||
export class ReportError extends Error {
|
||||
/**
|
||||
* @param {string} status - The error status code for CSV output
|
||||
* @param {string} message - The error message
|
||||
*/
|
||||
constructor(status, message) {
|
||||
super(message)
|
||||
this.status = status
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets the product ID from a Stripe Subscription Item
|
||||
*
|
||||
* @param {Stripe.SubscriptionItem | Stripe.SubscriptionSchedule.Phase.Item} item
|
||||
* @returns {string}
|
||||
*/
|
||||
export function getProductIdFromItem(item) {
|
||||
const product =
|
||||
typeof item.price === 'string'
|
||||
? null
|
||||
: 'product' in item.price
|
||||
? item.price.product
|
||||
: null
|
||||
return typeof product === 'string' ? product : (product?.id ?? '')
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets the price ID from a Stripe Subscription Item
|
||||
*
|
||||
* @param {Stripe.SubscriptionItem | Stripe.SubscriptionSchedule.Phase.Item} item
|
||||
* @returns {string}
|
||||
*/
|
||||
export function getPriceIdFromItem(item) {
|
||||
return typeof item.price === 'string' ? item.price : (item.price?.id ?? '')
|
||||
}
|
||||
|
||||
/**
|
||||
* Gets the product ID from a Stripe Price object
|
||||
*
|
||||
* @param {Stripe.Price} price
|
||||
* @returns {string}
|
||||
*/
|
||||
export function getProductIdFromPrice(price) {
|
||||
return typeof price.product === 'string'
|
||||
? price.product
|
||||
: (price.product?.id ?? '')
|
||||
}
|
||||
@@ -0,0 +1,585 @@
|
||||
#!/usr/bin/env node
|
||||
|
||||
/**
|
||||
* Rollback pending price changes for Stripe subscriptions
|
||||
*
|
||||
* This script removes pending subscription schedule changes that were created by the
|
||||
* change_existing_subscription_prices.mjs script. It only removes schedules that
|
||||
* are purely price changes on an existing plan - if the user has made any other
|
||||
* modifications (plan change, add-on changes), the pending change is left untouched.
|
||||
*
|
||||
* Usage:
|
||||
* node scripts/stripe/rollback_price_changes.mjs [OPTIONS] [INPUT-FILE]
|
||||
*
|
||||
* Options:
|
||||
* --region REGION Either 'uk' or 'us' (required)
|
||||
* --output PATH Output file path (default: /tmp/rollback_prices_output_<timestamp>.csv)
|
||||
* Use '-' to write to stdout
|
||||
* --commit Apply changes (without this flag, runs in dry-run mode)
|
||||
* --throttle DURATION Minimum time (in ms) between subscriptions processed (default: 100)
|
||||
* --help Show a help message
|
||||
*
|
||||
* CSV Input Format:
|
||||
* The CSV must have the following columns (same format as change_existing_subscription_prices.mjs):
|
||||
* - subscription_id: Stripe subscription id
|
||||
* - current_lookup_key: Current price lookup key
|
||||
* - new_lookup_key: New price lookup key
|
||||
* - current_add_on_lookup_key: Current price lookup key for add-on (optional)
|
||||
* - new_add_on_lookup_key: New price lookup key for add-on (optional)
|
||||
*
|
||||
* Output:
|
||||
* Writes a CSV with columns:
|
||||
* - subscription_id: The subscription id processed
|
||||
* - status: Result status (rolled-back, skipped, validated, not-found, or error)
|
||||
* - note: Additional information about the status
|
||||
*
|
||||
* The script will SKIP (not rollback) a subscription if:
|
||||
* - There is no active subscription schedule
|
||||
* - The schedule involves a plan change (user downgrade/upgrade)
|
||||
* - The schedule involves add-on additions/removals
|
||||
* - The schedule involves add-on quantity changes
|
||||
* - The prices don't match what we expect from the CSV
|
||||
*
|
||||
* Running on a Pod:
|
||||
* This script may run for multiple days. When running using `rake run:longpod[ENV,web]`,
|
||||
* use one of these strategies to preserve output:
|
||||
*
|
||||
* 1. Tail the output file from another session:
|
||||
* kubectl exec -it <pod-name> -- tail -f /tmp/rollback_prices_output_<timestamp>.csv > local_backup.csv
|
||||
*
|
||||
* 2. Periodically copy the output file to your laptop:
|
||||
* kubectl cp <pod-name>:/tmp/rollback_prices_output_<timestamp>.csv ./backup.csv
|
||||
*
|
||||
* 3. Write to stdout and capture locally:
|
||||
* kubectl exec -it <pod-name> -- node scripts/stripe/rollback_price_changes.mjs \
|
||||
* --region us --commit --output - input.csv > output.csv
|
||||
*
|
||||
* Examples:
|
||||
* # Dry run (preview only)
|
||||
* node scripts/stripe/rollback_price_changes.mjs --region us input.csv
|
||||
*
|
||||
* # Actually perform the rollback
|
||||
* node scripts/stripe/rollback_price_changes.mjs --region us --commit input.csv
|
||||
*/
|
||||
|
||||
import fs from 'node:fs'
|
||||
import path from 'node:path'
|
||||
import { setTimeout } from 'node:timers/promises'
|
||||
import * as csv from 'csv'
|
||||
import minimist from 'minimist'
|
||||
import AnalyticsManager from '../../app/src/Features/Analytics/AnalyticsManager.mjs'
|
||||
import { z } from '../../app/src/infrastructure/Validation.mjs'
|
||||
import { scriptRunner } from '../lib/ScriptRunner.mjs'
|
||||
import { getRegionClient } from '../../modules/subscriptions/app/src/StripeClient.mjs'
|
||||
import { ReportError, getProductIdFromItem } from './helpers.mjs'
|
||||
|
||||
/**
|
||||
* @import { CSVSubscriptionChange, StripeClient } from './helpers.mjs'
|
||||
* @import Stripe from 'stripe'
|
||||
* @import { ReadStream } from 'node:fs'
|
||||
* @import { Parser } from 'csv-parse'
|
||||
* @import { Stringifier } from 'csv-stringify'
|
||||
*/
|
||||
|
||||
// 100 ms corresponds to 10 requests per second (cautious rate within Stripe's 100 req/s limit)
|
||||
const DEFAULT_THROTTLE = 100
|
||||
|
||||
/**
|
||||
* Print usage information to stderr
|
||||
*/
|
||||
function usage() {
|
||||
console.error(`Usage: node scripts/stripe/rollback_price_changes.mjs [OPTIONS] [INPUT-FILE]
|
||||
|
||||
Rollback pending price changes for Stripe subscriptions.
|
||||
|
||||
This script only removes pending subscription schedules that are purely price changes on an
|
||||
existing plan. If a user has made any other modifications (plan change, add-on
|
||||
changes), the subscription is skipped.
|
||||
|
||||
Options:
|
||||
--region REGION Either 'uk' or 'us' (required)
|
||||
--output PATH Output file path (default: /tmp/rollback_prices_output_<timestamp>.csv)
|
||||
Use '-' to write to stdout
|
||||
--commit Apply changes (without this, runs in dry-run mode)
|
||||
--throttle DURATION Minimum time between requests in ms (default: ${DEFAULT_THROTTLE})
|
||||
--help Show this help message
|
||||
|
||||
See the source file header for detailed documentation on CSV format and pod usage.
|
||||
`)
|
||||
}
|
||||
|
||||
/**
|
||||
* Main script entry point
|
||||
* @param {function(string): Promise<void>} trackProgress - Function to log progress messages
|
||||
*/
|
||||
async function main(trackProgress) {
|
||||
const opts = parseArgs()
|
||||
const timestamp = new Date().toISOString().replace(/[:.]/g, '-')
|
||||
const outputFile =
|
||||
opts.output ?? `/tmp/rollback_prices_output_${timestamp}.csv`
|
||||
|
||||
const stripeClient = getRegionClient(opts.region)
|
||||
|
||||
await trackProgress('Starting price rollback script for Stripe')
|
||||
await trackProgress(`Region: ${opts.region}`)
|
||||
await trackProgress(`Run mode: ${opts.commit ? 'COMMIT' : 'DRY RUN'}`)
|
||||
await trackProgress(`Throttle: ${opts.throttle}ms between requests`)
|
||||
|
||||
const inputStream = opts.inputFile
|
||||
? fs.createReadStream(opts.inputFile)
|
||||
: process.stdin
|
||||
const csvReader = getCsvReader(inputStream)
|
||||
const csvWriter = getCsvWriter(outputFile)
|
||||
|
||||
await trackProgress(`Output: ${outputFile === '-' ? 'stdout' : outputFile}`)
|
||||
|
||||
let processedCount = 0
|
||||
let successCount = 0
|
||||
let skippedCount = 0
|
||||
let errorCount = 0
|
||||
|
||||
let lastLoopTimestamp = 0
|
||||
for await (const record of csvReader) {
|
||||
const timeSinceLastLoop = Date.now() - lastLoopTimestamp
|
||||
if (timeSinceLastLoop < opts.throttle) {
|
||||
await setTimeout(opts.throttle - timeSinceLastLoop)
|
||||
}
|
||||
lastLoopTimestamp = Date.now()
|
||||
|
||||
processedCount++
|
||||
|
||||
try {
|
||||
const result = await processRollback(record, stripeClient, opts.commit)
|
||||
|
||||
if (opts.commit && result.subscription) {
|
||||
try {
|
||||
const userId = result.subscription.customer.metadata?.userId
|
||||
await AnalyticsManager.recordEventForUser(
|
||||
userId,
|
||||
'script_price_change_reversed',
|
||||
{
|
||||
subscriptionId: record.subscription_id,
|
||||
}
|
||||
)
|
||||
} catch (err) {
|
||||
await trackProgress(
|
||||
`Warning: failed to record analytics event after successful price rollback for ${record.subscription_id}: ${err.message}`
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
csvWriter.write({
|
||||
subscription_id: record.subscription_id,
|
||||
status: result.status,
|
||||
note: result.note || '',
|
||||
})
|
||||
|
||||
if (result.status === 'skipped') {
|
||||
skippedCount++
|
||||
} else {
|
||||
successCount++
|
||||
}
|
||||
|
||||
if (processedCount % 10 === 0) {
|
||||
await trackProgress(
|
||||
`Processed ${processedCount} subscriptions (${successCount} ${opts.commit ? 'rolled-back' : 'validated'}, ${skippedCount} skipped, ${errorCount} errors)`
|
||||
)
|
||||
}
|
||||
} catch (err) {
|
||||
errorCount++
|
||||
if (err instanceof ReportError) {
|
||||
csvWriter.write({
|
||||
subscription_id: record.subscription_id,
|
||||
status: err.status,
|
||||
note: err.message,
|
||||
})
|
||||
} else {
|
||||
csvWriter.write({
|
||||
subscription_id: record.subscription_id,
|
||||
status: 'error',
|
||||
note: err.message,
|
||||
})
|
||||
await trackProgress(
|
||||
`Error processing ${record.subscription_id}: ${err.message}`
|
||||
)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
await trackProgress('\n✨ FINAL SUMMARY ✨')
|
||||
await trackProgress(`📊 Total processed: ${processedCount}`)
|
||||
if (opts.commit) {
|
||||
await trackProgress(`✅ Successfully rolled back: ${successCount}`)
|
||||
} else {
|
||||
await trackProgress(`✅ Successfully validated: ${successCount}`)
|
||||
await trackProgress('ℹ️ DRY RUN: No changes were applied to Stripe')
|
||||
}
|
||||
await trackProgress(`⏭️ Skipped: ${skippedCount}`)
|
||||
await trackProgress(`❌ Errors: ${errorCount}`)
|
||||
await trackProgress('🎉 Script completed!')
|
||||
|
||||
csvWriter.end()
|
||||
}
|
||||
|
||||
/**
|
||||
* Get a CSV parser configured for subscription change input
|
||||
* @param {ReadStream | NodeJS.ReadableStream} inputStream - The input stream to parse
|
||||
* @returns {Parser} The configured CSV parser
|
||||
*/
|
||||
function getCsvReader(inputStream) {
|
||||
const parser = csv.parse({
|
||||
columns: true,
|
||||
})
|
||||
inputStream.pipe(parser)
|
||||
return parser
|
||||
}
|
||||
|
||||
/**
|
||||
* Get a CSV stringifier configured for output
|
||||
* @param {string} outputFile - The output file path to write to, or '-' for stdout
|
||||
* @returns {Stringifier} The configured CSV stringifier
|
||||
*/
|
||||
function getCsvWriter(outputFile) {
|
||||
let outputStream
|
||||
if (outputFile === '-') {
|
||||
outputStream = process.stdout
|
||||
} else {
|
||||
fs.mkdirSync(path.dirname(outputFile), { recursive: true })
|
||||
outputStream = fs.createWriteStream(outputFile)
|
||||
}
|
||||
const writer = csv.stringify({
|
||||
columns: ['subscription_id', 'status', 'note'],
|
||||
header: true,
|
||||
})
|
||||
writer.on('error', err => {
|
||||
console.error(err)
|
||||
process.exit(1)
|
||||
})
|
||||
writer.pipe(outputStream)
|
||||
return writer
|
||||
}
|
||||
|
||||
/**
|
||||
* Process a single subscription rollback
|
||||
* @param {CSVSubscriptionChange} record - The subscription record to process
|
||||
* @param {StripeClient} stripeClient - The Stripe client for the region
|
||||
* @param {boolean} commit - Whether to commit changes or run in dry-run mode
|
||||
* @returns {Promise<{status: string, note: string, subscription?: Stripe.Subscription}>} The result of the rollback
|
||||
*/
|
||||
async function processRollback(record, stripeClient, commit) {
|
||||
const subscription = await fetchSubscription(
|
||||
record.subscription_id,
|
||||
stripeClient
|
||||
)
|
||||
|
||||
// Validate this is a price-only change that we created
|
||||
const validation = await validatePriceOnlyChange(
|
||||
record,
|
||||
subscription,
|
||||
stripeClient
|
||||
)
|
||||
|
||||
if (!validation.isPriceOnly) {
|
||||
return {
|
||||
status: 'skipped',
|
||||
note: `${validation.reason}: ${validation.detail || 'N/A'}`,
|
||||
}
|
||||
}
|
||||
|
||||
if (!commit) {
|
||||
return {
|
||||
status: 'validated',
|
||||
note: `Would release subscription schedule: ${validation.scheduleId}`,
|
||||
}
|
||||
}
|
||||
|
||||
// Safe to release - this is a price-only change matching our expected values
|
||||
await stripeClient.stripe.subscriptionSchedules.release(validation.scheduleId)
|
||||
|
||||
return {
|
||||
status: 'rolled-back',
|
||||
note: `Released subscription schedule: ${validation.scheduleId}`,
|
||||
subscription,
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Fetch a subscription from Stripe
|
||||
* @param {string} subscriptionId - The Stripe subscription id
|
||||
* @param {StripeClient} stripeClient - The Stripe client for the region
|
||||
* @returns {Promise<Stripe.Subscription>} The subscription with expanded schedule
|
||||
* @throws {ReportError} If subscription is not found
|
||||
*/
|
||||
async function fetchSubscription(subscriptionId, stripeClient) {
|
||||
try {
|
||||
const subscription = await stripeClient.stripe.subscriptions.retrieve(
|
||||
subscriptionId,
|
||||
{
|
||||
expand: [
|
||||
'schedule',
|
||||
'schedule.phases.items.price',
|
||||
'items.data.price',
|
||||
'customer',
|
||||
],
|
||||
}
|
||||
)
|
||||
return subscription
|
||||
} catch (err) {
|
||||
if (err.type === 'StripeInvalidRequestError' && err.statusCode === 404) {
|
||||
throw new ReportError('not-found', 'subscription not found')
|
||||
}
|
||||
throw err
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Validate that the subscription schedule is a price-only change created by our
|
||||
* price increase script, and not a user-initiated plan change or add-on modification.
|
||||
*
|
||||
* @param {CSVSubscriptionChange} record - The CSV record with expected values
|
||||
* @param {Stripe.Subscription} subscription - The Stripe subscription
|
||||
* @param {StripeClient} stripeClient - The Stripe client for the region
|
||||
* @returns {Promise<{ isPriceOnly: boolean, reason?: string, detail?: string, scheduleId?: string }>}
|
||||
*/
|
||||
async function validatePriceOnlyChange(record, subscription, stripeClient) {
|
||||
// Must have a subscription schedule
|
||||
if (!subscription.schedule) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'no-pending-change',
|
||||
detail: 'subscription has no schedule to rollback',
|
||||
}
|
||||
}
|
||||
|
||||
// Subscription must be updatable
|
||||
const inactiveStatuses = [
|
||||
'incomplete',
|
||||
'incomplete_expired',
|
||||
'canceled',
|
||||
'trialing',
|
||||
]
|
||||
if (inactiveStatuses.includes(subscription.status)) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'inactive',
|
||||
detail: `subscription status: ${subscription.status}`,
|
||||
}
|
||||
}
|
||||
|
||||
// Subscription should not be scheduled to be canceled
|
||||
if (subscription.cancel_at_period_end) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'inactive',
|
||||
detail: 'subscription is scheduled to be canceled at period end',
|
||||
}
|
||||
}
|
||||
|
||||
// Get the schedule (already expanded in subscription fetch)
|
||||
const schedule =
|
||||
typeof subscription.schedule === 'string' ? null : subscription.schedule
|
||||
|
||||
if (!schedule) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'no-pending-change',
|
||||
detail: 'subscription schedule not expanded',
|
||||
}
|
||||
}
|
||||
|
||||
const scheduleId = schedule.id
|
||||
|
||||
// Schedule must not be released already
|
||||
if (schedule.status === 'released') {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'schedule-released',
|
||||
detail: 'schedule has already been released',
|
||||
}
|
||||
}
|
||||
|
||||
// Schedule must have exactly 2 phases (current + future)
|
||||
if (schedule.phases.length !== 2) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'unexpected-schedule-structure',
|
||||
detail: `expected 2 phases, got ${schedule.phases.length}`,
|
||||
}
|
||||
}
|
||||
|
||||
const currentPhase = schedule.phases[0]
|
||||
const nextPhase = schedule.phases[1]
|
||||
|
||||
// All phases must have same number of items (no add-ons added/removed)
|
||||
const currentPhaseItemCount = currentPhase.items.length
|
||||
const subscriptionItemCount = subscription.items.data.length
|
||||
const nextPhaseItemCount = nextPhase.items.length
|
||||
|
||||
if (currentPhaseItemCount !== subscriptionItemCount) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'item-count-mismatch',
|
||||
detail: `current phase has ${currentPhaseItemCount} items, subscription has ${subscriptionItemCount}`,
|
||||
}
|
||||
}
|
||||
|
||||
if (nextPhaseItemCount !== currentPhaseItemCount) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'addon-change-detected',
|
||||
detail: `next phase has ${nextPhaseItemCount} items, current has ${currentPhaseItemCount}`,
|
||||
}
|
||||
}
|
||||
|
||||
// Verify current lookup keys match expected
|
||||
const currentLookupKeys = new Set(
|
||||
subscription.items.data.map(item => item.price.lookup_key)
|
||||
)
|
||||
|
||||
if (!currentLookupKeys.has(record.current_lookup_key)) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'current-price-mismatch',
|
||||
detail: `expected current_lookup_key ${record.current_lookup_key} not found in subscription`,
|
||||
}
|
||||
}
|
||||
|
||||
if (
|
||||
record.current_add_on_lookup_key &&
|
||||
!currentLookupKeys.has(record.current_add_on_lookup_key)
|
||||
) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'current-addon-price-mismatch',
|
||||
detail: `expected current_add_on_lookup_key ${record.current_add_on_lookup_key} not found in subscription`,
|
||||
}
|
||||
}
|
||||
|
||||
// Verify next phase price IDs match expected
|
||||
const nextPhaseLookupKeys = new Set(
|
||||
nextPhase.items.map(item => item.price.lookup_key)
|
||||
)
|
||||
|
||||
if (!nextPhaseLookupKeys.has(record.new_lookup_key)) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'pending-price-mismatch',
|
||||
detail: `expected new_lookup_key ${record.new_lookup_key} not found in next phase`,
|
||||
}
|
||||
}
|
||||
|
||||
if (
|
||||
record.new_add_on_lookup_key &&
|
||||
!nextPhaseLookupKeys.has(record.new_add_on_lookup_key)
|
||||
) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'pending-addon-price-mismatch',
|
||||
detail: `expected new_add_on_lookup_key ${record.new_add_on_lookup_key} not found in next phase`,
|
||||
}
|
||||
}
|
||||
|
||||
// Verify quantities remain the same and products match (no plan/add-on changes)
|
||||
for (const currentItem of currentPhase.items) {
|
||||
const currentLookupKey = currentItem.price.lookup_key
|
||||
const currentProductId = getProductIdFromItem(currentItem)
|
||||
|
||||
const nextItem = nextPhase.items.find(item => {
|
||||
const nextLookupKey = item.price.lookup_key
|
||||
const nextProductId = getProductIdFromItem(item)
|
||||
// Match by product ID to handle price changes
|
||||
return (
|
||||
nextLookupKey === currentLookupKey || nextProductId === currentProductId
|
||||
)
|
||||
})
|
||||
|
||||
if (!nextItem) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'item-mismatch',
|
||||
detail: `current phase item ${currentLookupKey} not found in next phase`,
|
||||
}
|
||||
}
|
||||
|
||||
// Verify quantity hasn't changed
|
||||
if (nextItem.quantity !== currentItem.quantity) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'quantity-change-detected',
|
||||
detail: `quantity change: ${currentItem.quantity} -> ${nextItem.quantity}`,
|
||||
}
|
||||
}
|
||||
|
||||
// Verify product hasn't changed
|
||||
const nextProductId = getProductIdFromItem(nextItem)
|
||||
|
||||
if (
|
||||
currentProductId &&
|
||||
nextProductId &&
|
||||
currentProductId !== nextProductId
|
||||
) {
|
||||
return {
|
||||
isPriceOnly: false,
|
||||
reason: 'product-change-detected',
|
||||
detail: `product change: ${currentProductId} -> ${nextProductId}`,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// All checks passed - this is a price-only change we created
|
||||
return { isPriceOnly: true, scheduleId }
|
||||
}
|
||||
|
||||
const paramsSchema = z.object({
|
||||
region: z.enum(['uk', 'us']),
|
||||
output: z.string().optional(),
|
||||
commit: z.boolean().default(false),
|
||||
throttle: z
|
||||
.string()
|
||||
.optional()
|
||||
.transform(val => (val ? parseInt(val, 10) : DEFAULT_THROTTLE)),
|
||||
_: z.array(z.string()).max(1),
|
||||
help: z.boolean().optional(),
|
||||
})
|
||||
|
||||
/**
|
||||
* Parse command line arguments
|
||||
* @returns {{inputFile: string | undefined, output: string | undefined, commit: boolean, throttle: number, region: 'uk' | 'us'}} Parsed options
|
||||
*/
|
||||
function parseArgs() {
|
||||
const argv = minimist(process.argv.slice(2), {
|
||||
string: ['throttle', 'output', 'region'],
|
||||
boolean: ['help', 'commit'],
|
||||
})
|
||||
|
||||
if (argv.help) {
|
||||
usage()
|
||||
process.exit(0)
|
||||
}
|
||||
|
||||
const parseResult = paramsSchema.safeParse(argv)
|
||||
|
||||
if (!parseResult.success) {
|
||||
console.error(`Invalid parameters: ${parseResult.error.message}`)
|
||||
usage()
|
||||
process.exit(1)
|
||||
}
|
||||
|
||||
const { region, output, commit, throttle, _ } = parseResult.data
|
||||
|
||||
return {
|
||||
inputFile: _[0],
|
||||
output,
|
||||
commit,
|
||||
throttle,
|
||||
region,
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
await scriptRunner(main)
|
||||
process.exit(0)
|
||||
} catch (error) {
|
||||
console.error(error)
|
||||
process.exit(1)
|
||||
}
|
||||
Reference in New Issue
Block a user