feat: add support for set product price action (#199)

.changeset/purple-crews-pump.md

@@ -0,0 +1,5 @@
+---
+"@labdigital/commercetools-mock": minor
+---
+
+add support for "setProductPriceCustomField" product update action

src/repositories/product/actions.ts

@@ -26,6 +26,8 @@ import type {
 	ProductSetMetaDescriptionAction,
 	ProductSetMetaKeywordsAction,
 	ProductSetMetaTitleAction,
+	ProductSetProductPriceCustomFieldAction,
+	ProductSetProductPriceCustomTypeAction,
 	ProductSetTaxCategoryAction,
 	ProductTransitionStateAction,
 	ProductUpdateAction,
@@ -35,8 +37,11 @@ import type {
 } from "@commercetools/platform-sdk";
 import { CommercetoolsError } from "~src/exceptions";
 import type { Writable } from "~src/types";
-import { AbstractUpdateHandler, RepositoryContext } from "../abstract";
-import { getReferenceFromResourceIdentifier } from "../helpers";
+import { AbstractUpdateHandler, type RepositoryContext } from "../abstract";
+import {
+	createCustomFields,
+	getReferenceFromResourceIdentifier,
+} from "../helpers";
 import {
 	checkForStagedChanges,
 	getVariant,
@@ -853,6 +858,87 @@ export class ProductUpdateHandler
 		return resource;
 	}
 
+	setProductPriceCustomField(
+		context: RepositoryContext,
+		resource: Writable<Product>,
+		{ name, value, staged, priceId }: ProductSetProductPriceCustomFieldAction,
+	) {
+		const updatePriceCustomFields = (data: Writable<ProductData>) => {
+			const price = [data.masterVariant, ...(data.variants ?? [])]
+				.flatMap((variant) => variant.prices ?? [])
+				.find((price) => price.id === priceId);
+
+			if (!price) {
+				throw new Error(
+					`Price with id ${priceId} not found on product ${resource.id}`,
+				);
+			}
+			if (price.custom) {
+				if (value === null) {
+					delete price.custom.fields[name];
+				} else {
+					price.custom.fields[name] = value;
+				}
+			}
+			return data;
+		};
+
+		resource.masterData.staged = updatePriceCustomFields(
+			resource.masterData.staged,
+		);
+
+		const onlyStaged = staged !== undefined ? staged : true;
+		if (!onlyStaged) {
+			resource.masterData.current = updatePriceCustomFields(
+				resource.masterData.current,
+			);
+		}
+		checkForStagedChanges(resource);
+		return resource;
+	}
+
+	setProductPriceCustomType(
+		context: RepositoryContext,
+		resource: Writable<Product>,
+		{ type, fields, staged, priceId }: ProductSetProductPriceCustomTypeAction,
+	) {
+		const updatePriceCustomType = (data: Writable<ProductData>) => {
+			const price = [data.masterVariant, ...(data.variants ?? [])]
+				.flatMap((variant) => variant.prices ?? [])
+				.find((price) => price.id === priceId);
+
+			if (price) {
+				if (type) {
+					price.custom = createCustomFields(
+						{ type, fields },
+						context.projectKey,
+						this._storage,
+					);
+				} else {
+					price.custom = undefined;
+				}
+			} else {
+				throw new Error(
+					`Price with id ${priceId} not found on product ${resource.id}`,
+				);
+			}
+			return data;
+		};
+
+		resource.masterData.staged = updatePriceCustomType(
+			resource.masterData.staged,
+		);
+
+		const onlyStaged = staged !== undefined ? staged : true;
+		if (!onlyStaged) {
+			resource.masterData.current = updatePriceCustomType(
+				resource.masterData.current,
+			);
+		}
+		checkForStagedChanges(resource);
+		return resource;
+	}
+
 	setTaxCategory(
 		context: RepositoryContext,
 		resource: Writable<Product>,
@@ -919,8 +1005,6 @@ export class ProductUpdateHandler
 	}
 
 	// 'setPrices': () => {},
-	// 'setProductPriceCustomType': () => {},
-	// 'setProductPriceCustomField': () => {},
 	// 'setDiscountedPrice': () => {},
 	// 'setAttributeInAllVariants': () => {},
 	// 'setCategoryOrderHint': () => {},

src/services/product.test.ts

@@ -12,6 +12,8 @@ import type {
 	StateDraft,
 	TaxCategory,
 	TaxCategoryDraft,
+	Type,
+	TypeDraft,
 } from "@commercetools/platform-sdk";
 import assert from "assert";
 import supertest from "supertest";
@@ -78,6 +80,30 @@ const productState2Draft: StateDraft = {
 	},
 };
 
+const productPriceTypeDraft: TypeDraft = {
+	key: "product-price",
+	name: {
+		en: "ProductPriceType",
+	},
+	description: {
+		en: "Product Price Type",
+	},
+	resourceTypeIds: ["product-price"],
+	fieldDefinitions: [
+		{
+			name: "lastModifiedAt",
+			label: {
+				en: "Last modified at",
+			},
+			required: false,
+			type: {
+				name: "DateTime",
+			},
+			inputHint: "SingleLine",
+		},
+	],
+};
+
 const publishedProductDraft: ProductDraft = {
 	name: {
 		"nl-NL": "test published product",
@@ -250,6 +276,7 @@ let taxCategory1: TaxCategory;
 let taxCategory2: TaxCategory;
 let productState1: State;
 let productState2: State;
+let productPriceType: Type;
 
 async function beforeAllProductTests(mock: CommercetoolsMock) {
 	let response;
@@ -308,6 +335,12 @@ async function beforeAllProductTests(mock: CommercetoolsMock) {
 
 	expect(response.status).toBe(201);
 	productState2 = response.body;
+
+	response = await supertest(mock.app)
+		.post("/dummy/types")
+		.send(productPriceTypeDraft);
+	expect(response.status).toBe(201);
+	productPriceType = response.body;
 }
 
 describe("Product", () => {
@@ -1418,4 +1451,38 @@ describe("Product update actions", () => {
 			id: productState2.id,
 		});
 	});
+
+	test("setProductPriceCustomField", async () => {
+		assert(productPublished, "product not created");
+		const priceId =
+			productPublished?.masterData.current.masterVariant.prices?.[0].id;
+		assert(priceId);
+
+		const response = await supertest(ctMock.app)
+			.post(`/dummy/products/${productPublished.id}`)
+			.send({
+				version: 1,
+				actions: [
+					{
+						action: "setProductPriceCustomType",
+						priceId,
+						type: {
+							typeId: "type",
+							key: productPriceType.key,
+						},
+					},
+					{
+						action: "setProductPriceCustomField",
+						name: "myCustomField",
+						value: "MyRandomValue",
+						priceId,
+					},
+				],
+			});
+		expect(response.status).toBe(200);
+		expect(
+			response.body.masterData.staged.masterVariant.prices?.[0].custom.fields
+				?.myCustomField,
+		).toBe("MyRandomValue");
+	});
 });