@labdigital/commercetools-mock 1.0.0 → 1.1.1

package/dist/index.js

@@ -2787,6 +2787,122 @@ var ProductRepository = class extends AbstractResourceRepository {
         }
         checkForStagedChanges(resource);
         return resource;
+      },
+      addPrice: (context, resource, { variantId, sku, price, staged }) => {
+        const addVariantPrice = (data) => {
+          const { variant, isMasterVariant, variantIndex } = getVariant(
+            data,
+            variantId,
+            sku
+          );
+          if (!variant) {
+            throw new Error(
+              `Variant with id ${variantId} or sku ${sku} not found on product ${resource.id}`
+            );
+          }
+          if (variant.prices === void 0) {
+            variant.prices = [priceFromDraft(price)];
+          } else {
+            variant.prices.push(priceFromDraft(price));
+          }
+          if (isMasterVariant) {
+            data.masterVariant = variant;
+          } else {
+            data.variants[variantIndex] = variant;
+          }
+        };
+        const onlyStaged = staged !== void 0 ? staged : true;
+        addVariantPrice(resource.masterData.staged);
+        if (!onlyStaged) {
+          addVariantPrice(resource.masterData.current);
+        }
+        checkForStagedChanges(resource);
+        return resource;
+      },
+      changePrice: (context, resource, { priceId, price, staged }) => {
+        const changeVariantPrice = (data) => {
+          var _a;
+          const allVariants = [data.masterVariant, ...data.variants ?? []];
+          const priceVariant = allVariants.find(
+            (variant2) => {
+              var _a2;
+              return (_a2 = variant2.prices) == null ? void 0 : _a2.some((x) => x.id === priceId);
+            }
+          );
+          if (!priceVariant) {
+            throw new Error(
+              `Price with id ${priceId} not found on product ${resource.id}`
+            );
+          }
+          const { variant, isMasterVariant, variantIndex } = getVariant(
+            data,
+            priceVariant.id,
+            priceVariant.sku
+          );
+          if (!variant) {
+            throw new Error(
+              `Variant with id ${priceVariant.id} or sku ${priceVariant.sku} not found on product ${resource.id}`
+            );
+          }
+          variant.prices = (_a = variant.prices) == null ? void 0 : _a.map((x) => {
+            if (x.id === priceId) {
+              return { ...x, ...price };
+            }
+            return x;
+          });
+          if (isMasterVariant) {
+            data.masterVariant = variant;
+          } else {
+            data.variants[variantIndex] = variant;
+          }
+        };
+        const onlyStaged = staged !== void 0 ? staged : true;
+        changeVariantPrice(resource.masterData.staged);
+        if (!onlyStaged) {
+          changeVariantPrice(resource.masterData.current);
+        }
+        checkForStagedChanges(resource);
+        return resource;
+      },
+      removePrice: (context, resource, { priceId, staged }) => {
+        const removeVariantPrice = (data) => {
+          var _a;
+          const allVariants = [data.masterVariant, ...data.variants ?? []];
+          const priceVariant = allVariants.find(
+            (variant2) => {
+              var _a2;
+              return (_a2 = variant2.prices) == null ? void 0 : _a2.some((x) => x.id === priceId);
+            }
+          );
+          if (!priceVariant) {
+            throw new Error(
+              `Price with id ${priceId} not found on product ${resource.id}`
+            );
+          }
+          const { variant, isMasterVariant, variantIndex } = getVariant(
+            data,
+            priceVariant.id,
+            priceVariant.sku
+          );
+          if (!variant) {
+            throw new Error(
+              `Variant with id ${priceVariant.id} or sku ${priceVariant.sku} not found on product ${resource.id}`
+            );
+          }
+          variant.prices = (_a = variant.prices) == null ? void 0 : _a.filter((x) => x.id !== priceId);
+          if (isMasterVariant) {
+            data.masterVariant = variant;
+          } else {
+            data.variants[variantIndex] = variant;
+          }
+        };
+        const onlyStaged = staged !== void 0 ? staged : true;
+        removeVariantPrice(resource.masterData.staged);
+        if (!onlyStaged) {
+          removeVariantPrice(resource.masterData.current);
+        }
+        checkForStagedChanges(resource);
+        return resource;
       }
     };
   }
@@ -2880,6 +2996,7 @@ var variantFromDraft = (variantId, variant) => {
 };
 var priceFromDraft = (draft) => ({
   id: (0, import_uuid6.v4)(),
+  country: draft.country,
   value: {
     currencyCode: draft.value.currencyCode,
     centAmount: draft.value.centAmount,