@labdigital/commercetools-mock 2.57.1 → 2.58.0

package/src/repositories/cart/index.ts

@@ -1,6 +1,9 @@
 import type {
 	BusinessUnit,
+	CentPrecisionMoney,
 	InvalidOperationError,
+	MissingTaxRateForCountryError,
+	ShippingMethodDoesNotMatchCartError,
 } from "@commercetools/platform-sdk";
 import type {
 	Cart,
@@ -12,17 +15,27 @@ import type {
 	LineItemDraft,
 	Product,
 	ProductPagedQueryResponse,
+	TaxPortion,
+	TaxedItemPrice,
 } from "@commercetools/platform-sdk";
+import { Decimal } from "decimal.js/decimal";
 import { v4 as uuidv4 } from "uuid";
 import type { Config } from "~src/config";
 import { CommercetoolsError } from "~src/exceptions";
 import { getBaseResourceProperties } from "~src/helpers";
+import { getShippingMethodsMatchingCart } from "~src/shipping";
 import type { Writable } from "~src/types";
 import {
 	AbstractResourceRepository,
 	type RepositoryContext,
 } from "../abstract";
-import { createAddress, createCustomFields } from "../helpers";
+import {
+	createAddress,
+	createCentPrecisionMoney,
+	createCustomFields,
+	createTypedMoney,
+	roundDecimal,
+} from "../helpers";
 import { CartUpdateHandler } from "./actions";
 import {
 	calculateCartTotalPrice,
@@ -33,7 +46,7 @@ import {
 export class CartRepository extends AbstractResourceRepository<"cart"> {
 	constructor(config: Config) {
 		super("cart", config);
-		this.actions = new CartUpdateHandler(this._storage);
+		this.actions = new CartUpdateHandler(this._storage, this);
 	}
 
 	create(context: RepositoryContext, draft: CartDraft): Cart {
@@ -128,6 +141,7 @@ export class CartRepository extends AbstractResourceRepository<"cart"> {
 					)
 				: undefined,
 			shipping: [],
+			shippingInfo: undefined,
 			origin: draft.origin ?? "Customer",
 			refusedGifts: [],
 			custom: createCustomFields(
@@ -139,7 +153,18 @@ export class CartRepository extends AbstractResourceRepository<"cart"> {
 		resource.totalPrice.centAmount = calculateCartTotalPrice(resource);
 		resource.store = context.storeKey
 			? { typeId: "store", key: context.storeKey }
-			: undefined;
+			: draft.store?.key
+				? { typeId: "store", key: draft.store.key }
+				: undefined;
+
+		// Set shipping info after resource is created
+		if (draft.shippingMethod) {
+			resource.shippingInfo = this.createShippingInfo(
+				context,
+				resource,
+				draft.shippingMethod,
+			);
+		}
 
 		return this.saveNew(context, resource);
 	}
@@ -249,4 +274,179 @@ export class CartRepository extends AbstractResourceRepository<"cart"> {
 			),
 		};
 	};
+
+	createShippingInfo(
+		context: RepositoryContext,
+		resource: Writable<Cart>,
+		shippingMethodRef: NonNullable<CartDraft["shippingMethod"]>,
+	): NonNullable<Cart["shippingInfo"]> {
+		if (resource.taxMode === "External") {
+			throw new Error("External tax rate is not supported");
+		}
+
+		const country = resource.shippingAddress?.country;
+
+		if (!country) {
+			throw new CommercetoolsError<InvalidOperationError>({
+				code: "InvalidOperation",
+				message: `The cart with ID '${resource.id}' does not have a shipping address set.`,
+			});
+		}
+
+		// Bit of a hack: calling this checks that the resource identifier is
+		// valid (i.e. id xor key) and that the shipping method exists.
+		this._storage.getByResourceIdentifier<"shipping-method">(
+			context.projectKey,
+			shippingMethodRef,
+		);
+
+		// getShippingMethodsMatchingCart does the work of determining whether the
+		// shipping method is allowed for the cart, and which shipping rate to use
+		const shippingMethods = getShippingMethodsMatchingCart(
+			context,
+			this._storage,
+			resource,
+			{
+				expand: ["zoneRates[*].zone"],
+			},
+		);
+
+		const method = shippingMethods.results.find((candidate) =>
+			shippingMethodRef.id
+				? candidate.id === shippingMethodRef.id
+				: candidate.key === shippingMethodRef.key,
+		);
+
+		// Not finding the method in the results means it's not allowed, since
+		// getShippingMethodsMatchingCart only returns allowed methods and we
+		// already checked that the method exists.
+		if (!method) {
+			throw new CommercetoolsError<ShippingMethodDoesNotMatchCartError>({
+				code: "ShippingMethodDoesNotMatchCart",
+				message: `The shipping method with ${shippingMethodRef.id ? `ID '${shippingMethodRef.id}'` : `key '${shippingMethodRef.key}'`} is not allowed for the cart with ID '${resource.id}'.`,
+			});
+		}
+
+		const taxCategory = this._storage.getByResourceIdentifier<"tax-category">(
+			context.projectKey,
+			method.taxCategory,
+		);
+
+		// TODO: match state in addition to country
+		const taxRate = taxCategory.rates.find((rate) => rate.country === country);
+
+		if (!taxRate) {
+			throw new CommercetoolsError<MissingTaxRateForCountryError>({
+				code: "MissingTaxRateForCountry",
+				message: `Tax category '${taxCategory.id}' is missing a tax rate for country '${country}'.`,
+				taxCategoryId: taxCategory.id,
+			});
+		}
+
+		// There should only be one zone rate matching the address, since
+		// Locations cannot be assigned to more than one zone.
+		// See https://docs.commercetools.com/api/projects/zones#location
+		const zoneRate = method.zoneRates.find((rate) =>
+			rate.zone.obj?.locations.some((loc) => loc.country === country),
+		);
+
+		if (!zoneRate) {
+			// This shouldn't happen because getShippingMethodsMatchingCart already
+			// filtered out shipping methods without any zones matching the address
+			throw new Error("Zone rate not found");
+		}
+
+		// Shipping rates are defined by currency, and getShippingMethodsMatchingCart
+		// also matches on currency, so there should only be one in the array.
+		// See https://docs.commercetools.com/api/projects/shippingMethods#zonerate
+		const shippingRate = zoneRate.shippingRates[0];
+		if (!shippingRate) {
+			// This shouldn't happen because getShippingMethodsMatchingCart already
+			// filtered out shipping methods without any matching rates
+			throw new Error("Shipping rate not found");
+		}
+
+		const shippingRateTier = shippingRate.tiers.find((tier) => tier.isMatching);
+		if (shippingRateTier && shippingRateTier.type !== "CartValue") {
+			throw new Error("Non-CartValue shipping rate tier is not supported");
+		}
+
+		let shippingPrice = shippingRateTier
+			? createCentPrecisionMoney(shippingRateTier.price)
+			: shippingRate.price;
+
+		// Handle freeAbove: if cart total is above the freeAbove threshold, shipping is free
+		if (
+			shippingRate.freeAbove &&
+			shippingRate.freeAbove.currencyCode ===
+				resource.totalPrice.currencyCode &&
+			resource.totalPrice.centAmount >= shippingRate.freeAbove.centAmount
+		) {
+			shippingPrice = {
+				...shippingPrice,
+				centAmount: 0,
+			};
+		}
+
+		// Calculate tax amounts
+		const totalGross: CentPrecisionMoney = taxRate.includedInPrice
+			? shippingPrice
+			: {
+					...shippingPrice,
+					centAmount: roundDecimal(
+						new Decimal(shippingPrice.centAmount).mul(1 + taxRate.amount),
+						resource.taxRoundingMode,
+					).toNumber(),
+				};
+
+		const totalNet: CentPrecisionMoney = taxRate.includedInPrice
+			? {
+					...shippingPrice,
+					centAmount: roundDecimal(
+						new Decimal(shippingPrice.centAmount).div(1 + taxRate.amount),
+						resource.taxRoundingMode,
+					).toNumber(),
+				}
+			: shippingPrice;
+
+		const taxPortions: TaxPortion[] = [
+			{
+				name: taxRate.name,
+				rate: taxRate.amount,
+				amount: {
+					...shippingPrice,
+					centAmount: totalGross.centAmount - totalNet.centAmount,
+				},
+			},
+		];
+
+		const totalTax: CentPrecisionMoney = {
+			...shippingPrice,
+			centAmount: taxPortions.reduce(
+				(acc, portion) => acc + portion.amount.centAmount,
+				0,
+			),
+		};
+
+		const taxedPrice: TaxedItemPrice = {
+			totalNet,
+			totalGross,
+			taxPortions,
+			totalTax,
+		};
+
+		return {
+			shippingMethod: {
+				typeId: "shipping-method" as const,
+				id: method.id,
+			},
+			shippingMethodName: method.name,
+			price: shippingPrice,
+			shippingRate,
+			taxedPrice,
+			taxRate,
+			taxCategory: method.taxCategory,
+			shippingMethodState: "MatchesCart",
+		};
+	}
 }