Add retrieval and update of product collections in update-product or…

…chestration
jamalsoueidan · Jun 29, 2024 · e54a46e · e54a46e
1 parent 3585bf5
commit e54a46e
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/src/functions/customer/orchestrations/product/update/update-product.spec.ts b/src/functions/customer/orchestrations/product/update/update-product.spec.ts
@@ -1,4 +1,5 @@
 import { faker } from "@faker-js/faker";
+import { CustomerProductServiceGet } from "~/functions/customer/services/product/get";
 import { OpenAIServiceProductCategorize } from "~/functions/openai/services/product-categorize";
 import { TimeUnit } from "~/functions/schedule";
 import { createUser } from "~/library/jest/helpers";
@@ -231,6 +232,13 @@ describe("CustomerProductUpdateOrchestration", () => {
       productId: product.productId,
     });
 
+    const updatedProduct = await CustomerProductServiceGet({
+      customerId,
+      productId: product.productId,
+    });
+
+    expect(updatedProduct.collectionIds).toHaveLength(2);
+
     const tags = [
       `user`,
       `user-${user.username}`,

diff --git a/src/functions/customer/orchestrations/product/update/update-product.ts b/src/functions/customer/orchestrations/product/update/update-product.ts
@@ -1,4 +1,5 @@
 import { CustomerServiceGet } from "~/functions/customer/services/customer/get";
+import { CustomerProductServiceGet } from "~/functions/customer/services/product/get";
 import { LocationModel } from "~/functions/location";
 import { OpenAIServiceProductCategorize } from "~/functions/openai/services/product-categorize";
 import { ScheduleModel } from "~/functions/schedule";
@@ -98,6 +99,29 @@ export const updateProduct = async ({
         tags.push(r.condition);
       });
     });
+
+    const { scheduleId, scheduleMetafieldId, scheduleName, ...oldProduct } =
+      await CustomerProductServiceGet({
+        customerId,
+        productId,
+      });
+
+    await ScheduleModel.updateOne(
+      {
+        customerId,
+        "products.productId": productId,
+      },
+      {
+        $set: {
+          "products.$": {
+            ...oldProduct,
+            collectionIds:
+              categories?.map((c) => GidFormat.parse(c.id)) ||
+              oldProduct.collectionIds,
+          },
+        },
+      }
+    );
   }
 
   const variables = {