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

import { notNullish } from "@/utils/genUtils.js";
import { advanceTestClock } from "@/utils/scriptUtils/testClockUtils.js";
import {
	AttachBranch,
	AttachPreview,
	OnIncrease,
	UsageModel,
} from "@autumn/shared";
import { addHours, addMonths } from "date-fns";
import { Decimal } from "decimal.js";
import Stripe from "stripe";
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
		) {
			continue;
		}
	}

	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();
};

export const advanceToNextInvoice = async ({
	stripeCli,
	testClockId,
}: {
	stripeCli: Stripe;
	testClockId: string;
}) => {
	return await advanceTestClock({
		stripeCli,
		testClockId,
		advanceTo: addHours(
			addMonths(new Date(), 1),
			hoursToFinalizeInvoice,
		).getTime(),
		waitForSeconds: 30,
	});
};