feat(tax): add support for updating tax rates (#6537)

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

@@ -1,2 +1,3 @@
 export * from "./create-tax-regions"
 export * from "./create-tax-rates"
+export * from "./update-tax-rates"

packages/core-flows/src/tax/steps/update-tax-rates.ts

@@ -0,0 +1,47 @@
+import { ModuleRegistrationName } from "@medusajs/modules-sdk"
+import {
+  FilterableTaxRateProps,
+  ITaxModuleService,
+  UpdateTaxRateDTO,
+} from "@medusajs/types"
+import { getSelectsAndRelationsFromObjectArray } from "@medusajs/utils"
+import { StepResponse, createStep } from "@medusajs/workflows-sdk"
+
+type UpdateTaxRatesStepInput = {
+  selector: FilterableTaxRateProps
+  update: UpdateTaxRateDTO
+}
+
+export const updateTaxRatesStepId = "update-tax-rates"
+export const updateTaxRatesStep = createStep(
+  updateTaxRatesStepId,
+  async (data: UpdateTaxRatesStepInput, { container }) => {
+    const service = container.resolve<ITaxModuleService>(
+      ModuleRegistrationName.TAX
+    )
+
+    const { selects, relations } = getSelectsAndRelationsFromObjectArray([
+      data.update,
+    ])
+
+    const prevData = await service.list(data.selector, {
+      select: selects,
+      relations,
+    })
+
+    const taxRates = await service.update(data.selector, data.update)
+
+    return new StepResponse(taxRates, prevData)
+  },
+  async (prevData, { container }) => {
+    if (!prevData?.length) {
+      return
+    }
+
+    const service = container.resolve<ITaxModuleService>(
+      ModuleRegistrationName.TAX
+    )
+
+    await service.upsert(prevData)
+  }
+)

packages/core-flows/src/tax/workflows/index.ts

@@ -1,2 +1,3 @@
 export * from "./create-tax-regions"
 export * from "./create-tax-rates"
+export * from "./update-tax-rates"

packages/core-flows/src/tax/workflows/update-tax-rates.ts

@@ -0,0 +1,22 @@
+import {
+  FilterableTaxRateProps,
+  TaxRateDTO,
+  UpdateTaxRateDTO,
+} from "@medusajs/types"
+import { WorkflowData, createWorkflow } from "@medusajs/workflows-sdk"
+import { updateTaxRatesStep } from "../steps"
+
+type UpdateTaxRatesStepInput = {
+  selector: FilterableTaxRateProps
+  update: UpdateTaxRateDTO
+}
+
+type WorkflowInput = UpdateTaxRatesStepInput
+
+export const updateTaxRatesWorkflowId = "update-tax-rates"
+export const updateTaxRatesWorkflow = createWorkflow(
+  updateTaxRatesWorkflowId,
+  (input: WorkflowData<WorkflowInput>): WorkflowData<TaxRateDTO[]> => {
+    return updateTaxRatesStep(input)
+  }
+)