cfw-autumn/server/tests/utils/testAttachUtils/testAttachUtils.ts

import {
	AttachBranch,
	type AttachPreview,
	OnIncrease,
	UsageModel,
} from "@autumn/shared";
import { addHours, addMonths } from "date-fns";
import { Decimal } from "decimal.js";
import type Stripe from "stripe";
import { advanceTestClock } from "@/utils/scriptUtils/testClockUtils.js";
import { hoursToFinalizeInvoice } from "../constants.js";

export const getCurrentOptions = ({
	preview,
	options,
}: {
	preview: AttachPreview;
	options?: any;
}) => {
	const currentOptions = structuredClone(options);
	if (!options) return currentOptions;

	const isUpdatePrepaidQuantity =
		preview?.branch === AttachBranch.UpdatePrepaidQuantity;

	for (const option of currentOptions || []) {
		const previewOption = preview?.options.find(
			(o: any) =>
				o.feature_id === option.feature_id || o.feature_id === option.featureId,
		);

		const currentQuantity = previewOption.current_quantity || 0;
		const newQuantity = option.quantity || 0;
		let difference = newQuantity - currentQuantity;
		difference = difference / previewOption.billing_units;

		const isDecrease = newQuantity < currentQuantity;
		const isIncrease = newQuantity > currentQuantity;

		if (isDecrease && previewOption.config.on_decrease === "none") {
			option.quantity = currentQuantity;
			continue;
		}

		if (
			isUpdatePrepaidQuantity &&
			isIncrease &&
			previewOption.config.on_increase === OnIncrease.ProrateNextCycle
		) {
		}
	}

	return currentOptions;
};

// 1. Calculate total
export const getAttachTotal = ({
	preview,
	options,
}: {
	preview: AttachPreview;
	options?: any;
}) => {
	const dueToday = preview?.due_today;

	let dueTodayTotal =
		dueToday?.line_items.reduce((acc: any, item: any) => {
			// Skip prepaid items that are already in the options
			if (
				item.usage_model === UsageModel.Prepaid &&
				options.some((o: any) => o.feature_id === item.feature_id)
			) {
				return acc;
			}

			if (item.amount) {
				return acc.plus(item.amount);
			}
			return acc;
		}, new Decimal(0)) || new Decimal(0);

	const isUpdatePrepaidQuantity =
		preview?.branch === AttachBranch.UpdatePrepaidQuantity;
	if (isUpdatePrepaidQuantity) {
		dueTodayTotal = new Decimal(0);
	}

	for (const option of options || []) {
		const previewOption = preview?.options.find(
			(o: any) =>
				o.feature_id === option.feature_id || o.feature_id === option.featureId,
		);

		const currentQuantity = previewOption.current_quantity || 0;
		const newQuantity = option.quantity || 0;
		let difference = newQuantity - currentQuantity;
		difference = difference / previewOption.billing_units;

		const isDecrease = newQuantity < currentQuantity;
		const isIncrease = newQuantity > currentQuantity;

		if (isDecrease && previewOption.config.on_decrease === "none") {
			option.quantity = currentQuantity;
			continue;
		}

		if (
			isUpdatePrepaidQuantity &&
			isIncrease &&
			previewOption.config.on_increase === OnIncrease.ProrateNextCycle
		) {
			continue;
		}

		const differenceAmount = new Decimal(previewOption.price).times(difference);
		dueTodayTotal = dueTodayTotal.plus(differenceAmount);

		// Prorated difference
		if (previewOption.proration_amount) {
			dueTodayTotal = dueTodayTotal.plus(
				new Decimal(previewOption.proration_amount),
			);
		}

		// let previewOption = preview?.options.find(
		//   (o: any) =>
		//     o.feature_id === option.feature_id || o.feature_id === option.featureId,
		// );

		// if (!previewOption) {
		//   continue;
		// }

		// const prepaidAmt = new Decimal(previewOption.price)
		//   .times(option.quantity)
		//   .dividedBy(previewOption.billing_units);

		// dueTodayTotal = dueTodayTotal.plus(prepaidAmt);
	}

	return dueTodayTotal.toDecimalPlaces(2).toNumber();
};

const isParallelRun = () => {
	const concurrency = Number(process.env.TEST_FILE_CONCURRENCY || "0");
	return concurrency > 1;
};

/**
 * Advances the test clock to the next invoice (1 month from current time).
 * Waits longer when running in parallel (TEST_FILE_CONCURRENCY > 1) to
 * account for increased Stripe webhook processing time.
 *
 * @param stripeCli - Stripe client
 * @param testClockId - Test clock ID
 * @param currentEpochMs - Current epoch in ms (use this for consecutive advances). If not provided, uses Date.now().
 * @param withPause - If true, advances in two steps (to month boundary, then to finalize)
 * @returns The new epoch time in ms after advancing
 */
export const advanceToNextInvoice = async ({
	stripeCli,
	testClockId,
	currentEpochMs,
	withPause = false,
}: {
	stripeCli: Stripe;
	testClockId: string;
	currentEpochMs?: number;
	withPause?: boolean;
}): Promise<number> => {
	const baseTime = currentEpochMs ? new Date(currentEpochMs) : new Date();
	const parallel = isParallelRun();

	if (withPause) {
		const newUnix = await advanceTestClock({
			stripeCli,
			testClockId,
			advanceTo: addMonths(baseTime, 1).getTime(),
			waitForSeconds: parallel ? 80 : 50,
		});

		await advanceTestClock({
			stripeCli,
			testClockId,
			advanceTo: addHours(newUnix, hoursToFinalizeInvoice).getTime(),
			waitForSeconds: 30,
		});

		return newUnix;
	}

	return await advanceTestClock({
		stripeCli,
		testClockId,
		advanceTo: addHours(
			addMonths(baseTime, 1),
			hoursToFinalizeInvoice,
		).getTime(),
		waitForSeconds: 30,
	});
};