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)
+  }
+)

packages/medusa/src/api-v2/admin/tax-rates/[id]/route.ts

@@ -5,6 +5,37 @@ import {
 
 import { defaultAdminTaxRateFields } from "../query-config"
 import { remoteQueryObjectFromString } from "@medusajs/utils"
+import { AdminPostTaxRatesTaxRateReq } from "../../../../api/routes/admin/tax-rates"
+import { updateTaxRatesWorkflow } from "@medusajs/core-flows"
+
+export const POST = async (
+  req: AuthenticatedMedusaRequest<AdminPostTaxRatesTaxRateReq>,
+  res: MedusaResponse
+) => {
+  const { errors } = await updateTaxRatesWorkflow(req.scope).run({
+    input: {
+      selector: { id: req.params.id },
+      update: req.validatedBody,
+    },
+    throwOnError: false,
+  })
+
+  if (Array.isArray(errors) && errors[0]) {
+    throw errors[0].error
+  }
+
+  const remoteQuery = req.scope.resolve("remoteQuery")
+
+  const queryObject = remoteQueryObjectFromString({
+    entryPoint: "tax_rate",
+    variables: { id: req.params.id },
+    fields: defaultAdminTaxRateFields,
+  })
+
+  const [taxRate] = await remoteQuery(queryObject)
+
+  res.status(200).json({ tax_rate: taxRate })
+}
 
 export const GET = async (
   req: AuthenticatedMedusaRequest,

packages/medusa/src/api-v2/admin/tax-rates/middlewares.ts

@@ -3,6 +3,7 @@ import * as QueryConfig from "./query-config"
 import {
   AdminGetTaxRatesTaxRateParams,
   AdminPostTaxRatesReq,
+  AdminPostTaxRatesTaxRateReq,
 } from "./validators"
 import { transformBody, transformQuery } from "../../../api/middlewares"
 
@@ -20,6 +21,11 @@ export const adminTaxRateRoutesMiddlewares: MiddlewareRoute[] = [
     matcher: "/admin/tax-rates",
     middlewares: [transformBody(AdminPostTaxRatesReq)],
   },
+  {
+    method: "POST",
+    matcher: "/admin/tax-rates/:id",
+    middlewares: [transformBody(AdminPostTaxRatesTaxRateReq)],
+  },
   {
     method: "GET",
     matcher: "/admin/tax-rates/:id",

packages/medusa/src/api-v2/admin/tax-rates/validators.ts

@@ -1,4 +1,3 @@
-// HEAD
 import { Type } from "class-transformer"
 import {
   IsBoolean,
@@ -52,3 +51,29 @@ export class AdminPostTaxRatesReq {
   @IsOptional()
   metadata?: Record<string, unknown>
 }
+
+export class AdminPostTaxRatesTaxRateReq {
+  @IsOptional()
+  @IsNumber()
+  rate?: number | null
+
+  @IsOptional()
+  @IsString()
+  code?: string | null
+
+  @IsString()
+  @IsOptional()
+  name?: string
+
+  @IsBoolean()
+  @IsOptional()
+  is_default?: boolean
+
+  @IsBoolean()
+  @IsOptional()
+  is_combinable?: boolean
+
+  @IsObject()
+  @IsOptional()
+  metadata?: Record<string, unknown>
+}

packages/tax/src/services/tax-module-service.ts

@@ -196,6 +196,27 @@ export default class TaxModuleService<
     return await this.taxRateService_.update({ selector, data }, sharedContext)
   }
 
+  async upsert(
+    data: TaxTypes.UpsertTaxRateDTO[],
+    sharedContext?: Context
+  ): Promise<TaxTypes.TaxRateDTO[]>
+  async upsert(
+    data: TaxTypes.UpsertTaxRateDTO,
+    sharedContext?: Context
+  ): Promise<TaxTypes.TaxRateDTO>
+
+  @InjectTransactionManager("baseRepository_")
+  async upsert(
+    data: TaxTypes.UpsertTaxRateDTO | TaxTypes.UpsertTaxRateDTO[],
+    @MedusaContext() sharedContext: Context = {}
+  ): Promise<TaxTypes.TaxRateDTO | TaxTypes.TaxRateDTO[]> {
+    const result = await this.taxRateService_.upsert(data, sharedContext)
+    const serialized = await this.baseRepository_.serialize<
+      TaxTypes.TaxRateDTO[]
+    >(result, { populate: true })
+    return Array.isArray(data) ? serialized : serialized[0]
+  }
+
   createTaxRegions(
     data: TaxTypes.CreateTaxRegionDTO,
     sharedContext?: Context

packages/types/src/tax/mutations.ts

@@ -9,6 +9,16 @@ export interface CreateTaxRateDTO {
   metadata?: Record<string, unknown>
 }
 
+export interface UpsertTaxRateDTO {
+  id?: string
+  rate?: number | null
+  code?: string | null
+  name?: string
+  is_default?: boolean
+  created_by?: string | null
+  metadata?: Record<string, unknown> | null
+}
+
 export interface UpdateTaxRateDTO {
   rate?: number | null
   code?: string | null

packages/types/src/tax/service.ts

@@ -20,6 +20,7 @@ import {
   CreateTaxRateDTO,
   CreateTaxRegionDTO,
   UpdateTaxRateDTO,
+  UpsertTaxRateDTO,
 } from "./mutations"
 
 export interface ITaxModuleService extends IModuleService {
@@ -63,6 +64,12 @@ export interface ITaxModuleService extends IModuleService {
     sharedContext?: Context
   ): Promise<TaxRateDTO[]>
 
+  upsert(data: UpsertTaxRateDTO, sharedContext?: Context): Promise<TaxRateDTO>
+  upsert(
+    data: UpsertTaxRateDTO[],
+    sharedContext?: Context
+  ): Promise<TaxRateDTO[]>
+
   delete(taxRateIds: string[], sharedContext?: Context): Promise<void>
   delete(taxRateId: string, sharedContext?: Context): Promise<void>