SetShippingMethod

import { getNormalizers } from "@alokai/connect/integration-kit";
import { ValidationError } from "@alokai/connect/middleware";
import { z } from "zod";

import { defineApi, getCartId, query } from "@vsf-enterprise/unified-api-magento";
import type { Cart } from "@vsf-enterprise/unified-api-magento/ecommerceTypes";

const shippingMethodIdSchema = z.string().regex(/^[^:]+:[^:]+$/, {
  message: "Expected a string in the format 'carrier_code:method_code'",
});

export const setShippingMethod = defineApi.setShippingMethod(async (context, args) => {
  const { api } = await context.getApiClient();
  const cartId = getCartId(context);
  const { carrier_code, method_code } = unwrapShippingMethodId(args.shippingMethodId);

  const updatedCart = await query(
    api.setShippingMethodsOnCart({
      cart_id: cartId,
      shipping_methods: [
        {
          carrier_code,
          method_code,
        },
      ],
    }),
  );

  const { normalizeCart } = getNormalizers(context);

  return normalizeCart(updatedCart.setShippingMethodsOnCart?.cart as Cart);
});

function unwrapShippingMethodId(input: string) {
  const result = shippingMethodIdSchema.safeParse(input);
  if (!result.success) {
    throw ValidationError.fromStandardSchemaError(
      result.error,
      `Invalid shipping method id "${input}"`,
    );
  }
  const [carrier_code, method_code] = input.split(":");
  return { carrier_code: carrier_code!, method_code: method_code! };
}