import { GraphQLRequest } from '@graphcommerce/graphql'
import { setContext } from '@graphcommerce/graphql/apollo'

const isMutation = (operation: GraphQLRequest) =>
  operation.query.definitions.some(
    (definition) =>
      definition.kind === 'OperationDefinition' && definition.operation === 'mutation',
  )

/** Apollo link that adds the Google reCAPTCHA token to the request context. */
export const recaptchaLink = setContext(async (operation, context) => {
  const recaptchaKey = import.meta.graphCommerce.googleRecaptchaKey
  if (!recaptchaKey || !globalThis.grecaptcha || !isMutation(operation)) return context

  await new Promise<void>((resolve) => {
    globalThis.grecaptcha?.ready(resolve)
  })

  let token: string | null = null
  let failure = 0
  while (failure < 5) {
    try {
      // eslint-disable-next-line no-await-in-loop
      token = await globalThis.grecaptcha.execute(recaptchaKey, { action: 'submit' })
      break
    } catch {
      failure++
    }
  }

  if (!token) {
    console.error('Failed to get reCAPTCHA token after 5 attempts')
    return context
  }

  if (!context.headers) context.headers = {}
  context.headers['X-ReCaptcha'] = token
  return context
})