fix(core-flows): set SalesChannels on Product update

.changeset/spicy-panthers-exist.md

@@ -0,0 +1,5 @@
+---
+"@medusajs/core-flows": patch
+---
+
+fix(core-flows): set SalesChannels for Products on update

packages/core/core-flows/src/product/steps/get-product-sales-channel-link.ts

@@ -0,0 +1,33 @@
+import { Modules } from "@medusajs/modules-sdk"
+import { ContainerRegistrationKeys } from "@medusajs/utils"
+import { StepResponse, createStep } from "@medusajs/workflows-sdk"
+
+type StepInput = {
+  ids: string[]
+}
+
+export const getProductSalesChannelLinkStepId = "get-product-sales-channel-link"
+export const getProductSalesChannelLinkStep = createStep(
+  getProductSalesChannelLinkStepId,
+  async (data: StepInput, { container }) => {
+    if (!data.ids.length) {
+      return new StepResponse([])
+    }
+
+    const remoteLink = container.resolve(ContainerRegistrationKeys.REMOTE_LINK)
+
+    const linkService = remoteLink.getLinkModule(
+      Modules.PRODUCT,
+      "product_id",
+      Modules.SALES_CHANNEL,
+      "sales_channel_id"
+    )
+
+    const existingItems = (await linkService.list(
+      { product_id: data.ids },
+      { select: ["product_id", "sales_channel_id"] }
+    )) as { product_id: string; sales_channel_id: string }[]
+
+    return new StepResponse(existingItems)
+  }
+)

packages/core/core-flows/src/product/workflows/update-products.ts

@@ -1,15 +1,32 @@
 import { ProductTypes } from "@medusajs/types"
-import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import {
+  WorkflowData,
+  createWorkflow,
+  transform,
+} from "@medusajs/workflows-sdk"
 import { updateProductsStep } from "../steps/update-products"
+import { getProductSalesChannelLinkStep } from "../steps/get-product-sales-channel-link"
+import {
+  associateProductsWithSalesChannelsStep,
+  detachProductsFromSalesChannelsStep,
+} from "../../sales-channel"
+
+type UpdateProductsStepInputSelector = {
+  selector: ProductTypes.FilterableProductProps
+  update: ProductTypes.UpdateProductDTO & {
+    sales_channels?: { id: string }[]
+  }
+}
+
+type UpdateProductsStepInputProducts = {
+  products: (ProductTypes.UpsertProductDTO & {
+    sales_channels?: { id: string }[]
+  })[]
+}
 
 type UpdateProductsStepInput =
-  | {
-      selector: ProductTypes.FilterableProductProps
-      update: ProductTypes.UpdateProductDTO
-    }
-  | {
-      products: ProductTypes.UpsertProductDTO[]
-    }
+  | UpdateProductsStepInputSelector
+  | UpdateProductsStepInputProducts
 
 type WorkflowInput = UpdateProductsStepInput
 
@@ -20,7 +37,87 @@ export const updateProductsWorkflow = createWorkflow(
     input: WorkflowData<WorkflowInput>
   ): WorkflowData<ProductTypes.ProductDTO[]> => {
     // TODO: Delete price sets for removed variants
-    // TODO Update sales channel links
-    return updateProductsStep(input)
+
+    const toUpdateInput = transform({ input }, (data) => {
+      if ((data.input as UpdateProductsStepInputProducts).products) {
+        return (data.input as UpdateProductsStepInputProducts).products.map(
+          (p) => ({
+            ...p,
+            sales_channels: undefined,
+          })
+        )
+      } else {
+        return {
+          selector: (data.input as UpdateProductsStepInputSelector).selector,
+          update: {
+            ...(data.input as UpdateProductsStepInputSelector).update,
+            sales_channels: undefined,
+          },
+        }
+      }
+    })
+
+    const updatedProducts = updateProductsStep(toUpdateInput as any) // TODO: type
+
+    const updatedProductIds = transform({ updatedProducts, input }, (data) => {
+      if (
+        !(data.input as UpdateProductsStepInputSelector).update.sales_channels
+      ) {
+        return []
+      }
+
+      let productIds = data.updatedProducts.map((p) => p.id)
+
+      ;(data.input as UpdateProductsStepInputProducts).products?.forEach(
+        (p) => {
+          if (!p.sales_channels) {
+            // these products don't update sales channels so we don't want to their previous channels
+            productIds = productIds.filter((id) => id !== p.id)
+          }
+        }
+      )
+
+      return productIds
+    })
+
+    const currentLinks = getProductSalesChannelLinkStep({
+      ids: updatedProductIds,
+    })
+
+    detachProductsFromSalesChannelsStep({ links: currentLinks })
+
+    const salesChannelLinks = transform({ input, updatedProducts }, (data) => {
+      if ((data.input as UpdateProductsStepInputSelector).selector) {
+        if (
+          !(data.input as UpdateProductsStepInputSelector).update.sales_channels
+        ) {
+          return []
+        }
+        return data.updatedProducts
+          .map((p) => {
+            return (
+              data.input as UpdateProductsStepInputSelector
+            ).update.sales_channels!.map((channel) => ({
+              sales_channel_id: channel.id,
+              product_id: p.id,
+            }))
+          })
+          .flat()
+      }
+
+      return (data.input as UpdateProductsStepInputProducts).products
+        .filter((input) => input.sales_channels)
+        .map((input) => {
+          return input.sales_channels!.map((salesChannel) => ({
+            sales_channel_id: salesChannel.id,
+            product_id: input.id as string,
+          }))
+        })
+        .flat()
+    })
+
+    associateProductsWithSalesChannelsStep({ links: salesChannelLinks })
+
+    return updatedProducts
   }
 )