import { gql } from '@vtex/gatsby-plugin-graphql'

import { request } from '../graphql/request'
import { SetRegionMutation } from './__generated__/SetRegionMutation.graphql'

export const setRegion = async ({
  postalCode,
  orderFormId,
}: {
  postalCode: string
  orderFormId: string
}) => {
  let result = await request({
    ...SetRegionMutation,
    variables: {
      orderFormId,
      postalCode,
    },
  })

  // TODO: it seems that sometimes the first request to set the address
  // either fails or just kind of "warms up" the order form shipping.
  // Anyway, retrying the request once seems to fix the problem.
  // This needs to be looked into though!
  if (!result?.updateSelectedAddress?.shipping?.selectedAddress) {
    result = await request({
      ...SetRegionMutation,
      variables: {
        orderFormId,
        postalCode,
      },
    })
  }

  return result
}

export const mutation = gql`
  mutation SetRegionMutation(
    $postalCode: String
    $country: String = "BRA"
    $addressType: VTEX_AddressType = residential
    $orderFormId: ID
  ) {
    updateSelectedAddress(
      input: {
        postalCode: $postalCode
        country: $country
        addressType: $addressType
        isDisposable: true
      }
      orderFormId: $orderFormId
    ) {
      ...OrderFormFragment_orderForm
    }
  }
`