cfw-autumn/server/tests/core/multiAttach/multiReward/multiRewardUtils.test.ts

import { DrizzleCli } from "@/db/initDrizzle.js";
import { AutumnInt } from "@/external/autumn/autumnCli.js";
import { ProductService } from "@/internal/products/ProductService.js";
import { RewardService } from "@/internal/rewards/RewardService.js";
import { constructFeatureItem } from "@/utils/scriptUtils/constructItem.js";
import { constructProduct } from "@/utils/scriptUtils/createTestProducts.js";
import {
  APIVersion,
  AppEnv,
  CouponDurationType,
  CreateReward,
  ProductItemInterval,
  RewardType,
} from "@autumn/shared";
import { TestFeature } from "tests/setup/v2Features.js";

export let premiumTrial = constructProduct({
  id: "multiReward_premiumTrial",
  group: "multiReward",
  items: [
    constructFeatureItem({ featureId: TestFeature.Words, includedUsage: 200 }),
  ],
  type: "premium",
  trial: true,
});

export let proTrial = constructProduct({
  id: "multiReward_proTrial",
  group: "multiReward",
  items: [
    constructFeatureItem({ featureId: TestFeature.Words, includedUsage: 300 }),
  ],
  type: "pro",
  trial: true,
});
export let multiRewardPremium = constructProduct({
  id: "multiReward_premium",
  group: "multiReward",
  items: [
    constructFeatureItem({ featureId: TestFeature.Words, includedUsage: 200 }),
  ],
  type: "premium",
});

export let multiRewardPro = constructProduct({
  id: "multiReward_pro",
  group: "multiReward",
  items: [
    constructFeatureItem({ featureId: TestFeature.Words, includedUsage: 300 }),
  ],
  type: "pro",
});

export const proReward: CreateReward = {
  id: "pro_reward",
  name: "pro_reward",
  promo_codes: [{ code: "pro_reward" }],
  type: RewardType.PercentageDiscount,
  discount_config: {
    discount_value: 50,
    duration_type: CouponDurationType.Months,
    duration_value: 3,
    should_rollover: true,
    apply_to_all: false,
    price_ids: [proTrial.id],
  },
};

export const premiumReward: CreateReward = {
  id: "premium_reward",
  name: "premium_reward",
  promo_codes: [{ code: "premium_reward" }],
  type: RewardType.PercentageDiscount,
  discount_config: {
    discount_value: 80,
    duration_type: CouponDurationType.Months,
    duration_value: 3,
    should_rollover: true,
    apply_to_all: false,
  },
};

export const setupMultiRewardBefore = async ({
  orgId,
  db,
  env,
}: {
  orgId: string;
  db: DrizzleCli;
  env: AppEnv;
}) => {
  const autumn = new AutumnInt({ version: APIVersion.v1_2 });
  for (const product of [
    proTrial,
    premiumTrial,
    multiRewardPro,
    multiRewardPremium,
  ]) {
    // let res = await autumn.products.get(product.id);

    // if (res.code === "product_not_found") {
    //   try {
    //     await autumn.products.create(product);
    //   } catch (error) {}
    // }

    try {
      await autumn.products.delete(product.id);
    } catch (error) {
      // console.log("Error deleting product:", error);
    }

    try {
      await autumn.products.create(product);
    } catch (error) {}
  }

  const products = await ProductService.listFull({
    db,
    orgId,
    env,
  });

  const proTrialPrice = products.find((p) => p.id === proTrial.id)?.prices[0];
  const premiumTrialPrice = products.find((p) => p.id === premiumTrial.id)
    ?.prices[0];
  const proProduct = products.find((p) => p.id === multiRewardPro.id);
  const premiumProduct = products.find((p) => p.id === multiRewardPremium.id);

  const proPriceIds = [proTrialPrice!.id, proProduct!.prices[0]!.id];
  const premiumPriceIds = [
    premiumTrialPrice!.id,
    premiumProduct!.prices[0]!.id,
  ];

  for (const reward of [proReward, premiumReward]) {
    let rewardRes = null;
    // try {
    //   rewardRes = await autumn.rewards.get(reward.id);
    // } catch (error) {}

    try {
      await autumn.rewards.delete(reward.id);
    } catch (error) {}

    if (!rewardRes) {
      try {
        await autumn.rewards.create({
          ...reward,
          discount_config: {
            ...reward.discount_config,
            price_ids:
              reward.id == proReward.id ? proPriceIds : premiumPriceIds,
          },
        });
      } catch (error) {}
    }
  }
};