feat: add payment authorization workflow

packages/core/core-flows/src/cart/workflows/complete-cart.ts

@@ -17,7 +17,7 @@ import {
   useRemoteQueryStep,
 } from "../../common"
 import { createOrdersStep } from "../../order/steps/create-orders"
-import { authorizePaymentSessionStep } from "../../payment/steps/authorize-payment-session"
+import { authorizePaymentSessionStep } from "../../payment-collection"
 import { registerUsageStep } from "../../promotion/steps/register-usage"
 import { updateCartsStep, validateCartPaymentsStep } from "../steps"
 import { reserveInventoryStep } from "../steps/reserve-inventory"
@@ -58,7 +58,6 @@ export const completeCartWorkflow = createWorkflow(
       // We choose the first payment session, as there will only be one active payment session
       // This might change in the future.
       id: paymentSessions[0].id,
-      context: { cart_id: cart.id },
     })
 
     const { variants, items, sales_channel_id } = transform(

packages/core/core-flows/src/order/workflows/mark-payment-collection-as-paid.ts

@@ -7,11 +7,11 @@ import {
   createWorkflow,
 } from "@medusajs/workflows-sdk"
 import { useRemoteQueryStep } from "../../common"
+import { capturePaymentWorkflow } from "../../payment"
 import {
   authorizePaymentSessionStep,
-  capturePaymentWorkflow,
-} from "../../payment"
-import { createPaymentSessionsWorkflow } from "../../payment-collection"
+  createPaymentSessionsWorkflow,
+} from "../../payment-collection"
 
 /**
  * This step validates that the payment collection is not_paid
@@ -63,7 +63,6 @@ export const markPaymentCollectionAsPaid = createWorkflow(
 
     const payment = authorizePaymentSessionStep({
       id: paymentSession.id,
-      context: { order_id: input.order_id },
     })
 
     capturePaymentWorkflow.runAsStep({

packages/core/core-flows/src/payment-collection/steps/authorize-payment-session.ts

@@ -0,0 +1,48 @@
+import { IPaymentModuleService, Logger } from "@medusajs/types"
+import {
+  ContainerRegistrationKeys,
+  ModuleRegistrationName,
+  PaymentSessionStatus,
+} from "@medusajs/utils"
+import { StepResponse, createStep } from "@medusajs/workflows-sdk"
+
+export type AuthorizePaymentSessionStepInput = {
+  id: string
+  provider_token?: string
+}
+
+export const authorizePaymentSessionStepId = "authorize-payment-session-step"
+/**
+ * This step authorizes a payment session.
+ */
+export const authorizePaymentSessionStep = createStep(
+  authorizePaymentSessionStepId,
+  async (input: AuthorizePaymentSessionStepInput, { container }) => {
+    const paymentModule = container.resolve<IPaymentModuleService>(
+      ModuleRegistrationName.PAYMENT
+    )
+    const payment = await paymentModule.authorizePaymentSession(input.id, {
+      provider_token: input.provider_token,
+    })
+    return new StepResponse(payment)
+  },
+  // If payment or any other part of complete cart fails post payment step, we cancel any payments made
+  async (payment, { container }) => {
+    if (!payment) {
+      return
+    }
+
+    const logger = container.resolve<Logger>(ContainerRegistrationKeys.LOGGER)
+    const paymentModule = container.resolve<IPaymentModuleService>(
+      ModuleRegistrationName.PAYMENT
+    )
+
+    try {
+      await paymentModule.cancelPayment(payment.id)
+    } catch (e) {
+      logger.error(
+        `Error was thrown trying to cancel payment - ${payment.id} - ${e}`
+      )
+    }
+  }
+)

packages/core/core-flows/src/payment-collection/steps/index.ts

@@ -6,3 +6,4 @@ export * from "./update-payment-collection"
 export * from "./update-refund-reasons"
 export * from "./validate-deleted-payment-sessions"
 export * from "./cancel-payment-session"
+export * from "./authorize-payment-session"

packages/core/core-flows/src/payment-collection/workflows/authorize-payment-session.ts

@@ -0,0 +1,19 @@
+import {
+  WorkflowData,
+  WorkflowResponse,
+  createWorkflow,
+} from "@medusajs/workflows-sdk"
+import { authorizePaymentSessionStep } from "../steps"
+
+export const authorizePaymentSessionWorkflowId =
+  "authorize-payment-session-workflow"
+
+/**
+ * This workflow authorizes a payment session.
+ */
+export const authorizePaymentSessionWorkflow = createWorkflow(
+  authorizePaymentSessionWorkflowId,
+  (input: WorkflowData<{ id: string; provider_token?: string }>) => {
+    return new WorkflowResponse(authorizePaymentSessionStep(input))
+  }
+)

packages/core/core-flows/src/payment-collection/workflows/cancel-payment-session.ts

@@ -3,7 +3,7 @@ import {
   WorkflowResponse,
   createWorkflow,
 } from "@medusajs/workflows-sdk"
-import { cancelPaymentSessionStep } from "../steps/cancel-payment-session"
+import { cancelPaymentSessionStep } from "../steps"
 
 export const cancelPaymentSessionWorkflowId = "cancel-payment-session-workflow"
 

packages/core/core-flows/src/payment/steps/index.ts

@@ -1,4 +1,3 @@
-export * from "./authorize-payment-session"
 export * from "./cancel-payment"
 export * from "./capture-payment"
 export * from "./refund-payment"