feat(core-flow, medusa): Create cart with Sales Channel (#6427)

packages/core-flows/src/definition/cart/steps/find-sales-channel.ts

@@ -0,0 +1,34 @@
+import { ModuleRegistrationName } from "@medusajs/modules-sdk"
+import { ISalesChannelModuleService, SalesChannelDTO } from "@medusajs/types"
+import { MedusaError } from "@medusajs/utils"
+import { StepResponse, createStep } from "@medusajs/workflows-sdk"
+
+interface StepInput {
+  salesChannelId?: string
+}
+
+export const findSalesChannelStepId = "find-sales-channel"
+export const findSalesChannelStep = createStep(
+  findSalesChannelStepId,
+  async (data: StepInput, { container }) => {
+    const salesChannelService = container.resolve<ISalesChannelModuleService>(
+      ModuleRegistrationName.SALES_CHANNEL
+    )
+
+    let salesChannel: SalesChannelDTO | undefined
+    if (data.salesChannelId) {
+      salesChannel = await salesChannelService.retrieve(data.salesChannelId)
+    } else {
+      // TODO: Find default sales channel from store
+    }
+
+    if (salesChannel?.is_disabled) {
+      throw new MedusaError(
+        MedusaError.Types.INVALID_DATA,
+        `Unable to assign cart to disabled Sales Channel: ${salesChannel.name}`
+      )
+    }
+
+    return new StepResponse(salesChannel)
+  }
+)

packages/core-flows/src/definition/cart/steps/index.ts

@@ -1,5 +1,5 @@
 export * from "./create-carts"
 export * from "./find-one-or-any-region"
 export * from "./find-or-create-customer"
+export * from "./find-sales-channel"
 export * from "./update-carts"
-

packages/core-flows/src/definition/cart/workflows/create-carts.ts

@@ -2,41 +2,54 @@ import { CartDTO, CreateCartWorkflowInputDTO } from "@medusajs/types"
 import {
   WorkflowData,
   createWorkflow,
+  parallelize,
   transform,
 } from "@medusajs/workflows-sdk"
 import {
   createCartsStep,
   findOneOrAnyRegionStep,
   findOrCreateCustomerStep,
+  findSalesChannelStep,
 } from "../steps"
 
 export const createCartWorkflowId = "create-cart"
 export const createCartWorkflow = createWorkflow(
   createCartWorkflowId,
   (input: WorkflowData<CreateCartWorkflowInputDTO>): WorkflowData<CartDTO> => {
-    const region = findOneOrAnyRegionStep({
-      regionId: input.region_id,
-    })
+    const [salesChannel, region, customerData] = parallelize(
+      findSalesChannelStep({
+        salesChannelId: input.sales_channel_id,
+      }),
+      findOneOrAnyRegionStep({
+        regionId: input.region_id,
+      }),
+      findOrCreateCustomerStep({
+        customerId: input.customer_id,
+        email: input.email,
+      })
+    )
 
-    const customerData = findOrCreateCustomerStep({
-      customerId: input.customer_id,
-      email: input.email,
-    })
+    const cartInput = transform(
+      { input, region, customerData, salesChannel },
+      (data) => {
+        const data_ = {
+          ...data.input,
+          currency_code: data.input.currency_code ?? data.region.currency_code,
+          region_id: data.region.id,
+        }
 
-    const cartInput = transform({ input, region, customerData }, (data) => {
-      const data_ = {
-        ...data.input,
-        currency_code: data.input.currency_code ?? data.region.currency_code,
-        region_id: data.region.id,
+        if (data.customerData.customer?.id) {
+          data_.customer_id = data.customerData.customer.id
+          data_.email = data.input?.email ?? data.customerData.customer.email
+        }
+
+        if (data.salesChannel?.id) {
+          data_.sales_channel_id = data.salesChannel.id
+        }
+
+        return data_
       }
-
-      if (data.customerData.customer?.id) {
-        data_.customer_id = data.customerData.customer.id
-        data_.email = data.input?.email ?? data.customerData.customer.email
-      }
-
-      return data_
-    })
+    )
 
     // TODO: Add line items